Update app.py

app.py

@@ -52,11 +52,14 @@ def invoke_tool(state):
     tool_calls = state['messages'][-1].additional_kwargs.get("tool_calls", [])
     for tool_call in tool_calls:
         if tool_call["function"]["name"] == "multiply":
+            # Extract and parse the arguments
             arguments = json.loads(tool_call["function"]["arguments"])
             result = multiply.invoke(arguments)
-            return {"messages": [ToolMessage(content=str(result))]}
+            # Return the tool result while maintaining tool_call_id for future use
+            return {"messages": [ToolMessage(content=str(result), name="multiply", tool_call_id=tool_call["id"])]}
     return {"messages": [ToolMessage(content="No valid tool input provided.")]}
 
+
 graph.add_node("tool", invoke_tool)
 
 # Router Node: Manual Addition