albertCHY commited on
Commit
1edc197
·
verified ·
1 Parent(s): 6de36de

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -3
agent.py CHANGED
@@ -198,14 +198,14 @@ model_with_tools = model.bind_tools(tools)
198
 
199
  def assistant(state: AgentState):
200
  response = model_with_tools.invoke(state["messages"])
 
 
 
201
  print("\n===== ASSISTANT RESPONSE =====")
202
  print("TYPE:", type(response))
203
  print("CONTENT:", response.content)
204
  print("TOOL CALLS:", response.tool_calls)
205
  print("===============================\n")
206
- # print("--- Thinking Section ---")
207
- # print(response.content[0].get("thinking").strip())
208
- # print("--- End of Thinking ---\n\n")
209
  return {
210
  "messages": [response],
211
  # "file_path": state["file_path"],
 
198
 
199
  def assistant(state: AgentState):
200
  response = model_with_tools.invoke(state["messages"])
201
+ print("\n===== Thinking Section =====")
202
+ print(response.content[0].get("thinking").strip())
203
+ print("===============================\n")
204
  print("\n===== ASSISTANT RESPONSE =====")
205
  print("TYPE:", type(response))
206
  print("CONTENT:", response.content)
207
  print("TOOL CALLS:", response.tool_calls)
208
  print("===============================\n")
 
 
 
209
  return {
210
  "messages": [response],
211
  # "file_path": state["file_path"],