Update agent.py

agent.py

@@ -178,13 +178,38 @@ def build_graph(provider: str = "google"):
         """Assistant node"""
         return {"messages": [llm_with_tools.invoke(state["messages"])]}
     
+    # 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}",
+       # )
+       # return {"messages": [sys_msg] + state["messages"] + [example_msg]}
+
     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}",
-        )
-        return {"messages": [sys_msg] + state["messages"] + [example_msg]}
+    """Retriever node"""
+        current_question = state["messages"][-1].content
+        similar_doc = vector_store.similarity_search(current_question, k=1)[0]
+        page_content = similar_doc.page_content.strip()
+
+        # Versuche, die Antwort aus dem bekannten Format zu extrahieren
+        # Erwartetes Format:
+        # "Question : ...\n\nFinal answer : ..."
+        try:
+            lines = page_content.splitlines()
+            final_line = next(line for line in lines if line.strip().lower().startswith("final answer"))
+            extracted_answer = final_line.split(":", 1)[1].strip()
+            return {
+                "messages": [HumanMessage(content=f"FINAL ANSWER: {extracted_answer}")],
+                "should_end": True
+            }
+        except Exception as e:
+            # Falls Parsing fehlschlägt, trotzdem sinnvoll weitermachen
+            return {
+                "messages": [sys_msg] + state["messages"] + [
+                    HumanMessage(content=f"(Retriever parsing failed, fallback triggered):\n\n{page_content}")
+                ]
+            }
 
     builder = StateGraph(MessagesState)
     builder.add_node("retriever", retriever)