diff --git a/run_agent.py b/run_agent.py index 59a547f0..c32d92d7 100644 --- a/run_agent.py +++ b/run_agent.py @@ -2567,7 +2567,8 @@ class AIAgent: tool_names.append(fn.get("name", "unknown")) msg["content"] = f"Calling the {', '.join(tool_names)} tool{'s' if len(tool_names) > 1 else ''}..." break - final_response = fallback + # Strip blocks from fallback content for user display + final_response = self._strip_think_blocks(fallback).strip() break # No fallback -- append the empty message as-is @@ -2596,6 +2597,9 @@ class AIAgent: if hasattr(self, '_empty_content_retries'): self._empty_content_retries = 0 + # Strip blocks from user-facing response (keep raw in messages for trajectory) + final_response = self._strip_think_blocks(final_response).strip() + final_msg = self._build_assistant_message(assistant_message, finish_reason) messages.append(final_msg)