graph test

app.py

@@ -47,8 +47,8 @@ Perform any final reasoning or calculation steps needed based on the collected d
 Double-check that the synthesized answer directly addresses the original question and respects ALL specific formatting or content details requested (e.g., rounding, order, units IF asked for).\n"
 
 4. Format Output:
-Response **MUST** start **EXACTLY** with `FINAL ANSWER: ` followed by the answer. NO EXCEPTIONS
-The part after `FINAL ANSWER: ` **MUST** contain **ONLY** the final answer, formatted precisely
+Response **MUST** start **EXACTLY** with `FINAL ANSWER: ` followed by the answer.
+OR with `TOOL: ` followed by the tool name and the request to the tool, example `TOOL: GoogleSearchAgent(request='search query')`
 
 Question: {question}
 """
@@ -59,6 +59,7 @@ class AnswerState(TypedDict):
     answer: Optional[str]
     messages: List[Dict[str, Any]]
     is_final_answer: bool
+    search_request: Optional[str]
 
 
 class BasicAgent:
@@ -92,6 +93,7 @@ class BasicAgent:
                 "answer": None,
                 "messages": [],
                 "is_final_answer": False,
+                "search_request": None,
             }
         )
         return res.get("answer", "N/A")
@@ -134,6 +136,8 @@ class BasicAgent:
             print(f"Agent returning answer: {state['answer']}")
             state["answer"] = answer
 
+            return "FINAL_ANSWER"
+
         return "FINAL_ANSWER"