pi2dsh 0.3.0 → 0.3.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (104) hide show
  1. package/README.md +31 -11
  2. package/README.zh.md +31 -11
  3. package/dist/all-Rt4K68QA.mjs +28512 -0
  4. package/dist/all-Rt4K68QA.mjs.map +1 -0
  5. package/dist/anthropic-messages-Beym0_aC.mjs +6438 -0
  6. package/dist/anthropic-messages-Beym0_aC.mjs.map +1 -0
  7. package/dist/azure-openai-responses-cHqZTgLD.mjs +209 -0
  8. package/dist/azure-openai-responses-cHqZTgLD.mjs.map +1 -0
  9. package/dist/{build-DsGYAgiT.mjs → build-BG7p2Lbr.mjs} +2 -2
  10. package/dist/{build-DsGYAgiT.mjs.map → build-BG7p2Lbr.mjs.map} +1 -1
  11. package/dist/cli.mjs +1 -1
  12. package/dist/compat/pi-ai.d.mts +25 -3
  13. package/dist/compat/pi-ai.d.mts.map +1 -1
  14. package/dist/compat/pi-ai.mjs +3 -2919
  15. package/dist/compat/pi-coding-agent.d.mts +12 -1
  16. package/dist/compat/pi-coding-agent.d.mts.map +1 -1
  17. package/dist/compat/pi-coding-agent.mjs +2 -2
  18. package/dist/credentials-oauth.d.mts +27 -0
  19. package/dist/credentials-oauth.d.mts.map +1 -0
  20. package/dist/credentials-oauth.mjs +91 -0
  21. package/dist/credentials-oauth.mjs.map +1 -0
  22. package/dist/deferred-tools-BD4g5QUx.mjs +32 -0
  23. package/dist/deferred-tools-BD4g5QUx.mjs.map +1 -0
  24. package/dist/diagnostics-CJye1UFh.mjs +99 -0
  25. package/dist/diagnostics-CJye1UFh.mjs.map +1 -0
  26. package/dist/dist-IQKOgQyt.mjs +10035 -0
  27. package/dist/dist-IQKOgQyt.mjs.map +1 -0
  28. package/dist/dist-QBBA-dL_.mjs +1246 -0
  29. package/dist/dist-QBBA-dL_.mjs.map +1 -0
  30. package/dist/error-body-DKA3lknR.mjs +104 -0
  31. package/dist/error-body-DKA3lknR.mjs.map +1 -0
  32. package/dist/github-copilot-headers-CZo2QHrt.mjs +25 -0
  33. package/dist/github-copilot-headers-CZo2QHrt.mjs.map +1 -0
  34. package/dist/google-generative-ai-B7OZs5lu.mjs +358 -0
  35. package/dist/google-generative-ai-B7OZs5lu.mjs.map +1 -0
  36. package/dist/google-shared-CRsAUItF.mjs +29042 -0
  37. package/dist/google-shared-CRsAUItF.mjs.map +1 -0
  38. package/dist/google-vertex-DmdaHQtB.mjs +402 -0
  39. package/dist/google-vertex-DmdaHQtB.mjs.map +1 -0
  40. package/dist/hash-Dbggamac.mjs +19 -0
  41. package/dist/hash-Dbggamac.mjs.map +1 -0
  42. package/dist/headers-CPzZhJQn.mjs +17 -0
  43. package/dist/headers-CPzZhJQn.mjs.map +1 -0
  44. package/dist/host.d.mts.map +1 -1
  45. package/dist/host.mjs +15 -3
  46. package/dist/host.mjs.map +1 -1
  47. package/dist/images-models-DJtHo9Rp.mjs +1068 -0
  48. package/dist/images-models-DJtHo9Rp.mjs.map +1 -0
  49. package/dist/index.d.mts.map +1 -1
  50. package/dist/index.mjs +1 -1
  51. package/dist/json-parse-CFyFZSLJ.mjs +384 -0
  52. package/dist/json-parse-CFyFZSLJ.mjs.map +1 -0
  53. package/dist/{mcp-config-gqg7GJEE.mjs → mcp-config-D2B3kv-Z.mjs} +34 -16
  54. package/dist/mcp-config-D2B3kv-Z.mjs.map +1 -0
  55. package/dist/mistral-conversations-HHZCcFv_.mjs +42007 -0
  56. package/dist/mistral-conversations-HHZCcFv_.mjs.map +1 -0
  57. package/dist/multipart-parser-CIa5BbwO.mjs +301 -0
  58. package/dist/multipart-parser-CIa5BbwO.mjs.map +1 -0
  59. package/dist/oauth-bridge-C6fL1qSv.mjs +375 -0
  60. package/dist/oauth-bridge-C6fL1qSv.mjs.map +1 -0
  61. package/dist/openai-DkzRh5_G.mjs +6732 -0
  62. package/dist/openai-DkzRh5_G.mjs.map +1 -0
  63. package/dist/openai-codex-responses-vmovNu-q.mjs +1169 -0
  64. package/dist/openai-codex-responses-vmovNu-q.mjs.map +1 -0
  65. package/dist/openai-completions-fXHhRolF.mjs +947 -0
  66. package/dist/openai-completions-fXHhRolF.mjs.map +1 -0
  67. package/dist/openai-prompt-cache-BDWtZRKh.mjs +11 -0
  68. package/dist/openai-prompt-cache-BDWtZRKh.mjs.map +1 -0
  69. package/dist/openai-responses-DKdCYf3h.mjs +247 -0
  70. package/dist/openai-responses-DKdCYf3h.mjs.map +1 -0
  71. package/dist/openai-responses-shared-D0Jq7yVP.mjs +572 -0
  72. package/dist/openai-responses-shared-D0Jq7yVP.mjs.map +1 -0
  73. package/dist/openrouter-images-fG3Aj6M4.mjs +127 -0
  74. package/dist/openrouter-images-fG3Aj6M4.mjs.map +1 -0
  75. package/dist/otel-BQSQRBLF.mjs +24935 -0
  76. package/dist/otel-BQSQRBLF.mjs.map +1 -0
  77. package/dist/pi-ai-DhYGgq4Y.mjs +2984 -0
  78. package/dist/pi-ai-DhYGgq4Y.mjs.map +1 -0
  79. package/dist/{pi-coding-agent-Z1hTs61i.mjs → pi-coding-agent-fPHwKP4d.mjs} +27 -3
  80. package/dist/{pi-coding-agent-Z1hTs61i.mjs.map → pi-coding-agent-fPHwKP4d.mjs.map} +1 -1
  81. package/dist/pi-messages-CV-Ts8An.mjs +306 -0
  82. package/dist/pi-messages-CV-Ts8An.mjs.map +1 -0
  83. package/dist/provider-env-C72pF4vP.mjs +40 -0
  84. package/dist/provider-env-C72pF4vP.mjs.map +1 -0
  85. package/dist/provider-retry-Cw6YJ4cr.mjs +80 -0
  86. package/dist/provider-retry-Cw6YJ4cr.mjs.map +1 -0
  87. package/dist/rolldown-runtime-CTfmNlz1.mjs +44 -0
  88. package/dist/{runtime-oLd2EInK.mjs → runtime-B9R7bDWT.mjs} +838 -345
  89. package/dist/runtime-B9R7bDWT.mjs.map +1 -0
  90. package/dist/runtime.d.mts.map +1 -1
  91. package/dist/runtime.mjs +1 -1
  92. package/dist/sanitize-unicode-wLA11KvZ.mjs +29 -0
  93. package/dist/sanitize-unicode-wLA11KvZ.mjs.map +1 -0
  94. package/dist/src-C3WBfOks.mjs +5031 -0
  95. package/dist/src-C3WBfOks.mjs.map +1 -0
  96. package/dist/transform-messages-DGq9mvhA.mjs +379 -0
  97. package/dist/transform-messages-DGq9mvhA.mjs.map +1 -0
  98. package/dist/uuid-DtFT7tOE.mjs +65 -0
  99. package/dist/uuid-DtFT7tOE.mjs.map +1 -0
  100. package/package.json +11 -1
  101. package/dist/compat/pi-ai.mjs.map +0 -1
  102. package/dist/mcp-config-gqg7GJEE.mjs.map +0 -1
  103. package/dist/rolldown-runtime-D-uZhY3_.mjs +0 -16
  104. package/dist/runtime-oLd2EInK.mjs.map +0 -1
