@mytegroupinc/myte-core 0.0.16 → 0.0.17

package/README.md

@@ -139,3 +139,5 @@ Examples:
 - `npx myte update-client --subject "Weekly client update" --body-file ./updates/week-12.md`
 
 This package is published under the org scope for governance; the public `myte` wrapper delegates here.
+
+`--json-response` sends both the Myte convenience flag (`myte_json_response: true`) and the OpenAI-compatible chat-completions shape (`response_format: { type: "json_object" }`). On the public developer API, that strict-JSON path also defaults to `medium` reasoning unless the caller explicitly overrides reasoning.

package/cli.js

@@ -275,7 +275,7 @@ function printHelp() {
     "  --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`",
-    "  --json-response       Ask the Myte AI gateway to return clean JSON only",
+    "  --json-response       Ask the Myte AI gateway to return clean JSON only and send OpenAI-compatible response_format",
     "  --max-output-tokens   Output token cap for `myte ai` simple queries",
     "  --temperature <num>   Temperature for `myte ai` simple queries",
     "  --output-dir <path>   Command Center output directory (default: <wrapper-root>/MyteCommandCenter)",
@@ -1067,9 +1067,15 @@ async function fetchQaqcSyncSnapshot({ apiBase, key, timeoutMs }) {
   );
 
   if (!resp.ok || body.status !== "success") {
+    const retryAfter = resp.headers?.get?.("retry-after");
     const msg = body?.message || `QAQC sync request failed (${resp.status})`;
-    const err = new Error(msg);
+    const err = new Error(
+      retryAfter
+        ? `${msg} Retry after ${retryAfter}s.`
+        : msg
+    );
     err.status = resp.status;
+    if (retryAfter) err.retryAfter = retryAfter;
     throw err;
   }
   return body.data || {};
@@ -1227,15 +1233,30 @@ async function fetchRunQaqcBatchStatus({ apiBase, key, timeoutMs, batchId }) {
   );
 
   if (!resp.ok || body.status !== "success") {
+    const retryAfter = resp.headers?.get?.("retry-after");
     const msg = body?.message || `Run QAQC status request failed (${resp.status})`;
-    const err = new Error(msg);
+    const err = new Error(
+      retryAfter
+        ? `${msg} Retry after ${retryAfter}s.`
+        : msg
+    );
     err.status = resp.status;
+    if (retryAfter) err.retryAfter = retryAfter;
     throw err;
   }
   return body.data || {};
 }
 
