graph test

app.py

@@ -137,20 +137,6 @@ class BasicAgent:
     def final_answer(self, state: AnswerState) -> Dict[str, Any]:
         print(f"[final_answer] Agent returning answer: {state['answer']}")
         answer = state["answer"] if state["answer"] else "I don't know"
-
-        if not state["is_final_answer"]:
-            state["messages"].append(
-                ChatMessage(
-                    role="user",
-                    content="Based on the information provided, please provide a final answer, no tool calls allowed.",
-                )
-            )
-            response = self.model.invoke(state["messages"])
-            if "FINAL ANSWER:" in response.content:
-                answer = response.text().split("FINAL ANSWER:")[-1].strip()
-                print(f"Agent final answer: {answer}")
-
-        answer = state["answer"] if state["answer"] else "I don't know"
         return {
             "answer": answer,
             "is_final_answer": True,
@@ -187,6 +173,21 @@ class BasicAgent:
         print(
             f"[exceeded_attempts] Exceeded max number of attempts ({state['attempt']})."
         )
+        state["messages"].append(
+            ChatMessage(
+                role="user",
+                content="Based on the information provided, please provide a final answer, no tool calls allowed.",
+            )
+        )
+        response = self.model.invoke(state["messages"])
+        if "FINAL ANSWER:" in response.content:
+            answer = response.text().split("FINAL ANSWER:")[-1].strip()
+            print(f"Agent final answer: {answer}")
+
+            return {
+                "answer": answer,
+                "is_final_answer": True,
+            }
         return {
             "answer": "Exceeded max number of attempts.",
             "is_final_answer": False,