Update app.py
Browse files
app.py
CHANGED
@@ -49,10 +49,14 @@ def invoke_tool(state):
|
|
49 |
arguments = json.loads(tool_call["function"]["arguments"])
|
50 |
result = multiply.invoke(arguments)
|
51 |
tool_results.append(
|
52 |
-
AIMessage(
|
|
|
|
|
|
|
53 |
)
|
54 |
return {"messages": tool_results or [AIMessage(content="No valid tool input provided.")]}
|
55 |
|
|
|
56 |
graph.add_node("tool", invoke_tool)
|
57 |
|
58 |
# Explicit Router Node
|
|
|
49 |
arguments = json.loads(tool_call["function"]["arguments"])
|
50 |
result = multiply.invoke(arguments)
|
51 |
tool_results.append(
|
52 |
+
AIMessage(
|
53 |
+
content=f"The result of multiplying {arguments['first_number']} and {arguments['second_number']} is {result}.",
|
54 |
+
additional_kwargs={"tool_call_id": tool_call["id"]}
|
55 |
+
)
|
56 |
)
|
57 |
return {"messages": tool_results or [AIMessage(content="No valid tool input provided.")]}
|
58 |
|
59 |
+
|
60 |
graph.add_node("tool", invoke_tool)
|
61 |
|
62 |
# Explicit Router Node
|