The _failure_callback at line 240 uses .get() instead of .pop() to retrieve the span, and doesn't check for streaming. If a streaming response fails mid-stream, after _success_callback has been called once (but not completed), a subsequent _failure_callback could attempt to exit the same span. While the fix in _success_callback is correct, the same pattern should be applied to _failure_callback for consistency and to prevent potential double exits when failures occur during streaming.

The test_streaming_chat_completion test is parameterized with send_default_pii and include_prompts parameters (lines 225-229) that control whether prompts and PII data are captured, but the assertions at lines 290-291 only check GEN_AI_RESPONSE_STREAMING. Unlike the non-streaming test which verifies GEN_AI_REQUEST_MESSAGES and GEN_AI_RESPONSE_TEXT based on these parameters, the streaming test doesn't assert on this behavior, reducing the test's effectiveness.