Update agent.py

agent.py

@@ -187,41 +187,44 @@ def build_graph(provider: str = "google"):
        # return {"messages": [sys_msg] + state["messages"] + [example_msg]}
 
     def retriever(state: MessagesState):
-        """Retriever node"""
+        """Retriever node – gibt direkt die extrahierte Antwort zurück"""
         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"))
+            final_line = next(line for line in lines if "final answer" in line.lower())
             extracted_answer = final_line.split(":", 1)[1].strip()
+
             return {
-                "messages": [HumanMessage(content=f"FINAL ANSWER: {extracted_answer}")],
+                "messages": [{"type": "system", "content": extracted_answer}],
                 "should_end": True
             }
-        except Exception as e:
-            # Falls Parsing fehlschlägt, trotzdem sinnvoll weitermachen
+        except Exception:
             return {
-                "messages": [sys_msg] + state["messages"] + [
-                    HumanMessage(content=f"(Retriever parsing failed, fallback triggered):\n\n{page_content}")
-                ]
+                "messages": [{"type": "system", "content": "Retriever error: no answer found."}],
+                "should_end": True
             }
 
+   # builder = StateGraph(MessagesState)
+    #builder.add_node("retriever", retriever)
+    #builder.add_node("assistant", assistant)
+    #builder.add_node("tools", ToolNode(tools))
+    #builder.add_edge(START, "retriever")
+    #builder.add_edge("retriever", "assistant")
+    #builder.add_conditional_edges(
+     #   "assistant",
+      #  tools_condition,
+    #)
+    #builder.add_edge("tools", "assistant")
+
     builder = StateGraph(MessagesState)
     builder.add_node("retriever", retriever)
-    builder.add_node("assistant", assistant)
-    builder.add_node("tools", ToolNode(tools))
-    builder.add_edge(START, "retriever")
-    builder.add_edge("retriever", "assistant")
-    builder.add_conditional_edges(
-        "assistant",
-        tools_condition,
-    )
-    builder.add_edge("tools", "assistant")
+
+    # Retriever ist Start und Endpunkt
+    builder.set_entry_point("retriever")
+    builder.set_finish_point("retriever")
 
     # Compile graph
     return builder.compile()