APEX/BrowserUse_and_ComputerUse_skills
5
0
Fork 0
Code Issues 6 Pull requests 1 Projects 1 Releases Packages Wiki Activity Actions

Merge pull request #174 from Bartok9/fix-think-block-leakage

Browse source 
fix: strip <think> blocks from final response to users
This commit is contained in:
Teknium 2026-02-28 16:43:47 -08:00 • committed by GitHub
commit 31a5cd185a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
run_agent.py
View file

@ -2569,7 +2569,8 @@ class AIAgent:
tool_names.append(fn.get("name", "unknown")) tool_names.append(fn.get("name", "unknown"))
msg["content"] = f"Calling the {', '.join(tool_names)} tool{'s' if len(tool_names) > 1 else ''}..." msg["content"] = f"Calling the {', '.join(tool_names)} tool{'s' if len(tool_names) > 1 else ''}..."
break break
final_response = fallback # Strip <think> blocks from fallback content for user display
final_response = self._strip_think_blocks(fallback).strip()
break break
# No fallback -- append the empty message as-is # No fallback -- append the empty message as-is
@ -2598,6 +2599,9 @@ class AIAgent:
if hasattr(self, '_empty_content_retries'): if hasattr(self, '_empty_content_retries'):
self._empty_content_retries = 0 self._empty_content_retries = 0
# Strip <think> 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) final_msg = self._build_assistant_message(assistant_message, finish_reason)
messages.append(final_msg) messages.append(final_msg)