Update agent.py

agent.py

@@ -192,16 +192,10 @@ model_with_tools = model.bind_tools(tools)
 
 def assistant(state: AgentState):
     response = model_with_tools.invoke(state["messages"])
-    print("~" * 50)
+    print("--- Thinking Section ---")
     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 ---")
-            print(message.tool_calls)
-        if message.type == "tool":
-            print(f"--- tool response ({message.name}) ---")
-            print(message.content[:200])
+    print("--- End of Thinking ---")
+    print(response)
     return {
         "messages": [response],
         "file_path": state["file_path"],