feat(streaming): Support ai_track decorator

[sentrivana]

Description

Make the @ai_track decorator work with span streaming.

sentry_data and sentry_tags will both set attributes in streaming mode, instead.

Issues

Closes #6893

Reminders

• Please add tests to validate your changes, and lint your code using uv run ruff.
• 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

[sentry]

Bug: Using span_kwargs.pop("op") mutates the shared span_kwargs dictionary, causing subsequent calls to the decorated function to lose the custom op and use a default value.
_{Severity: MEDIUM}

Suggested Fix

Instead of mutating span_kwargs with .pop(), use .get() to retrieve the op value. To prevent passing op along with other keyword arguments, create a copy of the dictionary for the start_span call, excluding the op key. For example: op = span_kwargs.get("op", ...) and other_kwargs = {k: v for k, v in span_kwargs.items() if k != 'op'}.

Prompt for AI Agent

Review the code at the location below. A potential bug has been identified by an AI
agent. Verify if this is a real issue. If it is, propose a fix; if not, explain why it's
not valid.

Location: sentry_sdk/ai/monitoring.py#L37

Potential issue: The `span_kwargs` dictionary is created once when the `@ai_track`
decorator is applied and is shared across all calls to the decorated function. The code
uses `span_kwargs.pop("op", ...)` to retrieve the operation name, which mutates the
dictionary by permanently removing the `op` key. Consequently, only the first call to a
decorated function with a custom `op` (e.g., `@ai_track("description", op="custom.op")`)
will use the correct operation. All subsequent calls will find the `op` key missing and
will incorrectly fall back to the default value, such as "ai.run".

_{Did we get this right? 👍 / 👎 to inform future reviews.}

[sentrivana]

Pre-existing issue

[sentry]

Bug: In streaming mode, extra keyword arguments passed to the @ai_track decorator, other than op, are silently ignored instead of being passed to start_span.
_{Severity: MEDIUM}

Suggested Fix

Create a copy of the span_kwargs dictionary, remove the op key, and then pass the remaining arguments to the start_span call in the streaming path. This can be done with other_kwargs = {k: v for k, v in span_kwargs.items() if k != 'op'} and then spreading them into the function call: start_span(..., **other_kwargs).

Prompt for AI Agent

Review the code at the location below. A potential bug has been identified by an AI
agent. Verify if this is a real issue. If it is, propose a fix; if not, explain why it's
not valid.

Location: sentry_sdk/ai/monitoring.py#L45-L48

Potential issue: In the new streaming mode code path for the `@ai_track` decorator, any
additional keyword arguments passed to the decorator (e.g., `origin`, `only_if_parent`)
are ignored. The code extracts the `op` but does not forward the rest of the
`span_kwargs` to the `sentry_sdk.traces.start_span` function. This creates an
inconsistency where the decorator's behavior changes depending on whether streaming is
active, silently dropping user-provided span keyword arguments in streaming mode while
correctly applying them in non-streaming mode.

_{Did we get this right? 👍 / 👎 to inform future reviews.}

[sentrivana]

Correct, intentional

[github-actions]

Codecov Results 📊

✅ 98293 passed | ⏭️ 6614 skipped | Total: 104907 | Pass Rate: 93.7% | Execution Time: 352m 29s

📊 Comparison with Base Branch

Metric	Change
Total Tests	📈 +90
Passed Tests	📈 +92
Failed Tests	📉 -2
Skipped Tests	—

➖ Removed Tests (2)

View removed tests

• test_circular_references_span_streaming
  • File: tests.tracing.test_misc
• test_continuous_profiler_auto_start_and_stop_sampled_span_streaming[non-experiment-thread]
  • File: tests.profiler.test_continuous_profiler

All tests are passing successfully.

❌ Patch coverage is 70.37%. Project has 2520 uncovered lines.
❌ Project coverage is 89.69%. Comparing base (base) to head (head).

Files with missing lines (1)

File	Patch %	Lines
sentry_sdk/ai/monitoring.py	70.37%	⚠️ 24 Missing

Coverage diff

@@            Coverage Diff             @@
##          main       #PR       +/-##
==========================================
- Coverage    89.72%    89.69%    -0.03%
==========================================
  Files          193       193         —
  Lines        24394     24437       +43
  Branches      8644      8660       +16
==========================================
+ Hits         21886     21917       +31
- Misses        2508      2520       +12
- Partials      1400      1400         —

---

Generated by Codecov Action