Upload 2 files

agent.py

@@ -183,9 +183,15 @@ def build_graph(provider: str = "google"):
     def retriever(state: MessagesState):
         """Retriever node"""
         similar_question = vector_store.similarity_search(state["messages"][0].content)
-        example_msg = HumanMessage(
-            content=f"Here I provide a similar question and answer for reference: \n\n{similar_question[0].page_content}",
-        )
+        if not similar_question:
+            # No similar documents found, fallback message
+            example_msg = HumanMessage(
+                content="Sorry, I could not find any similar questions in the vector store."
+            )
+        else:
+            example_msg = HumanMessage(
+                content=f"Here I provide a similar question and answer for reference: \n\n{similar_question[0].page_content}",
+            )
         return {"messages": [sys_msg] + state["messages"] + [example_msg]}
 
     builder = StateGraph(MessagesState)