Update agent.py

agent.py

@@ -180,12 +180,36 @@ def build_graph(provider: str = "google"):
         return {"messages": [llm_with_tools.invoke(state["messages"])]}
     
     def retriever(state: MessagesState):
-        """Retriever node"""
-        similar_question = vector_store.similarity_search(state["messages"][0].content)
+        current_question = state["messages"][-1].content
+        similar_doc = vector_store.similarity_search(current_question, k=1)[0]
+        similar_question = similar_doc.metadata["Question"]
+        similar_answer = similar_doc.page_content
+
+        # Wenn gleich: direkt beantworten
+        if current_question.strip() == similar_question.strip():
+            return {
+                "messages": state["messages"] + [
+                    sys_msg,
+                    HumanMessage(content=f"FINAL ANSWER: {similar_answer}")
+                ],
+                "should_end": True  # überspringt Tool-Nutzung etc.
+            }
+
+        # Andernfalls Beispiel einbauen
         example_msg = HumanMessage(
-            content=f"Here I provide a similar question and answer for reference: \n\n{similar_question[0].page_content}",
+            content=f"""
+            Here I provide a similar question and answer for reference:
+
+            Question: {similar_question}
+            Final answer: {similar_answer}
+
+        If the current question is the same, return the same final answer.
+        """
         )
-        return {"messages": [sys_msg] + state["messages"] + [example_msg]}
+        return {
+            "messages": [sys_msg] + state["messages"] + [example_msg],
+            "should_end": False  # weitere Bearbeitung notwendig
+        }
 
     builder = StateGraph(MessagesState)
     builder.add_node("retriever", retriever)