graph test

app.py

@@ -137,6 +137,20 @@ 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,