@thanh01.pmt/curriculum-kit 1.0.12 → 1.0.13

package/dist/index.cjs

@@ -13525,6 +13525,78 @@ var DeterministicPipelineRunner = class {
 
 // src/services/prefillService.ts
 init_streamRunner();
+var DEFAULT_STREAM_IDLE_MS = 45e3;
+var DEFAULT_STREAM_TOTAL_MS = 3e5;
+var LAYER_TOTAL_BUDGET_MS = {
+  1: 18e4,
+  2: 36e4,
+  3: 36e4
+};
+function resolveStreamBudget(layer, opts) {
+  const envIdle = Number(process.env.WIZARD_PREFILL_IDLE_TIMEOUT_MS);
+  const envTotal = Number(process.env.WIZARD_PREFILL_TOTAL_TIMEOUT_MS);
+  const layerTotal = layer !== void 0 ? LAYER_TOTAL_BUDGET_MS[layer] : void 0;
+  return {
+    idleMs: opts?.idleMs ?? (Number.isFinite(envIdle) && envIdle > 0 ? envIdle : DEFAULT_STREAM_IDLE_MS),
+    totalMs: opts?.totalMs ?? (Number.isFinite(envTotal) && envTotal > 0 ? envTotal : layerTotal ?? DEFAULT_STREAM_TOTAL_MS)
+  };
+}
+function extractStreamChunk(part) {
+  if (!part || typeof part !== "object") return {};
+  if (part.type === "reasoning-delta" || part.type === "reasoning") {
+    const thought = part.text ?? part.delta ?? part.reasoning ?? "";
+    return thought ? { thought } : {};
+  }
+  if (part.type === "text-delta") {
+    const content = part.text ?? part.delta ?? "";
+    return content ? { content } : {};
+  }
+  if (part.type === "raw") {
+    const raw = part.rawValue;
+    const delta = raw?.choices?.[0]?.delta;
+    const reasoning = delta?.reasoning_content ?? delta?.reasoning ?? raw?.delta?.reasoning_content ?? raw?.delta?.reasoning;
+    if (typeof reasoning === "string" && reasoning) return { thought: reasoning };
+    const text = delta?.content ?? raw?.delta?.content;
+    if (typeof text === "string" && text) return { content: text };
+    return {};
+  }
+  return {};
+}
+function createStreamAbortSignal(budget) {
+  const controller = new AbortController();
+  let idleTimer = null;
+  let totalTimer = null;
+  const armIdle = () => {
+    if (idleTimer) clearTimeout(idleTimer);
+    if (budget.idleMs > 0) {
+      idleTimer = setTimeout(
+        () => controller.abort(new Error(`Idle timeout: no data for ${Math.round(budget.idleMs / 1e3)}s`)),
+        budget.idleMs
+      );
+      idleTimer?.unref?.();
+    }
+  };
+  armIdle();
+  if (budget.totalMs > 0) {
+    totalTimer = setTimeout(
+      () => controller.abort(new Error(`Total stream timeout after ${Math.round(budget.totalMs / 1e3)}s`)),
+      budget.totalMs
+    );
+    totalTimer?.unref?.();
+  }
+  return {
+    signal: controller.signal,
+    /** Reset the idle window — call on EVERY received stream part. */
+    kick: armIdle,
+    /** Clear both timers once the stream lifecycle is over. */
+    dispose: () => {
+      if (idleTimer) clearTimeout(idleTimer);
+      if (totalTimer) clearTimeout(totalTimer);
+      idleTimer = null;
+      totalTimer = null;
+    }
+  };
+}
 function safeParseJson(rawText) {
   if (!rawText) return null;
   const cleaned = rawText.replace(/```json\s*/gi, "").replace(/```\s*/g, "").trim();
@@ -13549,7 +13621,7 @@ function safeParseJson(rawText) {
     return null;
   }
 }
-async function streamLLMWithFallback(systemPrompt, userPrompt, apiKeys = {}, onChunk, preferredModel, preferredProvider) {
+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;
   const nvidiaKey = apiKeys.nvidia || process.env.NVIDIA_API_KEY;
   const deepseekKey = apiKeys.deepseek || process.env.DEEPSEEK_API_KEY;
@@ -13627,8 +13699,13 @@ async function streamLLMWithFallback(systemPrompt, userPrompt, apiKeys = {}, onC
   const systemInstructions = `${systemPrompt}
 
 THINKING DIRECTIVE: Keep reasoning concise, structured, and focused strictly on the core pedagogical trade-offs in 4-6 concise bullet points (under 120 words). Then output the JSON immediately.`;
+  const resolvedBudget = {
+    idleMs: budget?.idleMs ?? DEFAULT_STREAM_IDLE_MS,
+    totalMs: budget?.totalMs ?? DEFAULT_STREAM_TOTAL_MS
+  };
   for (const candidate of candidates) {
     const t0 = Date.now();
+    const abort = createStreamAbortSignal(resolvedBudget);
     try {
       const modelInstance = getAIModel({
         provider: candidate.provider,
@@ -13641,26 +13718,16 @@ THINKING DIRECTIVE: Keep reasoning concise, structured, and focused strictly on
         prompt: userPrompt,
         temperature: 0.2,
         includeRawChunks: true,
-        abortSignal: AbortSignal.timeout(6e4)
+        abortSignal: abort.signal
       });
       let fullContent = "";
       for await (const part of streamResult.fullStream) {
-        if (part.type === "reasoning-delta") {
-          const thoughtText = part.text ?? part.delta ?? "";
-          if (thoughtText) onChunk?.(thoughtText, "thought");
-        } else if (part.type === "raw") {
-          const raw = part.rawValue;
-          const delta = raw?.choices?.[0]?.delta;
-          const reasoning = delta?.reasoning_content || delta?.reasoning;
-          if (reasoning) {
-            onChunk?.(reasoning, "thought");
-          }
-        } else if (part.type === "text-delta") {
-          const textDelta = part.text ?? part.delta ?? "";
-          if (textDelta) {
-            fullContent += textDelta;
-            onChunk?.(textDelta, "content");
-          }
+        abort.kick();
+        const extracted = extractStreamChunk(part);
+        if (extracted.thought) onChunk?.(extracted.thought, "thought");
+        if (extracted.content) {
+          fullContent += extracted.content;
+          onChunk?.(extracted.content, "content");
         }
       }
       if (fullContent.trim()) {
@@ -13669,6 +13736,8 @@ THINKING DIRECTIVE: Keep reasoning concise, structured, and focused strictly on
       }
     } catch (e) {
       console.warn(`[curriculum-kit:VercelAI] Candidate ${candidate.provider} (${candidate.model}) failed in ${Date.now() - t0}ms:`, e?.message || e);
+    } finally {
+      abort.dispose();
     }
   }
   return null;
@@ -13709,7 +13778,11 @@ Return concise JSON matching:
     (chunk, type) => {
       if (type === "content") rawContent += chunk;
       onChunk?.(chunk, type);
-    }
+    },
+    options.model,
+    options.provider,
+    // Research may run live web grounding — generous budget, still idle-guarded.
+    resolveStreamBudget(void 0, { idleMs: options.idleTimeoutMs, totalMs: options.timeoutMs })
   );
   let parsedResearch = {
     groundTruthSummary: `Verified ecosystem standards for ${projectName}`,
@@ -13911,7 +13984,10 @@ REQUIREMENT: Synthesize ALL parameters above \u2014 every single line in FULL AC
       onChunk?.(chunk, type);
     },
     options.model,
-    options.provider
+    options.provider,
+    // RC-W1: layer-aware budget — Layer 2 has the largest output and free-tier
+    // models may think/stream for minutes. Idle window still catches hangs.
+    resolveStreamBudget(targetLayer, { idleMs: options.idleTimeoutMs, totalMs: options.timeoutMs })
   );
   const parsed = safeParseJson(rawContent);
   return {
@@ -17190,6 +17266,8 @@ exports.CurriculumPlanSchema = CurriculumPlanSchema;
 exports.CurriculumQualityReportSchema = CurriculumQualityReportSchema;
 exports.DEFAULT_ENABLED_PROVIDERS = DEFAULT_ENABLED_PROVIDERS;
 exports.DEFAULT_GATE_SETTINGS = DEFAULT_GATE_SETTINGS;
+exports.DEFAULT_STREAM_IDLE_MS = DEFAULT_STREAM_IDLE_MS;
+exports.DEFAULT_STREAM_TOTAL_MS = DEFAULT_STREAM_TOTAL_MS;
 exports.DependencyEdgeSchema = DependencyEdgeSchema;
 exports.DepthAssignmentSchema = DepthAssignmentSchema;
 exports.DepthLevelSchema = DepthLevelSchema;
@@ -17234,6 +17312,7 @@ exports.HandoutSectionSchema = HandoutSectionSchema;
 exports.InstructionSectionSchema = InstructionSectionSchema;
 exports.InstructionStepSchema = InstructionStepSchema;
 exports.JudgeCriterionSchema = JudgeCriterionSchema;
+exports.LAYER_TOTAL_BUDGET_MS = LAYER_TOTAL_BUDGET_MS;
 exports.LESSON_PLAN_TEMPLATE = LESSON_PLAN_TEMPLATE;
 exports.LLMJudgeEngine = LLMJudgeEngine;
 exports.LabTierTaskSchema = LabTierTaskSchema;
@@ -17363,6 +17442,7 @@ exports.contentTools = contentTools;
 exports.convertRoadmapToFoundationSot = convertRoadmapToFoundationSot;
 exports.createAiInferenceError = createAiInferenceError;
 exports.createCurriculumStorage = createCurriculumStorage;
+exports.createStreamAbortSignal = createStreamAbortSignal;
 exports.curateMediaLedger = curateMediaLedger;
 exports.designerTools = designerTools;
 exports.detectProjectPedagogy = detectProjectPedagogy;
@@ -17375,6 +17455,7 @@ exports.exportAllProjectQuizzes = exportAllProjectQuizzes;
 exports.expositionCacheKey = expositionCacheKey;
 exports.extractScopeSequenceRows = extractScopeSequenceRows;
 exports.extractSessionSlice = extractSessionSlice;
+exports.extractStreamChunk = extractStreamChunk;
 exports.extractThoughtAndContent = extractThoughtAndContent;
 exports.formatQuizzesToCsv = formatQuizzesToCsv;
 exports.fulfillMediaLedger = fulfillMediaLedger;
@@ -17457,6 +17538,7 @@ exports.repairMasterLessonStep = repairMasterLessonStep;
 exports.researcherTools = researcherTools;
 exports.resolveGateSettings = resolveGateSettings;
 exports.resolveStandardsPacks = resolveStandardsPacks;
+exports.resolveStreamBudget = resolveStreamBudget;
 exports.resolveTranslationTargets = resolveTranslationTargets;
 exports.reviewerTools = reviewerTools;
 exports.runCurriculumAIInference = runCurriculumAIInference;