Update agent.py

agent.py

@@ -181,6 +181,12 @@ def build_graph(provider: str = "groq"):
     def retriever(state: MessagesState):
         """Retriever node"""
         similar_question = vector_store.similarity_search(state["messages"][0].content)
+        if not similar_questions:
+            # Aucun document trouvé => on en informe l'utilisateur et on passe la main à l'LLM
+            fallback_msg = HumanMessage(
+                content="Désolé, aucune question similaire n’a été trouvée dans la base de données. Je vais essayer de répondre sans contexte."
+            )
+            return {"messages": [sys_msg] + state["messages"] + [fallback_msg]}
         example_msg = HumanMessage(
             content=f"Here I provide a similar question and answer for reference: \n\n{similar_question[0].page_content}",
         )