@openai/agents-openai 0.4.15 → 0.5.0

package/dist/defaults.d.ts

@@ -1,14 +1,18 @@
 import { OpenAI } from 'openai';
 export declare const DEFAULT_OPENAI_API = "responses";
 export declare const DEFAULT_OPENAI_MODEL = "gpt-4.1";
+export declare const DEFAULT_OPENAI_RESPONSES_TRANSPORT = "http";
 export declare function setTracingExportApiKey(key: string): void;
 export declare function getTracingExportApiKey(): string | undefined;
 export declare function shouldUseResponsesByDefault(): boolean;
+export declare function shouldUseResponsesWebSocketByDefault(): boolean;
 export declare function setOpenAIAPI(value: 'chat_completions' | 'responses'): void;
+export declare function setOpenAIResponsesTransport(value: 'http' | 'websocket'): void;
 export declare function setDefaultOpenAIClient(client: OpenAI): void;
 export declare function getDefaultOpenAIClient(): OpenAI | undefined;
 export declare function setDefaultOpenAIKey(key: string): void;
 export declare function getDefaultOpenAIKey(): string | undefined;
+export declare function getDefaultOpenAIWebSocketBaseURL(): string | undefined;
 export declare const HEADERS: {
     'User-Agent': string;
 };

package/dist/defaults.js

@@ -3,20 +3,25 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.HEADERS = exports.DEFAULT_OPENAI_MODEL = exports.DEFAULT_OPENAI_API = void 0;
+exports.HEADERS = exports.DEFAULT_OPENAI_RESPONSES_TRANSPORT = exports.DEFAULT_OPENAI_MODEL = exports.DEFAULT_OPENAI_API = void 0;
 exports.setTracingExportApiKey = setTracingExportApiKey;
 exports.getTracingExportApiKey = getTracingExportApiKey;
 exports.shouldUseResponsesByDefault = shouldUseResponsesByDefault;
+exports.shouldUseResponsesWebSocketByDefault = shouldUseResponsesWebSocketByDefault;
 exports.setOpenAIAPI = setOpenAIAPI;
+exports.setOpenAIResponsesTransport = setOpenAIResponsesTransport;
 exports.setDefaultOpenAIClient = setDefaultOpenAIClient;
 exports.getDefaultOpenAIClient = getDefaultOpenAIClient;
 exports.setDefaultOpenAIKey = setDefaultOpenAIKey;
 exports.getDefaultOpenAIKey = getDefaultOpenAIKey;
+exports.getDefaultOpenAIWebSocketBaseURL = getDefaultOpenAIWebSocketBaseURL;
 const _shims_1 = require("@openai/agents-core/_shims");
 const metadata_1 = __importDefault(require("./metadata.js"));
 exports.DEFAULT_OPENAI_API = 'responses';
 exports.DEFAULT_OPENAI_MODEL = 'gpt-4.1';
+exports.DEFAULT_OPENAI_RESPONSES_TRANSPORT = 'http';
 let _defaultOpenAIAPI = exports.DEFAULT_OPENAI_API;
+let _defaultOpenAIResponsesTransport = exports.DEFAULT_OPENAI_RESPONSES_TRANSPORT;
 let _defaultOpenAIClient;
 let _defaultOpenAIKey = undefined;
 let _defaultTracingApiKey = undefined;