@@ -0,0 +1,1169 @@
1
+
2
+ import { a as AssistantMessageEventStream, i as formatThrownValue, n as createAssistantMessageDiagnostic, t as appendAssistantMessageDiagnostic } from "./diagnostics-CJye1UFh.mjs";
3
+ import { o as clampThinkingLevel } from "./images-models-DJtHo9Rp.mjs";
4
+ import { t as splitDeferredTools } from "./deferred-tools-BD4g5QUx.mjs";
5
+ import { t as headersToRecord } from "./headers-CPzZhJQn.mjs";
6
+ import { t as getProviderEnvValue } from "./provider-env-C72pF4vP.mjs";
7
+ import { c as createGrammarToolInputProperties, i as buildBaseOptions } from "./transform-messages-DGq9mvhA.mjs";
8
+ import { n as normalizeProviderError, t as formatProviderError } from "./error-body-DKA3lknR.mjs";
9
+ import { t as clampOpenAIPromptCacheKey } from "./openai-prompt-cache-BDWtZRKh.mjs";
10
+ import { n as convertResponsesTools, r as processResponsesStream, t as convertResponsesMessages } from "./openai-responses-shared-D0Jq7yVP.mjs";
11
+ import { r as registerSessionResourceCleanup, t as uuidv7 } from "./uuid-DtFT7tOE.mjs";
12
+ //#region node_modules/.pnpm/@earendil-works+pi-ai@0.84.1_ws@8.21.3_zod@4.4.3/node_modules/@earendil-works/pi-ai/dist/utils/abort-signals.js
13
+ function combineAbortSignals(signals) {
14
+ const activeSignals = signals.filter((signal) => signal !== void 0);
15
+ if (activeSignals.length === 0) return { cleanup: () => {} };
16
+ if (activeSignals.length === 1) return {
17
+ signal: activeSignals[0],
18
+ cleanup: () => {}
19
+ };
20
+ const controller = new AbortController();
21
+ const listeners = [];
22
+ const abort = (signal) => {
23
+ if (!controller.signal.aborted) controller.abort(signal.reason);
24
+ };
25
+ for (const signal of activeSignals) {
26
+ if (signal.aborted) {
27
+ abort(signal);
28
+ break;
29
+ }
30
+ const listener = () => abort(signal);
31
+ signal.addEventListener("abort", listener, { once: true });
32
+ listeners.push({
33
+ signal,
34
+ listener
35
+ });
36
+ }
37
+ return {
38
+ signal: controller.signal,
39
+ cleanup: () => {
40
+ for (const { signal, listener } of listeners) signal.removeEventListener("abort", listener);
41
+ }
42
+ };
43
+ }
44
+ //#endregion
45
+ //#region node_modules/.pnpm/@earendil-works+pi-ai@0.84.1_ws@8.21.3_zod@4.4.3/node_modules/@earendil-works/pi-ai/dist/utils/node-http-proxy.js
46
+ const DEFAULT_PROXY_PORTS = {
47
+ ftp: 21,
48
+ gopher: 70,
49
+ http: 80,
50
+ https: 443,
51
+ ws: 80,
52
+ wss: 443
53
+ };
54
+ function getProxyEnv(key, env) {
55
+ const lowercaseKey = key.toLowerCase();
56
+ const uppercaseKey = key.toUpperCase();
57
+ return env?.[lowercaseKey] || env?.[uppercaseKey] || getProviderEnvValue(lowercaseKey) || getProviderEnvValue(uppercaseKey) || "";
58
+ }
59
+ function parseProxyTargetUrl(targetUrl) {
60
+ if (targetUrl instanceof URL) return targetUrl;
61
+ try {
62
+ return new URL(targetUrl);
63
+ } catch {
64
+ return;
65
+ }
66
+ }
67
+ function shouldProxyHostname(hostname, port, env) {
68
+ const noProxy = getProxyEnv("no_proxy", env).toLowerCase();
69
+ if (!noProxy) return true;
70
+ if (noProxy === "*") return false;
71
+ return noProxy.split(/[,\s]/).every((proxy) => {
72
+ if (!proxy) return true;
73
+ const parsedProxy = proxy.match(/^(.+):(\d+)$/);
74
+ let proxyHostname = parsedProxy ? parsedProxy[1] : proxy;
75
+ const proxyPort = parsedProxy ? Number.parseInt(parsedProxy[2], 10) : 0;
76
+ if (proxyPort && proxyPort !== port) return true;
77
+ if (!/^[.*]/.test(proxyHostname)) return hostname !== proxyHostname;
78
+ if (proxyHostname.startsWith("*")) proxyHostname = proxyHostname.slice(1);
79
+ return !hostname.endsWith(proxyHostname);
80
+ });
81
+ }
82
+ function getProxyForUrl(targetUrl, env) {
83
+ const parsedUrl = parseProxyTargetUrl(targetUrl);
84
+ if (!parsedUrl?.protocol || !parsedUrl.host) return "";
85
+ const protocol = parsedUrl.protocol.split(":", 1)[0];
86
+ if (!shouldProxyHostname(parsedUrl.host.replace(/:\d*$/, ""), Number.parseInt(parsedUrl.port, 10) || DEFAULT_PROXY_PORTS[protocol] || 0, env)) return "";
87
+ let proxy = getProxyEnv(`${protocol}_proxy`, env) || getProxyEnv("all_proxy", env);
88
+ if (proxy && !proxy.includes("://")) proxy = `${protocol}://${proxy}`;
89
+ return proxy;
90
+ }
91
+ const UNSUPPORTED_PROXY_PROTOCOL_MESSAGE = "Unsupported proxy protocol. SOCKS and PAC proxy URLs are not supported; use an HTTP or HTTPS proxy URL.";
92
+ function resolveHttpProxyUrlForTarget(targetUrl, env) {
93
+ const proxy = getProxyForUrl(targetUrl, env);
94
+ if (!proxy) return;
95
+ let proxyUrl;
96
+ try {
97
+ proxyUrl = new URL(proxy);
98
+ } catch (error) {
99
+ throw new Error(`Invalid proxy URL ${JSON.stringify(proxy)}: ${error instanceof Error ? error.message : String(error)}`);
100
+ }
101
+ if (proxyUrl.protocol !== "http:" && proxyUrl.protocol !== "https:") throw new Error(`${UNSUPPORTED_PROXY_PROTOCOL_MESSAGE} Got ${proxyUrl.protocol}`);
102
+ return proxyUrl;
103
+ }
104
+ //#endregion
105
+ //#region node_modules/.pnpm/@earendil-works+pi-ai@0.84.1_ws@8.21.3_zod@4.4.3/node_modules/@earendil-works/pi-ai/dist/api/openai-codex-responses.js
106
+ function loadNodeOs() {
107
+ if (typeof process === "undefined" || !(process.versions?.node || process.versions?.bun)) return null;
108
+ return process.getBuiltinModule?.("node:os") ?? null;
109
+ }
110
+ const _os = loadNodeOs();
111
+ const DEFAULT_CODEX_BASE_URL = "https://chatgpt.com/backend-api";
112
+ const JWT_CLAIM_PATH = "https://api.openai.com/auth";
113
+ const DEFAULT_MAX_RETRIES = 0;
114
+ const BASE_DELAY_MS = 1e3;
115
+ const DEFAULT_MAX_RETRY_DELAY_MS = 6e4;
116
+ const DEFAULT_WEBSOCKET_CONNECT_TIMEOUT_MS = 15e3;
117
+ const REQUEST_COMPRESSION_ZSTD_LEVEL = 3;
118
+ const CODEX_TOOL_CALL_PROVIDERS = /* @__PURE__ */ new Set([
119
+ "openai",
120
+ "openai-codex",
121
+ "opencode"
122
+ ]);
123
+ const WEBSOCKET_MESSAGE_TOO_BIG_CLOSE_CODE = 1009;
124
+ const WEBSOCKET_CONNECTION_LIMIT_REACHED_CODE = "websocket_connection_limit_reached";
125
+ const PREVIOUS_RESPONSE_NOT_FOUND_CODE = "previous_response_not_found";
126
+ const CODEX_RESPONSE_STATUSES = /* @__PURE__ */ new Set([
127
+ "completed",
128
+ "incomplete",
129
+ "failed",
130
+ "cancelled",
131
+ "queued",
132
+ "in_progress"
133
+ ]);
134
+ function assertSuccessfulOutput(output) {
135
+ if (output.stopReason === "pending") throw new Error("Codex stream ended without a stop reason");
136
+ if (output.stopReason === "error" || output.stopReason === "aborted") throw new Error(output.errorMessage || "An unknown error occurred");
137
+ }
138
+ function isTerminalRateLimitError(errorText) {
139
+ return /GoUsageLimitError|FreeUsageLimitError|Monthly usage limit reached|available balance|insufficient_quota|out of budget|quota exceeded|billing/i.test(errorText);
140
+ }
141
+ function isRetryableError(status, errorText) {
142
+ if (status === 429 && isTerminalRateLimitError(errorText)) return false;
143
+ if (status === 429 || status === 500 || status === 502 || status === 503 || status === 504) return true;
144
+ return /rate.?limit|overloaded|service.?unavailable|upstream.?connect|connection.?refused/i.test(errorText);
145
+ }
146
+ function getRetryAfterDelayMs(headers) {
147
+ const retryAfterMs = headers.get("retry-after-ms");
148
+ if (retryAfterMs !== null) {
149
+ const millis = Number(retryAfterMs);
150
+ if (Number.isFinite(millis)) return Math.max(0, millis);
151
+ }
152
+ const retryAfter = headers.get("retry-after");
153
+ if (!retryAfter) return;
154
+ const seconds = Number(retryAfter);
155
+ if (Number.isFinite(seconds)) return Math.max(0, seconds * 1e3);
156
+ const date = Date.parse(retryAfter);
157
+ if (!Number.isNaN(date)) return Math.max(0, date - Date.now());
158
+ }
159
+ var RetryDelayExceededError = class extends Error {};
160
+ function validateRetryDelayMs(delayMs, options) {
161
+ const maxRetryDelayMs = options?.maxRetryDelayMs ?? DEFAULT_MAX_RETRY_DELAY_MS;
162
+ if (maxRetryDelayMs > 0 && delayMs > maxRetryDelayMs) throw new RetryDelayExceededError(`Server requested ${Math.ceil(delayMs / 1e3)}s retry delay (max: ${Math.ceil(maxRetryDelayMs / 1e3)}s)`);
163
+ return delayMs;
164
+ }
165
+ function sleep(ms, signal) {
166
+ return new Promise((resolve, reject) => {
167
+ if (signal?.aborted) {
168
+ reject(/* @__PURE__ */ new Error("Request was aborted"));
169
+ return;
170
+ }
171
+ const timeout = setTimeout(resolve, ms);
172
+ signal?.addEventListener("abort", () => {
173
+ clearTimeout(timeout);
174
+ reject(/* @__PURE__ */ new Error("Request was aborted"));
175
+ });
176
+ });
177
+ }
178
+ function normalizeTimeoutMs(value) {
179
+ if (value === void 0) return void 0;
180
+ if (!Number.isFinite(value) || value < 0) throw new Error(`Invalid timeoutMs: ${String(value)}`);
181
+ return Math.floor(value);
182
+ }
183
+ function loadNodeZlib() {
184
+ if (typeof process === "undefined" || !(process.versions?.node || process.versions?.bun)) return null;
185
+ return process.getBuiltinModule?.("node:zlib") ?? null;
186
+ }
187
+ function compressRequestBodyZstd(bodyJson) {
188
+ const zlib = loadNodeZlib();
189
+ if (!zlib || typeof zlib.zstdCompressSync !== "function") return null;
190
+ try {
191
+ const compressed = zlib.zstdCompressSync(bodyJson, { params: { [zlib.constants.ZSTD_c_compressionLevel]: REQUEST_COMPRESSION_ZSTD_LEVEL } });
192
+ return new Uint8Array(compressed.buffer, compressed.byteOffset, compressed.byteLength);
193
+ } catch {
194
+ return null;
195
+ }
196
+ }
197
+ const stream = (model, context, options) => {
198
+ const stream = new AssistantMessageEventStream();
199
+ (async () => {
200
+ const output = {
201
+ role: "assistant",
202
+ content: [],
203
+ api: "openai-codex-responses",
204
+ provider: model.provider,
205
+ model: model.id,
206
+ usage: {
207
+ input: 0,
208
+ output: 0,
209
+ cacheRead: 0,
210
+ cacheWrite: 0,
211
+ totalTokens: 0,
212
+ cost: {
213
+ input: 0,
214
+ output: 0,
215
+ cacheRead: 0,
216
+ cacheWrite: 0,
217
+ total: 0
218
+ }
219
+ },
220
+ stopReason: "pending",
221
+ timestamp: Date.now()
222
+ };
223
+ try {
224
+ const apiKey = options?.apiKey;
225
+ if (!apiKey) throw new Error(`No API key for provider: ${model.provider}`);
226
+ const accountId = extractAccountId(apiKey);
227
+ const grammarToolInputProperties = createGrammarToolInputProperties(context.tools, model.compat?.supportsOpenAIGrammarTools ?? false);
228
+ const cacheSessionId = options?.cacheRetention === "none" ? void 0 : options?.sessionId;
229
+ const codexSessionId = clampOpenAIPromptCacheKey(cacheSessionId);
230
+ let body = buildRequestBody(model, context, options, codexSessionId, grammarToolInputProperties);
231
+ const nextBody = await options?.onPayload?.(body, model);
232
+ if (nextBody !== void 0) body = nextBody;
233
+ const websocketRequestId = codexSessionId || uuidv7();
234
+ const sseHeaders = buildSSEHeaders(model.headers, options?.headers, accountId, apiKey, codexSessionId);
235
+ const websocketHeaders = buildWebSocketHeaders(model.headers, options?.headers, accountId, apiKey, websocketRequestId);
236
+ const bodyJson = JSON.stringify(body);
237
+ const httpTimeoutMs = normalizeTimeoutMs(options?.timeoutMs);
238
+ const websocketConnectTimeoutMs = normalizeTimeoutMs(options?.websocketConnectTimeoutMs);
239
+ const transport = options?.transport || "auto";
240
+ let startEmitted = false;
241
+ const websocketDisabledForSession = transport !== "sse" && isWebSocketSseFallbackActive(cacheSessionId);
242
+ if (websocketDisabledForSession) recordWebSocketSseFallback(cacheSessionId);
243
+ if (transport !== "sse" && !websocketDisabledForSession) {
244
+ let websocketStarted = false;
245
+ let retriedWebSocketConnectionLimit = false;
246
+ let retriedMissingWebSocketContinuation = false;
247
+ while (true) {
248
+ websocketStarted = false;
249
+ try {
250
+ await processWebSocketStream(resolveCodexWebSocketUrl(model.baseUrl), body, websocketHeaders, output, stream, model, () => {
251
+ websocketStarted = true;
252
+ if (!startEmitted) {
253
+ startEmitted = true;
254
+ stream.push({
255
+ type: "start",
256
+ partial: output
257
+ });
258
+ }
259
+ }, httpTimeoutMs, websocketConnectTimeoutMs, cacheSessionId, accountId, grammarToolInputProperties, options);
260
+ if (options?.signal?.aborted) throw new Error("Request was aborted");
261
+ assertSuccessfulOutput(output);
262
+ stream.push({
263
+ type: "done",
264
+ reason: output.stopReason,
265
+ message: output
266
+ });
267
+ stream.end();
268
+ return;
269
+ } catch (error) {
270
+ const aborted = options?.signal?.aborted;
271
+ const connectionLimitBeforeStart = !websocketStarted && isWebSocketConnectionLimitReachedError(error);
272
+ const previousResponseNotFound = isPreviousResponseNotFoundError(error);
273
+ if (!aborted && previousResponseNotFound && !retriedMissingWebSocketContinuation) {
274
+ retriedMissingWebSocketContinuation = true;
275
+ continue;
276
+ }
277
+ if (!aborted && connectionLimitBeforeStart && !retriedWebSocketConnectionLimit) {
278
+ retriedWebSocketConnectionLimit = true;
279
+ continue;
280
+ }
281
+ if (aborted || isCodexNonTransportError(error) && !connectionLimitBeforeStart) throw error;
282
+ appendAssistantMessageDiagnostic(output, createAssistantMessageDiagnostic("provider_transport_failure", error, {
283
+ configuredTransport: transport,
284
+ fallbackTransport: websocketStarted ? void 0 : "sse",
285
+ eventsEmitted: websocketStarted,
286
+ phase: websocketStarted ? "after_message_stream_start" : "before_message_stream_start",
287
+ requestBytes: new TextEncoder().encode(bodyJson).byteLength
288
+ }));
289
+ recordWebSocketFailure(cacheSessionId, error);
290
+ if (websocketStarted) throw error;
291
+ recordWebSocketSseFallback(cacheSessionId);
292
+ break;
293
+ }
294
+ }
295
+ }
296
+ const compressedBody = compressRequestBodyZstd(bodyJson);
297
+ if (compressedBody) sseHeaders.set("content-encoding", "zstd");
298
+ const sseBody = compressedBody ?? bodyJson;
299
+ let response;
300
+ let lastError;
301
+ const maxRetries = options?.maxRetries ?? DEFAULT_MAX_RETRIES;
302
+ for (let attempt = 0; attempt <= maxRetries; attempt++) {
303
+ if (options?.signal?.aborted) throw new Error("Request was aborted");
304
+ try {
305
+ const headerTimeoutSignal = httpTimeoutMs !== void 0 && httpTimeoutMs > 0 ? AbortSignal.timeout(httpTimeoutMs) : void 0;
306
+ const combinedSignal = combineAbortSignals([options?.signal, headerTimeoutSignal]);
307
+ try {
308
+ response = await (options?.fetch ?? globalThis.fetch)(resolveCodexUrl(model.baseUrl), {
309
+ method: "POST",
310
+ headers: sseHeaders,
311
+ body: sseBody,
312
+ signal: combinedSignal.signal
313
+ });
314
+ } catch (error) {
315
+ if (headerTimeoutSignal?.aborted && !options?.signal?.aborted) throw new Error(`Codex SSE response headers timed out after ${httpTimeoutMs}ms`);
316
+ throw error;
317
+ } finally {
318
+ combinedSignal.cleanup();
319
+ }
320
+ await options?.onResponse?.({
321
+ status: response.status,
322
+ headers: headersToRecord(response.headers)
323
+ }, model);
324
+ if (response.ok) break;
325
+ const errorText = await response.text();
326
+ if (attempt < maxRetries && isRetryableError(response.status, errorText)) {
327
+ const retryAfterDelayMs = getRetryAfterDelayMs(response.headers);
328
+ await sleep(retryAfterDelayMs === void 0 ? BASE_DELAY_MS * 2 ** attempt : validateRetryDelayMs(retryAfterDelayMs, options), options?.signal);
329
+ continue;
330
+ }
331
+ const info = await parseErrorResponse(new Response(errorText, {
332
+ status: response.status,
333
+ statusText: response.statusText
334
+ }));
335
+ throw new Error(info.friendlyMessage || info.message);
336
+ } catch (error) {
337
+ if (error instanceof Error) {
338
+ if (error.name === "AbortError" || error.message === "Request was aborted") throw new Error("Request was aborted");
339
+ }
340
+ lastError = error instanceof Error ? error : new Error(String(error));
341
+ if (attempt < maxRetries && !(lastError instanceof RetryDelayExceededError) && !lastError.message.includes("usage limit")) {
342
+ await sleep(BASE_DELAY_MS * 2 ** attempt, options?.signal);
343
+ continue;
344
+ }
345
+ throw lastError;
346
+ }
347
+ }
348
+ if (!response?.ok) throw lastError ?? /* @__PURE__ */ new Error("Failed after retries");
349
+ if (!response.body) throw new Error("No response body");
350
+ if (!startEmitted) {
351
+ startEmitted = true;
352
+ stream.push({
353
+ type: "start",
354
+ partial: output
355
+ });
356
+ }
357
+ await processStream(response, output, stream, model, grammarToolInputProperties, options);
358
+ if (options?.signal?.aborted) throw new Error("Request was aborted");
359
+ assertSuccessfulOutput(output);
360
+ stream.push({
361
+ type: "done",
362
+ reason: output.stopReason,
363
+ message: output
364
+ });
365
+ stream.end();
366
+ } catch (error) {
367
+ for (const block of output.content) {
368
+ delete block.partialJson;
369
+ delete block.customInput;
370
+ }
371
+ output.stopReason = options?.signal?.aborted ? "aborted" : "error";
372
+ output.errorMessage = formatProviderError(normalizeProviderError(error));
373
+ stream.push({
374
+ type: "error",
375
+ reason: output.stopReason,
376
+ error: output
377
+ });
378
+ stream.end();
379
+ }
380
+ })();
381
+ return stream;
382
+ };
383
+ const streamSimple = (model, context, options) => {
384
+ const apiKey = options?.apiKey;
385
+ if (!apiKey) throw new Error(`No API key for provider: ${model.provider}`);
386
+ const base = buildBaseOptions(model, context, options, apiKey);
387
+ const clampedReasoning = options?.reasoning ? clampThinkingLevel(model, options.reasoning) : void 0;
388
+ const reasoningEffort = clampedReasoning === "off" ? void 0 : clampedReasoning;
389
+ return stream(model, context, {
390
+ ...base,
391
+ reasoningEffort
392
+ });
393
+ };
394
+ function buildRequestBody(model, context, options, cacheSessionId, grammarToolInputProperties = createGrammarToolInputProperties(context.tools, model.compat?.supportsOpenAIGrammarTools ?? false)) {
395
+ const supportsStrictMode = model.compat?.supportsStrictMode ?? true;
396
+ const supportsOpenAIGrammarTools = model.compat?.supportsOpenAIGrammarTools ?? false;
397
+ const toolPlacement = splitDeferredTools(context, model.compat?.supportsToolSearch ?? false);
398
+ const messages = convertResponsesMessages(model, context, CODEX_TOOL_CALL_PROVIDERS, {
399
+ includeSystemPrompt: false,
400
+ grammarToolInputProperties,
401
+ deferredTools: toolPlacement.deferred,
402
+ toolOptions: {
403
+ strict: null,
404
+ supportsStrictMode,
405
+ supportsOpenAIGrammarTools
406
+ }
407
+ });
408
+ const body = {
409
+ model: model.id,
410
+ store: false,
411
+ stream: true,
412
+ instructions: context.systemPrompt || "You are a helpful assistant.",
413
+ input: messages,
414
+ text: { verbosity: options?.textVerbosity || "low" },
415
+ include: ["reasoning.encrypted_content"],
416
+ prompt_cache_key: cacheSessionId,
417
+ tool_choice: options?.toolChoice ?? "auto",
418
+ parallel_tool_calls: true
419
+ };
420
+ if (options?.temperature !== void 0) body.temperature = options.temperature;
421
+ if (options?.serviceTier !== void 0) body.service_tier = options.serviceTier;
422
+ if (toolPlacement.immediate.length > 0) body.tools = convertResponsesTools(toolPlacement.immediate, {
423
+ strict: null,
424
+ supportsStrictMode,
425
+ supportsOpenAIGrammarTools
426
+ });
427
+ if (options?.reasoningEffort !== void 0) {
428
+ const effort = options.reasoningEffort === "none" ? model.thinkingLevelMap?.off ?? "none" : model.thinkingLevelMap?.[options.reasoningEffort] ?? options.reasoningEffort;
429
+ if (effort !== null) body.reasoning = {
430
+ effort,
431
+ summary: options.reasoningSummary ?? "auto"
432
+ };
433
+ }
434
+ return body;
435
+ }
436
+ function getServiceTierCostMultiplier(model, serviceTier) {
437
+ switch (serviceTier) {
438
+ case "flex": return .5;
439
+ case "priority": return model.id === "gpt-5.5" ? 2.5 : 2;
440
+ default: return 1;
441
+ }
442
+ }
443
+ function applyServiceTierPricing(usage, serviceTier, model) {
444
+ const multiplier = getServiceTierCostMultiplier(model, serviceTier);
445
+ if (multiplier === 1) return;
446
+ usage.cost.input *= multiplier;
447
+ usage.cost.output *= multiplier;
448
+ usage.cost.cacheRead *= multiplier;
449
+ usage.cost.cacheWrite *= multiplier;
450
+ usage.cost.total = usage.cost.input + usage.cost.output + usage.cost.cacheRead + usage.cost.cacheWrite;
451
+ }
452
+ function resolveCodexServiceTier(responseServiceTier, requestServiceTier) {
453
+ if (responseServiceTier === "default" && (requestServiceTier === "flex" || requestServiceTier === "priority")) return requestServiceTier;
454
+ return responseServiceTier ?? requestServiceTier;
455
+ }
456
+ function resolveCodexUrl(baseUrl) {
457
+ const normalized = (baseUrl && baseUrl.trim().length > 0 ? baseUrl : DEFAULT_CODEX_BASE_URL).replace(/\/+$/, "");
458
+ if (normalized.endsWith("/codex/responses")) return normalized;
459
+ if (normalized.endsWith("/codex")) return `${normalized}/responses`;
460
+ return `${normalized}/codex/responses`;
461
+ }
462
+ function resolveCodexWebSocketUrl(baseUrl) {
463
+ const url = new URL(resolveCodexUrl(baseUrl));
464
+ if (url.protocol === "https:") url.protocol = "wss:";
465
+ if (url.protocol === "http:") url.protocol = "ws:";
466
+ return url.toString();
467
+ }
468
+ async function processStream(response, output, stream, model, grammarToolInputProperties, options) {
469
+ await processResponsesStream(mapCodexEvents(parseSSE(response, options?.signal)), output, stream, model, {
470
+ serviceTier: options?.serviceTier,
471
+ grammarToolInputProperties,
472
+ resolveServiceTier: resolveCodexServiceTier,
473
+ applyServiceTierPricing: (usage, serviceTier) => applyServiceTierPricing(usage, serviceTier, model)
474
+ });
475
+ }
476
+ var CodexApiError = class extends Error {
477
+ code;
478
+ payload;
479
+ constructor(message, options) {
480
+ super(message);
481
+ this.name = "CodexApiError";
482
+ this.code = options?.code;
483
+ this.payload = options?.payload;
484
+ this.cause = options?.cause;
485
+ }
486
+ };
487
+ var CodexProtocolError = class extends Error {
488
+ payload;
489
+ constructor(message, options) {
490
+ super(message);
491
+ this.name = "CodexProtocolError";
492
+ this.payload = options?.payload;
493
+ this.cause = options?.cause;
494
+ }
495
+ };
496
+ function isCodexNonTransportError(error) {
497
+ return error instanceof CodexApiError || error instanceof CodexProtocolError;
498
+ }
499
+ function isWebSocketConnectionLimitReachedError(error) {
500
+ return error instanceof CodexApiError && error.code === WEBSOCKET_CONNECTION_LIMIT_REACHED_CODE;
501
+ }
502
+ function isPreviousResponseNotFoundError(error) {
503
+ return error instanceof CodexApiError && error.code === PREVIOUS_RESPONSE_NOT_FOUND_CODE;
504
+ }
505
+ function extractCodexEventError(event) {
506
+ const nested = event.error && typeof event.error === "object" ? event.error : void 0;
507
+ return {
508
+ code: typeof event.code === "string" ? event.code : typeof nested?.code === "string" ? nested.code : void 0,
509
+ message: typeof event.message === "string" ? event.message : typeof nested?.message === "string" ? nested.message : void 0
510
+ };
511
+ }
512
+ async function* mapCodexEvents(events) {
513
+ for await (const event of events) {
514
+ const type = typeof event.type === "string" ? event.type : void 0;
515
+ if (!type) continue;
516
+ if (type === "error") {
517
+ const { code, message } = extractCodexEventError(event);
518
+ throw new CodexApiError(`Codex error: ${message || code || JSON.stringify(event)}`, {
519
+ code,
520
+ payload: event
521
+ });
522
+ }
523
+ if (type === "response.failed") {
524
+ const response = event.response;
525
+ const code = response?.error?.code;
526
+ const message = response?.error?.message;
527
+ throw new CodexApiError(message || "Codex response failed", {
528
+ code,
529
+ payload: event
530
+ });
531
+ }
532
+ if (type === "response.done" || type === "response.completed" || type === "response.incomplete") {
533
+ const response = event.response;
534
+ const normalizedResponse = response ? {
535
+ ...response,
536
+ status: normalizeCodexStatus(response.status)
537
+ } : response;
538
+ yield {
539
+ ...event,
540
+ type: "response.completed",
541
+ response: normalizedResponse
542
+ };
543
+ return;
544
+ }
545
+ yield event;
546
+ }
547
+ }
548
+ function normalizeCodexStatus(status) {
549
+ if (typeof status !== "string") return void 0;
550
+ return CODEX_RESPONSE_STATUSES.has(status) ? status : void 0;
551
+ }
552
+ async function* parseSSE(response, signal) {
553
+ if (!response.body) return;
554
+ const reader = response.body.getReader();
555
+ const decoder = new TextDecoder();
556
+ let buffer = "";
557
+ const onAbort = () => {
558
+ reader.cancel().catch(() => {});
559
+ };
560
+ signal?.addEventListener("abort", onAbort, { once: true });
561
+ try {
562
+ while (true) {
563
+ if (signal?.aborted) throw new Error("Request was aborted");
564
+ const { done, value } = await reader.read();
565
+ if (signal?.aborted) throw new Error("Request was aborted");
566
+ if (done) break;
567
+ buffer += decoder.decode(value, { stream: true });
568
+ let idx = buffer.indexOf("\n\n");
569
+ while (idx !== -1) {
570
+ const chunk = buffer.slice(0, idx);
571
+ buffer = buffer.slice(idx + 2);
572
+ const dataLines = chunk.split("\n").filter((l) => l.startsWith("data:")).map((l) => l.slice(5).trim());
573
+ if (dataLines.length > 0) {
574
+ const data = dataLines.join("\n").trim();
575
+ if (data && data !== "[DONE]") try {
576
+ yield JSON.parse(data);
577
+ } catch (cause) {
578
+ throw new CodexProtocolError(`Invalid Codex SSE JSON: ${formatThrownValue(cause)}`, {
579
+ cause,
580
+ payload: data
581
+ });
582
+ }
583
+ }
584
+ idx = buffer.indexOf("\n\n");
585
+ }
586
+ }
587
+ } finally {
588
+ signal?.removeEventListener("abort", onAbort);
589
+ try {
590
+ await reader.cancel();
591
+ } catch {}
592
+ try {
593
+ reader.releaseLock();
594
+ } catch {}
595
+ }
596
+ }
597
+ const OPENAI_BETA_RESPONSES_WEBSOCKETS = "responses_websockets=2026-02-06";
598
+ const SESSION_WEBSOCKET_CACHE_TTL_MS = 3e5;
599
+ const SESSION_WEBSOCKET_MAX_AGE_MS = 33e5;
600
+ const websocketSessionCache = /* @__PURE__ */ new Map();
601
+ const websocketDebugStats = /* @__PURE__ */ new Map();
602
+ const websocketSseFallbackSessions = /* @__PURE__ */ new Set();
603
+ function getOrCreateWebSocketDebugStats(sessionId) {
604
+ let stats = websocketDebugStats.get(sessionId);
605
+ if (!stats) {
606
+ stats = {
607
+ requests: 0,
608
+ connectionsCreated: 0,
609
+ connectionsReused: 0,
610
+ cachedContextRequests: 0,
611
+ storeTrueRequests: 0,
612
+ fullContextRequests: 0,
613
+ deltaRequests: 0,
614
+ lastInputItems: 0,
615
+ websocketFailures: 0,
616
+ sseFallbacks: 0
617
+ };
618
+ websocketDebugStats.set(sessionId, stats);
619
+ }
620
+ return stats;
621
+ }
622
+ function getOpenAICodexWebSocketDebugStats(sessionId) {
623
+ const stats = websocketDebugStats.get(sessionId);
624
+ return stats ? { ...stats } : void 0;
625
+ }
626
+ function resetOpenAICodexWebSocketDebugStats(sessionId) {
627
+ if (sessionId) {
628
+ websocketDebugStats.delete(sessionId);
629
+ websocketSseFallbackSessions.delete(sessionId);
630
+ return;
631
+ }
632
+ websocketDebugStats.clear();
633
+ websocketSseFallbackSessions.clear();
634
+ }
635
+ function closeOpenAICodexWebSocketSessions(sessionId) {
636
+ const closeEntry = (entry) => {
637
+ if (entry.idleTimer) clearTimeout(entry.idleTimer);
638
+ closeWebSocketSilently(entry.socket, 1e3, "debug_close");
639
+ };
640
+ if (sessionId) {
641
+ for (const entry of websocketSessionCache.get(sessionId)?.values() ?? []) closeEntry(entry);
642
+ websocketSessionCache.delete(sessionId);
643
+ return;
644
+ }
645
+ for (const accountEntries of websocketSessionCache.values()) for (const entry of accountEntries.values()) closeEntry(entry);
646
+ websocketSessionCache.clear();
647
+ }
648
+ registerSessionResourceCleanup(closeOpenAICodexWebSocketSessions);
649
+ function isWebSocketSseFallbackActive(sessionId) {
650
+ return sessionId ? websocketSseFallbackSessions.has(sessionId) : false;
651
+ }
652
+ function recordWebSocketSseFallback(sessionId) {
653
+ if (!sessionId) return;
654
+ const stats = getOrCreateWebSocketDebugStats(sessionId);
655
+ stats.sseFallbacks++;
656
+ stats.websocketFallbackActive = isWebSocketSseFallbackActive(sessionId);
657
+ }
658
+ function recordWebSocketFailure(sessionId, error) {
659
+ if (!sessionId) return;
660
+ websocketSseFallbackSessions.add(sessionId);
661
+ const stats = getOrCreateWebSocketDebugStats(sessionId);
662
+ stats.websocketFailures++;
663
+ stats.lastWebSocketError = formatThrownValue(error);
664
+ stats.websocketFallbackActive = true;
665
+ }
666
+ let _cachedWebsocket = null;
667
+ async function getWebSocketConstructor(env) {
668
+ if (!env && _cachedWebsocket) return _cachedWebsocket;
669
+ if (typeof process !== "undefined" && process.versions?.bun) {
670
+ const WebSocketWithProxy = class extends WebSocket {
671
+ constructor(url, options) {
672
+ let _opts = {};
673
+ if (Array.isArray(options) || typeof options === "string") _opts = { protocols: options };
674
+ else _opts = { ...options };
675
+ const proxyUrl = resolveHttpProxyUrlForTarget(url.toString().replace(/^wss:/, "https:").replace(/^ws:/, "http:"), env);
676
+ super(url, {
677
+ ..._opts,
678
+ ...proxyUrl ? { proxy: proxyUrl.toString() } : {}
679
+ });
680
+ }
681
+ };
682
+ if (!env) _cachedWebsocket = WebSocketWithProxy;
683
+ return WebSocketWithProxy;
684
+ }
685
+ const ctor = globalThis.WebSocket;
686
+ if (typeof ctor !== "function") return null;
687
+ return ctor;
688
+ }
689
+ var WebSocketCloseError = class extends Error {
690
+ code;
691
+ reason;
692
+ wasClean;
693
+ constructor(message, options) {
694
+ super(message);
695
+ this.name = "WebSocketCloseError";
696
+ this.code = options?.code;
697
+ this.reason = options?.reason;
698
+ this.wasClean = options?.wasClean;
699
+ }
700
+ };
701
+ function getWebSocketReadyState(socket) {
702
+ const readyState = socket.readyState;
703
+ return typeof readyState === "number" ? readyState : void 0;
704
+ }
705
+ function isWebSocketReusable(socket) {
706
+ const readyState = getWebSocketReadyState(socket);
707
+ return readyState === void 0 || readyState === 1;
708
+ }
709
+ function isWebSocketSessionExpired(entry) {
710
+ return Date.now() - entry.createdAt >= SESSION_WEBSOCKET_MAX_AGE_MS;
711
+ }
712
+ function closeWebSocketSilently(socket, code = 1e3, reason = "done") {
713
+ try {
714
+ socket.close(code, reason);
715
+ } catch {}
716
+ }
717
+ function scheduleSessionWebSocketExpiry(sessionId, accountId, entry) {
718
+ if (entry.idleTimer) clearTimeout(entry.idleTimer);
719
+ entry.idleTimer = setTimeout(() => {
720
+ if (entry.busy) return;
721
+ closeWebSocketSilently(entry.socket, 1e3, "idle_timeout");
722
+ const accountEntries = websocketSessionCache.get(sessionId);
723
+ if (accountEntries?.get(accountId) === entry) accountEntries.delete(accountId);
724
+ if (accountEntries?.size === 0) websocketSessionCache.delete(sessionId);
725
+ }, SESSION_WEBSOCKET_CACHE_TTL_MS);
726
+ }
727
+ async function connectWebSocket(url, headers, signal, connectTimeoutMs = DEFAULT_WEBSOCKET_CONNECT_TIMEOUT_MS, env) {
728
+ const WebSocketCtor = await getWebSocketConstructor(env);
729
+ if (!WebSocketCtor) throw new Error("WebSocket transport is not available in this runtime");
730
+ const wsHeaders = headersToRecord(headers);
731
+ delete wsHeaders["OpenAI-Beta"];
732
+ return new Promise((resolve, reject) => {
733
+ let settled = false;
734
+ let timeout;
735
+ let socket;
736
+ try {
737
+ socket = new WebSocketCtor(url, { headers: wsHeaders });
738
+ } catch (error) {
739
+ reject(error instanceof Error ? error : new Error(String(error)));
740
+ return;
741
+ }
742
+ const cleanup = () => {
743
+ if (timeout) {
744
+ clearTimeout(timeout);
745
+ timeout = void 0;
746
+ }
747
+ socket.removeEventListener("open", onOpen);
748
+ socket.removeEventListener("error", onError);
749
+ socket.removeEventListener("close", onClose);
750
+ signal?.removeEventListener("abort", onAbort);
751
+ };
752
+ const fail = (error, closeReason) => {
753
+ if (settled) return;
754
+ settled = true;
755
+ cleanup();
756
+ if (closeReason) closeWebSocketSilently(socket, 1e3, closeReason);
757
+ reject(error);
758
+ };
759
+ const onOpen = () => {
760
+ if (settled) return;
761
+ settled = true;
762
+ cleanup();
763
+ resolve(socket);
764
+ };
765
+ const onError = (event) => {
766
+ fail(extractWebSocketError(event));
767
+ };
768
+ const onClose = (event) => {
769
+ fail(extractWebSocketCloseError(event));
770
+ };
771
+ const onAbort = () => {
772
+ fail(/* @__PURE__ */ new Error("Request was aborted"), "aborted");
773
+ };
774
+ socket.addEventListener("open", onOpen);
775
+ socket.addEventListener("error", onError);
776
+ socket.addEventListener("close", onClose);
777
+ signal?.addEventListener("abort", onAbort);
778
+ if (connectTimeoutMs > 0) timeout = setTimeout(() => {
779
+ fail(/* @__PURE__ */ new Error(`WebSocket connect timeout after ${connectTimeoutMs}ms`), "connect_timeout");
780
+ }, connectTimeoutMs);
781
+ if (signal?.aborted) onAbort();
782
+ });
783
+ }
784
+ async function acquireWebSocket(url, headers, sessionId, accountId, signal, connectTimeoutMs, env) {
785
+ if (!sessionId) {
786
+ const socket = await connectWebSocket(url, headers, signal, connectTimeoutMs, env);
787
+ return {
788
+ socket,
789
+ reused: false,
790
+ release: () => closeWebSocketSilently(socket)
791
+ };
792
+ }
793
+ let accountEntries = websocketSessionCache.get(sessionId);
794
+ const cached = accountEntries?.get(accountId);
795
+ if (cached) {
796
+ if (cached.idleTimer) {
797
+ clearTimeout(cached.idleTimer);
798
+ cached.idleTimer = void 0;
799
+ }
800
+ if (!cached.busy && isWebSocketSessionExpired(cached)) {
801
+ closeWebSocketSilently(cached.socket, 1e3, "connection_age_limit");
802
+ accountEntries?.delete(accountId);
803
+ if (accountEntries?.size === 0) websocketSessionCache.delete(sessionId);
804
+ } else if (!cached.busy && isWebSocketReusable(cached.socket)) {
805
+ cached.busy = true;
806
+ return {
807
+ socket: cached.socket,
808
+ entry: cached,
809
+ reused: true,
810
+ release: ({ keep } = {}) => {
811
+ if (!keep || !isWebSocketReusable(cached.socket)) {
812
+ closeWebSocketSilently(cached.socket);
813
+ const currentEntries = websocketSessionCache.get(sessionId);
814
+ if (currentEntries?.get(accountId) === cached) currentEntries.delete(accountId);
815
+ if (currentEntries?.size === 0) websocketSessionCache.delete(sessionId);
816
+ return;
817
+ }
818
+ cached.busy = false;
819
+ scheduleSessionWebSocketExpiry(sessionId, accountId, cached);
820
+ }
821
+ };
822
+ }
823
+ if (cached.busy) {
824
+ const socket = await connectWebSocket(url, headers, signal, connectTimeoutMs, env);
825
+ return {
826
+ socket,
827
+ reused: false,
828
+ release: () => {
829
+ closeWebSocketSilently(socket);
830
+ }
831
+ };
832
+ }
833
+ if (!isWebSocketReusable(cached.socket)) {
834
+ closeWebSocketSilently(cached.socket);
835
+ accountEntries?.delete(accountId);
836
+ if (accountEntries?.size === 0) websocketSessionCache.delete(sessionId);
837
+ }
838
+ }
839
+ const socket = await connectWebSocket(url, headers, signal, connectTimeoutMs, env);
840
+ const entry = {
841
+ socket,
842
+ busy: true,
843
+ createdAt: Date.now()
844
+ };
845
+ accountEntries = websocketSessionCache.get(sessionId);
846
+ if (!accountEntries) {
847
+ accountEntries = /* @__PURE__ */ new Map();
848
+ websocketSessionCache.set(sessionId, accountEntries);
849
+ }
850
+ accountEntries.set(accountId, entry);
851
+ return {
852
+ socket,
853
+ entry,
854
+ reused: false,
855
+ release: ({ keep } = {}) => {
856
+ if (!keep || !isWebSocketReusable(entry.socket)) {
857
+ closeWebSocketSilently(entry.socket);
858
+ if (entry.idleTimer) clearTimeout(entry.idleTimer);
859
+ const currentEntries = websocketSessionCache.get(sessionId);
860
+ if (currentEntries?.get(accountId) === entry) currentEntries.delete(accountId);
861
+ if (currentEntries?.size === 0) websocketSessionCache.delete(sessionId);
862
+ return;
863
+ }
864
+ entry.busy = false;
865
+ scheduleSessionWebSocketExpiry(sessionId, accountId, entry);
866
+ }
867
+ };
868
+ }
869
+ function extractWebSocketError(event) {
870
+ if (event && typeof event === "object") {
871
+ const message = "message" in event ? event.message : void 0;
872
+ if (typeof message === "string" && message.length > 0) return new Error(message);
873
+ const nestedError = "error" in event ? event.error : void 0;
874
+ if (nestedError instanceof Error && nestedError.message.length > 0) return nestedError;
875
+ if (nestedError && typeof nestedError === "object" && "message" in nestedError) {
876
+ const nestedMessage = nestedError.message;
877
+ if (typeof nestedMessage === "string" && nestedMessage.length > 0) return new Error(nestedMessage);
878
+ }
879
+ }
880
+ return /* @__PURE__ */ new Error("WebSocket error");
881
+ }
882
+ function extractWebSocketCloseError(event) {
883
+ if (event && typeof event === "object") {
884
+ const code = "code" in event ? event.code : void 0;
885
+ const reason = "reason" in event ? event.reason : void 0;
886
+ const wasClean = "wasClean" in event ? event.wasClean : void 0;
887
+ const codeText = typeof code === "number" ? ` ${code}` : "";
888
+ let reasonText = typeof reason === "string" && reason.length > 0 ? ` ${reason}` : "";
889
+ if (!reasonText && code === WEBSOCKET_MESSAGE_TOO_BIG_CLOSE_CODE) reasonText = " message too big";
890
+ return new WebSocketCloseError(`WebSocket closed${codeText}${reasonText}`.trim(), {
891
+ code: typeof code === "number" ? code : void 0,
892
+ reason: typeof reason === "string" && reason.length > 0 ? reason : void 0,
893
+ wasClean: typeof wasClean === "boolean" ? wasClean : void 0
894
+ });
895
+ }
896
+ return /* @__PURE__ */ new Error("WebSocket closed");
897
+ }
898
+ async function decodeWebSocketData(data) {
899
+ if (typeof data === "string") return data;
900
+ if (data instanceof ArrayBuffer) return new TextDecoder().decode(new Uint8Array(data));
901
+ if (ArrayBuffer.isView(data)) {
902
+ const view = data;
903
+ return new TextDecoder().decode(new Uint8Array(view.buffer, view.byteOffset, view.byteLength));
904
+ }
905
+ if (data && typeof data === "object" && "arrayBuffer" in data) {
906
+ const arrayBuffer = await data.arrayBuffer();
907
+ return new TextDecoder().decode(new Uint8Array(arrayBuffer));
908
+ }
909
+ return null;
910
+ }
911
+ async function* parseWebSocket(socket, signal, idleTimeoutMs) {
912
+ const queue = [];
913
+ let pending = null;
914
+ let done = false;
915
+ let failed = null;
916
+ let sawCompletion = false;
917
+ const wake = () => {
918
+ if (!pending) return;
919
+ const resolve = pending;
920
+ pending = null;
921
+ resolve();
922
+ };
923
+ const onMessage = (event) => {
924
+ (async () => {
925
+ let text = null;
926
+ try {
927
+ if (!event || typeof event !== "object" || !("data" in event)) return;
928
+ text = await decodeWebSocketData(event.data);
929
+ if (!text) return;
930
+ const parsed = JSON.parse(text);
931
+ const type = typeof parsed.type === "string" ? parsed.type : "";
932
+ if (type === "response.completed" || type === "response.done" || type === "response.incomplete") {
933
+ sawCompletion = true;
934
+ done = true;
935
+ }
936
+ queue.push(parsed);
937
+ wake();
938
+ } catch (cause) {
939
+ failed = new CodexProtocolError(`Invalid Codex WebSocket JSON: ${formatThrownValue(cause)}`, {
940
+ cause,
941
+ payload: text
942
+ });
943
+ done = true;
944
+ wake();
945
+ }
946
+ })();
947
+ };
948
+ const onError = (event) => {
949
+ failed = extractWebSocketError(event);
950
+ done = true;
951
+ wake();
952
+ };
953
+ const onClose = (event) => {
954
+ if (sawCompletion) {
955
+ done = true;
956
+ wake();
957
+ return;
958
+ }
959
+ if (!failed) failed = extractWebSocketCloseError(event);
960
+ done = true;
961
+ wake();
962
+ };
963
+ const onAbort = () => {
964
+ failed = /* @__PURE__ */ new Error("Request was aborted");
965
+ done = true;
966
+ wake();
967
+ };
968
+ socket.addEventListener("message", onMessage);
969
+ socket.addEventListener("error", onError);
970
+ socket.addEventListener("close", onClose);
971
+ signal?.addEventListener("abort", onAbort);
972
+ try {
973
+ while (true) {
974
+ if (signal?.aborted) throw new Error("Request was aborted");
975
+ if (queue.length > 0) {
976
+ yield queue.shift();
977
+ continue;
978
+ }
979
+ if (done) break;
980
+ let timeout;
981
+ await new Promise((resolve, reject) => {
982
+ pending = resolve;
983
+ if (idleTimeoutMs !== void 0 && idleTimeoutMs > 0) timeout = setTimeout(() => {
984
+ const error = /* @__PURE__ */ new Error(`WebSocket idle timeout after ${idleTimeoutMs}ms`);
985
+ failed = error;
986
+ done = true;
987
+ pending = null;
988
+ closeWebSocketSilently(socket, 1e3, "idle_timeout");
989
+ reject(error);
990
+ }, idleTimeoutMs);
991
+ }).finally(() => {
992
+ if (timeout) clearTimeout(timeout);
993
+ });
994
+ }
995
+ if (failed) throw failed;
996
+ if (!sawCompletion) throw new Error("WebSocket stream closed before response.completed");
997
+ } finally {
998
+ socket.removeEventListener("message", onMessage);
999
+ socket.removeEventListener("error", onError);
1000
+ socket.removeEventListener("close", onClose);
1001
+ signal?.removeEventListener("abort", onAbort);
1002
+ }
1003
+ }
1004
+ function requestBodyWithoutInput(body) {
1005
+ const { input: _input, previous_response_id: _previousResponseId, ...rest } = body;
1006
+ return rest;
1007
+ }
1008
+ function responseInputsEqual(a, b) {
1009
+ return JSON.stringify(a ?? []) === JSON.stringify(b ?? []);
1010
+ }
1011
+ function requestBodiesMatchExceptInput(a, b) {
1012
+ return JSON.stringify(requestBodyWithoutInput(a)) === JSON.stringify(requestBodyWithoutInput(b));
1013
+ }
1014
+ function getCachedWebSocketInputDelta(body, continuation) {
1015
+ if (!requestBodiesMatchExceptInput(body, continuation.lastRequestBody)) return;
1016
+ const currentInput = body.input ?? [];
1017
+ const baseline = [...continuation.lastRequestBody.input ?? [], ...continuation.lastResponseItems];
1018
+ if (currentInput.length < baseline.length) return;
1019
+ if (!responseInputsEqual(currentInput.slice(0, baseline.length), baseline)) return;
1020
+ return currentInput.slice(baseline.length);
1021
+ }
1022
+ function buildCachedWebSocketRequestBody(entry, body) {
1023
+ const continuation = entry.continuation;
1024
+ if (!continuation) return body;
1025
+ const delta = getCachedWebSocketInputDelta(body, continuation);
1026
+ if (!delta || !continuation.lastResponseId) {
1027
+ entry.continuation = void 0;
1028
+ return body;
1029
+ }
1030
+ return {
1031
+ ...body,
1032
+ previous_response_id: continuation.lastResponseId,
1033
+ input: delta
1034
+ };
1035
+ }
1036
+ async function* startWebSocketOutputOnFirstEvent(events, onStart) {
1037
+ let started = false;
1038
+ for await (const event of events) {
1039
+ if (!started) {
1040
+ started = true;
1041
+ onStart();
1042
+ }
1043
+ yield event;
1044
+ }
1045
+ }
1046
+ async function processWebSocketStream(url, body, headers, output, stream, model, onStart, idleTimeoutMs, websocketConnectTimeoutMs, cacheSessionId, accountId, grammarToolInputProperties, options) {
1047
+ const { socket, entry, reused, release } = await acquireWebSocket(url, headers, cacheSessionId, accountId, options?.signal, websocketConnectTimeoutMs, options?.env);
1048
+ let keepConnection = true;
1049
+ const useCachedContext = options?.transport === "websocket-cached" || options?.transport === "auto";
1050
+ const fullBody = body;
1051
+ const requestBody = useCachedContext && entry ? buildCachedWebSocketRequestBody(entry, fullBody) : fullBody;
1052
+ const stats = cacheSessionId ? getOrCreateWebSocketDebugStats(cacheSessionId) : void 0;
1053
+ if (stats) {
1054
+ stats.requests++;
1055
+ if (reused) stats.connectionsReused++;
1056
+ else stats.connectionsCreated++;
1057
+ if (useCachedContext) stats.cachedContextRequests++;
1058
+ if (requestBody.store === true) stats.storeTrueRequests++;
1059
+ stats.lastInputItems = requestBody.input?.length ?? 0;
1060
+ if (requestBody.previous_response_id) {
1061
+ stats.deltaRequests++;
1062
+ stats.lastDeltaInputItems = requestBody.input?.length ?? 0;
1063
+ stats.lastPreviousResponseId = requestBody.previous_response_id;
1064
+ } else {
1065
+ stats.fullContextRequests++;
1066
+ stats.lastDeltaInputItems = void 0;
1067
+ stats.lastPreviousResponseId = void 0;
1068
+ }
1069
+ }
1070
+ try {
1071
+ socket.send(JSON.stringify({
1072
+ type: "response.create",
1073
+ ...requestBody
1074
+ }));
1075
+ await processResponsesStream(startWebSocketOutputOnFirstEvent(mapCodexEvents(parseWebSocket(socket, options?.signal, idleTimeoutMs)), onStart), output, stream, model, {
1076
+ serviceTier: options?.serviceTier,
1077
+ grammarToolInputProperties,
1078
+ resolveServiceTier: resolveCodexServiceTier,
1079
+ applyServiceTierPricing: (usage, serviceTier) => applyServiceTierPricing(usage, serviceTier, model)
1080
+ });
1081
+ if (options?.signal?.aborted) keepConnection = false;
1082
+ else if (useCachedContext && entry && output.responseId) {
1083
+ const responseItems = convertResponsesMessages(model, { messages: [output] }, CODEX_TOOL_CALL_PROVIDERS, {
1084
+ includeSystemPrompt: false,
1085
+ grammarToolInputProperties
1086
+ }).filter((item) => item.type !== "function_call_output" && item.type !== "custom_tool_call_output");
1087
+ entry.continuation = {
1088
+ lastRequestBody: fullBody,
1089
+ lastResponseId: output.responseId,
1090
+ lastResponseItems: responseItems
1091
+ };
1092
+ }
1093
+ } catch (error) {
1094
+ if (entry) entry.continuation = void 0;
1095
+ keepConnection = false;
1096
+ throw error;
1097
+ } finally {
1098
+ release({ keep: keepConnection });
1099
+ }
1100
+ }
1101
+ async function parseErrorResponse(response) {
1102
+ const raw = await response.text();
1103
+ let message = raw || response.statusText || "Request failed";
1104
+ let friendlyMessage;
1105
+ try {
1106
+ const err = JSON.parse(raw)?.error;
1107
+ if (err) {
1108
+ const code = err.code || err.type || "";
1109
+ if (/usage_limit_reached|usage_not_included|rate_limit_exceeded/i.test(code) || response.status === 429) {
1110
+ const plan = err.plan_type ? ` (${err.plan_type.toLowerCase()} plan)` : "";
1111
+ const mins = err.resets_at ? Math.max(0, Math.round((err.resets_at * 1e3 - Date.now()) / 6e4)) : void 0;
1112
+ friendlyMessage = `You have hit your ChatGPT usage limit${plan}.${mins !== void 0 ? ` Try again in ~${mins} min.` : ""}`.trim();
1113
+ }
1114
+ message = err.message || friendlyMessage || message;
1115
+ }
1116
+ } catch {}
1117
+ return {
1118
+ message,
1119
+ friendlyMessage
1120
+ };
1121
+ }
1122
+ function extractAccountId(token) {
1123
+ try {
1124
+ const parts = token.split(".");
1125
+ if (parts.length !== 3) throw new Error("Invalid token");
1126
+ const accountId = JSON.parse(atob(parts[1]))?.[JWT_CLAIM_PATH]?.chatgpt_account_id;
1127
+ if (!accountId) throw new Error("No account ID in token");
1128
+ return accountId;
1129
+ } catch {
1130
+ throw new Error("Failed to extract accountId from token");
1131
+ }
1132
+ }
1133
+ function buildBaseCodexHeaders(initHeaders, additionalHeaders, accountId, token) {
1134
+ const headers = new Headers(initHeaders);
1135
+ for (const [key, value] of Object.entries(additionalHeaders || {})) if (value === null) headers.delete(key);
1136
+ else headers.set(key, value);
1137
+ headers.set("Authorization", `Bearer ${token}`);
1138
+ headers.set("chatgpt-account-id", accountId);
1139
+ headers.set("originator", "pi");
1140
+ const userAgent = _os ? `pi (${_os.platform()} ${_os.release()}; ${_os.arch()})` : "pi (browser)";
1141
+ headers.set("User-Agent", userAgent);
1142
+ return headers;
1143
+ }
1144
+ function buildSSEHeaders(initHeaders, additionalHeaders, accountId, token, sessionId) {
1145
+ const headers = buildBaseCodexHeaders(initHeaders, additionalHeaders, accountId, token);
1146
+ headers.set("OpenAI-Beta", "responses=experimental");
1147
+ headers.set("accept", "text/event-stream");
1148
+ headers.set("content-type", "application/json");
1149
+ if (sessionId) {
1150
+ headers.set("session-id", sessionId);
1151
+ headers.set("x-client-request-id", sessionId);
1152
+ }
1153
+ return headers;
1154
+ }
1155
+ function buildWebSocketHeaders(initHeaders, additionalHeaders, accountId, token, requestId) {
1156
+ const headers = buildBaseCodexHeaders(initHeaders, additionalHeaders, accountId, token);
1157
+ headers.delete("accept");
1158
+ headers.delete("content-type");
1159
+ headers.delete("OpenAI-Beta");
1160
+ headers.delete("openai-beta");
1161
+ headers.set("OpenAI-Beta", OPENAI_BETA_RESPONSES_WEBSOCKETS);
1162
+ headers.set("x-client-request-id", requestId);
1163
+ headers.set("session-id", requestId);
1164
+ return headers;
1165
+ }
1166
+ //#endregion
1167
+ export { closeOpenAICodexWebSocketSessions, getOpenAICodexWebSocketDebugStats, resetOpenAICodexWebSocketDebugStats, stream, streamSimple };
1168
+
1169
+ //# sourceMappingURL=openai-codex-responses-vmovNu-q.mjs.map