Refactor answer_message function to clean up final response handling by removing redundant print statement and trimming the response content for better formatting.

agent.py

@@ -132,10 +132,9 @@ def answer_message(state: AgentState) -> AgentState:
 
     final_response = model_with_tools.invoke(messages)
 
-    # Step 3: Pass results back to model for final response
-    print(f"Messages: {messages}")
     # final_response = model_with_tools.invoke(messages)
     print(f"Final response: {final_response}")
+    final_response.content.split('</think>')[1].trim()
 
     # Append the model's answer to the messages list
     return {"messages": [final_response]}