test(node): Fix flaky Vercel AI integration test

Fixes #20997

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');