@mytegroupinc/myte-core 0.0.21 → 0.0.22

package/cli.js

@@ -22,6 +22,7 @@ const {
 } = require("./lib/ai-gateway");
 
 const DEFAULT_API_BASE = "https://api.myte.dev";
+const DEFAULT_DIFF_LIMIT_CHARS = 500_000;
 const REMOVED_COMMAND_MESSAGES = {
   ask: "The `ask` alias has been removed. Use `myte query \"...\"`.",
   chat: "The `chat` command has been removed. Use repeated `myte query` calls instead.",
@@ -314,7 +315,7 @@ function printHelp() {
     "",
     "Options:",
     "  --with-diff           Include deterministic git diffs (project-scoped; fails fast if no project repos are configured or resolved)",
-    "  --diff-limit <chars>  Truncate diff context to N chars (default: 200000)",
+    "  --diff-limit <chars>  Truncate diff context to N chars (default: 500000)",
     "  --timeout-ms <ms>     Request timeout (default: 300000)",
     "  --base-url <url>      API base (default: https://api.myte.dev)",
     "  --payload-file <path> Raw OpenAI-style chat-completions payload for `myte ai`",
@@ -604,6 +605,8 @@ async function fetchJsonWithTimeout(fetchFn, url, options, timeoutMs) {
     } catch {
       const err = new Error(`Non-JSON response (${resp.status}): ${text.slice(0, 500)}`);
       err.status = resp.status;
+      const retryAfter = resp.headers?.get?.("retry-after");
+      if (retryAfter) err.retryAfter = retryAfter;
       throw err;
     }
     return { resp, body };
@@ -1881,6 +1884,10 @@ function resolveRetryAfterMs(err, fallbackMs = 5_000) {
   return fallbackMs;
 }
 
+function isTransientQueryStatusError(err) {
+  return [408, 429, 500, 502, 503, 504].includes(Number(err?.status));
+}
+
 async function createAssistantQueryJob({ apiBase, key, payload, timeoutMs, endpoint = "/project-assistant/query" }) {
   const fetchFn = await getFetch();
   const url = `${apiBase}${endpoint}`;
@@ -5002,8 +5009,9 @@ async function runQuery(args) {
   const timeoutMs = Number.isFinite(timeoutParsed) ? timeoutParsed : 300_000;
 
   const charLimitRaw = args["diff-limit"] || args.diffLimit || args.diff_limit;
-  const charLimitParsed = charLimitRaw !== undefined && charLimitRaw !== null ? Number(charLimitRaw) : 200_000;
-  const diffLimit = Number.isFinite(charLimitParsed) ? charLimitParsed : 200_000;
+  const charLimitParsed =
+    charLimitRaw !== undefined && charLimitRaw !== null ? Number(charLimitRaw) : DEFAULT_DIFF_LIMIT_CHARS;
+  const diffLimit = Number.isFinite(charLimitParsed) ? charLimitParsed : DEFAULT_DIFF_LIMIT_CHARS;
 
   const baseRaw = args["base-url"] || args.baseUrl || args.base_url || process.env.MYTE_API_BASE || DEFAULT_API_BASE;
   const apiBase = normalizeApiBase(baseRaw);
@@ -5077,13 +5085,19 @@ async function runQuery(args) {
       const pollTimeoutMs = Math.max(timeoutMs, 900_000);
       const startedAt = Date.now();
       let finalStatus = null;
+      let pollDelayMs = 2_000;
       do {
-        await sleep(2_000);
+        await sleep(pollDelayMs);
         try {
           finalStatus = await fetchAssistantQueryJobStatus({ apiBase, key, timeoutMs, jobId });
+          pollDelayMs = Math.min(10_000, Math.ceil(pollDelayMs * 1.25));
         } catch (err) {
-          if (Number(err?.status) === 429) {
-            await sleep(resolveRetryAfterMs(err));
+          if (isTransientQueryStatusError(err)) {
+            const fallbackMs =
+              Number(err?.status) === 429 ? 5_000 : Math.min(30_000, Math.max(5_000, pollDelayMs * 2));
+            const waitMs = resolveRetryAfterMs(err, fallbackMs);
+            await sleep(waitMs);
+            pollDelayMs = Math.min(30_000, Math.max(pollDelayMs, waitMs));
             continue;
           }
           throw err;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mytegroupinc/myte-core",
-  "version": "0.0.21",
+  "version": "0.0.22",
   "description": "Myte CLI core implementation.",
   "type": "commonjs",
   "main": "cli.js",