@thanh01.pmt/curriculum-kit 1.0.14 → 1.0.16

package/dist/index.cjs

@@ -140,6 +140,10 @@ var init_errors = __esm({
 // src/ai/streamRunner.ts
 var streamRunner_exports = {};
 __export(streamRunner_exports, {
+  DEFAULT_ARTIFACT_STREAM_IDLE_MS: () => exports.DEFAULT_ARTIFACT_STREAM_IDLE_MS,
+  DEFAULT_ARTIFACT_STREAM_TOTAL_MS: () => exports.DEFAULT_ARTIFACT_STREAM_TOTAL_MS,
+  createIdleAbortController: () => createIdleAbortController,
+  createStreamAbortController: () => createStreamAbortController,
   extractThoughtAndContent: () => extractThoughtAndContent,
   getDesignatedFallbackChain: () => getDesignatedFallbackChain,
   getDesignatedFallbackConfig: () => getDesignatedFallbackConfig,
@@ -291,29 +295,44 @@ function extractThoughtAndContent(rawText) {
     content: content.trim()
   };
 }
-function createIdleAbortController(idleMs) {
+function createStreamAbortController(options) {
+  const envIdle = Number(process.env.ARTIFACT_STREAM_IDLE_TIMEOUT_MS || process.env.STREAM_IDLE_TIMEOUT_MS);
+  const envTotal = Number(process.env.ARTIFACT_STREAM_TOTAL_TIMEOUT_MS || process.env.STREAM_TOTAL_TIMEOUT_MS);
+  const idleMs = Number.isFinite(options?.idleMs) && options.idleMs > 0 ? options.idleMs : Number.isFinite(envIdle) && envIdle > 0 ? envIdle : exports.DEFAULT_ARTIFACT_STREAM_IDLE_MS;
+  const totalMs = Number.isFinite(options?.totalMs) && options.totalMs > 0 ? options.totalMs : Number.isFinite(envTotal) && envTotal > 0 ? envTotal : exports.DEFAULT_ARTIFACT_STREAM_TOTAL_MS;
   const controller = new AbortController();
-  let timer = null;
-  const arm = () => {
-    if (timer) clearTimeout(timer);
-    timer = setTimeout(
+  let idleTimer = null;
+  let totalTimer = null;
+  const armIdle = () => {
+    if (idleTimer) clearTimeout(idleTimer);
+    idleTimer = setTimeout(
       () => controller.abort(new Error(`Idle timeout: no data for ${Math.round(idleMs / 1e3)}s`)),
       idleMs
     );
-    timer?.unref?.();
+    idleTimer?.unref?.();
   };
-  arm();
+  armIdle();
+  if (totalMs > 0) {
+    totalTimer = setTimeout(
+      () => controller.abort(new Error(`Total stream timeout after ${Math.round(totalMs / 1e3)}s`)),
+      totalMs
+    );
+    totalTimer?.unref?.();
+  }
   return {
     signal: controller.signal,
-    /** Reset the idle window (call on every received chunk). */
-    kick: arm,
-    /** Clear the pending timer once the request lifecycle is over. */
+    kick: armIdle,
     dispose: () => {
-      if (timer) clearTimeout(timer);
-      timer = null;
+      if (idleTimer) clearTimeout(idleTimer);
+      if (totalTimer) clearTimeout(totalTimer);
+      idleTimer = null;
+      totalTimer = null;
     }
   };
 }
+function createIdleAbortController(idleMs, totalMs) {
+  return createStreamAbortController({ idleMs, totalMs });
+}
 async function processOpenAISSEStream(response, onChunk, idle, toolCallCollector) {
   if (!response.body) return "";
   const reader = response.body.getReader();
@@ -497,7 +516,7 @@ Guidelines:
     if (isNvidiaPreferred || uniqueNvidiaModels.length > 0) {
       for (const model of uniqueNvidiaModels) {
         try {
-          const idle = createIdleAbortController(18e4);
+          const idle = createStreamAbortController({ idleMs: options.idleTimeoutMs, totalMs: options.timeoutMs });
           const res = await fetch("https://integrate.api.nvidia.com/v1/chat/completions", {
             method: "POST",
             headers: {
@@ -541,7 +560,7 @@ Guidelines:
     const uniqueOrModels = Array.from(new Set(openrouterModels));
     for (const model of uniqueOrModels) {
       try {
-        const idle = createIdleAbortController(18e4);
+        const idle = createStreamAbortController({ idleMs: options.idleTimeoutMs, totalMs: options.timeoutMs });
         const res = await fetch("https://openrouter.ai/api/v1/chat/completions", {
           method: "POST",
           headers: {
@@ -580,7 +599,7 @@ Guidelines:
     const dsModel = options.model?.includes("deepseek-reasoner") ? "deepseek-reasoner" : "deepseek-chat";
     if (isModelAllowed(dsModel)) {
       try {
-        const idle = createIdleAbortController(18e4);
+        const idle = createStreamAbortController({ idleMs: options.idleTimeoutMs, totalMs: options.timeoutMs });
         const res = await fetch("https://api.deepseek.com/chat/completions", {
           method: "POST",
           headers: {
@@ -615,7 +634,7 @@ Guidelines:
     const baseUrl = alibabaKey.startsWith("sk-sp-") ? "https://token-plan.ap-southeast-1.maas.aliyuncs.com/compatible-mode/v1/chat/completions" : "https://dashscope-intl.aliyuncs.com/compatible-mode/v1/chat/completions";
     for (const qwenModel of qwenModels) {
       try {
-        const idle = createIdleAbortController(18e4);
+        const idle = createStreamAbortController({ idleMs: options.idleTimeoutMs, totalMs: options.timeoutMs });
         const payload = {
           model: qwenModel,
           messages: formattedMessages,
@@ -658,7 +677,7 @@ Guidelines:
     ];
     for (const gModel of geminiModels) {
       try {
-        const idle = createIdleAbortController(18e4);
+        const idle = createStreamAbortController({ idleMs: options.idleTimeoutMs, totalMs: options.timeoutMs });
         const res = await fetch(`https://generativelanguage.googleapis.com/v1beta/models/${gModel}:streamGenerateContent?alt=sse&key=${geminiKey}`, {
           method: "POST",
           headers: { "Content-Type": "application/json" },
@@ -694,7 +713,7 @@ Guidelines:
       const { provider, model } = target;
       if (provider === "nvidia" && nvidiaKey && isProviderEnabled("nvidia")) {
         try {
-          const idle = createIdleAbortController(18e4);
+          const idle = createStreamAbortController({ idleMs: options.idleTimeoutMs, totalMs: options.timeoutMs });
           const res = await fetch("https://integrate.api.nvidia.com/v1/chat/completions", {
             method: "POST",
             headers: {
@@ -726,7 +745,7 @@ Guidelines:
         }
       } else if (provider === "openrouter" && openrouterKey && isProviderEnabled("openrouter")) {
         try {
-          const idle = createIdleAbortController(18e4);
+          const idle = createStreamAbortController({ idleMs: options.idleTimeoutMs, totalMs: options.timeoutMs });
           const res = await fetch("https://openrouter.ai/api/v1/chat/completions", {
             method: "POST",
             headers: {
@@ -761,7 +780,7 @@ Guidelines:
       } else if ((provider === "alibaba" || provider === "dashscope") && alibabaKey && isProviderEnabled("alibaba")) {
         const baseUrl = alibabaKey.startsWith("sk-sp-") ? "https://token-plan.ap-southeast-1.maas.aliyuncs.com/compatible-mode/v1/chat/completions" : "https://dashscope-intl.aliyuncs.com/compatible-mode/v1/chat/completions";
         try {
-          const idle = createIdleAbortController(18e4);
+          const idle = createStreamAbortController({ idleMs: options.idleTimeoutMs, totalMs: options.timeoutMs });
           const res = await fetch(baseUrl, {
             method: "POST",
             headers: {
@@ -793,7 +812,7 @@ Guidelines:
         }
       } else if ((provider === "google" || provider === "gemini") && geminiKey && (isProviderEnabled("gemini") || isProviderEnabled("google"))) {
         try {
-          const idle = createIdleAbortController(18e4);
+          const idle = createStreamAbortController({ idleMs: options.idleTimeoutMs, totalMs: options.timeoutMs });
           const res = await fetch(
             `https://generativelanguage.googleapis.com/v1beta/models/${model}:streamGenerateContent?alt=sse&key=${geminiKey}`,
             {
@@ -844,9 +863,12 @@ Guidelines:
 async function runCurriculumAIInference(messages, projectContext, options = {}, onChunk) {
   return await streamCurriculumAIInference(messages, projectContext, options, onChunk);
 }
+exports.DEFAULT_ARTIFACT_STREAM_IDLE_MS = void 0; exports.DEFAULT_ARTIFACT_STREAM_TOTAL_MS = void 0;
 var init_streamRunner = __esm({
   "src/ai/streamRunner.ts"() {
     init_errors();
+    exports.DEFAULT_ARTIFACT_STREAM_IDLE_MS = 45e3;
+    exports.DEFAULT_ARTIFACT_STREAM_TOTAL_MS = 42e4;
   }
 });
 
@@ -13525,11 +13547,11 @@ var DeterministicPipelineRunner = class {
 // src/services/prefillService.ts
 init_streamRunner();
 var DEFAULT_STREAM_IDLE_MS = 45e3;
-var DEFAULT_STREAM_TOTAL_MS = 3e5;
+var DEFAULT_STREAM_TOTAL_MS = 42e4;
 var LAYER_TOTAL_BUDGET_MS = {
-  1: 18e4,
-  2: 36e4,
-  3: 36e4
+  1: 42e4,
+  2: 42e4,
+  3: 42e4
 };
 function resolveStreamBudget(layer, opts) {
   const envIdle = Number(process.env.WIZARD_PREFILL_IDLE_TIMEOUT_MS);
@@ -13596,29 +13618,111 @@ function createStreamAbortSignal(budget) {
     }
   };
 }
+function closeTruncatedJson(candidate) {
+  let inStr = false;
+  let esc2 = false;
+  const stack = [];
+  for (let i = 0; i < candidate.length; i++) {
+    const ch = candidate[i];
+    if (esc2) {
+      esc2 = false;
+      continue;
+    }
+    if (inStr) {
+      if (ch === "\\") esc2 = true;
+      else if (ch === '"') inStr = false;
+      continue;
+    }
+    if (ch === '"') inStr = true;
+    else if (ch === "{" || ch === "[") stack.push(ch);
+    else if (ch === "}" || ch === "]") stack.pop();
+  }
+  if (stack.length === 0 && !inStr) return null;
+  let out = candidate;
+  out = out.replace(/,\s*$/, "");
+  if (inStr) {
+    out = out.replace(/\\+$/, "");
+    out += '"';
+  }
+  while (stack.length > 0) {
+    out += stack.pop() === "{" ? "}" : "]";
+  }
+  return out;
+}
+function stripLlmJsonWrappers(rawText) {
+  let cleaned = rawText.replace(/<think>[\s\S]*?<\/think>/gi, "").replace(/```json\s*/gi, "").replace(/```\s*/g, "").trim();
+  const firstBrace = cleaned.indexOf("{");
+  if (firstBrace > 0) {
+    cleaned = cleaned.slice(firstBrace);
+  }
+  return cleaned.trim();
+}
 function safeParseJson(rawText) {
   if (!rawText) return null;
-  const cleaned = rawText.replace(/```json\s*/gi, "").replace(/```\s*/g, "").trim();
+  const cleaned = stripLlmJsonWrappers(rawText);
+  if (!cleaned) return null;
   try {
     return JSON.parse(cleaned);
   } catch {
-    const firstBrace = cleaned.indexOf("{");
+  }
+  const firstBrace = cleaned.indexOf("{");
+  if (firstBrace === -1) return null;
+  const buildCandidates = () => {
+    const list = [];
+    const closed = closeTruncatedJson(cleaned.slice(firstBrace));
+    if (closed) list.push(closed);
     const lastBrace = cleaned.lastIndexOf("}");
-    if (firstBrace !== -1 && lastBrace !== -1 && lastBrace > firstBrace) {
-      const candidate = cleaned.slice(firstBrace, lastBrace + 1);
-      try {
-        return JSON.parse(candidate);
-      } catch {
+    if (lastBrace > firstBrace) {
+      list.push(cleaned.slice(firstBrace, lastBrace + 1));
+    }
+    return list;
+  };
+  for (const candidate of buildCandidates()) {
+    try {
+      return JSON.parse(candidate);
+    } catch {
+    }
+    try {
+      return JSON.parse(jsonrepair.jsonrepair(candidate));
+    } catch {
+    }
+  }
+  let depth = 0;
+  let inStr = false;
+  let esc2 = false;
+  for (let i = firstBrace; i < cleaned.length; i++) {
+    const ch = cleaned[i];
+    if (esc2) {
+      esc2 = false;
+      continue;
+    }
+    if (inStr) {
+      if (ch === "\\") esc2 = true;
+      else if (ch === '"') inStr = false;
+      continue;
+    }
+    if (ch === '"') inStr = true;
+    else if (ch === "{") depth++;
+    else if (ch === "}") {
+      depth--;
+      if (depth === 0) {
+        const candidate = cleaned.slice(firstBrace, i + 1);
+        try {
+          return JSON.parse(candidate);
+        } catch {
+        }
         try {
-          const repaired = jsonrepair.jsonrepair(candidate);
-          return JSON.parse(repaired);
+          return JSON.parse(jsonrepair.jsonrepair(candidate));
         } catch {
-          return null;
         }
       }
     }
-    return null;
   }
+  try {
+    return JSON.parse(jsonrepair.jsonrepair(cleaned));
+  } catch {
+  }
+  return null;
 }
 async function streamLLMWithFallback(systemPrompt, userPrompt, apiKeys = {}, onChunk, preferredModel, preferredProvider, budget) {
   const alibabaKey = apiKeys.alibaba || apiKeys.dashscope || process.env.ALIBABA_API_KEY || process.env.DASHSCOPE_API_KEY;
@@ -16131,6 +16235,10 @@ function resolveApiKey2(options) {
   }
   return { provider, key };
 }
+function resolveImageTimeout(options) {
+  const envTimeout = Number(process.env.IMAGE_GEN_TIMEOUT_MS || process.env.ARTIFACT_STREAM_TOTAL_TIMEOUT_MS);
+  return Number.isFinite(options.timeoutMs) && options.timeoutMs > 0 ? options.timeoutMs : Number.isFinite(envTimeout) && envTimeout > 0 ? envTimeout : 42e4;
+}
 async function generateWithGemini(options, key) {
   const finalPrompt = buildImagePrompt(options);
   const model = "gemini-2.0-flash-exp";
@@ -16141,7 +16249,7 @@ async function generateWithGemini(options, key) {
       contents: [{ parts: [{ text: finalPrompt }] }],
       generationConfig: { responseModalities: ["TEXT", "IMAGE"] }
     }),
-    signal: AbortSignal.timeout(9e4)
+    signal: AbortSignal.timeout(resolveImageTimeout(options))
   });
   if (!res.ok) {
     const body = await res.text().catch(() => "");
@@ -16174,7 +16282,7 @@ async function generateWithOpenAI(options, key) {
       size: sizeMap[options.aspectRatio || "1:1"] || "1024x1024",
       n: 1
     }),
-    signal: AbortSignal.timeout(9e4)
+    signal: AbortSignal.timeout(resolveImageTimeout(options))
   });
   if (!res.ok) {
     const body = await res.text().catch(() => "");
@@ -16790,6 +16898,7 @@ exports.buildStandardsContext = buildStandardsContext;
 exports.buildStandardsContextBlock = buildStandardsContextBlock;
 exports.buildTeacherGuidePrompt = buildTeacherGuidePrompt;
 exports.buildWorksheetPrompt = buildWorksheetPrompt;
+exports.closeTruncatedJson = closeTruncatedJson;
 exports.computeContentHash = computeContentHash;
 exports.computePackingSpec = computePackingSpec;
 exports.conductPreliminaryResearch = conductPreliminaryResearch;
@@ -16797,6 +16906,8 @@ exports.contentTools = contentTools;
 exports.convertRoadmapToFoundationSot = convertRoadmapToFoundationSot;
 exports.createAiInferenceError = createAiInferenceError;
 exports.createCurriculumStorage = createCurriculumStorage;
+exports.createIdleAbortController = createIdleAbortController;
+exports.createStreamAbortController = createStreamAbortController;
 exports.createStreamAbortSignal = createStreamAbortSignal;
 exports.curateMediaLedger = curateMediaLedger;
 exports.designerTools = designerTools;
@@ -16896,6 +17007,7 @@ exports.streamLLMWithFallback = streamLLMWithFallback;
 exports.streamLayerPrefillWithFallback = streamLayerPrefillWithFallback;
 exports.streamLessonMasterFlow = streamLessonMasterFlow;
 exports.streamSelfLabFlow = streamSelfLabFlow;
+exports.stripLlmJsonWrappers = stripLlmJsonWrappers;
 exports.techSmeTools = techSmeTools;
 exports.topoSort = topoSort;
 exports.uploadAssetToBucket = uploadAssetToBucket;