@lenylvt/pi-ai 0.64.0

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 (167) hide show
  1. package/README.md +203 -0
  2. package/dist/api-registry.d.ts +20 -0
  3. package/dist/api-registry.d.ts.map +1 -0
  4. package/dist/api-registry.js +44 -0
  5. package/dist/api-registry.js.map +1 -0
  6. package/dist/cli.d.ts +3 -0
  7. package/dist/cli.d.ts.map +1 -0
  8. package/dist/cli.js +119 -0
  9. package/dist/cli.js.map +1 -0
  10. package/dist/env-api-keys.d.ts +7 -0
  11. package/dist/env-api-keys.d.ts.map +1 -0
  12. package/dist/env-api-keys.js +13 -0
  13. package/dist/env-api-keys.js.map +1 -0
  14. package/dist/index.d.ts +20 -0
  15. package/dist/index.d.ts.map +1 -0
  16. package/dist/index.js +14 -0
  17. package/dist/index.js.map +1 -0
  18. package/dist/models.d.ts +24 -0
  19. package/dist/models.d.ts.map +1 -0
  20. package/dist/models.generated.d.ts +2332 -0
  21. package/dist/models.generated.d.ts.map +1 -0
  22. package/dist/models.generated.js +2186 -0
  23. package/dist/models.generated.js.map +1 -0
  24. package/dist/models.js +60 -0
  25. package/dist/models.js.map +1 -0
  26. package/dist/oauth.d.ts +2 -0
  27. package/dist/oauth.d.ts.map +1 -0
  28. package/dist/oauth.js +2 -0
  29. package/dist/oauth.js.map +1 -0
  30. package/dist/providers/anthropic.d.ts +40 -0
  31. package/dist/providers/anthropic.d.ts.map +1 -0
  32. package/dist/providers/anthropic.js +749 -0
  33. package/dist/providers/anthropic.js.map +1 -0
  34. package/dist/providers/faux.d.ts +56 -0
  35. package/dist/providers/faux.d.ts.map +1 -0
  36. package/dist/providers/faux.js +367 -0
  37. package/dist/providers/faux.js.map +1 -0
  38. package/dist/providers/github-copilot-headers.d.ts +8 -0
  39. package/dist/providers/github-copilot-headers.d.ts.map +1 -0
  40. package/dist/providers/github-copilot-headers.js +29 -0
  41. package/dist/providers/github-copilot-headers.js.map +1 -0
  42. package/dist/providers/openai-codex-responses.d.ts +9 -0
  43. package/dist/providers/openai-codex-responses.d.ts.map +1 -0
  44. package/dist/providers/openai-codex-responses.js +741 -0
  45. package/dist/providers/openai-codex-responses.js.map +1 -0
  46. package/dist/providers/openai-completions.d.ts +15 -0
  47. package/dist/providers/openai-completions.d.ts.map +1 -0
  48. package/dist/providers/openai-completions.js +687 -0
  49. package/dist/providers/openai-completions.js.map +1 -0
  50. package/dist/providers/openai-responses-shared.d.ts +17 -0
  51. package/dist/providers/openai-responses-shared.d.ts.map +1 -0
  52. package/dist/providers/openai-responses-shared.js +458 -0
  53. package/dist/providers/openai-responses-shared.js.map +1 -0
  54. package/dist/providers/openai-responses.d.ts +13 -0
  55. package/dist/providers/openai-responses.d.ts.map +1 -0
  56. package/dist/providers/openai-responses.js +190 -0
  57. package/dist/providers/openai-responses.js.map +1 -0
  58. package/dist/providers/register-builtins.d.ts +16 -0
  59. package/dist/providers/register-builtins.d.ts.map +1 -0
  60. package/dist/providers/register-builtins.js +140 -0
  61. package/dist/providers/register-builtins.js.map +1 -0
  62. package/dist/providers/simple-options.d.ts +8 -0
  63. package/dist/providers/simple-options.d.ts.map +1 -0
  64. package/dist/providers/simple-options.js +35 -0
  65. package/dist/providers/simple-options.js.map +1 -0
  66. package/dist/providers/transform-messages.d.ts +8 -0
  67. package/dist/providers/transform-messages.d.ts.map +1 -0
  68. package/dist/providers/transform-messages.js +155 -0
  69. package/dist/providers/transform-messages.js.map +1 -0
  70. package/dist/stream.d.ts +8 -0
  71. package/dist/stream.d.ts.map +1 -0
  72. package/dist/stream.js +27 -0
  73. package/dist/stream.js.map +1 -0
  74. package/dist/types.d.ts +283 -0
  75. package/dist/types.d.ts.map +1 -0
  76. package/dist/types.js +2 -0
  77. package/dist/types.js.map +1 -0
  78. package/dist/utils/event-stream.d.ts +21 -0
  79. package/dist/utils/event-stream.d.ts.map +1 -0
  80. package/dist/utils/event-stream.js +81 -0
  81. package/dist/utils/event-stream.js.map +1 -0
  82. package/dist/utils/hash.d.ts +3 -0
  83. package/dist/utils/hash.d.ts.map +1 -0
  84. package/dist/utils/hash.js +14 -0
  85. package/dist/utils/hash.js.map +1 -0
  86. package/dist/utils/json-parse.d.ts +9 -0
  87. package/dist/utils/json-parse.d.ts.map +1 -0
  88. package/dist/utils/json-parse.js +29 -0
  89. package/dist/utils/json-parse.js.map +1 -0
  90. package/dist/utils/oauth/anthropic.d.ts +25 -0
  91. package/dist/utils/oauth/anthropic.d.ts.map +1 -0
  92. package/dist/utils/oauth/anthropic.js +335 -0
  93. package/dist/utils/oauth/anthropic.js.map +1 -0
  94. package/dist/utils/oauth/github-copilot.d.ts +30 -0
  95. package/dist/utils/oauth/github-copilot.d.ts.map +1 -0
  96. package/dist/utils/oauth/github-copilot.js +292 -0
  97. package/dist/utils/oauth/github-copilot.js.map +1 -0
  98. package/dist/utils/oauth/index.d.ts +36 -0
  99. package/dist/utils/oauth/index.d.ts.map +1 -0
  100. package/dist/utils/oauth/index.js +92 -0
  101. package/dist/utils/oauth/index.js.map +1 -0
  102. package/dist/utils/oauth/oauth-page.d.ts +3 -0
  103. package/dist/utils/oauth/oauth-page.d.ts.map +1 -0
  104. package/dist/utils/oauth/oauth-page.js +105 -0
  105. package/dist/utils/oauth/oauth-page.js.map +1 -0
  106. package/dist/utils/oauth/openai-codex.d.ts +34 -0
  107. package/dist/utils/oauth/openai-codex.d.ts.map +1 -0
  108. package/dist/utils/oauth/openai-codex.js +373 -0
  109. package/dist/utils/oauth/openai-codex.js.map +1 -0
  110. package/dist/utils/oauth/pkce.d.ts +13 -0
  111. package/dist/utils/oauth/pkce.d.ts.map +1 -0
  112. package/dist/utils/oauth/pkce.js +31 -0
  113. package/dist/utils/oauth/pkce.js.map +1 -0
  114. package/dist/utils/oauth/types.d.ts +47 -0
  115. package/dist/utils/oauth/types.d.ts.map +1 -0
  116. package/dist/utils/oauth/types.js +2 -0
  117. package/dist/utils/oauth/types.js.map +1 -0
  118. package/dist/utils/overflow.d.ts +53 -0
  119. package/dist/utils/overflow.d.ts.map +1 -0
  120. package/dist/utils/overflow.js +119 -0
  121. package/dist/utils/overflow.js.map +1 -0
  122. package/dist/utils/sanitize-unicode.d.ts +22 -0
  123. package/dist/utils/sanitize-unicode.d.ts.map +1 -0
  124. package/dist/utils/sanitize-unicode.js +26 -0
  125. package/dist/utils/sanitize-unicode.js.map +1 -0
  126. package/dist/utils/typebox-helpers.d.ts +17 -0
  127. package/dist/utils/typebox-helpers.d.ts.map +1 -0
  128. package/dist/utils/typebox-helpers.js +21 -0
  129. package/dist/utils/typebox-helpers.js.map +1 -0
  130. package/dist/utils/validation.d.ts +18 -0
  131. package/dist/utils/validation.d.ts.map +1 -0
  132. package/dist/utils/validation.js +80 -0
  133. package/dist/utils/validation.js.map +1 -0
  134. package/package.json +89 -0
  135. package/src/api-registry.ts +98 -0
  136. package/src/cli.ts +136 -0
  137. package/src/env-api-keys.ts +22 -0
  138. package/src/index.ts +29 -0
  139. package/src/models.generated.ts +2188 -0
  140. package/src/models.ts +82 -0
  141. package/src/oauth.ts +1 -0
  142. package/src/providers/anthropic.ts +905 -0
  143. package/src/providers/faux.ts +498 -0
  144. package/src/providers/github-copilot-headers.ts +37 -0
  145. package/src/providers/openai-codex-responses.ts +929 -0
  146. package/src/providers/openai-completions.ts +811 -0
  147. package/src/providers/openai-responses-shared.ts +513 -0
  148. package/src/providers/openai-responses.ts +251 -0
  149. package/src/providers/register-builtins.ts +232 -0
  150. package/src/providers/simple-options.ts +46 -0
  151. package/src/providers/transform-messages.ts +172 -0
  152. package/src/stream.ts +59 -0
  153. package/src/types.ts +294 -0
  154. package/src/utils/event-stream.ts +87 -0
  155. package/src/utils/hash.ts +13 -0
  156. package/src/utils/json-parse.ts +28 -0
  157. package/src/utils/oauth/anthropic.ts +402 -0
  158. package/src/utils/oauth/github-copilot.ts +396 -0
  159. package/src/utils/oauth/index.ts +123 -0
  160. package/src/utils/oauth/oauth-page.ts +109 -0
  161. package/src/utils/oauth/openai-codex.ts +450 -0
  162. package/src/utils/oauth/pkce.ts +34 -0
  163. package/src/utils/oauth/types.ts +59 -0
  164. package/src/utils/overflow.ts +125 -0
  165. package/src/utils/sanitize-unicode.ts +25 -0
  166. package/src/utils/typebox-helpers.ts +24 -0
  167. package/src/utils/validation.ts +93 -0
