Update app.py

app.py

@@ -78,7 +78,8 @@ def route_agent(state: AgentState, agent_out):
     Return "tools" if the LLM output is a dict with a valid "tool" key,
     otherwise return "final".
     """
-    if isinstance(agent_out, dict) and agent_out.get("tool") in {"ocr_image", "parse_excel"}:
+    print(f"Route agent received agent_out: {agent_out}")
+    if isinstance(agent_out, dict) and agent_out.get("tool") in {"ocr_image", "parse_excel", "web_search"}:
         return "tools"
     return "final"
 
@@ -95,6 +96,7 @@ compiled_graph = graph.compile()
 def respond_to_input(user_input: str) -> str:
     print(f"Reached: User input: {user_input}")
     initial_state: AgentState = {"messages": []}
+    print("reached respond_to_input")
     return compiled_graph.invoke(initial_state, user_input)