Update agent.py

agent.py

@@ -121,25 +121,27 @@ with open("system_prompt.txt", "r", encoding="utf-8") as f:
 # System message
 sys_msg = SystemMessage(content=system_prompt)
 
-# build a retriever
-embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2") #  dim=768
-supabase = create_client(
-    os.environ.get("SUPABASE_URL"), 
-    os.environ.get("SUPABASE_SERVICE_KEY"))
-vector_store = SupabaseVectorStore(
-    client=supabase,
-    embedding= embeddings,
-    table_name="docs",
-    query_name="match_docs",
-)
-create_retriever_tool = create_retriever_tool(
-    retriever=vector_store.as_retriever(),
-    name="Question Search",
-    description="A tool to retrieve similar questions from a vector store.",
-)
-
-
-        
+
+def create_retriever_tool(persist_directory="vector_store"):
+    # Initialize embeddings - same as your original
+    embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2") #  dim=768
+    
+    # Create/load Chroma vector store
+    vector_store = Chroma(
+        embedding_function=embeddings,
+        persist_directory=""/home/wendy/Downloads"
+    )
+    
+    # Create retriever tool
+    retriever_tool = create_retriever_tool(
+        retriever=vector_store.as_retriever(),
+        name="Question Search",
+        description="A tool to retrieve similar questions from a vector store.",
+    )
+    
+    return retriever_tool, vector_store
+
+       
 
 
 tools = [