@@ -0,0 +1,741 @@
1
+ // NEVER convert to top-level runtime imports - breaks browser/Vite builds (web-ui)
2
+ let _os = null;
3
+ const dynamicImport = (specifier) => import(specifier);
4
+ const NODE_OS_SPECIFIER = "node:" + "os";
5
+ if (typeof process !== "undefined" && (process.versions?.node || process.versions?.bun)) {
6
+ dynamicImport(NODE_OS_SPECIFIER).then((m) => {
7
+ _os = m;
8
+ });
9
+ }
10
+ import { getEnvApiKey } from "../env-api-keys.js";
11
+ import { supportsXhigh } from "../models.js";
12
+ import { AssistantMessageEventStream } from "../utils/event-stream.js";
13
+ import { convertResponsesMessages, convertResponsesTools, processResponsesStream } from "./openai-responses-shared.js";
14
+ import { buildBaseOptions, clampReasoning } from "./simple-options.js";
15
+ // ============================================================================
16
+ // Configuration
17
+ // ============================================================================
18
+ const DEFAULT_CODEX_BASE_URL = "https://chatgpt.com/backend-api";
19
+ const JWT_CLAIM_PATH = "https://api.openai.com/auth";
20
+ const MAX_RETRIES = 3;
21
+ const BASE_DELAY_MS = 1000;
22
+ const CODEX_TOOL_CALL_PROVIDERS = new Set(["openai-codex"]);
23
+ const CODEX_RESPONSE_STATUSES = new Set([
24
+ "completed",
25
+ "incomplete",
26
+ "failed",
27
+ "cancelled",
28
+ "queued",
29
+ "in_progress",
30
+ ]);
31
+ // ============================================================================
32
+ // Retry Helpers
33
+ // ============================================================================
34
+ function isRetryableError(status, errorText) {
35
+ if (status === 429 || status === 500 || status === 502 || status === 503 || status === 504) {
36
+ return true;
37
+ }
38
+ return /rate.?limit|overloaded|service.?unavailable|upstream.?connect|connection.?refused/i.test(errorText);
39
+ }
40
+ function sleep(ms, signal) {
41
+ return new Promise((resolve, reject) => {
42
+ if (signal?.aborted) {
43
+ reject(new Error("Request was aborted"));
44
+ return;
45
+ }
46
+ const timeout = setTimeout(resolve, ms);
47
+ signal?.addEventListener("abort", () => {
48
+ clearTimeout(timeout);
49
+ reject(new Error("Request was aborted"));
50
+ });
51
+ });
52
+ }
53
+ // ============================================================================
54
+ // Main Stream Function
55
+ // ============================================================================
56
+ export const streamOpenAICodexResponses = (model, context, options) => {
57
+ const stream = new AssistantMessageEventStream();
58
+ (async () => {
59
+ const output = {
60
+ role: "assistant",
61
+ content: [],
62
+ api: "openai-codex-responses",
63
+ provider: model.provider,
64
+ model: model.id,
65
+ usage: {
66
+ input: 0,
67
+ output: 0,
68
+ cacheRead: 0,
69
+ cacheWrite: 0,
70
+ totalTokens: 0,
71
+ cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
72
+ },
73
+ stopReason: "stop",
74
+ timestamp: Date.now(),
75
+ };
76
+ try {
77
+ const apiKey = options?.apiKey || getEnvApiKey(model.provider) || "";
78
+ if (!apiKey) {
79
+ throw new Error(`No API key for provider: ${model.provider}`);
80
+ }
81
+ const accountId = extractAccountId(apiKey);
82
+ let body = buildRequestBody(model, context, options);
83
+ const nextBody = await options?.onPayload?.(body, model);
84
+ if (nextBody !== undefined) {
85
+ body = nextBody;
86
+ }
87
+ const websocketRequestId = options?.sessionId || createCodexRequestId();
88
+ const sseHeaders = buildSSEHeaders(model.headers, options?.headers, accountId, apiKey, options?.sessionId);
89
+ const websocketHeaders = buildWebSocketHeaders(model.headers, options?.headers, accountId, apiKey, websocketRequestId);
90
+ const bodyJson = JSON.stringify(body);
91
+ const transport = options?.transport || "sse";
92
+ if (transport !== "sse") {
93
+ let websocketStarted = false;
94
+ try {
95
+ await processWebSocketStream(resolveCodexWebSocketUrl(model.baseUrl), body, websocketHeaders, output, stream, model, () => {
96
+ websocketStarted = true;
97
+ }, options);
98
+ if (options?.signal?.aborted) {
99
+ throw new Error("Request was aborted");
100
+ }
101
+ stream.push({
102
+ type: "done",
103
+ reason: output.stopReason,
104
+ message: output,
105
+ });
106
+ stream.end();
107
+ return;
108
+ }
109
+ catch (error) {
110
+ if (transport === "websocket" || websocketStarted) {
111
+ throw error;
112
+ }
113
+ }
114
+ }
115
+ // Fetch with retry logic for rate limits and transient errors
116
+ let response;
117
+ let lastError;
118
+ for (let attempt = 0; attempt <= MAX_RETRIES; attempt++) {
119
+ if (options?.signal?.aborted) {
120
+ throw new Error("Request was aborted");
121
+ }
122
+ try {
123
+ response = await fetch(resolveCodexUrl(model.baseUrl), {
124
+ method: "POST",
125
+ headers: sseHeaders,
126
+ body: bodyJson,
127
+ signal: options?.signal,
128
+ });
129
+ if (response.ok) {
130
+ break;
131
+ }
132
+ const errorText = await response.text();
133
+ if (attempt < MAX_RETRIES && isRetryableError(response.status, errorText)) {
134
+ const delayMs = BASE_DELAY_MS * 2 ** attempt;
135
+ await sleep(delayMs, options?.signal);
136
+ continue;
137
+ }
138
+ // Parse error for friendly message on final attempt or non-retryable error
139
+ const fakeResponse = new Response(errorText, {
140
+ status: response.status,
141
+ statusText: response.statusText,
142
+ });
143
+ const info = await parseErrorResponse(fakeResponse);
144
+ throw new Error(info.friendlyMessage || info.message);
145
+ }
146
+ catch (error) {
147
+ if (error instanceof Error) {
148
+ if (error.name === "AbortError" || error.message === "Request was aborted") {
149
+ throw new Error("Request was aborted");
150
+ }
151
+ }
152
+ lastError = error instanceof Error ? error : new Error(String(error));
153
+ // Network errors are retryable
154
+ if (attempt < MAX_RETRIES && !lastError.message.includes("usage limit")) {
155
+ const delayMs = BASE_DELAY_MS * 2 ** attempt;
156
+ await sleep(delayMs, options?.signal);
157
+ continue;
158
+ }
159
+ throw lastError;
160
+ }
161
+ }
162
+ if (!response?.ok) {
163
+ throw lastError ?? new Error("Failed after retries");
164
+ }
165
+ if (!response.body) {
166
+ throw new Error("No response body");
167
+ }
168
+ stream.push({ type: "start", partial: output });
169
+ await processStream(response, output, stream, model);
170
+ if (options?.signal?.aborted) {
171
+ throw new Error("Request was aborted");
172
+ }
173
+ stream.push({ type: "done", reason: output.stopReason, message: output });
174
+ stream.end();
175
+ }
176
+ catch (error) {
177
+ output.stopReason = options?.signal?.aborted ? "aborted" : "error";
178
+ output.errorMessage = error instanceof Error ? error.message : String(error);
179
+ stream.push({ type: "error", reason: output.stopReason, error: output });
180
+ stream.end();
181
+ }
182
+ })();
183
+ return stream;
184
+ };
185
+ export const streamSimpleOpenAICodexResponses = (model, context, options) => {
186
+ const apiKey = options?.apiKey || getEnvApiKey(model.provider);
187
+ if (!apiKey) {
188
+ throw new Error(`No API key for provider: ${model.provider}`);
189
+ }
190
+ const base = buildBaseOptions(model, options, apiKey);
191
+ const reasoningEffort = supportsXhigh(model) ? options?.reasoning : clampReasoning(options?.reasoning);
192
+ return streamOpenAICodexResponses(model, context, {
193
+ ...base,
194
+ reasoningEffort,
195
+ });
196
+ };
197
+ // ============================================================================
198
+ // Request Building
199
+ // ============================================================================
200
+ function buildRequestBody(model, context, options) {
201
+ const messages = convertResponsesMessages(model, context, CODEX_TOOL_CALL_PROVIDERS, {
202
+ includeSystemPrompt: false,
203
+ });
204
+ const body = {
205
+ model: model.id,
206
+ store: false,
207
+ stream: true,
208
+ instructions: context.systemPrompt,
209
+ input: messages,
210
+ text: { verbosity: options?.textVerbosity || "medium" },
211
+ include: ["reasoning.encrypted_content"],
212
+ prompt_cache_key: options?.sessionId,
213
+ tool_choice: "auto",
214
+ parallel_tool_calls: true,
215
+ };
216
+ if (options?.temperature !== undefined) {
217
+ body.temperature = options.temperature;
218
+ }
219
+ if (context.tools) {
220
+ body.tools = convertResponsesTools(context.tools, { strict: null });
221
+ }
222
+ if (options?.reasoningEffort !== undefined) {
223
+ body.reasoning = {
224
+ effort: clampReasoningEffort(model.id, options.reasoningEffort),
225
+ summary: options.reasoningSummary ?? "auto",
226
+ };
227
+ }
228
+ return body;
229
+ }
230
+ function clampReasoningEffort(modelId, effort) {
231
+ const id = modelId.includes("/") ? modelId.split("/").pop() : modelId;
232
+ if ((id.startsWith("gpt-5.2") || id.startsWith("gpt-5.3") || id.startsWith("gpt-5.4")) && effort === "minimal")
233
+ return "low";
234
+ if (id === "gpt-5.1" && effort === "xhigh")
235
+ return "high";
236
+ if (id === "gpt-5.1-codex-mini")
237
+ return effort === "high" || effort === "xhigh" ? "high" : "medium";
238
+ return effort;
239
+ }
240
+ function resolveCodexUrl(baseUrl) {
241
+ const raw = baseUrl && baseUrl.trim().length > 0 ? baseUrl : DEFAULT_CODEX_BASE_URL;
242
+ const normalized = raw.replace(/\/+$/, "");
243
+ if (normalized.endsWith("/codex/responses"))
244
+ return normalized;
245
+ if (normalized.endsWith("/codex"))
246
+ return `${normalized}/responses`;
247
+ return `${normalized}/codex/responses`;
248
+ }
249
+ function resolveCodexWebSocketUrl(baseUrl) {
250
+ const url = new URL(resolveCodexUrl(baseUrl));
251
+ if (url.protocol === "https:")
252
+ url.protocol = "wss:";
253
+ if (url.protocol === "http:")
254
+ url.protocol = "ws:";
255
+ return url.toString();
256
+ }
257
+ // ============================================================================
258
+ // Response Processing
259
+ // ============================================================================
260
+ async function processStream(response, output, stream, model) {
261
+ await processResponsesStream(mapCodexEvents(parseSSE(response)), output, stream, model);
262
+ }
263
+ async function* mapCodexEvents(events) {
264
+ for await (const event of events) {
265
+ const type = typeof event.type === "string" ? event.type : undefined;
266
+ if (!type)
267
+ continue;
268
+ if (type === "error") {
269
+ const code = event.code || "";
270
+ const message = event.message || "";
271
+ throw new Error(`Codex error: ${message || code || JSON.stringify(event)}`);
272
+ }
273
+ if (type === "response.failed") {
274
+ const msg = event.response?.error?.message;
275
+ throw new Error(msg || "Codex response failed");
276
+ }
277
+ if (type === "response.done" || type === "response.completed" || type === "response.incomplete") {
278
+ const response = event.response;
279
+ const normalizedResponse = response
280
+ ? { ...response, status: normalizeCodexStatus(response.status) }
281
+ : response;
282
+ yield { ...event, type: "response.completed", response: normalizedResponse };
283
+ return;
284
+ }
285
+ yield event;
286
+ }
287
+ }
288
+ function normalizeCodexStatus(status) {
289
+ if (typeof status !== "string")
290
+ return undefined;
291
+ return CODEX_RESPONSE_STATUSES.has(status) ? status : undefined;
292
+ }
293
+ // ============================================================================
294
+ // SSE Parsing
295
+ // ============================================================================
296
+ async function* parseSSE(response) {
297
+ if (!response.body)
298
+ return;
299
+ const reader = response.body.getReader();
300
+ const decoder = new TextDecoder();
301
+ let buffer = "";
302
+ try {
303
+ while (true) {
304
+ const { done, value } = await reader.read();
305
+ if (done)
306
+ break;
307
+ buffer += decoder.decode(value, { stream: true });
308
+ let idx = buffer.indexOf("\n\n");
309
+ while (idx !== -1) {
310
+ const chunk = buffer.slice(0, idx);
311
+ buffer = buffer.slice(idx + 2);
312
+ const dataLines = chunk
313
+ .split("\n")
314
+ .filter((l) => l.startsWith("data:"))
315
+ .map((l) => l.slice(5).trim());
316
+ if (dataLines.length > 0) {
317
+ const data = dataLines.join("\n").trim();
318
+ if (data && data !== "[DONE]") {
319
+ try {
320
+ yield JSON.parse(data);
321
+ }
322
+ catch { }
323
+ }
324
+ }
325
+ idx = buffer.indexOf("\n\n");
326
+ }
327
+ }
328
+ }
329
+ finally {
330
+ try {
331
+ await reader.cancel();
332
+ }
333
+ catch { }
334
+ try {
335
+ reader.releaseLock();
336
+ }
337
+ catch { }
338
+ }
339
+ }
340
+ // ============================================================================
341
+ // WebSocket Parsing
342
+ // ============================================================================
343
+ const OPENAI_BETA_RESPONSES_WEBSOCKETS = "responses_websockets=2026-02-06";
344
+ const SESSION_WEBSOCKET_CACHE_TTL_MS = 5 * 60 * 1000;
345
+ const websocketSessionCache = new Map();
346
+ function getWebSocketConstructor() {
347
+ const ctor = globalThis.WebSocket;
348
+ if (typeof ctor !== "function")
349
+ return null;
350
+ return ctor;
351
+ }
352
+ function headersToRecord(headers) {
353
+ const out = {};
354
+ for (const [key, value] of headers.entries()) {
355
+ out[key] = value;
356
+ }
357
+ return out;
358
+ }
359
+ function getWebSocketReadyState(socket) {
360
+ const readyState = socket.readyState;
361
+ return typeof readyState === "number" ? readyState : undefined;
362
+ }
363
+ function isWebSocketReusable(socket) {
364
+ const readyState = getWebSocketReadyState(socket);
365
+ // If readyState is unavailable, assume the runtime keeps it open/reusable.
366
+ return readyState === undefined || readyState === 1;
367
+ }
368
+ function closeWebSocketSilently(socket, code = 1000, reason = "done") {
369
+ try {
370
+ socket.close(code, reason);
371
+ }
372
+ catch { }
373
+ }
374
+ function scheduleSessionWebSocketExpiry(sessionId, entry) {
375
+ if (entry.idleTimer) {
376
+ clearTimeout(entry.idleTimer);
377
+ }
378
+ entry.idleTimer = setTimeout(() => {
379
+ if (entry.busy)
380
+ return;
381
+ closeWebSocketSilently(entry.socket, 1000, "idle_timeout");
382
+ websocketSessionCache.delete(sessionId);
383
+ }, SESSION_WEBSOCKET_CACHE_TTL_MS);
384
+ }
385
+ async function connectWebSocket(url, headers, signal) {
386
+ const WebSocketCtor = getWebSocketConstructor();
387
+ if (!WebSocketCtor) {
388
+ throw new Error("WebSocket transport is not available in this runtime");
389
+ }
390
+ const wsHeaders = headersToRecord(headers);
391
+ delete wsHeaders["OpenAI-Beta"];
392
+ return new Promise((resolve, reject) => {
393
+ let settled = false;
394
+ let socket;
395
+ try {
396
+ socket = new WebSocketCtor(url, { headers: wsHeaders });
397
+ }
398
+ catch (error) {
399
+ reject(error instanceof Error ? error : new Error(String(error)));
400
+ return;
401
+ }
402
+ const onOpen = () => {
403
+ if (settled)
404
+ return;
405
+ settled = true;
406
+ cleanup();
407
+ resolve(socket);
408
+ };
409
+ const onError = (event) => {
410
+ const error = extractWebSocketError(event);
411
+ if (settled)
412
+ return;
413
+ settled = true;
414
+ cleanup();
415
+ reject(error);
416
+ };
417
+ const onClose = (event) => {
418
+ const error = extractWebSocketCloseError(event);
419
+ if (settled)
420
+ return;
421
+ settled = true;
422
+ cleanup();
423
+ reject(error);
424
+ };
425
+ const onAbort = () => {
426
+ if (settled)
427
+ return;
428
+ settled = true;
429
+ cleanup();
430
+ socket.close(1000, "aborted");
431
+ reject(new Error("Request was aborted"));
432
+ };
433
+ const cleanup = () => {
434
+ socket.removeEventListener("open", onOpen);
435
+ socket.removeEventListener("error", onError);
436
+ socket.removeEventListener("close", onClose);
437
+ signal?.removeEventListener("abort", onAbort);
438
+ };
439
+ socket.addEventListener("open", onOpen);
440
+ socket.addEventListener("error", onError);
441
+ socket.addEventListener("close", onClose);
442
+ signal?.addEventListener("abort", onAbort);
443
+ });
444
+ }
445
+ async function acquireWebSocket(url, headers, sessionId, signal) {
446
+ if (!sessionId) {
447
+ const socket = await connectWebSocket(url, headers, signal);
448
+ return {
449
+ socket,
450
+ release: ({ keep } = {}) => {
451
+ if (keep === false) {
452
+ closeWebSocketSilently(socket);
453
+ return;
454
+ }
455
+ closeWebSocketSilently(socket);
456
+ },
457
+ };
458
+ }
459
+ const cached = websocketSessionCache.get(sessionId);
460
+ if (cached) {
461
+ if (cached.idleTimer) {
462
+ clearTimeout(cached.idleTimer);
463
+ cached.idleTimer = undefined;
464
+ }
465
+ if (!cached.busy && isWebSocketReusable(cached.socket)) {
466
+ cached.busy = true;
467
+ return {
468
+ socket: cached.socket,
469
+ release: ({ keep } = {}) => {
470
+ if (!keep || !isWebSocketReusable(cached.socket)) {
471
+ closeWebSocketSilently(cached.socket);
472
+ websocketSessionCache.delete(sessionId);
473
+ return;
474
+ }
475
+ cached.busy = false;
476
+ scheduleSessionWebSocketExpiry(sessionId, cached);
477
+ },
478
+ };
479
+ }
480
+ if (cached.busy) {
481
+ const socket = await connectWebSocket(url, headers, signal);
482
+ return {
483
+ socket,
484
+ release: () => {
485
+ closeWebSocketSilently(socket);
486
+ },
487
+ };
488
+ }
489
+ if (!isWebSocketReusable(cached.socket)) {
490
+ closeWebSocketSilently(cached.socket);
491
+ websocketSessionCache.delete(sessionId);
492
+ }
493
+ }
494
+ const socket = await connectWebSocket(url, headers, signal);
495
+ const entry = { socket, busy: true };
496
+ websocketSessionCache.set(sessionId, entry);
497
+ return {
498
+ socket,
499
+ release: ({ keep } = {}) => {
500
+ if (!keep || !isWebSocketReusable(entry.socket)) {
501
+ closeWebSocketSilently(entry.socket);
502
+ if (entry.idleTimer)
503
+ clearTimeout(entry.idleTimer);
504
+ if (websocketSessionCache.get(sessionId) === entry) {
505
+ websocketSessionCache.delete(sessionId);
506
+ }
507
+ return;
508
+ }
509
+ entry.busy = false;
510
+ scheduleSessionWebSocketExpiry(sessionId, entry);
511
+ },
512
+ };
513
+ }
514
+ function extractWebSocketError(event) {
515
+ if (event && typeof event === "object" && "message" in event) {
516
+ const message = event.message;
517
+ if (typeof message === "string" && message.length > 0) {
518
+ return new Error(message);
519
+ }
520
+ }
521
+ return new Error("WebSocket error");
522
+ }
523
+ function extractWebSocketCloseError(event) {
524
+ if (event && typeof event === "object") {
525
+ const code = "code" in event ? event.code : undefined;
526
+ const reason = "reason" in event ? event.reason : undefined;
527
+ const codeText = typeof code === "number" ? ` ${code}` : "";
528
+ const reasonText = typeof reason === "string" && reason.length > 0 ? ` ${reason}` : "";
529
+ return new Error(`WebSocket closed${codeText}${reasonText}`.trim());
530
+ }
531
+ return new Error("WebSocket closed");
532
+ }
533
+ async function decodeWebSocketData(data) {
534
+ if (typeof data === "string")
535
+ return data;
536
+ if (data instanceof ArrayBuffer) {
537
+ return new TextDecoder().decode(new Uint8Array(data));
538
+ }
539
+ if (ArrayBuffer.isView(data)) {
540
+ const view = data;
541
+ return new TextDecoder().decode(new Uint8Array(view.buffer, view.byteOffset, view.byteLength));
542
+ }
543
+ if (data && typeof data === "object" && "arrayBuffer" in data) {
544
+ const blobLike = data;
545
+ const arrayBuffer = await blobLike.arrayBuffer();
546
+ return new TextDecoder().decode(new Uint8Array(arrayBuffer));
547
+ }
548
+ return null;
549
+ }
550
+ async function* parseWebSocket(socket, signal) {
551
+ const queue = [];
552
+ let pending = null;
553
+ let done = false;
554
+ let failed = null;
555
+ let sawCompletion = false;
556
+ const wake = () => {
557
+ if (!pending)
558
+ return;
559
+ const resolve = pending;
560
+ pending = null;
561
+ resolve();
562
+ };
563
+ const onMessage = (event) => {
564
+ void (async () => {
565
+ if (!event || typeof event !== "object" || !("data" in event))
566
+ return;
567
+ const text = await decodeWebSocketData(event.data);
568
+ if (!text)
569
+ return;
570
+ try {
571
+ const parsed = JSON.parse(text);
572
+ const type = typeof parsed.type === "string" ? parsed.type : "";
573
+ if (type === "response.completed" || type === "response.done" || type === "response.incomplete") {
574
+ sawCompletion = true;
575
+ done = true;
576
+ }
577
+ queue.push(parsed);
578
+ wake();
579
+ }
580
+ catch { }
581
+ })();
582
+ };
583
+ const onError = (event) => {
584
+ failed = extractWebSocketError(event);
585
+ done = true;
586
+ wake();
587
+ };
588
+ const onClose = (event) => {
589
+ if (sawCompletion) {
590
+ done = true;
591
+ wake();
592
+ return;
593
+ }
594
+ if (!failed) {
595
+ failed = extractWebSocketCloseError(event);
596
+ }
597
+ done = true;
598
+ wake();
599
+ };
600
+ const onAbort = () => {
601
+ failed = new Error("Request was aborted");
602
+ done = true;
603
+ wake();
604
+ };
605
+ socket.addEventListener("message", onMessage);
606
+ socket.addEventListener("error", onError);
607
+ socket.addEventListener("close", onClose);
608
+ signal?.addEventListener("abort", onAbort);
609
+ try {
610
+ while (true) {
611
+ if (signal?.aborted) {
612
+ throw new Error("Request was aborted");
613
+ }
614
+ if (queue.length > 0) {
615
+ yield queue.shift();
616
+ continue;
617
+ }
618
+ if (done)
619
+ break;
620
+ await new Promise((resolve) => {
621
+ pending = resolve;
622
+ });
623
+ }
624
+ if (failed) {
625
+ throw failed;
626
+ }
627
+ if (!sawCompletion) {
628
+ throw new Error("WebSocket stream closed before response.completed");
629
+ }
630
+ }
631
+ finally {
632
+ socket.removeEventListener("message", onMessage);
633
+ socket.removeEventListener("error", onError);
634
+ socket.removeEventListener("close", onClose);
635
+ signal?.removeEventListener("abort", onAbort);
636
+ }
637
+ }
638
+ async function processWebSocketStream(url, body, headers, output, stream, model, onStart, options) {
639
+ const { socket, release } = await acquireWebSocket(url, headers, options?.sessionId, options?.signal);
640
+ let keepConnection = true;
641
+ try {
642
+ socket.send(JSON.stringify({ type: "response.create", ...body }));
643
+ onStart();
644
+ stream.push({ type: "start", partial: output });
645
+ await processResponsesStream(mapCodexEvents(parseWebSocket(socket, options?.signal)), output, stream, model);
646
+ if (options?.signal?.aborted) {
647
+ keepConnection = false;
648
+ }
649
+ }
650
+ catch (error) {
651
+ keepConnection = false;
652
+ throw error;
653
+ }
654
+ finally {
655
+ release({ keep: keepConnection });
656
+ }
657
+ }
658
+ // ============================================================================
659
+ // Error Handling
660
+ // ============================================================================
661
+ async function parseErrorResponse(response) {
662
+ const raw = await response.text();
663
+ let message = raw || response.statusText || "Request failed";
664
+ let friendlyMessage;
665
+ try {
666
+ const parsed = JSON.parse(raw);
667
+ const err = parsed?.error;
668
+ if (err) {
669
+ const code = err.code || err.type || "";
670
+ if (/usage_limit_reached|usage_not_included|rate_limit_exceeded/i.test(code) || response.status === 429) {
671
+ const plan = err.plan_type ? ` (${err.plan_type.toLowerCase()} plan)` : "";
672
+ const mins = err.resets_at
673
+ ? Math.max(0, Math.round((err.resets_at * 1000 - Date.now()) / 60000))
674
+ : undefined;
675
+ const when = mins !== undefined ? ` Try again in ~${mins} min.` : "";
676
+ friendlyMessage = `You have hit your ChatGPT usage limit${plan}.${when}`.trim();
677
+ }
678
+ message = err.message || friendlyMessage || message;
679
+ }
680
+ }
681
+ catch { }
682
+ return { message, friendlyMessage };
683
+ }
684
+ // ============================================================================
685
+ // Auth & Headers
686
+ // ============================================================================
687
+ function extractAccountId(token) {
688
+ try {
689
+ const parts = token.split(".");
690
+ if (parts.length !== 3)
691
+ throw new Error("Invalid token");
692
+ const payload = JSON.parse(atob(parts[1]));
693
+ const accountId = payload?.[JWT_CLAIM_PATH]?.chatgpt_account_id;
694
+ if (!accountId)
695
+ throw new Error("No account ID in token");
696
+ return accountId;
697
+ }
698
+ catch {
699
+ throw new Error("Failed to extract accountId from token");
700
+ }
701
+ }
702
+ function createCodexRequestId() {
703
+ if (typeof globalThis.crypto?.randomUUID === "function") {
704
+ return globalThis.crypto.randomUUID();
705
+ }
706
+ return `codex_${Date.now()}_${Math.random().toString(36).slice(2, 10)}`;
707
+ }
708
+ function buildBaseCodexHeaders(initHeaders, additionalHeaders, accountId, token) {
709
+ const headers = new Headers(initHeaders);
710
+ for (const [key, value] of Object.entries(additionalHeaders || {})) {
711
+ headers.set(key, value);
712
+ }
713
+ headers.set("Authorization", `Bearer ${token}`);
714
+ headers.set("chatgpt-account-id", accountId);
715
+ headers.set("originator", "pi");
716
+ const userAgent = _os ? `pi (${_os.platform()} ${_os.release()}; ${_os.arch()})` : "pi (browser)";
717
+ headers.set("User-Agent", userAgent);
718
+ return headers;
719
+ }
720
+ function buildSSEHeaders(initHeaders, additionalHeaders, accountId, token, sessionId) {
721
+ const headers = buildBaseCodexHeaders(initHeaders, additionalHeaders, accountId, token);
722
+ headers.set("OpenAI-Beta", "responses=experimental");
723
+ headers.set("accept", "text/event-stream");
724
+ headers.set("content-type", "application/json");
725
+ if (sessionId) {
726
+ headers.set("session_id", sessionId);
727
+ }
728
+ return headers;
729
+ }
730
+ function buildWebSocketHeaders(initHeaders, additionalHeaders, accountId, token, requestId) {
731
+ const headers = buildBaseCodexHeaders(initHeaders, additionalHeaders, accountId, token);
732
+ headers.delete("accept");
733
+ headers.delete("content-type");
734
+ headers.delete("OpenAI-Beta");
735
+ headers.delete("openai-beta");
736
+ headers.set("OpenAI-Beta", OPENAI_BETA_RESPONSES_WEBSOCKETS);
737
+ headers.set("x-client-request-id", requestId);
738
+ headers.set("session_id", requestId);
739
+ return headers;
740
+ }
741
+ //# sourceMappingURL=openai-codex-responses.js.map