Python: Skip get_final_response in OTel _finalize_stream when stream errored

[Dineshsuriya D (droideronline)]

Closes#5231

Problem

AgentTelemetryLayer registers _finalize_stream as a cleanup hook on the ResponseStream. Cleanup hooks run on both normal completion and streaming errors. Inside _finalize_stream, get_final_response() was called unconditionally — and get_final_response() always runs all registered result hooks, including _post_hook in _agents.py which calls _run_after_providers.

This caused after_run context providers to fire on error paths, which is incorrect.

The call chain on streaming error:

stream raises Exception
→ ResponseStream.__anext__ runs _run_cleanup_hooks()
→ _finalize_stream() called
→ get_final_response() called unconditionally ← bug
→ result hooks fire (e.g. _post_hook)
→ _run_after_providers called ← wrong on error

Fix

_types.py — ResponseStream

• Store the exception in _stream_error when __anext__ catches one, then clear it after _run_cleanup_hooks() completes to avoid retaining the traceback and any large object graphs it references.
• Expose two public properties: consumed: bool (True only after normal StopAsyncIteration) and stream_error: Exception | None.

observability.py — both _finalize_stream closures (chat and agent layers)

• Key the early-return off result_stream.stream_error is not None — this is precise to actual errors rather than any non-normal completion state, so edge cases where consumed is False but no error is set still proceed to get_final_response().
• Call capture_exception(span, result_stream.stream_error, ...) before returning so the OTel span records the failure rather than closing silently.

Notes

Affects both _trace_chat_client_invocation (chat layer) and _trace_agent_invocation (agent layer) — both had the same pattern and are fixed here.

[Copilot]

Pull request overview

Note

Copilot was unable to run its full agentic suite in this review.

Fixes OTel streaming instrumentation so result hooks (e.g., after_run context providers) don’t fire when a stream errors, by skipping get_final_response() unless the stream completed normally.

Changes:

• Add a _consumed guard in the chat stream _finalize_stream to avoid calling get_final_response() on error paths.
• Add the same guard in the agent stream _finalize_stream.

[Copilot]

Pull request overview

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

[Dineshsuriya D (droideronline)]

Evan Mattson (@moonbox3)Eduard van Valkenburg (@eavanvalkenburg)Tao Chen (@TaoChenOSU) - Would appreciate a look when you get a chance.

[Eduard van Valkenburg (eavanvalkenburg)]

small comment, otherwise looks good and thanks

[Evan Mattson (moonbox3)]

Python Test Coverage Report •

File	Stmts	Miss	Cover	Missing
packages/core/agent_framework
_types.py	1086	91	91%	58, 67–68, 122, 127, 146, 148, 152, 156, 158, 160, 162, 180, 184, 210, 232, 237, 242, 246, 276, 687–688, 847–848, 1233, 1305, 1340, 1360, 1370, 1422, 1554–1556, 1738, 1841–1846, 1871, 1959, 1967–1969, 1974, 2065, 2077, 2100, 2355, 2379, 2474, 2723, 2930, 3003, 3014, 3016–3020, 3022, 3025–3033, 3043, 3113, 3247–3249, 3252–3254, 3258, 3263, 3267, 3351–3353, 3382, 3459–3463
observability.py	755	84	88%	377, 379–380, 383, 386, 389–390, 395–396, 402–403, 409–410, 417, 419–421, 424–426, 431–432, 438–439, 445–446, 453, 610–611, 739, 743–745, 747, 751–752, 756, 794, 796, 807–809, 811–813, 817, 825, 949–950, 1112, 1355–1356, 1454–1459, 1466–1469, 1473–1481, 1488, 1553–1557, 1616–1617, 1751, 1843, 2040, 2258, 2260
TOTAL	27264	3203	88%

Python Unit Test Overview

Tests	Skipped	Failures	Errors	Time
5486	20 💤	0 ❌	0 🔥	1m 24s ⏱️

[Dineshsuriya D (droideronline)]

Evan Mattson (@moonbox3) - seems like a rate limit error in ms learn MCP.