diff --git a/src/agents/openai-transport-stream.test.ts b/src/agents/openai-transport-stream.test.ts index bc25eaf614e..42e38dda375 100644 --- a/src/agents/openai-transport-stream.test.ts +++ b/src/agents/openai-transport-stream.test.ts @@ -348,7 +348,6 @@ describe("openai transport stream", () => { }); it("passes provider request timeouts to OpenAI SDK clients", () => { - const context = { systemPrompt: "system", messages: [], tools: [] } as never; const requestTimeoutMs = 900_000; const responsesModel = { @@ -376,24 +375,9 @@ describe("openai transport stream", () => { reasoning: false, } satisfies Model<"openai-completions"> & { requestTimeoutMs: number }; - expect( - ( - __testing.createOpenAIResponsesClient(responsesModel, context, "test-key") as { - timeout: number; - } - ).timeout, - ).toBe(requestTimeoutMs); - expect( - (__testing.createAzureOpenAIClient(azureModel, context, "test-key") as { timeout: number }) - .timeout, - ).toBe(requestTimeoutMs); - expect( - ( - __testing.createOpenAICompletionsClient(completionsModel, context, "test-key") as { - timeout: number; - } - ).timeout, - ).toBe(requestTimeoutMs); + expect(__testing.buildOpenAISdkClientOptions(responsesModel).timeout).toBe(requestTimeoutMs); + expect(__testing.buildOpenAISdkClientOptions(azureModel).timeout).toBe(requestTimeoutMs); + expect(__testing.buildOpenAISdkClientOptions(completionsModel).timeout).toBe(requestTimeoutMs); }); it("passes provider request timeouts to OpenAI SDK per-request options", () => { diff --git a/src/agents/openai-transport-stream.ts b/src/agents/openai-transport-stream.ts index c966ca3c1ff..39f090ccd81 100644 --- a/src/agents/openai-transport-stream.ts +++ b/src/agents/openai-transport-stream.ts @@ -1869,6 +1869,7 @@ function mapStopReason(reason: string | null) { } export const __testing = { + buildOpenAISdkClientOptions, buildOpenAISdkRequestOptions, createAzureOpenAIClient, createOpenAICompletionsClient,