Update agent.py

agent.py

@@ -192,17 +192,16 @@ model_with_tools = model.bind_tools(tools)
 
 def assistant(state: AgentState):
     response = model_with_tools.invoke(state["messages"])
+    print("~" * 50)
+    print(response.content[0].get("thinking"))
+    print("~" * 50)
     for message in response["messages"]:
         if hasattr(message, "tool_calls") and message.tool_calls:
             print("--- Agent tool call ---")
-            for call in message.tool_calls:
-                print(f"tool name: {call['name']}")
-                print(f"parameter: {call['args']}")
+            print(message.tool_calls)
         if message.type == "tool":
             print(f"--- tool response ({message.name}) ---")
             print(message.content[:200])
-    print(response.content[0].get("thinking"))
-    print("-" * 50)
     return {
         "messages": [response],
         "file_path": state["file_path"],