Share stream manager wrappers between instrumentations

[lmolkova]

The openai and anthropic instrumentations each had their own copy of the stream-manager lifecycle, a close-finalizing proxy for stream.response, and the stream/iterator rebind. This moves them into opentelemetry.util.genai.stream as SyncStreamManagerWrapper / AsyncStreamManagerWrapper and finalize_on_close / finalize_on_aclose, and onboards both packages.

Also fixes a span leak for streams exposing aclose instead of close — Google's async generate_content_stream emitted no span when closed before being drained.

Follow-ups: the duplicated with_raw_response proxy moves to the util next, then generator-backed stream support for #386.

[Copilot]

Pull request overview

Centralizes stream lifecycle handling in opentelemetry-util-genai and adopts it across OpenAI and Anthropic instrumentations.

Changes:

• Adds shared stream-manager and close-finalization wrappers.
• Migrates OpenAI and Anthropic implementations.
• Supports successful finalization of aclose-only Google streams.

Reviewed changes

Copilot reviewed 10 out of 10 changed files in this pull request and generated 3 comments.

Show a summary per file

File	Description
util/opentelemetry-util-genai/src/opentelemetry/util/genai/stream.py	Adds shared wrappers and aclose support.
util/opentelemetry-util-genai/tests/test_stream.py	Tests shared lifecycle behavior.
util/opentelemetry-util-genai/.changelog/390.added	Records new public helpers.
util/opentelemetry-util-genai/.changelog/390.changed	Records aclose finalization.
instrumentation/opentelemetry-instrumentation-google-genai/tests/generate_content/test_async_streaming.py	Tests early async stream closure.
instrumentation/opentelemetry-instrumentation-google-genai/.changelog/390.fixed	Documents the span leak fix.
instrumentation/opentelemetry-instrumentation-genai-openai/src/opentelemetry/instrumentation/genai/openai/response_wrappers.py	Adopts shared wrappers.
instrumentation/opentelemetry-instrumentation-genai-openai/tests/test_response_wrappers.py	Updates OpenAI wrapper tests.
instrumentation/opentelemetry-instrumentation-genai-anthropic/src/opentelemetry/instrumentation/genai/anthropic/wrappers.py	Adopts shared wrappers.
instrumentation/opentelemetry-instrumentation-genai-anthropic/tests/test_async_wrappers.py	Updates Anthropic wrapper tests.

Suppressed comments (1)

util/opentelemetry-util-genai/src/opentelemetry/util/genai/stream.py:536

• After manager.__aexit__ succeeds, the SDK manager has already awaited the stream's close (as the OpenAI and Anthropic async managers do). Calling stream_wrapper.__aexit__ then closes it a second time, which changes the SDK lifecycle and may fail for non-idempotent streams. Finalize telemetry directly here instead.

            if suppressed:
                # See SyncStreamManagerWrapper.__exit__.
                await stream_wrapper.__aexit__(None, None, None)
            else:
                await stream_wrapper.__aexit__(exc_type, exc_val, exc_tb)

---

💡 Add a code-review agent skill or configure MCP servers for context-aware, tailored reviews. Learn more in the docs.

[lmolkova]

moved to utils

[opentelemetry-pr-dashboard]

Pull request dashboard status

Waiting on reviewers · refreshed 2026-08-14 22:13 UTC

Review the latest changes.

Status above doesn't look right?

• Just replied or pushed? Anything around or after the refresh time above may not be picked up yet — give it a few minutes.
• Anything look wrong? Report it with what you expected; it helps us improve the dashboard.