Update app.py

app.py

@@ -121,7 +121,7 @@ def excel_tool(state: AgentState) -> AgentState:
 
 # ------------- final answer -------------
 
-def final_answer(state: AgentState) -> AgentState:
+def final_node(state: AgentState) -> AgentState:
     wrap = SystemMessage(
         content="Using everything so far, reply ONLY with {'final_answer':'…'}. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."
     )
@@ -141,7 +141,7 @@ for name, fn in [
     ("wiki_tool", wiki_tool),
     ("ocr_tool", ocr_tool),
     ("audio_tool", audio_tool),
-    ("final_answer", final_answer),
+    ("final_node", final_node),
 ]:
     graph.add_node(name, fn)
 
@@ -153,8 +153,9 @@ def dispatch(state: AgentState) -> str:
         "wiki": "wiki_tool",
         "ocr": "ocr_tool",
         "audio": "audio_tool",
-        "final": "final_answer",
-    }.get(state.next_action, "final_answer")
+        "excel": "excel_tool",
+        "final": "final_node",
+    }.get(state.next_action, "final_node")
 
 graph.add_conditional_edges(
     "tool_selector",
@@ -164,7 +165,7 @@ graph.add_conditional_edges(
         "ocr_tool": "ocr_tool",
         "audio_tool": "audio_tool",
         "excel_tool": "excel_tool",
-        "final_answer": "final_answer",
+        "final_node": "final_node",
     },
 )
 
@@ -173,13 +174,13 @@ for tool_name in ("wiki_tool", "ocr_tool", "audio_tool", "excel_tool"):
     graph.add_edge(tool_name, "tool_selector")
 
 # final_answer → END
-graph.add_edge("final_answer", END)
+graph.add_edge("final_node", END)
 
 compiled_graph = graph.compile()
 
 # ─────────────────────────── Public API  ────────────────────────────────
 
-def answer(question: str, *, task_id: Optional[str] = None) -> str:
+def answer(question: str, task_id: Optional[str] = None) -> str:
     state = AgentState(user_question=question, task_id=task_id)
     state.add(SystemMessage(content="You are a helpful assistant."))
     state.add(HumanMessage(content=question))
@@ -210,7 +211,7 @@ class BasicAgent:
         
         print(f"Agent received question: {question}")
         print()
-        return respond_to_input(question, task_id)
+        return answer(question, task_id)
         # return fixed_answer