fix(litellm): Avoid double span exits when streaming

[alexander-alderman-webb]

Description

Only exit the span on the final invocation of _success_callback when litellm streams a response. Remove the shared span reference in the metadata dictionary immediately before the span is finished, to avoid race conditions between concurrent _success_callback invocations.

The litellm.success_callback callbacks are fired multiple times when streaming a response with litellm.

Avoid an unhandled SDK exception by using litellm like a user would in the relevant test.

Issues

Reminders

• Please add tests to validate your changes, and lint your code using tox -e linters.
• Add GH Issue ID & Linear ID (if applicable)
• PR title should use conventional commit style (feat:, fix:, ref:, meta:)
• For external contributors: CONTRIBUTING.md, Sentry SDK development docs, Discord community

[github-actions]

Semver Impact of This PR

🟢 Patch (bug fixes)

📋 Changelog Preview

This is how your changes will appear in the changelog.
Entries from this PR are highlighted with a left border (blockquote style).

---

New Features ✨

• (ai) Redact base64 data URLs in image_url content blocks by ericapisani in #5953
• (integrations) Instrument pyreqwest tracing by servusdei2018 in #5682

Bug Fixes 🐛

Anthropic

• Capture exceptions for stream() calls by alexander-alderman-webb in #5950
• Stop setting transaction status when child span fails by alexander-alderman-webb in #5717
• Only finish relevant spans in .create() patches by alexander-alderman-webb in #5716

Pydantic Ai

• Adapt import for new library versions by alexander-alderman-webb in #5984
• Use first-class hooks when available by alexander-alderman-webb in #5947

Other

• (huggingface_hub) Stop setting transaction status when a child span fails by Zenithatic in #5952

• (litellm) Avoid double span exits when streaming by alexander-alderman-webb in #5933

• (wsgi) Respect HTTP_X_FORWARDED_PROTO in request.url construction by sl0thentr0py in #5963

Internal Changes 🔧

• (ai) Remove gen_ai.tool.type span attribute by ericapisani in #5964
• (anthropic) Separate sync and async .create() patches by alexander-alderman-webb in #5715
• (litellm) Remove dead attributes by alexander-alderman-webb in #5985
• (openai) Split token counting by API for easier deprecation by ericapisani in #5930
• (openai-agents) Remove error attributes by alexander-alderman-webb in #5986
• (opentelemetry) Ignore mypy error by alexander-alderman-webb in #5927
• 🤖 Update test matrix with new releases (04/13) by github-actions in #5983
• Fix license metadata in setup.py by sl0thentr0py in #5934
• Update validate-pr workflow by stephanie-anderson in #5931

Other

• Handle None span context in the span processor and pin tokenizers version for anthropic tests on Python 3.8 by alexander-alderman-webb in #5967

---

_{🤖 This preview updates automatically when you update the PR.}

[github-actions]

Codecov Results 📊

✅ 2090 passed | ⏭️ 169 skipped | Total: 2259 | Pass Rate: 92.52% | Execution Time: 4m 4s

All tests are passing successfully.

❌ Patch coverage is 0.00%. Project has 12422 uncovered lines.

Files with missing lines (1)

File	Patch %	Lines
litellm.py	0.00%	⚠️ 139 Missing

---

Generated by Codecov Action