test(node): Fix flaky Vercel AI integration test (#21076)

The `creates ai related spans with sendDefaultPii: false` test in the
Vercel AI integration suite was looking up `firstGenerateContentSpan`
via `find()` by name + `vercel.ai.operationId` +
`GEN_AI_INPUT_MESSAGES_ATTRIBUTE === undefined`.

With `sendDefaultPii: false`, both the first and the third
`generate_content mock-model-id` spans have undefined input messages
(the first because telemetry isn't explicitly enabled, the third —
the tool-call generation — for the same reason). Depending on the
order spans arrive in the container, `find()` could match the third
span (which has `inputTokens: 15`) and the subsequent
`expect(...INPUT_TOKENS).toBe(10)` would fail.

Tightens the find query to also require
`GEN_AI_USAGE_INPUT_TOKENS_ATTRIBUTE === 10`,
mirroring the disambiguation already used for `firstInvokeAgentSpan`
in the same test.

The same pattern is also present in `v5/test.ts` and `v6/test.ts` —
fixed there as well.

Fixes #20997

Co-authored-by: claude[bot] <41898282+claude[bot]@users.noreply.github.com>
Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com>

Author: github-actions[bot]

dev-packages/node-integration-tests/suites/tracing/vercelai/test.ts

@@ -58,7 +58,8 @@ describe('Vercel AI integration', () => {
               span =>
                 span.name === 'generate_content mock-model-id' &&
                 span.attributes['vercel.ai.operationId'].value === 'ai.generateText.doGenerate' &&
-                span.attributes[GEN_AI_INPUT_MESSAGES_ATTRIBUTE] === undefined,
+                span.attributes[GEN_AI_INPUT_MESSAGES_ATTRIBUTE] === undefined &&
+                span.attributes[GEN_AI_USAGE_INPUT_TOKENS_ATTRIBUTE].value === 10,
             );
             expect(firstGenerateContentSpan).toBeDefined();
             expect(firstGenerateContentSpan!.name).toBe('generate_content mock-model-id');

dev-packages/node-integration-tests/suites/tracing/vercelai/v5/test.ts

@@ -58,7 +58,8 @@ describe('Vercel AI integration (V5)', () => {
                 span =>
                   span.name === 'generate_content mock-model-id' &&
                   span.attributes['vercel.ai.operationId'].value === 'ai.generateText.doGenerate' &&
-                  span.attributes[GEN_AI_INPUT_MESSAGES_ATTRIBUTE] === undefined,
+                  span.attributes[GEN_AI_INPUT_MESSAGES_ATTRIBUTE] === undefined &&
+                  span.attributes[GEN_AI_USAGE_INPUT_TOKENS_ATTRIBUTE].value === 10,
               );
               expect(firstGenerateContentSpan).toBeDefined();
               expect(firstGenerateContentSpan!.name).toBe('generate_content mock-model-id');

dev-packages/node-integration-tests/suites/tracing/vercelai/v6/test.ts

@@ -59,7 +59,8 @@ describe('Vercel AI integration (V6)', () => {
                 span =>
                   span.name === 'generate_content mock-model-id' &&
                   span.attributes['vercel.ai.operationId'].value === 'ai.generateText.doGenerate' &&
-                  span.attributes[GEN_AI_INPUT_MESSAGES_ATTRIBUTE] === undefined,
+                  span.attributes[GEN_AI_INPUT_MESSAGES_ATTRIBUTE] === undefined &&
+                  span.attributes[GEN_AI_USAGE_INPUT_TOKENS_ATTRIBUTE].value === 10,
               );
               expect(firstGenerateContentSpan).toBeDefined();
               expect(firstGenerateContentSpan!.name).toBe('generate_content mock-model-id');