Commit graph

1 commit

Author SHA1 Message Date
Abhay Singh
0cbc0e8e54
fix(proxy/openai): replay incremental events in buffered Responses SSE (#2410) (#2415)
## Description

Fixes #2410.

When a streaming `/v1/responses` request has `headroom_retrieve`
available, Headroom forces a non-streaming (`stream:false`) upstream
call so CCR retrieval can be resolved server-side, then reconstructs the
complete response as SSE for the client. GitHub Copilot returns 200 with
real output tokens, but OpenCode shows no assistant response.

Root cause: `_openai_responses_to_sse` emitted only two events —
`response.created` and `response.completed`:

```python
created_response = {**response, "status": "in_progress", "output": []}
events = [("response.created", created_response), ("response.completed", response)]
```

Clients that read the whole answer off the terminal `response.completed`
event work, but OpenCode and the Vercel AI SDK render output from the
**incremental** item/text events (`response.output_item.added`,
`response.output_text.delta`, ...). With those absent, the SDK displays
nothing.

## Fix

Reconstruct the real Responses event sequence:

```
response.created            (status in_progress, empty output)
response.in_progress
for each output item:
    response.output_item.added        (message items start with empty content)
    for each message content part:
        response.content_part.added   (text blanked)
        response.output_text.delta    (the text)
        response.output_text.done
        response.content_part.done
    response.output_item.done         (full item)
response.completed          (full response)
data: [DONE]
```

Non-message items (reasoning, function_call, ...) get
`output_item.added` + `output_item.done` with the full item. Every event
carries a contiguous `sequence_number`. The terminal
`response.completed` still carries the full response, so clients that
key off it are unaffected; clients that stream now receive the deltas
they need.

## Type of Change

- [x] Bug fix (non-breaking change that fixes an issue)
- [ ] New feature (non-breaking change that adds functionality)
- [ ] Breaking change (fix or feature that would cause existing
functionality to change)
- [ ] Documentation update
- [ ] Performance improvement
- [ ] Code refactoring (no functional changes)

## Changes Made

- `headroom/proxy/handlers/openai.py`: rewrite
`_openai_responses_to_sse` to replay the incremental
output-item/content-part/output-text events between
`response.created`/`response.in_progress` and `response.completed`.
- `tests/test_openai_responses_buffered_sse.py`: new test asserting the
incremental `output_text.delta` (visible text), the per-item sequence
for message vs non-message items, the empty-output case, and contiguous
sequence numbers.

## Testing

- [x] Unit tests pass (`pytest`)
- [x] Linting passes (`ruff check .`)
- [x] Type checking passes (`mypy headroom`)
- [x] New tests added for new functionality
- [ ] Manual testing performed

### Test Output

```text
$ uvx ruff@0.15.17 check headroom/proxy/handlers/openai.py tests/test_openai_responses_buffered_sse.py
All checks passed!
$ uvx mypy@1.20.2 --ignore-missing-imports headroom/proxy/handlers/openai.py
# no errors in the changed file
# _openai_responses_to_sse is a pure module-level function, so I ran the new
# tests against the real code in the project venv (uv sync): 3 passed.
```

## Real Behavior Proof

- Environment: Windows 11, Python 3.12, project venv (`uv sync --extra
proxy`), `uvx ruff@0.15.17` / `uvx mypy@1.20.2`.
- Exact command / steps: fed the real `_openai_responses_to_sse` a
completed response with a reasoning item and a message item whose
content is `output_text: "Hello world"`, plus an empty-output response
and a function_call-only response.
- Observed result: the stream now contains `response.output_text.delta`
with `"Hello world"` at `output_index=1, content_index=0`, wrapped by
`content_part.added/done` and `output_item.added/done`, with the
reasoning and function_call items emitted as `output_item.added/done`
and preserved whole; `response.created`/`in_progress` carry empty output
while `response.completed` carries the full output; sequence numbers are
`0..n`. Ran against the actual module.
- Not tested: a live OpenCode -> Copilot Responses round trip; the added
tests assert the event stream directly.

## Review Readiness

- [x] I have performed a self-review
- [x] This PR is ready for human review

## Checklist

- [x] My code follows the project's style guidelines
- [x] I have performed a self-review of my code
- [x] I have commented my code, particularly in hard-to-understand areas
- [ ] I have made corresponding changes to the documentation
- [x] My changes generate no new warnings
- [x] I have added tests that prove my fix is effective or that my
feature works
- [x] New and existing unit tests pass locally with my changes
- [ ] I have updated the CHANGELOG.md if applicable

## Additional Notes

`_openai_responses_to_sse` is a pure function, so I verified the fix
against the real code in the venv (output above) in addition to the unit
tests. This mirrors the incremental replay the Anthropic buffered path
already does in `StreamingMixin._response_to_sse`
(content_block_start/delta/stop), bringing the Responses buffered-CCR
path to the same fidelity.
2026-07-19 22:18:47 -07:00