Update agent.py

agent.py

@@ -123,24 +123,23 @@ sys_msg = SystemMessage(content=system_prompt)
 
 
 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
+    # Initialize embeddings
+    embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2") #dim=768
     
-    # Create/load Chroma vector store
+    # Create/load Chroma vector store - must use same parameter name as in your function
     vector_store = Chroma(
         embedding_function=embeddings,
-        persist_directory=""/home/wendy/Downloads"
+        persist_directory=persist_directory  # Use the parameter, not hardcoded path
     )
     
     # Create retriever tool
-    retriever_tool = create_retriever_tool(
+    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.",
     )
     
     return retriever_tool, vector_store
-