Update app.py

app.py

@@ -588,16 +588,21 @@ class BasicAgent:
         ]
 
         # Pre-initialize RAG vector store
-        self.vector_store = create_vector_store()
+        try:
+            self.vector_store = create_vector_store()
+        except Exception as e:
+            print(f"ERROR: Failed to create vector store: {str(e)}")
+            self.vector_store = None
+            
         self.workflow = create_agent_workflow(self.tools)
-
+    
     def __call__(self, question: str) -> str:
         print(f"\n--- Agent received question: {question[:50]}{'...' if len(question) > 50 else ''} ---")
 
         state = {
             "question": question,
             "context": {
-                "vector_store": self.vector_store
+                "vector_store": self.vector_store  # Include vector store in context
             },
             "reasoning": "",
             "iterations": 0,
@@ -605,9 +610,9 @@ class BasicAgent:
             "final_answer": None,
             "current_task": "Understand the question and plan the next step.",
             "current_thoughts": "",
-            "tools": self.tools
+            "tools": self.tools 
         }
-
+        
         try:
             # The invoke method returns an iterator, so we need to consume it to get the final state
             final_state = self.workflow.invoke(state, {"recursion_limit": 20})