Skip to content

Commit 5b710e9

Browse files
committed
fix
1 parent 458de55 commit 5b710e9

File tree

1 file changed

+21
-4
lines changed

1 file changed

+21
-4
lines changed

packages/agents-openai/test/openaiResponsesModel.test.ts

Lines changed: 21 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ import {
77
withTrace,
88
type ResponseStreamEvent,
99
} from '@openai/agents-core';
10+
import type { ResponseStreamEvent as OpenAIResponseStreamEvent } from 'openai/resources/responses/responses';
1011

1112
describe('OpenAIResponsesModel', () => {
1213
beforeAll(() => {
@@ -412,11 +413,20 @@ describe('OpenAIResponsesModel', () => {
412413
it('getStreamedResponse yields events and calls client with stream flag', async () => {
413414
await withTrace('test', async () => {
414415
const fakeResponse = { id: 'res2', usage: {}, output: [] };
415-
const events: ResponseStreamEvent[] = [
416-
{ type: 'response.created', response: fakeResponse as any },
416+
const events: OpenAIResponseStreamEvent[] = [
417+
{
418+
type: 'response.created',
419+
response: fakeResponse as any,
420+
sequence_number: 0,
421+
},
417422
{
418423
type: 'response.output_text.delta',
424+
content_index: 0,
419425
delta: 'delta',
426+
item_id: 'item-1',
427+
logprobs: [],
428+
output_index: 0,
429+
sequence_number: 1,
420430
} as any,
421431
];
422432
async function* fakeStream() {
@@ -462,6 +472,11 @@ describe('OpenAIResponsesModel', () => {
462472
type: 'output_text_delta',
463473
delta: 'delta',
464474
providerData: {
475+
content_index: 0,
476+
item_id: 'item-1',
477+
logprobs: [],
478+
output_index: 0,
479+
sequence_number: 1,
465480
type: 'response.output_text.delta',
466481
},
467482
},
@@ -475,11 +490,12 @@ describe('OpenAIResponsesModel', () => {
475490

476491
it('getStreamedResponse maps streamed usage data onto response_done events', async () => {
477492
await withTrace('test', async () => {
478-
const createdEvent: ResponseStreamEvent = {
493+
const createdEvent: OpenAIResponseStreamEvent = {
479494
type: 'response.created',
480495
response: { id: 'res-stream-init' } as any,
496+
sequence_number: 0,
481497
};
482-
const completedEvent: ResponseStreamEvent = {
498+
const completedEvent: OpenAIResponseStreamEvent = {
483499
type: 'response.completed',
484500
response: {
485501
id: 'res-stream',
@@ -492,6 +508,7 @@ describe('OpenAIResponsesModel', () => {
492508
output_tokens_details: { reasoning_tokens: 3 },
493509
},
494510
},
511+
sequence_number: 1,
495512
} as any;
496513
async function* fakeStream() {
497514
yield createdEvent;

0 commit comments

Comments
 (0)