albertCHY commited on
Commit
4df88d3
·
verified ·
1 Parent(s): d7149b1

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +8 -3
agent.py CHANGED
@@ -181,9 +181,14 @@ model_with_tools = model.bind_tools(tools)
181
 
182
  def assistant(state: AgentState):
183
  response = model_with_tools.invoke(state["messages"])
184
- print("--- Thinking Section ---")
185
- print(response.content[0].get("thinking").strip())
186
- print("--- End of Thinking ---\n\n")
 
 
 
 
 
187
  return {
188
  "messages": [response],
189
  # "file_path": state["file_path"],
 
181
 
182
  def assistant(state: AgentState):
183
  response = model_with_tools.invoke(state["messages"])
184
+ print("\n===== ASSISTANT RESPONSE =====")
185
+ print("TYPE:", type(response))
186
+ print("CONTENT:", response.content)
187
+ print("TOOL CALLS:", response.tool_calls)
188
+ print("===============================\n")
189
+ # print("--- Thinking Section ---")
190
+ # print(response.content[0].get("thinking").strip())
191
+ # print("--- End of Thinking ---\n\n")
192
  return {
193
  "messages": [response],
194
  # "file_path": state["file_path"],