fix(core): backfill OpenAI responses model for streaming responses

Author: meronogbai

packages/core/src/tracing/openai/streaming.ts

@@ -1,11 +1,14 @@
 import { captureException } from '../../exports';
 import { SPAN_STATUS_ERROR } from '../../tracing';
 import type { Span } from '../../types-hoist/span';
+import { updateSpanName } from '../../utils/spanUtils';
 import {
+  GEN_AI_REQUEST_MODEL_ATTRIBUTE,
   GEN_AI_RESPONSE_FINISH_REASONS_ATTRIBUTE,
   GEN_AI_RESPONSE_STREAMING_ATTRIBUTE,
   GEN_AI_RESPONSE_TEXT_ATTRIBUTE,
   GEN_AI_RESPONSE_TOOL_CALLS_ATTRIBUTE,
+  OPENAI_OPERATIONS,
 } from '../ai/gen-ai-attributes';
 import { RESPONSE_EVENT_TYPES } from './constants';
 import type {
@@ -246,6 +249,12 @@ export async function* instrumentStream<T>(
     }
   } finally {
     setCommonResponseAttributes(span, state.responseId, state.responseModel, state.responseTimestamp);
+    if (state.responseModel) {
+      span.setAttributes({
+        [GEN_AI_REQUEST_MODEL_ATTRIBUTE]: state.responseModel,
+      });
+      updateSpanName(span, `${OPENAI_OPERATIONS.CHAT} ${state.responseModel}`);
+    }
     setTokenUsageAttributes(span, state.promptTokens, state.completionTokens, state.totalTokens);
 
     span.setAttributes({

packages/core/test/lib/utils/openai-streaming.test.ts

@@ -0,0 +1,46 @@
+import { describe, expect, it, vi } from 'vitest';
+import { instrumentStream } from '../../../src/tracing/openai/streaming';
+import type { ResponseStreamingEvent } from '../../../src/tracing/openai/types';
+
+async function collectStream<T>(stream: AsyncIterable<T>): Promise<T[]> {
+  const events: T[] = [];
+  for await (const event of stream) {
+    events.push(event);
+  }
+  return events;
+}
+
+describe('openai-streaming', () => {
+  it('should backfill the request model and span name for streamed Responses API events', async () => {
+    async function* createStream(): AsyncGenerator<ResponseStreamingEvent> {
+      yield {
+        type: 'response.completed',
+        response: {
+          object: 'response',
+          id: 'resp_123',
+          model: 'gpt-4.1-mini',
+          created_at: 1704067200,
+          status: 'completed',
+        },
+      } as ResponseStreamingEvent;
+    }
+
+    const span = {
+      setAttributes: vi.fn(),
+      setStatus: vi.fn(),
+      updateName: vi.fn(),
+      end: vi.fn(),
+    };
+
+    const events = await collectStream(
+      instrumentStream(createStream(), span as unknown as Parameters<typeof instrumentStream>[1], false),
+    );
+
+    expect(events).toHaveLength(1);
+    expect(span.setAttributes).toHaveBeenCalledWith({
+      'gen_ai.request.model': 'gpt-4.1-mini',
+    });
+    expect(span.updateName).toHaveBeenCalledWith('chat gpt-4.1-mini');
+    expect(span.end).toHaveBeenCalled();
+  });
+});