Update agent.py

agent.py

@@ -181,14 +181,24 @@ def build_graph(provider: str = "groq"):
 
     def retriever(state: MessagesState):
         query = state["messages"][-1].content
-        similar_doc = vector_store.similarity_search(query, k=1)[0]
-
+        similar_docs = vector_store.similarity_search(query, k=1)
+        
+        # Handle empty results
+        if not similar_docs:
+            return {"messages": [AIMessage(content="I don't have information about this topic in my knowledge base. Please try a different question.")]}
+        
+        similar_doc = similar_docs[0]
         content = similar_doc.page_content
+        
         if "Final answer :" in content:
             answer = content.split("Final answer :")[-1].strip()
         else:
             answer = content.strip()
-
+        
+        # Ensure answer is not empty
+        if not answer:
+            answer = "I found related information but couldn't extract a clear answer. Please rephrase your question."
+    
         return {"messages": [AIMessage(content=answer)]}