-async function callAssistantQuery({ apiBase, key, payload, timeoutMs, endpoint = "/project-assistant/query" }) {
+function resolveRetryAfterMs(err, fallbackMs = 5_000) {
+  const retryAfterRaw = firstNonEmptyString(err?.retryAfter);
+  const retryAfterSeconds = Number.parseInt(String(retryAfterRaw || ""), 10);
+  if (Number.isFinite(retryAfterSeconds) && retryAfterSeconds > 0) {
+    return Math.min(retryAfterSeconds * 1_000, 60_000);
+  }
+  return fallbackMs;
+}
+
+async function createAssistantQueryJob({ apiBase, key, payload, timeoutMs, endpoint = "/project-assistant/query" }) {
   const fetchFn = await getFetch();
   const url = `${apiBase}${endpoint}`;
   const { resp, body } = await fetchJsonWithTimeout(
@@ -1253,9 +1274,35 @@ async function callAssistantQuery({ apiBase, key, payload, timeoutMs, endpoint =
   );
 
   if (!resp.ok || body.status !== "success") {
+    const retryAfter = resp.headers?.get?.("retry-after");
     const msg = body?.message || `Query failed (${resp.status})`;
-    const err = new Error(msg);
+    const err = new Error(retryAfter ? `${msg} Retry after ${retryAfter}s.` : msg);
+    err.status = resp.status;
+    if (retryAfter) err.retryAfter = retryAfter;
+    throw err;
+  }
+  return body.data || {};
+}
+
+async function fetchAssistantQueryJobStatus({ apiBase, key, timeoutMs, jobId }) {
+  const fetchFn = await getFetch();
+  const url = `${apiBase}/project-assistant/query/${encodeURIComponent(String(jobId || ""))}`;
+  const { resp, body } = await fetchJsonWithTimeout(
+    fetchFn,
+    url,
+    {
+      method: "GET",
+      headers: { Authorization: `Bearer ${key}` },
+    },
+    timeoutMs
+  );
+
+  if (!resp.ok || body.status !== "success") {
+    const retryAfter = resp.headers?.get?.("retry-after");
+    const msg = body?.message || `Query status request failed (${resp.status})`;
+    const err = new Error(retryAfter ? `${msg} Retry after ${retryAfter}s.` : msg);
     err.status = resp.status;
+    if (retryAfter) err.retryAfter = retryAfter;
     throw err;
   }
   return body.data || {};
@@ -1326,6 +1373,10 @@ async function runRunQaqc(args) {
       try {
         finalStatus = await fetchRunQaqcBatchStatus({ apiBase, key, timeoutMs, batchId: data.batch_id });
       } catch (err) {
+        if (Number(err?.status) === 429) {
+          await sleep(resolveRetryAfterMs(err));
+          continue;
+        }
         console.error("Failed to fetch QAQC batch status:", err?.message || err);
         process.exit(1);
       }
@@ -1341,11 +1392,18 @@ async function runRunQaqc(args) {
 
     if (shouldSync && ["completed", "completed_with_errors"].includes(String(finalStatus.status || "").trim())) {
       let snapshot;
-      try {
-        snapshot = await fetchQaqcSyncSnapshot({ apiBase, key, timeoutMs });
-      } catch (err) {
-        console.error("Failed to fetch QAQC sync snapshot after batch completion:", err?.message || err);
-        process.exit(1);
+      for (let attempt = 0; attempt < 5; attempt += 1) {
+        try {
+          snapshot = await fetchQaqcSyncSnapshot({ apiBase, key, timeoutMs });
+          break;
+        } catch (err) {
+          if (Number(err?.status) === 429 && attempt < 4) {
+            await sleep(resolveRetryAfterMs(err));
+            continue;
+          }
+          console.error("Failed to fetch QAQC sync snapshot after batch completion:", err?.message || err);
+          process.exit(1);
+        }
       }
 
       let resolved;
@@ -1668,6 +1726,20 @@ function stableItemId(item, keys, fallback) {
   return fallback;
 }
 
+function bootstrapPathId(item, preferredKeys, fallbackKeys, fallback) {
+  return stableItemId(item, preferredKeys, stableItemId(item, fallbackKeys, fallback));
+}
+
+function bootstrapScopedPathId(item, preferredKeys, scopeKeys, fallbackKeys, fallback) {
+  const preferred = stableItemId(item, preferredKeys, "");
+  if (preferred) return preferred;
+  const fallbackId = stableItemId(item, fallbackKeys, fallback);
+  const scopeParts = (Array.isArray(scopeKeys) ? scopeKeys : [])
+    .map((key) => String(item?.[key] || "").trim())
+    .filter(Boolean);
+  return [...scopeParts, fallbackId].filter(Boolean).join("__") || fallbackId;
+}
+
 function stringifyYaml(value) {
   // eslint-disable-next-line global-require
   const YAML = require("yaml");
@@ -1889,20 +1961,42 @@ function writeBootstrapSnapshot({ snapshot, wrapperRoot, outputDir }) {
   const missions = Array.isArray(snapshot.missions) ? snapshot.missions.map((item) => scrubBootstrapValue(item)) : [];
 
   phases.forEach((phase, index) => {
-    const phaseId = stableItemId(phase, ["phase_id", "id"], `P${String(index + 1).padStart(3, "0")}`);
-    writeYamlFile(path.join(phasesDir, `${phaseId}.yml`), phase);
+    const phasePathId = bootstrapPathId(
+      phase,
+      ["phase_key", "id"],
+      ["phase_id"],
+      `phase-${String(index + 1).padStart(3, "0")}`
+    );
+    writeYamlFile(path.join(phasesDir, `${phasePathId}.yml`), phase);
   });
   epics.forEach((epic, index) => {
-    const epicId = stableItemId(epic, ["epic_id", "id"], `E${String(index + 1).padStart(3, "0")}`);
-    writeYamlFile(path.join(epicsDir, `${epicId}.yml`), epic);
+    const epicPathId = bootstrapScopedPathId(
+      epic,
+      ["epic_key", "id"],
+      ["phase_key", "phase_id"],
+      ["epic_id"],
+      `epic-${String(index + 1).padStart(3, "0")}`
+    );
+    writeYamlFile(path.join(epicsDir, `${epicPathId}.yml`), epic);
   });
   stories.forEach((story, index) => {
-    const storyId = stableItemId(story, ["story_id", "id"], `S${String(index + 1).padStart(3, "0")}`);
-    writeYamlFile(path.join(storiesDir, `${storyId}.yml`), story);
+    const storyPathId = bootstrapScopedPathId(
+      story,
+      ["story_key", "id"],
+      ["phase_key", "phase_id", "epic_key", "epic_id"],
+      ["story_id"],
+      `story-${String(index + 1).padStart(3, "0")}`
+    );
+    writeYamlFile(path.join(storiesDir, `${storyPathId}.yml`), story);
   });
   missions.forEach((mission, index) => {
-    const missionId = stableItemId(mission, ["mission_id", "id"], `M${String(index + 1).padStart(3, "0")}`);
-    writeYamlFile(path.join(missionsDir, `${missionId}.yml`), mission);
+    const missionPathId = bootstrapPathId(
+      mission,
+      ["mission_key", "id"],
+      ["mission_id"],
+      `mission-${String(index + 1).padStart(3, "0")}`
+    );
+    writeYamlFile(path.join(missionsDir, `${missionPathId}.yml`), mission);
   });
 
   if (snapshot.project && typeof snapshot.project === "object") {
@@ -3815,7 +3909,46 @@ async function runQuery(args) {
 
   let data;
   try {
-    data = await callAssistantQuery({ apiBase, key, payload, timeoutMs });
+    const queued = await createAssistantQueryJob({ apiBase, key, payload, timeoutMs });
+    if (queued.answer) {
+      data = queued;
+    } else {
+      const jobId = firstNonEmptyString(queued.job_id, queued.id);
+      if (!jobId) {
+        throw new Error("Query job was accepted without a job_id.");
+      }
+      const pollTimeoutMs = Math.max(timeoutMs, 900_000);
+      const startedAt = Date.now();
+      let finalStatus = null;
+      do {
+        await sleep(2_000);
+        try {
+          finalStatus = await fetchAssistantQueryJobStatus({ apiBase, key, timeoutMs, jobId });
+        } catch (err) {
+          if (Number(err?.status) === 429) {
+            await sleep(resolveRetryAfterMs(err));
+            continue;
+          }
+          throw err;
+        }
+        if (["completed", "failed"].includes(String(finalStatus.status || "").trim())) {
+          break;
+        }
+      } while (Date.now() - startedAt < pollTimeoutMs);
+
+      if (!finalStatus || !["completed", "failed"].includes(String(finalStatus.status || "").trim())) {
+        throw new Error(`Timed out waiting for query job ${jobId}`);
+      }
+      if (String(finalStatus.status || "").trim() === "failed") {
+        const detail = firstNonEmptyString(finalStatus?.error?.message, finalStatus?.error?.code);
+        throw new Error(detail || `Query job ${jobId} failed`);
+      }
+      data = {
+        answer: finalStatus.answer,
+        context_blocks: finalStatus.context_blocks,
+        telemetry: finalStatus.telemetry,
+      };
+    }
   } catch (err) {
     if (err?.name === "AbortError") {
       console.error(`Request timed out after ${timeoutMs}ms`);

package/lib/ai-gateway.js

@@ -16,7 +16,10 @@ function buildSimpleAiPayload({ query, jsonResponse = false, maxOutputTokens, te
   const payload = {
     messages: [{ role: "user", content: String(query || "").trim() }],
   };
-  if (jsonResponse) payload.myte_json_response = true;
+  if (jsonResponse) {
+    payload.myte_json_response = true;
+    payload.response_format = { type: "json_object" };
+  }
   if (Number.isFinite(Number(maxOutputTokens)) && Number(maxOutputTokens) > 0) {
     payload.max_tokens = Number(maxOutputTokens);
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mytegroupinc/myte-core",
-  "version": "0.0.16",
+  "version": "0.0.17",
   "description": "Myte CLI core implementation (Project Assistant + Myte AI gateway).",
   "type": "commonjs",
   "main": "cli.js",