Update agent.py

agent.py

@@ -187,25 +187,9 @@ def build_graph(provider: str = "google"):
        # return {"messages": [sys_msg] + state["messages"] + [example_msg]}
 
     def retriever(state: MessagesState):
-        """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()
-
-        try:
-            lines = page_content.splitlines()
-            final_line = next(line for line in lines if "final answer" in line.lower())
-            extracted_answer = final_line.split(":", 1)[1].strip()
-
-            return {
-                "messages": [{"type": "system", "content": extracted_answer}],
-                "should_end": True
-            }
-        except Exception:
-            return {
-                "messages": [{"type": "system", "content": "Retriever error: no answer found."}],
-                "should_end": True
-            }
+        question = state["messages"][-1].content
+        similar_doc = vector_store.similarity_search(question, k=1)[0]
+        return {"output": similar_doc.page_content.strip()}
 
    # builder = StateGraph(MessagesState)
     #builder.add_node("retriever", retriever)