ai-zero-token 2.0.4 → 2.0.6

package/dist/core/providers/openai-codex/chat.js

@@ -1,7 +1,8 @@
 #!/usr/bin/env node
 import { DEFAULT_CODEX_MODEL } from "../../models/openai-codex-models.js";
-import { requestText } from "../http-client.js";
+import { requestStream, requestText } from "../http-client.js";
 const CODEX_RESPONSES_URL = "https://chatgpt.com/backend-api/codex/responses";
+const CODEX_RESPONSES_COMPACT_URL = `${CODEX_RESPONSES_URL}/compact`;
 const URL_KEY_RE = /(url|uri|href|download|preview|thumbnail|image|asset|file)/i;
 const REFERENCE_KEY_RE = /(image|asset|file|media|blob|artifact|download|preview|thumbnail)/i;
 const REFERENCE_VALUE_RE = /^(file|asset|image|img|media|blob)-[\w-]+$/i;
@@ -39,21 +40,65 @@ function parseOptionalText(value) {
 function parseUpstreamErrorBody(body) {
   try {
     const parsed = JSON.parse(body);
-    const error = parsed.error;
-    if (!error || typeof error !== "object") {
+    const record = asRecord(parsed.error);
+    if (!record) {
       return void 0;
     }
-    const record = error;
+    const eligiblePromo = asRecord(record.eligible_promo);
     return {
       message: typeof record.message === "string" ? record.message : void 0,
       type: typeof record.type === "string" ? record.type : void 0,
       code: typeof record.code === "string" ? record.code : void 0,
-      param: typeof record.param === "string" || record.param === null ? record.param : void 0
+      param: typeof record.param === "string" || record.param === null ? record.param : void 0,
+      planType: typeof record.plan_type === "string" ? record.plan_type : void 0,
+      resetsAt: typeof record.resets_at === "number" && Number.isFinite(record.resets_at) ? record.resets_at : void 0,
+      resetsInSeconds: typeof record.resets_in_seconds === "number" && Number.isFinite(record.resets_in_seconds) ? record.resets_in_seconds : void 0,
+      promoCampaignId: typeof eligiblePromo?.campaign_id === "string" ? eligiblePromo.campaign_id : void 0,
+      promoMessage: typeof eligiblePromo?.message === "string" ? eligiblePromo.message : void 0
     };
   } catch {
     return void 0;
   }
 }
+function buildCodexRequestHeaders(profile) {
+  return {
+    Accept: "text/event-stream",
+    "Content-Type": "application/json",
+    Authorization: `Bearer ${profile.access}`,
+    "ChatGPT-Account-Id": profile.accountId,
+    "OpenAI-Beta": "responses=experimental",
+    Originator: "pi",
+    "User-Agent": "pi (bun demo)"
+  };
+}
+function createCodexUpstreamError(status, body, transport, quota, requestId) {
+  const upstreamError = parseUpstreamErrorBody(body);
+  const error = new Error(`\u8C03\u7528 Responses API \u5931\u8D25: HTTP ${status} via ${transport} ${body}`);
+  error.quota = quota ?? createUsageLimitQuotaSnapshot(status, upstreamError, requestId);
+  error.upstreamStatus = status;
+  error.upstreamErrorCode = upstreamError?.code;
+  error.upstreamErrorType = upstreamError?.type;
+  error.upstreamErrorMessage = upstreamError?.message;
+  return error;
+}
+function createUsageLimitQuotaSnapshot(status, upstreamError, requestId) {
+  const errorKind = `${upstreamError?.type ?? ""} ${upstreamError?.code ?? ""}`.toLowerCase();
+  if (status !== 429 && !errorKind.includes("usage_limit_reached")) {
+    return void 0;
+  }
+  return {
+    capturedAt: Date.now(),
+    sourceRequestId: requestId,
+    planType: upstreamError?.planType,
+    primaryUsedPercent: 100,
+    primaryResetAt: upstreamError?.resetsAt,
+    primaryResetAfterSeconds: upstreamError?.resetsInSeconds,
+    creditsHasCredits: false,
+    creditsBalance: "0",
+    promoCampaignId: upstreamError?.promoCampaignId,
+    promoMessage: upstreamError?.promoMessage
+  };
+}
 function extractCodexQuotaSnapshot(headers, requestId) {
   const activeLimit = parseOptionalText(headers["x-codex-active-limit"]);
   const planType = parseOptionalText(headers["x-codex-plan-type"]);
@@ -362,34 +407,50 @@ async function askOpenAICodex(params) {
   const response = await requestText({
     method: "POST",
     url: CODEX_RESPONSES_URL,
-    headers: {
-      Accept: "text/event-stream",
-      "Content-Type": "application/json",
-      Authorization: `Bearer ${params.profile.access}`,
-      "ChatGPT-Account-Id": params.profile.accountId,
-      "OpenAI-Beta": "responses=experimental",
-      Originator: "pi",
-      "User-Agent": "pi (bun demo)"
-    },
+    headers: buildCodexRequestHeaders(params.profile),
     body: JSON.stringify(requestBody)
   });
   const quota = extractCodexQuotaSnapshot(response.headers, response.requestId);
   if (response.status < 200 || response.status >= 300) {
-    const upstreamError = parseUpstreamErrorBody(response.body);
-    const error = new Error(`\u8C03\u7528 Responses API \u5931\u8D25: HTTP ${response.status} via ${response.transport} ${response.body}`);
-    error.quota = quota;
-    error.upstreamStatus = response.status;
-    error.upstreamErrorCode = upstreamError?.code;
-    error.upstreamErrorType = upstreamError?.type;
-    error.upstreamErrorMessage = upstreamError?.message;
-    throw error;
+    throw createCodexUpstreamError(response.status, response.body, response.transport, quota, response.requestId);
   }
   return {
     ...extractCodexText(response.body, requestBody),
     quota
   };
 }
+async function streamOpenAICodex(params) {
+  const requestBody = params.passthroughBody ? { ...params.bodyOverride ?? {} } : {
+    ...buildDefaultRequestBody(params),
+    ...params.bodyOverride ?? {}
+  };
+  if (!params.passthroughBody && typeof requestBody.input === "undefined") {
+    throw new Error("Codex \u8BF7\u6C42\u7F3A\u5C11 input\u3002\u8BF7\u63D0\u4F9B prompt \u6216\u5728\u5B9E\u9A8C\u8BF7\u6C42\u4F53\u91CC\u663E\u5F0F\u4F20\u5165 input\u3002");
+  }
+  const response = await requestStream({
+    method: "POST",
+    url: params.endpoint === "responses/compact" ? CODEX_RESPONSES_COMPACT_URL : CODEX_RESPONSES_URL,
+    headers: buildCodexRequestHeaders(params.profile),
+    body: JSON.stringify(requestBody),
+    signal: params.signal
+  });
+  const headers = response.headers;
+  const requestId = headers["x-request-id"] ?? response.requestId;
+  const quota = extractCodexQuotaSnapshot(headers, requestId);
+  if (response.status < 200 || response.status >= 300) {
+    const body = await new Response(response.body).text();
+    throw createCodexUpstreamError(response.status, body, response.transport, quota, requestId);
+  }
+  return {
+    body: response.body,
+    headers,
+    quota,
+    requestId,
+    status: response.status
+  };
+}
 export {
   askOpenAICodex,
-  extractCodexQuotaSnapshot
+  extractCodexQuotaSnapshot,
+  streamOpenAICodex
 };