Update app.py

app.py

@@ -47,8 +47,12 @@ graph.add_edge("tool", END)
 graph.set_entry_point("agent")
 
 def router(state):
-    calls = state['messages'][-1].additional_kwargs.get("tool_calls", [])
-    return "multiply" if calls else "end"
+    tool_calls = state['messages'][-1].additional_kwargs.get("tool_calls", [])
+    if len(tool_calls):  # Check if tool_calls exist
+        return "multiply"  # Route to the tool node
+    else:
+        return "end"  # Route to the end node
+
 
 graph.add_conditional_edges("agent", router, {"multiply": "tool", "end": END})
 app_graph = graph.compile()