@prestyj/ai 5.11.0 → 5.12.0

package/dist/index.cjs

@@ -1226,7 +1226,7 @@ function parseToolArguments(argsJson) {
 var NON_STREAMING_TIMEOUT_MS = 60 * 60 * 1e3;
 var anthropicClientCache = /* @__PURE__ */ new Map();
 function fineGrainedToolStreamingEnabled() {
-  const raw = process.env.GG_FINE_GRAINED_TOOL_STREAMING ?? process.env.CLAUDE_CODE_ENABLE_FINE_GRAINED_TOOL_STREAMING;
+  const raw = process.env.EZ_FINE_GRAINED_TOOL_STREAMING ?? process.env.CLAUDE_CODE_ENABLE_FINE_GRAINED_TOOL_STREAMING;
   if (!raw) return false;
   const v = raw.trim().toLowerCase();
   return v === "1" || v === "true" || v === "yes" || v === "on";
@@ -2091,7 +2091,15 @@ async function* runStream2(options) {
       if (chunk.usage) {
         ({ inputTokens, outputTokens, cacheRead, cacheWrite } = extractOpenAIUsage(chunk.usage));
       }
-      if (!choice) continue;
+      if (!choice) {
+        const gatewayError = classifyChoicelessFrame(chunk);
+        if (gatewayError) {
+          throw new ProviderError(providerName, gatewayError.message, {
+            statusCode: gatewayError.statusCode
+          });
+        }
+        continue;
+      }
       if (choice.finish_reason) {
         finishReason = choice.finish_reason;
       }
@@ -2275,6 +2283,31 @@ function completionToResponse(completion, endpointKey) {
     }
   };
 }
+function classifyChoicelessFrame(frame) {
+  if (!frame || typeof frame !== "object" || Array.isArray(frame)) return null;
+  const rec = frame;
+  if (Array.isArray(rec.choices)) return null;
+  const statusOf = (value) => {
+    const n = typeof value === "string" ? Number(value) : value;
+    return typeof n === "number" && Number.isFinite(n) && n >= 400 && n <= 599 ? n : void 0;
+  };
+  const statusCode = statusOf(rec.status) ?? statusOf(rec.statusCode) ?? statusOf(rec.code);
+  const typeIsError = typeof rec.type === "string" && rec.type.toLowerCase() === "error";
+  let detailText;
+  const detail = rec.detail;
+  if (typeof detail === "string" && detail.trim()) {
+    detailText = detail.trim();
+  } else if (Array.isArray(detail)) {
+    const parts = detail.map(
+      (d) => d && typeof d === "object" && typeof d.msg === "string" ? d.msg : typeof d === "string" ? d : ""
+    ).filter(Boolean);
+    if (parts.length) detailText = parts.join("; ");
+  }
+  if (statusCode === void 0 && !typeIsError && !detailText) return null;
+  const rawMessage = (typeof rec.message === "string" && rec.message.trim() ? rec.message.trim() : void 0) ?? detailText ?? (typeof rec.error === "string" && rec.error.trim() ? rec.error.trim() : void 0) ?? (statusCode !== void 0 ? `Gateway returned status ${statusCode}.` : "Gateway error.");
+  const message = rawMessage.slice(0, 500);
+  return { message, statusCode };
+}
 function classifyOpenAICompatLimit(args) {
   const { status, code, type, message } = args;
   const codeType = `${code ?? ""} ${type ?? ""}`.toLowerCase();
@@ -2286,6 +2319,7 @@ function classifyOpenAICompatLimit(args) {
   return null;
 }
 function toError2(err, provider = "openai") {
+  if (err instanceof ProviderError) return err;
   if (err instanceof import_openai.default.APIError) {
     const body = err.error;
     const bodyMessage = typeof body?.message === "string" && body.message.trim() ? body.message.trim() : void 0;
@@ -3646,6 +3680,8 @@ function sanitizeMessagesForWire(messages) {
 // src/stream.ts
 var GLM_CODING_BASE_URL = "https://api.z.ai/api/coding/paas/v4";
 var KIMI_CODE_USER_AGENT = `kimi-code-cli/${process.env.KIMI_CODE_VERSION ?? "1.0.11"}`;
+var GROK_CLI_PROXY_HOST = "cli-chat-proxy.grok.com";
+var GROK_CLI_VERSION = process.env.GROK_CLI_VERSION ?? "0.2.101";
 providerRegistry.register("anthropic", {
   stream: (options) => streamAnthropic(options)
 });
@@ -3712,13 +3748,25 @@ providerRegistry.register("xai", {
   // xAI's public API (console.x.ai key) is OpenAI-compatible — ride the Chat
   // Completions transport like Moonshot/DeepSeek. Grok reasoning models take
   // top-level `reasoning_effort` (low/medium/high), which the shared thinking
-  // path already sends. xAI's OAuth path exists but only via the Grok CLI's
-  // private Responses proxy (cli-chat-proxy.grok.com) with reverse-engineered
-  // attribution headers and account-tier gating — intentionally not wired.
-  stream: (options) => streamOpenAI({
-    ...options,
-    baseUrl: options.baseUrl ?? "https://api.x.ai/v1"
-  })
+  // path already sends.
+  //
+  // Subscription OAuth (SuperGrok / X Premium) routes to the Grok CLI chat proxy
+  // instead, which speaks the same Chat Completions wire but gates on Grok-CLI
+  // client identity. Inject those headers centrally here — exactly as the Kimi
+  // endpoint above — so EVERY stream (agent loop, compaction, title-gen,
+  // sub-agents) is accepted rather than depending on each call site to thread
+  // headers. Caller-provided headers still win on collision.
+  stream: (options) => {
+    const baseUrl = options.baseUrl ?? "https://api.x.ai/v1";
+    const defaultHeaders = baseUrl.includes(GROK_CLI_PROXY_HOST) ? {
+      "X-XAI-Token-Auth": "xai-grok-cli",
+      "x-grok-client-version": GROK_CLI_VERSION,
+      "x-grok-client-identifier": "ezcoder",
+      "x-grok-model-override": options.model,
+      ...options.defaultHeaders
+    } : options.defaultHeaders;
+    return streamOpenAI({ ...options, baseUrl, defaultHeaders });
+  }
 });
 providerRegistry.register("minimax", {
   stream: (options) => streamAnthropic({