@@ -29,9 +34,15 @@ function getTracingExportApiKey() {
 function shouldUseResponsesByDefault() {
     return _defaultOpenAIAPI === 'responses';
 }
+function shouldUseResponsesWebSocketByDefault() {
+    return _defaultOpenAIResponsesTransport === 'websocket';
+}
 function setOpenAIAPI(value) {
     _defaultOpenAIAPI = value;
 }
+function setOpenAIResponsesTransport(value) {
+    _defaultOpenAIResponsesTransport = value;
+}
 function setDefaultOpenAIClient(client) {
     _defaultOpenAIClient = client;
 }
@@ -44,6 +55,9 @@ function setDefaultOpenAIKey(key) {
 function getDefaultOpenAIKey() {
     return _defaultOpenAIKey ?? (0, _shims_1.loadEnv)().OPENAI_API_KEY;
 }
+function getDefaultOpenAIWebSocketBaseURL() {
+    return (0, _shims_1.loadEnv)().OPENAI_WEBSOCKET_BASE_URL;
+}
 exports.HEADERS = {
     'User-Agent': `Agents/JavaScript ${metadata_1.default.version}`,
 };

package/dist/defaults.js.map

@@ -1 +1 @@
-{"version":3,"file":"defaults.js","sourceRoot":"","sources":["../src/defaults.ts"],"names":[],"mappings":";;;;;;AAYA,wDAEC;AAED,wDAEC;AAED,kEAEC;AAED,oCAEC;AAED,wDAEC;AAED,wDAEC;AAED,kDAEC;AAED,kDAEC;AAzCD,uDAAqD;AACrD,6DAAkC;AAErB,QAAA,kBAAkB,GAAG,WAAW,CAAC;AACjC,QAAA,oBAAoB,GAAG,SAAS,CAAC;AAE9C,IAAI,iBAAiB,GAAG,0BAAkB,CAAC;AAC3C,IAAI,oBAAwC,CAAC;AAC7C,IAAI,iBAAiB,GAAuB,SAAS,CAAC;AACtD,IAAI,qBAAqB,GAAuB,SAAS,CAAC;AAE1D,SAAgB,sBAAsB,CAAC,GAAW;IAChD,qBAAqB,GAAG,GAAG,CAAC;AAC9B,CAAC;AAED,SAAgB,sBAAsB;IACpC,OAAO,qBAAqB,IAAI,IAAA,gBAAO,GAAE,CAAC,cAAc,CAAC;AAC3D,CAAC;AAED,SAAgB,2BAA2B;IACzC,OAAO,iBAAiB,KAAK,WAAW,CAAC;AAC3C,CAAC;AAED,SAAgB,YAAY,CAAC,KAAuC;IAClE,iBAAiB,GAAG,KAAK,CAAC;AAC5B,CAAC;AAED,SAAgB,sBAAsB,CAAC,MAAc;IACnD,oBAAoB,GAAG,MAAM,CAAC;AAChC,CAAC;AAED,SAAgB,sBAAsB;IACpC,OAAO,oBAAoB,CAAC;AAC9B,CAAC;AAED,SAAgB,mBAAmB,CAAC,GAAW;IAC7C,iBAAiB,GAAG,GAAG,CAAC;AAC1B,CAAC;AAED,SAAgB,mBAAmB;IACjC,OAAO,iBAAiB,IAAI,IAAA,gBAAO,GAAE,CAAC,cAAc,CAAC;AACvD,CAAC;AAEY,QAAA,OAAO,GAAG;IACrB,YAAY,EAAE,qBAAqB,kBAAQ,CAAC,OAAO,EAAE;CACtD,CAAC"}
+{"version":3,"file":"defaults.js","sourceRoot":"","sources":["../src/defaults.ts"],"names":[],"mappings":";;;;;;AAcA,wDAEC;AAED,wDAEC;AAED,kEAEC;AAED,oFAEC;AAED,oCAEC;AAED,kEAEC;AAED,wDAEC;AAED,wDAEC;AAED,kDAEC;AAED,kDAEC;AAED,4EAEC;AAvDD,uDAAqD;AACrD,6DAAkC;AAErB,QAAA,kBAAkB,GAAG,WAAW,CAAC;AACjC,QAAA,oBAAoB,GAAG,SAAS,CAAC;AACjC,QAAA,kCAAkC,GAAG,MAAM,CAAC;AAEzD,IAAI,iBAAiB,GAAG,0BAAkB,CAAC;AAC3C,IAAI,gCAAgC,GAAG,0CAAkC,CAAC;AAC1E,IAAI,oBAAwC,CAAC;AAC7C,IAAI,iBAAiB,GAAuB,SAAS,CAAC;AACtD,IAAI,qBAAqB,GAAuB,SAAS,CAAC;AAE1D,SAAgB,sBAAsB,CAAC,GAAW;IAChD,qBAAqB,GAAG,GAAG,CAAC;AAC9B,CAAC;AAED,SAAgB,sBAAsB;IACpC,OAAO,qBAAqB,IAAI,IAAA,gBAAO,GAAE,CAAC,cAAc,CAAC;AAC3D,CAAC;AAED,SAAgB,2BAA2B;IACzC,OAAO,iBAAiB,KAAK,WAAW,CAAC;AAC3C,CAAC;AAED,SAAgB,oCAAoC;IAClD,OAAO,gCAAgC,KAAK,WAAW,CAAC;AAC1D,CAAC;AAED,SAAgB,YAAY,CAAC,KAAuC;IAClE,iBAAiB,GAAG,KAAK,CAAC;AAC5B,CAAC;AAED,SAAgB,2BAA2B,CAAC,KAA2B;IACrE,gCAAgC,GAAG,KAAK,CAAC;AAC3C,CAAC;AAED,SAAgB,sBAAsB,CAAC,MAAc;IACnD,oBAAoB,GAAG,MAAM,CAAC;AAChC,CAAC;AAED,SAAgB,sBAAsB;IACpC,OAAO,oBAAoB,CAAC;AAC9B,CAAC;AAED,SAAgB,mBAAmB,CAAC,GAAW;IAC7C,iBAAiB,GAAG,GAAG,CAAC;AAC1B,CAAC;AAED,SAAgB,mBAAmB;IACjC,OAAO,iBAAiB,IAAI,IAAA,gBAAO,GAAE,CAAC,cAAc,CAAC;AACvD,CAAC;AAED,SAAgB,gCAAgC;IAC9C,OAAO,IAAA,gBAAO,GAAE,CAAC,yBAAyB,CAAC;AAC7C,CAAC;AAEY,QAAA,OAAO,GAAG;IACrB,YAAY,EAAE,qBAAqB,kBAAQ,CAAC,OAAO,EAAE;CACtD,CAAC"}

package/dist/defaults.mjs

@@ -2,7 +2,9 @@ import { loadEnv } from '@openai/agents-core/_shims';
 import METADATA from "./metadata.mjs";
 export const DEFAULT_OPENAI_API = 'responses';
 export const DEFAULT_OPENAI_MODEL = 'gpt-4.1';
+export const DEFAULT_OPENAI_RESPONSES_TRANSPORT = 'http';
 let _defaultOpenAIAPI = DEFAULT_OPENAI_API;
+let _defaultOpenAIResponsesTransport = DEFAULT_OPENAI_RESPONSES_TRANSPORT;
 let _defaultOpenAIClient;
 let _defaultOpenAIKey = undefined;
 let _defaultTracingApiKey = undefined;
@@ -15,9 +17,15 @@ export function getTracingExportApiKey() {
 export function shouldUseResponsesByDefault() {
     return _defaultOpenAIAPI === 'responses';
 }
+export function shouldUseResponsesWebSocketByDefault() {
+    return _defaultOpenAIResponsesTransport === 'websocket';
+}
 export function setOpenAIAPI(value) {
     _defaultOpenAIAPI = value;
 }
+export function setOpenAIResponsesTransport(value) {
+    _defaultOpenAIResponsesTransport = value;
+}
 export function setDefaultOpenAIClient(client) {
     _defaultOpenAIClient = client;
 }
@@ -30,6 +38,9 @@ export function setDefaultOpenAIKey(key) {
 export function getDefaultOpenAIKey() {
     return _defaultOpenAIKey ?? loadEnv().OPENAI_API_KEY;
 }
+export function getDefaultOpenAIWebSocketBaseURL() {
+    return loadEnv().OPENAI_WEBSOCKET_BASE_URL;
+}
 export const HEADERS = {
     'User-Agent': `Agents/JavaScript ${METADATA.version}`,
 };

package/dist/defaults.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"defaults.mjs","sourceRoot":"","sources":["../src/defaults.ts"],"names":[],"mappings":"OACO,EAAE,OAAO,EAAE,MAAM,4BAA4B;OAC7C,QAAQ;AAEf,MAAM,CAAC,MAAM,kBAAkB,GAAG,WAAW,CAAC;AAC9C,MAAM,CAAC,MAAM,oBAAoB,GAAG,SAAS,CAAC;AAE9C,IAAI,iBAAiB,GAAG,kBAAkB,CAAC;AAC3C,IAAI,oBAAwC,CAAC;AAC7C,IAAI,iBAAiB,GAAuB,SAAS,CAAC;AACtD,IAAI,qBAAqB,GAAuB,SAAS,CAAC;AAE1D,MAAM,UAAU,sBAAsB,CAAC,GAAW;IAChD,qBAAqB,GAAG,GAAG,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,sBAAsB;IACpC,OAAO,qBAAqB,IAAI,OAAO,EAAE,CAAC,cAAc,CAAC;AAC3D,CAAC;AAED,MAAM,UAAU,2BAA2B;IACzC,OAAO,iBAAiB,KAAK,WAAW,CAAC;AAC3C,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,KAAuC;IAClE,iBAAiB,GAAG,KAAK,CAAC;AAC5B,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,MAAc;IACnD,oBAAoB,GAAG,MAAM,CAAC;AAChC,CAAC;AAED,MAAM,UAAU,sBAAsB;IACpC,OAAO,oBAAoB,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,GAAW;IAC7C,iBAAiB,GAAG,GAAG,CAAC;AAC1B,CAAC;AAED,MAAM,UAAU,mBAAmB;IACjC,OAAO,iBAAiB,IAAI,OAAO,EAAE,CAAC,cAAc,CAAC;AACvD,CAAC;AAED,MAAM,CAAC,MAAM,OAAO,GAAG;IACrB,YAAY,EAAE,qBAAqB,QAAQ,CAAC,OAAO,EAAE;CACtD,CAAC"}
+{"version":3,"file":"defaults.mjs","sourceRoot":"","sources":["../src/defaults.ts"],"names":[],"mappings":"OACO,EAAE,OAAO,EAAE,MAAM,4BAA4B;OAC7C,QAAQ;AAEf,MAAM,CAAC,MAAM,kBAAkB,GAAG,WAAW,CAAC;AAC9C,MAAM,CAAC,MAAM,oBAAoB,GAAG,SAAS,CAAC;AAC9C,MAAM,CAAC,MAAM,kCAAkC,GAAG,MAAM,CAAC;AAEzD,IAAI,iBAAiB,GAAG,kBAAkB,CAAC;AAC3C,IAAI,gCAAgC,GAAG,kCAAkC,CAAC;AAC1E,IAAI,oBAAwC,CAAC;AAC7C,IAAI,iBAAiB,GAAuB,SAAS,CAAC;AACtD,IAAI,qBAAqB,GAAuB,SAAS,CAAC;AAE1D,MAAM,UAAU,sBAAsB,CAAC,GAAW;IAChD,qBAAqB,GAAG,GAAG,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,sBAAsB;IACpC,OAAO,qBAAqB,IAAI,OAAO,EAAE,CAAC,cAAc,CAAC;AAC3D,CAAC;AAED,MAAM,UAAU,2BAA2B;IACzC,OAAO,iBAAiB,KAAK,WAAW,CAAC;AAC3C,CAAC;AAED,MAAM,UAAU,oCAAoC;IAClD,OAAO,gCAAgC,KAAK,WAAW,CAAC;AAC1D,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,KAAuC;IAClE,iBAAiB,GAAG,KAAK,CAAC;AAC5B,CAAC;AAED,MAAM,UAAU,2BAA2B,CAAC,KAA2B;IACrE,gCAAgC,GAAG,KAAK,CAAC;AAC3C,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,MAAc;IACnD,oBAAoB,GAAG,MAAM,CAAC;AAChC,CAAC;AAED,MAAM,UAAU,sBAAsB;IACpC,OAAO,oBAAoB,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,GAAW;IAC7C,iBAAiB,GAAG,GAAG,CAAC;AAC1B,CAAC;AAED,MAAM,UAAU,mBAAmB;IACjC,OAAO,iBAAiB,IAAI,OAAO,EAAE,CAAC,cAAc,CAAC;AACvD,CAAC;AAED,MAAM,UAAU,gCAAgC;IAC9C,OAAO,OAAO,EAAE,CAAC,yBAAyB,CAAC;AAC7C,CAAC;AAED,MAAM,CAAC,MAAM,OAAO,GAAG;IACrB,YAAY,EAAE,qBAAqB,QAAQ,CAAC,OAAO,EAAE;CACtD,CAAC"}

package/dist/index.d.ts

@@ -1,7 +1,8 @@
 export { OpenAIProvider } from './openaiProvider';
-export { OpenAIResponsesModel } from './openaiResponsesModel';
+export { withResponsesWebSocketSession, type ResponsesWebSocketSession, type ResponsesWebSocketSessionOptions, } from './responsesWebSocketSession';
+export { OpenAIResponsesModel, OpenAIResponsesWSModel, } from './openaiResponsesModel';
 export { OpenAIChatCompletionsModel } from './openaiChatCompletionsModel';
-export { setDefaultOpenAIClient, setOpenAIAPI, setDefaultOpenAIKey, setTracingExportApiKey, } from './defaults';
+export { setDefaultOpenAIClient, setOpenAIAPI, setOpenAIResponsesTransport, setDefaultOpenAIKey, setTracingExportApiKey, } from './defaults';
 export { setDefaultOpenAITracingExporter, OpenAITracingExporter, OpenAITracingExporterOptions, } from './openaiTracingExporter';
 export { webSearchTool, fileSearchTool, codeInterpreterTool, imageGenerationTool, } from './tools';
 export { OpenAIConversationsSession, startOpenAIConversationsSession, type OpenAIConversationsSessionOptions, } from './memory/openaiConversationsSession';

package/dist/index.js

@@ -1,15 +1,19 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.OpenAIResponsesCompactionSession = exports.startOpenAIConversationsSession = exports.OpenAIConversationsSession = exports.imageGenerationTool = exports.codeInterpreterTool = exports.fileSearchTool = exports.webSearchTool = exports.OpenAITracingExporter = exports.setDefaultOpenAITracingExporter = exports.setTracingExportApiKey = exports.setDefaultOpenAIKey = exports.setOpenAIAPI = exports.setDefaultOpenAIClient = exports.OpenAIChatCompletionsModel = exports.OpenAIResponsesModel = exports.OpenAIProvider = void 0;
+exports.OpenAIResponsesCompactionSession = exports.startOpenAIConversationsSession = exports.OpenAIConversationsSession = exports.imageGenerationTool = exports.codeInterpreterTool = exports.fileSearchTool = exports.webSearchTool = exports.OpenAITracingExporter = exports.setDefaultOpenAITracingExporter = exports.setTracingExportApiKey = exports.setDefaultOpenAIKey = exports.setOpenAIResponsesTransport = exports.setOpenAIAPI = exports.setDefaultOpenAIClient = exports.OpenAIChatCompletionsModel = exports.OpenAIResponsesWSModel = exports.OpenAIResponsesModel = exports.withResponsesWebSocketSession = exports.OpenAIProvider = void 0;
 var openaiProvider_1 = require("./openaiProvider.js");
 Object.defineProperty(exports, "OpenAIProvider", { enumerable: true, get: function () { return openaiProvider_1.OpenAIProvider; } });
+var responsesWebSocketSession_1 = require("./responsesWebSocketSession.js");
+Object.defineProperty(exports, "withResponsesWebSocketSession", { enumerable: true, get: function () { return responsesWebSocketSession_1.withResponsesWebSocketSession; } });
 var openaiResponsesModel_1 = require("./openaiResponsesModel.js");
 Object.defineProperty(exports, "OpenAIResponsesModel", { enumerable: true, get: function () { return openaiResponsesModel_1.OpenAIResponsesModel; } });
+Object.defineProperty(exports, "OpenAIResponsesWSModel", { enumerable: true, get: function () { return openaiResponsesModel_1.OpenAIResponsesWSModel; } });
 var openaiChatCompletionsModel_1 = require("./openaiChatCompletionsModel.js");
 Object.defineProperty(exports, "OpenAIChatCompletionsModel", { enumerable: true, get: function () { return openaiChatCompletionsModel_1.OpenAIChatCompletionsModel; } });
 var defaults_1 = require("./defaults.js");
 Object.defineProperty(exports, "setDefaultOpenAIClient", { enumerable: true, get: function () { return defaults_1.setDefaultOpenAIClient; } });
 Object.defineProperty(exports, "setOpenAIAPI", { enumerable: true, get: function () { return defaults_1.setOpenAIAPI; } });
+Object.defineProperty(exports, "setOpenAIResponsesTransport", { enumerable: true, get: function () { return defaults_1.setOpenAIResponsesTransport; } });
 Object.defineProperty(exports, "setDefaultOpenAIKey", { enumerable: true, get: function () { return defaults_1.setDefaultOpenAIKey; } });
 Object.defineProperty(exports, "setTracingExportApiKey", { enumerable: true, get: function () { return defaults_1.setTracingExportApiKey; } });
 var openaiTracingExporter_1 = require("./openaiTracingExporter.js");

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,sDAAkD;AAAzC,gHAAA,cAAc,OAAA;AACvB,kEAA8D;AAArD,4HAAA,oBAAoB,OAAA;AAC7B,8EAA0E;AAAjE,wIAAA,0BAA0B,OAAA;AACnC,0CAKoB;AAJlB,kHAAA,sBAAsB,OAAA;AACtB,wGAAA,YAAY,OAAA;AACZ,+GAAA,mBAAmB,OAAA;AACnB,kHAAA,sBAAsB,OAAA;AAExB,oEAIiC;AAH/B,wIAAA,+BAA+B,OAAA;AAC/B,8HAAA,qBAAqB,OAAA;AAGvB,oCAKiB;AAJf,sGAAA,aAAa,OAAA;AACb,uGAAA,cAAc,OAAA;AACd,4GAAA,mBAAmB,OAAA;AACnB,4GAAA,mBAAmB,OAAA;AAErB,qFAI6C;AAH3C,wIAAA,0BAA0B,OAAA;AAC1B,6IAAA,+BAA+B,OAAA;AAGjC,iGAKmD;AAJjD,oJAAA,gCAAgC,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,sDAAkD;AAAzC,gHAAA,cAAc,OAAA;AACvB,4EAIqC;AAHnC,0IAAA,6BAA6B,OAAA;AAI/B,kEAGgC;AAF9B,4HAAA,oBAAoB,OAAA;AACpB,8HAAA,sBAAsB,OAAA;AAExB,8EAA0E;AAAjE,wIAAA,0BAA0B,OAAA;AACnC,0CAMoB;AALlB,kHAAA,sBAAsB,OAAA;AACtB,wGAAA,YAAY,OAAA;AACZ,uHAAA,2BAA2B,OAAA;AAC3B,+GAAA,mBAAmB,OAAA;AACnB,kHAAA,sBAAsB,OAAA;AAExB,oEAIiC;AAH/B,wIAAA,+BAA+B,OAAA;AAC/B,8HAAA,qBAAqB,OAAA;AAGvB,oCAKiB;AAJf,sGAAA,aAAa,OAAA;AACb,uGAAA,cAAc,OAAA;AACd,4GAAA,mBAAmB,OAAA;AACnB,4GAAA,mBAAmB,OAAA;AAErB,qFAI6C;AAH3C,wIAAA,0BAA0B,OAAA;AAC1B,6IAAA,+BAA+B,OAAA;AAGjC,iGAKmD;AAJjD,oJAAA,gCAAgC,OAAA"}

package/dist/index.mjs

@@ -1,7 +1,8 @@
 export { OpenAIProvider } from "./openaiProvider.mjs";
-export { OpenAIResponsesModel } from "./openaiResponsesModel.mjs";
+export { withResponsesWebSocketSession, } from "./responsesWebSocketSession.mjs";
+export { OpenAIResponsesModel, OpenAIResponsesWSModel, } from "./openaiResponsesModel.mjs";
 export { OpenAIChatCompletionsModel } from "./openaiChatCompletionsModel.mjs";
-export { setDefaultOpenAIClient, setOpenAIAPI, setDefaultOpenAIKey, setTracingExportApiKey, } from "./defaults.mjs";
+export { setDefaultOpenAIClient, setOpenAIAPI, setOpenAIResponsesTransport, setDefaultOpenAIKey, setTracingExportApiKey, } from "./defaults.mjs";
 export { setDefaultOpenAITracingExporter, OpenAITracingExporter, } from "./openaiTracingExporter.mjs";
 export { webSearchTool, fileSearchTool, codeInterpreterTool, imageGenerationTool, } from "./tools.mjs";
 export { OpenAIConversationsSession, startOpenAIConversationsSession, } from "./memory/openaiConversationsSession.mjs";

package/dist/index.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.mjs","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"OAAO,EAAE,cAAc,EAAE;OAClB,EAAE,oBAAoB,EAAE;OACxB,EAAE,0BAA0B,EAAE;OAC9B,EACL,sBAAsB,EACtB,YAAY,EACZ,mBAAmB,EACnB,sBAAsB,GACvB;OACM,EACL,+BAA+B,EAC/B,qBAAqB,GAEtB;OACM,EACL,aAAa,EACb,cAAc,EACd,mBAAmB,EACnB,mBAAmB,GACpB;OACM,EACL,0BAA0B,EAC1B,+BAA+B,GAEhC;OACM,EACL,gCAAgC,GAIjC"}
+{"version":3,"file":"index.mjs","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"OAAO,EAAE,cAAc,EAAE;OAClB,EACL,6BAA6B,GAG9B;OACM,EACL,oBAAoB,EACpB,sBAAsB,GACvB;OACM,EAAE,0BAA0B,EAAE;OAC9B,EACL,sBAAsB,EACtB,YAAY,EACZ,2BAA2B,EAC3B,mBAAmB,EACnB,sBAAsB,GACvB;OACM,EACL,+BAA+B,EAC/B,qBAAqB,GAEtB;OACM,EACL,aAAa,EACb,cAAc,EACd,mBAAmB,EACnB,mBAAmB,GACpB;OACM,EACL,0BAA0B,EAC1B,+BAA+B,GAEhC;OACM,EACL,gCAAgC,GAIjC"}

package/dist/metadata.js

@@ -4,9 +4,9 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.METADATA = void 0;
 exports.METADATA = {
     "name": "@openai/agents-openai",
-    "version": "0.4.15",
+    "version": "0.5.0",
     "versions": {
-        "@openai/agents-openai": "0.4.15",
+        "@openai/agents-openai": "0.5.0",
         "@openai/agents-core": "workspace:*",
         "openai": "^6.20.0"
     }

package/dist/metadata.js.map

@@ -1 +1 @@
-{"version":3,"file":"metadata.js","sourceRoot":"","sources":["../src/metadata.ts"],"names":[],"mappings":";AACA,uCAAuC;;;AAE1B,QAAA,QAAQ,GAAG;IACtB,MAAM,EAAE,uBAAuB;IAC/B,SAAS,EAAE,QAAQ;IACnB,UAAU,EAAE;QACV,uBAAuB,EAAE,QAAQ;QACjC,qBAAqB,EAAE,aAAa;QACpC,QAAQ,EAAE,SAAS;KACpB;CACF,CAAC;AAEF,kBAAe,gBAAQ,CAAC"}
+{"version":3,"file":"metadata.js","sourceRoot":"","sources":["../src/metadata.ts"],"names":[],"mappings":";AACA,uCAAuC;;;AAE1B,QAAA,QAAQ,GAAG;IACtB,MAAM,EAAE,uBAAuB;IAC/B,SAAS,EAAE,OAAO;IAClB,UAAU,EAAE;QACV,uBAAuB,EAAE,OAAO;QAChC,qBAAqB,EAAE,aAAa;QACpC,QAAQ,EAAE,SAAS;KACpB;CACF,CAAC;AAEF,kBAAe,gBAAQ,CAAC"}

package/dist/metadata.mjs

@@ -1,9 +1,9 @@
 // This file is automatically generated
 export const METADATA = {
     "name": "@openai/agents-openai",
-    "version": "0.4.15",
+    "version": "0.5.0",
     "versions": {
-        "@openai/agents-openai": "0.4.15",
+        "@openai/agents-openai": "0.5.0",
         "@openai/agents-core": "workspace:*",
         "openai": "^6.20.0"
     }

package/dist/metadata.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"metadata.mjs","sourceRoot":"","sources":["../src/metadata.ts"],"names":[],"mappings":"AACA,uCAAuC;AAEvC,MAAM,CAAC,MAAM,QAAQ,GAAG;IACtB,MAAM,EAAE,uBAAuB;IAC/B,SAAS,EAAE,QAAQ;IACnB,UAAU,EAAE;QACV,uBAAuB,EAAE,QAAQ;QACjC,qBAAqB,EAAE,aAAa;QACpC,QAAQ,EAAE,SAAS;KACpB;CACF,CAAC;AAEF,eAAe,QAAQ,CAAC"}
+{"version":3,"file":"metadata.mjs","sourceRoot":"","sources":["../src/metadata.ts"],"names":[],"mappings":"AACA,uCAAuC;AAEvC,MAAM,CAAC,MAAM,QAAQ,GAAG;IACtB,MAAM,EAAE,uBAAuB;IAC/B,SAAS,EAAE,OAAO;IAClB,UAAU,EAAE;QACV,uBAAuB,EAAE,OAAO;QAChC,qBAAqB,EAAE,aAAa;QACpC,QAAQ,EAAE,SAAS;KACpB;CACF,CAAC;AAEF,eAAe,QAAQ,CAAC"}

package/dist/openaiProvider.d.ts

@@ -6,9 +6,12 @@ import OpenAI from 'openai';
 export type OpenAIProviderOptions = {
     apiKey?: string;
     baseURL?: string;
+    websocketBaseURL?: string;
+    cacheResponsesWebSocketModels?: boolean;
     organization?: string;
     project?: string;
     useResponses?: boolean;
+    useResponsesWebSocket?: boolean;
     openAIClient?: OpenAI;
 };
 /**
@@ -18,4 +21,8 @@ export declare class OpenAIProvider implements ModelProvider {
     #private;
     constructor(options?: OpenAIProviderOptions);
     getModel(modelName?: string | undefined): Promise<Model>;
+    /**
+     * Closes cached model wrappers (for example websocket-backed responses models) and clears cache.
+     */
+    close(): Promise<void>;
 }

package/dist/openaiProvider.js

@@ -15,6 +15,10 @@ const openaiChatCompletionsModel_1 = require("./openaiChatCompletionsModel.js");
 class OpenAIProvider {
     #client;
     #useResponses;
+    #useResponsesWebSocket;
+    #websocketBaseURL;
+    #cacheResponsesWebSocketModels;
+    #modelCache = new Map();
     #options;
     constructor(options = {}) {
         this.#options = options;
@@ -25,9 +29,16 @@ class OpenAIProvider {
             if (this.#options.baseURL) {
                 throw new Error('Cannot provide both baseURL and openAIClient');
             }
+            if (this.#options.websocketBaseURL) {
+                throw new Error('Cannot provide both websocketBaseURL and openAIClient');
+            }
             this.#client = this.#options.openAIClient;
         }
         this.#useResponses = this.#options.useResponses;
+        this.#useResponsesWebSocket = this.#options.useResponsesWebSocket;
+        this.#websocketBaseURL = this.#options.websocketBaseURL;
+        this.#cacheResponsesWebSocketModels =
+            this.#options.cacheResponsesWebSocketModels ?? true;
     }
     /**
      * Lazy loads the OpenAI client to not throw an error if you don't have an API key set but
@@ -49,13 +60,84 @@ class OpenAIProvider {
         }
         return this.#client;
     }
+    #getWebSocketBaseURLForResponsesModel(client) {
+        if (typeof this.#websocketBaseURL !== 'undefined') {
+            return this.#websocketBaseURL;
+        }
+        // Preserve configured client/provider endpoints rather than redirecting via env websocket base URL.
+        if (this.#options.openAIClient || this.#options.baseURL) {
+            return undefined;
+        }
+        const defaultClient = (0, defaults_1.getDefaultOpenAIClient)();
+        if (defaultClient && client === defaultClient) {
+            return undefined;
+        }
+        return (0, defaults_1.getDefaultOpenAIWebSocketBaseURL)();
+    }
     async getModel(modelName) {
         const model = modelName || (0, agents_core_1.getDefaultModel)();
         const useResponses = this.#useResponses ?? (0, defaults_1.shouldUseResponsesByDefault)();
+        const useResponsesWebSocket = this.#useResponsesWebSocket ?? (0, defaults_1.shouldUseResponsesWebSocketByDefault)();
+        const shouldCacheModelWrapper = !(useResponses &&
+            useResponsesWebSocket &&
+            !this.#cacheResponsesWebSocketModels);
+        const cacheKey = JSON.stringify([
+            model,
+            useResponses,
+            useResponsesWebSocket,
+        ]);
+        if (shouldCacheModelWrapper) {
+            const cachedModel = this.#modelCache.get(cacheKey);
+            if (cachedModel) {
+                return cachedModel;
+            }
+        }
+        let resolvedModel;
         if (useResponses) {
-            return new openaiResponsesModel_1.OpenAIResponsesModel(this.#getClient(), model);
+            const client = this.#getClient();
+            resolvedModel = useResponsesWebSocket
+                ? new openaiResponsesModel_1.OpenAIResponsesWSModel(client, model, {
+                    websocketBaseURL: this.#getWebSocketBaseURLForResponsesModel(client),
+                    reuseConnection: shouldCacheModelWrapper,
+                })
+                : new openaiResponsesModel_1.OpenAIResponsesModel(client, model);
+        }
+        else {
+            resolvedModel = new openaiChatCompletionsModel_1.OpenAIChatCompletionsModel(this.#getClient(), model);
+        }
+        if (shouldCacheModelWrapper) {
+            this.#modelCache.set(cacheKey, resolvedModel);
+        }
+        return resolvedModel;
+    }
+    /**
+     * Closes cached model wrappers (for example websocket-backed responses models) and clears cache.
+     */
+    async close() {
+        const cachedModels = Array.from(new Set(this.#modelCache.values()));
+        this.#modelCache.clear();
+        const closeErrors = [];
+        await Promise.all(cachedModels.map(async (model) => {
+            const maybeClose = model
+                .close;
+            if (typeof maybeClose !== 'function') {
+                return;
+            }
+            try {
+                await maybeClose.call(model);
+            }
+            catch (error) {
+                closeErrors.push(error);
+            }
+        }));
+        if (closeErrors.length === 1) {
+            throw closeErrors[0];
+        }
+        if (closeErrors.length > 1) {
+            const error = new Error('Failed to close OpenAIProvider.');
+            error.causes = closeErrors;
+            throw error;
         }
-        return new openaiChatCompletionsModel_1.OpenAIChatCompletionsModel(this.#getClient(), model);
     }
 }
 exports.OpenAIProvider = OpenAIProvider;

package/dist/openaiProvider.js.map

@@ -1 +1 @@
-{"version":3,"file":"openaiProvider.js","sourceRoot":"","sources":["../src/openaiProvider.ts"],"names":[],"mappings":";;;;;;AAAA,qDAA4E;AAC5E,oDAA4B;AAC5B,4CAIoB;AACpB,oEAA8D;AAC9D,gFAA0E;AAc1E;;GAEG;AACH,MAAa,cAAc;IACzB,OAAO,CAAU;IACjB,aAAa,CAAW;IACxB,QAAQ,CAAwB;IAEhC,YAAY,UAAiC,EAAE;QAC7C,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;QACxB,IAAI,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,CAAC;YAC/B,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;gBACzB,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;YACjE,CAAC;YACD,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;gBAC1B,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;YAClE,CAAC;YACD,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC;QAC5C,CAAC;QACD,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC;IAClD,CAAC;IAED;;;OAGG;IACH,UAAU;QACR,wDAAwD;QACxD,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,CAAC,OAAO;gBACV,6DAA6D;gBAC7D,IAAA,iCAAsB,GAAE;oBACxB,uCAAuC;oBACvC,IAAI,gBAAM,CAAC;wBACT,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,MAAM,IAAI,IAAA,8BAAmB,GAAE;wBACrD,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,OAAO;wBAC9B,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY;wBACxC,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,OAAO;qBAC/B,CAAC,CAAC;QACP,CAAC;QACD,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,SAA8B;QAC3C,MAAM,KAAK,GAAG,SAAS,IAAI,IAAA,6BAAe,GAAE,CAAC;QAC7C,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,IAAI,IAAA,sCAA2B,GAAE,CAAC;QAEzE,IAAI,YAAY,EAAE,CAAC;YACjB,OAAO,IAAI,2CAAoB,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,KAAK,CAAC,CAAC;QAC5D,CAAC;QAED,OAAO,IAAI,uDAA0B,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,KAAK,CAAC,CAAC;IAClE,CAAC;CACF;AAlDD,wCAkDC"}
+{"version":3,"file":"openaiProvider.js","sourceRoot":"","sources":["../src/openaiProvider.ts"],"names":[],"mappings":";;;;;;AAAA,qDAA4E;AAC5E,oDAA4B;AAC5B,4CAMoB;AACpB,oEAGgC;AAChC,gFAA0E;AAiB1E;;GAEG;AACH,MAAa,cAAc;IACzB,OAAO,CAAU;IACjB,aAAa,CAAW;IACxB,sBAAsB,CAAW;IACjC,iBAAiB,CAAU;IAC3B,8BAA8B,CAAU;IACxC,WAAW,GAAG,IAAI,GAAG,EAAiB,CAAC;IACvC,QAAQ,CAAwB;IAEhC,YAAY,UAAiC,EAAE;QAC7C,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;QACxB,IAAI,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,CAAC;YAC/B,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;gBACzB,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;YACjE,CAAC;YACD,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;gBAC1B,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;YAClE,CAAC;YACD,IAAI,IAAI,CAAC,QAAQ,CAAC,gBAAgB,EAAE,CAAC;gBACnC,MAAM,IAAI,KAAK,CACb,uDAAuD,CACxD,CAAC;YACJ,CAAC;YACD,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC;QAC5C,CAAC;QACD,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC;QAChD,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAC,QAAQ,CAAC,qBAAqB,CAAC;QAClE,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC;QACxD,IAAI,CAAC,8BAA8B;YACjC,IAAI,CAAC,QAAQ,CAAC,6BAA6B,IAAI,IAAI,CAAC;IACxD,CAAC;IAED;;;OAGG;IACH,UAAU;QACR,wDAAwD;QACxD,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,CAAC,OAAO;gBACV,6DAA6D;gBAC7D,IAAA,iCAAsB,GAAE;oBACxB,uCAAuC;oBACvC,IAAI,gBAAM,CAAC;wBACT,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,MAAM,IAAI,IAAA,8BAAmB,GAAE;wBACrD,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,OAAO;wBAC9B,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY;wBACxC,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,OAAO;qBAC/B,CAAC,CAAC;QACP,CAAC;QACD,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,qCAAqC,CAAC,MAAc;QAClD,IAAI,OAAO,IAAI,CAAC,iBAAiB,KAAK,WAAW,EAAE,CAAC;YAClD,OAAO,IAAI,CAAC,iBAAiB,CAAC;QAChC,CAAC;QAED,oGAAoG;QACpG,IAAI,IAAI,CAAC,QAAQ,CAAC,YAAY,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YACxD,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,aAAa,GAAG,IAAA,iCAAsB,GAAE,CAAC;QAC/C,IAAI,aAAa,IAAI,MAAM,KAAK,aAAa,EAAE,CAAC;YAC9C,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO,IAAA,2CAAgC,GAAE,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,SAA8B;QAC3C,MAAM,KAAK,GAAG,SAAS,IAAI,IAAA,6BAAe,GAAE,CAAC;QAC7C,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,IAAI,IAAA,sCAA2B,GAAE,CAAC;QACzE,MAAM,qBAAqB,GACzB,IAAI,CAAC,sBAAsB,IAAI,IAAA,+CAAoC,GAAE,CAAC;QACxE,MAAM,uBAAuB,GAAG,CAAC,CAC/B,YAAY;YACZ,qBAAqB;YACrB,CAAC,IAAI,CAAC,8BAA8B,CACrC,CAAC;QACF,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC;YAC9B,KAAK;YACL,YAAY;YACZ,qBAAqB;SACtB,CAAC,CAAC;QAEH,IAAI,uBAAuB,EAAE,CAAC;YAC5B,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACnD,IAAI,WAAW,EAAE,CAAC;gBAChB,OAAO,WAAW,CAAC;YACrB,CAAC;QACH,CAAC;QAED,IAAI,aAAoB,CAAC;QAEzB,IAAI,YAAY,EAAE,CAAC;YACjB,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;YACjC,aAAa,GAAG,qBAAqB;gBACnC,CAAC,CAAC,IAAI,6CAAsB,CAAC,MAAM,EAAE,KAAK,EAAE;oBACxC,gBAAgB,EACd,IAAI,CAAC,qCAAqC,CAAC,MAAM,CAAC;oBACpD,eAAe,EAAE,uBAAuB;iBACzC,CAAC;gBACJ,CAAC,CAAC,IAAI,2CAAoB,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QAC9C,CAAC;aAAM,CAAC;YACN,aAAa,GAAG,IAAI,uDAA0B,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,KAAK,CAAC,CAAC;QAC3E,CAAC;QAED,IAAI,uBAAuB,EAAE,CAAC;YAC5B,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QAChD,CAAC;QACD,OAAO,aAAa,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,MAAM,YAAY,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACpE,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QAEzB,MAAM,WAAW,GAAc,EAAE,CAAC;QAClC,MAAM,OAAO,CAAC,GAAG,CACf,YAAY,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;YAC/B,MAAM,UAAU,GAAI,KAAgD;iBACjE,KAAK,CAAC;YACT,IAAI,OAAO,UAAU,KAAK,UAAU,EAAE,CAAC;gBACrC,OAAO;YACT,CAAC;YACD,IAAI,CAAC;gBACH,MAAM,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC/B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC,CAAC,CACH,CAAC;QAEF,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,MAAM,WAAW,CAAC,CAAC,CAAC,CAAC;QACvB,CAAC;QACD,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3B,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;YAC1D,KAAwC,CAAC,MAAM,GAAG,WAAW,CAAC;YAC/D,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;CACF;AAnJD,wCAmJC"}

package/dist/openaiProvider.mjs

@@ -1,7 +1,7 @@
 import { getDefaultModel } from '@openai/agents-core';
 import OpenAI from 'openai';
-import { getDefaultOpenAIClient, getDefaultOpenAIKey, shouldUseResponsesByDefault, } from "./defaults.mjs";
-import { OpenAIResponsesModel } from "./openaiResponsesModel.mjs";
+import { getDefaultOpenAIClient, getDefaultOpenAIKey, getDefaultOpenAIWebSocketBaseURL, shouldUseResponsesByDefault, shouldUseResponsesWebSocketByDefault, } from "./defaults.mjs";
+import { OpenAIResponsesModel, OpenAIResponsesWSModel, } from "./openaiResponsesModel.mjs";
 import { OpenAIChatCompletionsModel } from "./openaiChatCompletionsModel.mjs";
 /**
  * The provider of OpenAI's models (or Chat Completions compatible ones)
@@ -9,6 +9,10 @@ import { OpenAIChatCompletionsModel } from "./openaiChatCompletionsModel.mjs";
 export class OpenAIProvider {
     #client;
     #useResponses;
+    #useResponsesWebSocket;
+    #websocketBaseURL;
+    #cacheResponsesWebSocketModels;
+    #modelCache = new Map();
     #options;
     constructor(options = {}) {
         this.#options = options;
@@ -19,9 +23,16 @@ export class OpenAIProvider {
             if (this.#options.baseURL) {
                 throw new Error('Cannot provide both baseURL and openAIClient');
             }
+            if (this.#options.websocketBaseURL) {
+                throw new Error('Cannot provide both websocketBaseURL and openAIClient');
+            }
             this.#client = this.#options.openAIClient;
         }
         this.#useResponses = this.#options.useResponses;
+        this.#useResponsesWebSocket = this.#options.useResponsesWebSocket;
+        this.#websocketBaseURL = this.#options.websocketBaseURL;
+        this.#cacheResponsesWebSocketModels =
+            this.#options.cacheResponsesWebSocketModels ?? true;
     }
     /**
      * Lazy loads the OpenAI client to not throw an error if you don't have an API key set but
@@ -43,13 +54,84 @@ export class OpenAIProvider {
         }
         return this.#client;
     }
+    #getWebSocketBaseURLForResponsesModel(client) {
+        if (typeof this.#websocketBaseURL !== 'undefined') {
+            return this.#websocketBaseURL;
+        }
+        // Preserve configured client/provider endpoints rather than redirecting via env websocket base URL.
+        if (this.#options.openAIClient || this.#options.baseURL) {
+            return undefined;
+        }
+        const defaultClient = getDefaultOpenAIClient();
+        if (defaultClient && client === defaultClient) {
+            return undefined;
+        }
+        return getDefaultOpenAIWebSocketBaseURL();
+    }
     async getModel(modelName) {
         const model = modelName || getDefaultModel();
         const useResponses = this.#useResponses ?? shouldUseResponsesByDefault();
+        const useResponsesWebSocket = this.#useResponsesWebSocket ?? shouldUseResponsesWebSocketByDefault();
+        const shouldCacheModelWrapper = !(useResponses &&
+            useResponsesWebSocket &&
+            !this.#cacheResponsesWebSocketModels);
+        const cacheKey = JSON.stringify([
+            model,
+            useResponses,
+            useResponsesWebSocket,
+        ]);
+        if (shouldCacheModelWrapper) {
+            const cachedModel = this.#modelCache.get(cacheKey);
+            if (cachedModel) {
+                return cachedModel;
+            }
+        }
+        let resolvedModel;
         if (useResponses) {
-            return new OpenAIResponsesModel(this.#getClient(), model);
+            const client = this.#getClient();
+            resolvedModel = useResponsesWebSocket
+                ? new OpenAIResponsesWSModel(client, model, {
+                    websocketBaseURL: this.#getWebSocketBaseURLForResponsesModel(client),
+                    reuseConnection: shouldCacheModelWrapper,
+                })
+                : new OpenAIResponsesModel(client, model);
+        }
+        else {
+            resolvedModel = new OpenAIChatCompletionsModel(this.#getClient(), model);
+        }
+        if (shouldCacheModelWrapper) {
+            this.#modelCache.set(cacheKey, resolvedModel);
+        }
+        return resolvedModel;
+    }
+    /**
+     * Closes cached model wrappers (for example websocket-backed responses models) and clears cache.
+     */
+    async close() {
+        const cachedModels = Array.from(new Set(this.#modelCache.values()));
+        this.#modelCache.clear();
+        const closeErrors = [];
+        await Promise.all(cachedModels.map(async (model) => {
+            const maybeClose = model
+                .close;
+            if (typeof maybeClose !== 'function') {
+                return;
+            }
+            try {
+                await maybeClose.call(model);
+            }
+            catch (error) {
+                closeErrors.push(error);
+            }
+        }));
+        if (closeErrors.length === 1) {
+            throw closeErrors[0];
+        }
+        if (closeErrors.length > 1) {
+            const error = new Error('Failed to close OpenAIProvider.');
+            error.causes = closeErrors;
+            throw error;
         }
-        return new OpenAIChatCompletionsModel(this.#getClient(), model);
     }
 }
 //# sourceMappingURL=openaiProvider.mjs.map

package/dist/openaiProvider.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"openaiProvider.mjs","sourceRoot":"","sources":["../src/openaiProvider.ts"],"names":[],"mappings":"OAAO,EAAwB,eAAe,EAAE,MAAM,qBAAqB;OACpE,MAAM,MAAM,QAAQ;OACpB,EACL,sBAAsB,EACtB,mBAAmB,EACnB,2BAA2B,GAC5B;OACM,EAAE,oBAAoB,EAAE;OACxB,EAAE,0BAA0B,EAAE;AAcrC;;GAEG;AACH,MAAM,OAAO,cAAc;IACzB,OAAO,CAAU;IACjB,aAAa,CAAW;IACxB,QAAQ,CAAwB;IAEhC,YAAY,UAAiC,EAAE;QAC7C,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;QACxB,IAAI,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,CAAC;YAC/B,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;gBACzB,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;YACjE,CAAC;YACD,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;gBAC1B,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;YAClE,CAAC;YACD,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC;QAC5C,CAAC;QACD,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC;IAClD,CAAC;IAED;;;OAGG;IACH,UAAU;QACR,wDAAwD;QACxD,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,CAAC,OAAO;gBACV,6DAA6D;gBAC7D,sBAAsB,EAAE;oBACxB,uCAAuC;oBACvC,IAAI,MAAM,CAAC;wBACT,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,MAAM,IAAI,mBAAmB,EAAE;wBACrD,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,OAAO;wBAC9B,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY;wBACxC,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,OAAO;qBAC/B,CAAC,CAAC;QACP,CAAC;QACD,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,SAA8B;QAC3C,MAAM,KAAK,GAAG,SAAS,IAAI,eAAe,EAAE,CAAC;QAC7C,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,IAAI,2BAA2B,EAAE,CAAC;QAEzE,IAAI,YAAY,EAAE,CAAC;YACjB,OAAO,IAAI,oBAAoB,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,KAAK,CAAC,CAAC;QAC5D,CAAC;QAED,OAAO,IAAI,0BAA0B,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,KAAK,CAAC,CAAC;IAClE,CAAC;CACF"}
+{"version":3,"file":"openaiProvider.mjs","sourceRoot":"","sources":["../src/openaiProvider.ts"],"names":[],"mappings":"OAAO,EAAwB,eAAe,EAAE,MAAM,qBAAqB;OACpE,MAAM,MAAM,QAAQ;OACpB,EACL,sBAAsB,EACtB,mBAAmB,EACnB,gCAAgC,EAChC,2BAA2B,EAC3B,oCAAoC,GACrC;OACM,EACL,oBAAoB,EACpB,sBAAsB,GACvB;OACM,EAAE,0BAA0B,EAAE;AAiBrC;;GAEG;AACH,MAAM,OAAO,cAAc;IACzB,OAAO,CAAU;IACjB,aAAa,CAAW;IACxB,sBAAsB,CAAW;IACjC,iBAAiB,CAAU;IAC3B,8BAA8B,CAAU;IACxC,WAAW,GAAG,IAAI,GAAG,EAAiB,CAAC;IACvC,QAAQ,CAAwB;IAEhC,YAAY,UAAiC,EAAE;QAC7C,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;QACxB,IAAI,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,CAAC;YAC/B,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;gBACzB,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;YACjE,CAAC;YACD,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;gBAC1B,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;YAClE,CAAC;YACD,IAAI,IAAI,CAAC,QAAQ,CAAC,gBAAgB,EAAE,CAAC;gBACnC,MAAM,IAAI,KAAK,CACb,uDAAuD,CACxD,CAAC;YACJ,CAAC;YACD,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC;QAC5C,CAAC;QACD,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC;QAChD,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAC,QAAQ,CAAC,qBAAqB,CAAC;QAClE,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC;QACxD,IAAI,CAAC,8BAA8B;YACjC,IAAI,CAAC,QAAQ,CAAC,6BAA6B,IAAI,IAAI,CAAC;IACxD,CAAC;IAED;;;OAGG;IACH,UAAU;QACR,wDAAwD;QACxD,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,CAAC,OAAO;gBACV,6DAA6D;gBAC7D,sBAAsB,EAAE;oBACxB,uCAAuC;oBACvC,IAAI,MAAM,CAAC;wBACT,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,MAAM,IAAI,mBAAmB,EAAE;wBACrD,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,OAAO;wBAC9B,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY;wBACxC,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,OAAO;qBAC/B,CAAC,CAAC;QACP,CAAC;QACD,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,qCAAqC,CAAC,MAAc;QAClD,IAAI,OAAO,IAAI,CAAC,iBAAiB,KAAK,WAAW,EAAE,CAAC;YAClD,OAAO,IAAI,CAAC,iBAAiB,CAAC;QAChC,CAAC;QAED,oGAAoG;QACpG,IAAI,IAAI,CAAC,QAAQ,CAAC,YAAY,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YACxD,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,aAAa,GAAG,sBAAsB,EAAE,CAAC;QAC/C,IAAI,aAAa,IAAI,MAAM,KAAK,aAAa,EAAE,CAAC;YAC9C,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO,gCAAgC,EAAE,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,SAA8B;QAC3C,MAAM,KAAK,GAAG,SAAS,IAAI,eAAe,EAAE,CAAC;QAC7C,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,IAAI,2BAA2B,EAAE,CAAC;QACzE,MAAM,qBAAqB,GACzB,IAAI,CAAC,sBAAsB,IAAI,oCAAoC,EAAE,CAAC;QACxE,MAAM,uBAAuB,GAAG,CAAC,CAC/B,YAAY;YACZ,qBAAqB;YACrB,CAAC,IAAI,CAAC,8BAA8B,CACrC,CAAC;QACF,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC;YAC9B,KAAK;YACL,YAAY;YACZ,qBAAqB;SACtB,CAAC,CAAC;QAEH,IAAI,uBAAuB,EAAE,CAAC;YAC5B,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACnD,IAAI,WAAW,EAAE,CAAC;gBAChB,OAAO,WAAW,CAAC;YACrB,CAAC;QACH,CAAC;QAED,IAAI,aAAoB,CAAC;QAEzB,IAAI,YAAY,EAAE,CAAC;YACjB,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;YACjC,aAAa,GAAG,qBAAqB;gBACnC,CAAC,CAAC,IAAI,sBAAsB,CAAC,MAAM,EAAE,KAAK,EAAE;oBACxC,gBAAgB,EACd,IAAI,CAAC,qCAAqC,CAAC,MAAM,CAAC;oBACpD,eAAe,EAAE,uBAAuB;iBACzC,CAAC;gBACJ,CAAC,CAAC,IAAI,oBAAoB,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QAC9C,CAAC;aAAM,CAAC;YACN,aAAa,GAAG,IAAI,0BAA0B,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,KAAK,CAAC,CAAC;QAC3E,CAAC;QAED,IAAI,uBAAuB,EAAE,CAAC;YAC5B,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QAChD,CAAC;QACD,OAAO,aAAa,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,MAAM,YAAY,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACpE,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QAEzB,MAAM,WAAW,GAAc,EAAE,CAAC;QAClC,MAAM,OAAO,CAAC,GAAG,CACf,YAAY,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;YAC/B,MAAM,UAAU,GAAI,KAAgD;iBACjE,KAAK,CAAC;YACT,IAAI,OAAO,UAAU,KAAK,UAAU,EAAE,CAAC;gBACrC,OAAO;YACT,CAAC;YACD,IAAI,CAAC;gBACH,MAAM,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC/B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC,CAAC,CACH,CAAC;QAEF,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,MAAM,WAAW,CAAC,CAAC,CAAC,CAAC;QACvB,CAAC;QACD,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3B,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;YAC1D,KAAwC,CAAC,MAAM,GAAG,WAAW,CAAC;YAC/D,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;CACF"}

package/dist/openaiResponsesModel.d.ts

@@ -2,9 +2,18 @@ import { Model, protocol } from '@openai/agents-core';
 import type { SerializedTool, ModelRequest, ModelResponse, ModelSettingsToolChoice, ResponseStreamEvent } from '@openai/agents-core';
 import OpenAI from 'openai';
 import { ToolChoiceFunction, ToolChoiceOptions, ToolChoiceTypes } from 'openai/resources/responses/responses';
+import { headerAccumulatorToSDKHeaders } from './responsesTransportUtils';
 type ToolChoice = ToolChoiceOptions | ToolChoiceTypes | {
     type: 'web_search';
 } | ToolChoiceFunction;
+type ResponsesCreateRequestSDKHeaders = ReturnType<typeof headerAccumulatorToSDKHeaders>;
+type BuiltResponsesCreateRequest = {
+    requestData: Record<string, any>;
+    sdkRequestHeaders: ResponsesCreateRequestSDKHeaders;
+    signal: AbortSignal | undefined;
+    transportExtraHeaders?: Record<string, unknown>;
+    transportExtraQuery?: Record<string, unknown>;
+};
 type ResponseOutputItemWithFunctionResult = OpenAI.Responses.ResponseOutputItem | (OpenAI.Responses.ResponseFunctionToolCallOutputItem & {
     name?: string;
     function_name?: string;
@@ -21,8 +30,11 @@ export { getToolChoice, converTool, getInputItems, convertToOutputItem };
  * Model implementation that uses OpenAI's Responses API to generate responses.
  */
 export declare class OpenAIResponsesModel implements Model {
-    #private;
+    protected readonly _client: OpenAI;
+    protected readonly _model: string;
     constructor(client: OpenAI, model: string);
+    protected _fetchResponse(request: ModelRequest, stream: false): Promise<OpenAI.Responses.Response>;
+    protected _buildResponsesCreateRequest(request: ModelRequest, stream: boolean): BuiltResponsesCreateRequest;
     /**
      * Get a response from the OpenAI model using the Responses API.
      * @param request - The request to send to the model.
@@ -36,3 +48,18 @@ export declare class OpenAIResponsesModel implements Model {
      */
     getStreamedResponse(request: ModelRequest): AsyncIterable<ResponseStreamEvent>;
 }
+export type OpenAIResponsesWSModelOptions = {
+    websocketBaseURL?: string;
+    reuseConnection?: boolean;
+};
+/**
+ * Model implementation that uses the OpenAI Responses API over a websocket transport.
+ *
+ * @see {@link https://developers.openai.com/api/docs/guides/websocket-mode}
+ */
+export declare class OpenAIResponsesWSModel extends OpenAIResponsesModel {
+    #private;
+    constructor(client: OpenAI, model: string, options?: OpenAIResponsesWSModelOptions);
+    protected _fetchResponse(request: ModelRequest, stream: false): Promise<OpenAI.Responses.Response>;
+    close(): Promise<void>;
+}