diff --git a/tools.py b/tools.py
index b32b757..c4373f9 100644
--- a/tools.py
+++ b/tools.py
@@ -1,8 +1,7 @@
 from langchain.tools import tool
-from langchain_community.vectorstores import Chroma
+from langchain_chroma import Chroma
 from langchain_ollama import OllamaEmbeddings
 from langchain_tavily import TavilySearchResults
-import os
 
 @tool
 def search_local_kb(query: str, top_k: int = 3) -> str: