strip
Browse files
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].
|
| 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]}
|