akadmin/headroom
1
0
Fork 0
mirror of https://github.com/headroomlabs-ai/headroom.git synced 2026-08-27 14:17:10 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: replace computer_call_output with apply_patch_call_output in output_shaper (#2250)

Browse source 
The _RESPONSES_TOOL_OUTPUT_TYPES frozenset in output_shaper.py had
computer_call_output instead of apply_patch_call_output, making it
inconsistent with the canonical definitions in handlers/openai.py and
output_turn_policy.py. This caused apply_patch_call_output items to be
misclassified, preventing effort routing optimization for those turns.

---------

Co-authored-by: JerrettDavis <mxjerrett@gmail.com>
This commit is contained in:
Semianchuk Vitalii 2026-07-15 22:33:25 +01:00 • committed by GitHub
parent 81d40a6437
commit 63f74aa3e6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
headroom/proxy/output_shaper.py
View file

@ -364,10 +364,10 @@ _RESPONSES_EFFORT_RANK = {"minimal": 0, "low": 1, "medium": 2, "high": 3, "xhigh
# model — the Responses counterpart of an Anthropic ``tool_result`` block.
_RESPONSES_TOOL_OUTPUT_TYPES = frozenset(
{
"function_call_output",
"custom_tool_call_output",
"function_call_output",
"local_shell_call_output",
"computer_call_output",
"apply_patch_call_output",
}
)

4
tests/test_output_shaper_responses.py
View file

@ -75,6 +75,10 @@ class TestClassifyResponsesTurn:
items = _mechanical_input()[:-1] + [_fn_output(item_type="custom_tool_call_output")]
assert classify_responses_turn(items) == TurnKind.MECHANICAL_CONTINUATION
def test_apply_patch_call_output_is_mechanical(self):
items = _mechanical_input()[:-1] + [_fn_output(item_type="apply_patch_call_output")]
assert classify_responses_turn(items) == TurnKind.MECHANICAL_CONTINUATION
def test_trailing_user_message_is_new_ask(self):
items = _mechanical_input() + [_user_message("also check bar.py")]
assert classify_responses_turn(items) == TurnKind.NEW_USER_ASK