DrishtiSharma commited on
Commit
bced22e
·
verified ·
1 Parent(s): 569f7b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
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(content=f"Tool Result: {result}", additional_kwargs={"tool_call_id": tool_call["id"]})
 
 
 
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