Commit 01d0a70
authored
feat(core): Migrate Vercel AI event processor to span streaming (#20608)
Migrates the Vercel AI event processor so it also works in the span
streaming path via the `processSpan` hook. The event processor currently
serves three purposes:
**Attribute renaming.** The Vercel AI SDK emits attributes under `ai.*`
names that need to be renamed to OpenTelemetry semantic conventions
(`gen_ai.*`) and the `vercel.ai.*` namespace. This logic is
straightforward to port. It's extracted into a shared
`processVercelAiSpanAttributes` helper that is now called from both the
legacy event processor path (for transactions) and the new `processSpan`
hook (for streamed spans).
**Token accumulation on parent spans.** The event processor aggregates
token usage from child spans onto their parent `invoke_agent` spans.
This is a cross-span operation that fundamentally doesn't work in the
streaming model where spans are processed individually. The [span
streaming implementation
guide](https://develop.sentry.dev/sdk/telemetry/spans/implementation/)
explicitly lists this as a case that cannot be replaced. Since the plan
is for parent-level token accumulation to go away regardless, we simply
drop it for the streaming path.
**Tool descriptions on execute_tool spans.** The event processor
iterates over all spans in a transaction to find
`gen_ai.request.available_tools` on `doGenerate` spans and applies the
matching description to sibling `execute_tool` spans. This cross-span
iteration doesn't work when spans are processed individually. Instead,
we parse and store tool descriptions in a map at `spanStart` time of the
`doGenerate` span. Since `execute_tool` spans are siblings of
`doGenerate` (both children of `invoke_agent`), we key the map by the
parent span ID so `execute_tool` spans can look up descriptions by their
own `parent_span_id`. This assumes a flat sibling hierarchy, which holds
for our test scenarios. If we encounter more complex cases down the
road, I think it's fine to address in a follow-up.
Closes #203771 parent 12cd3e5 commit 01d0a70
17 files changed
Lines changed: 1095 additions & 64 deletions
File tree
- dev-packages/node-integration-tests/suites/tracing/vercelai
- span-streaming-v4
- span-streaming-v6
- packages/core/src/tracing/vercel-ai
Lines changed: 2 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
35 | 35 | | |
36 | 36 | | |
37 | 37 | | |
| 38 | + | |
| 39 | + | |
38 | 40 | | |
39 | 41 | | |
40 | 42 | | |
Lines changed: 2 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
75 | 75 | | |
76 | 76 | | |
77 | 77 | | |
| 78 | + | |
| 79 | + | |
78 | 80 | | |
79 | 81 | | |
80 | 82 | | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
8 | 8 | | |
9 | 9 | | |
10 | 10 | | |
| 11 | + | |
11 | 12 | | |
Lines changed: 11 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
| 1 | + | |
| 2 | + | |
| 3 | + | |
| 4 | + | |
| 5 | + | |
| 6 | + | |
| 7 | + | |
| 8 | + | |
| 9 | + | |
| 10 | + | |
| 11 | + | |
Lines changed: 46 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
| 1 | + | |
| 2 | + | |
| 3 | + | |
| 4 | + | |
| 5 | + | |
| 6 | + | |
| 7 | + | |
| 8 | + | |
| 9 | + | |
| 10 | + | |
| 11 | + | |
| 12 | + | |
| 13 | + | |
| 14 | + | |
| 15 | + | |
| 16 | + | |
| 17 | + | |
| 18 | + | |
| 19 | + | |
| 20 | + | |
| 21 | + | |
| 22 | + | |
| 23 | + | |
| 24 | + | |
| 25 | + | |
| 26 | + | |
| 27 | + | |
| 28 | + | |
| 29 | + | |
| 30 | + | |
| 31 | + | |
| 32 | + | |
| 33 | + | |
| 34 | + | |
| 35 | + | |
| 36 | + | |
| 37 | + | |
| 38 | + | |
| 39 | + | |
| 40 | + | |
| 41 | + | |
| 42 | + | |
| 43 | + | |
| 44 | + | |
| 45 | + | |
| 46 | + | |
Lines changed: 82 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
| 1 | + | |
| 2 | + | |
| 3 | + | |
| 4 | + | |
| 5 | + | |
| 6 | + | |
| 7 | + | |
| 8 | + | |
| 9 | + | |
| 10 | + | |
| 11 | + | |
| 12 | + | |
| 13 | + | |
| 14 | + | |
| 15 | + | |
| 16 | + | |
| 17 | + | |
| 18 | + | |
| 19 | + | |
| 20 | + | |
| 21 | + | |
| 22 | + | |
| 23 | + | |
| 24 | + | |
| 25 | + | |
| 26 | + | |
| 27 | + | |
| 28 | + | |
| 29 | + | |
| 30 | + | |
| 31 | + | |
| 32 | + | |
| 33 | + | |
| 34 | + | |
| 35 | + | |
| 36 | + | |
| 37 | + | |
| 38 | + | |
| 39 | + | |
| 40 | + | |
| 41 | + | |
| 42 | + | |
| 43 | + | |
| 44 | + | |
| 45 | + | |
| 46 | + | |
| 47 | + | |
| 48 | + | |
| 49 | + | |
| 50 | + | |
| 51 | + | |
| 52 | + | |
| 53 | + | |
| 54 | + | |
| 55 | + | |
| 56 | + | |
| 57 | + | |
| 58 | + | |
| 59 | + | |
| 60 | + | |
| 61 | + | |
| 62 | + | |
| 63 | + | |
| 64 | + | |
| 65 | + | |
| 66 | + | |
| 67 | + | |
| 68 | + | |
| 69 | + | |
| 70 | + | |
| 71 | + | |
| 72 | + | |
| 73 | + | |
| 74 | + | |
| 75 | + | |
| 76 | + | |
| 77 | + | |
| 78 | + | |
| 79 | + | |
| 80 | + | |
| 81 | + | |
| 82 | + | |
0 commit comments