Paperbag commited on
Commit
5f0c38f
·
1 Parent(s): f02dcff
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -134,7 +134,7 @@ def answer_message(state: AgentState) -> AgentState:
134
 
135
  # final_response = model_with_tools.invoke(messages)
136
  print(f"Final response: {final_response}")
137
- final_response = final_response.content.split('</think>')[1].trim()
138
 
139
  # Append the model's answer to the messages list
140
  return {"messages": [final_response]}
 
134
 
135
  # final_response = model_with_tools.invoke(messages)
136
  print(f"Final response: {final_response}")
137
+ final_response = final_response.content.split('</think>')[1].strip()
138
 
139
  # Append the model's answer to the messages list
140
  return {"messages": [final_response]}