@thanh01.pmt/curriculum-kit 1.4.51 → 1.4.52

package/dist/pipeline/index.mjs

@@ -6,6 +6,7 @@ import fs2 from 'fs';
 import path from 'path';
 import { z } from 'zod';
 import { fileURLToPath } from 'url';
+import { jsonrepair } from 'jsonrepair';
 import pLimit from 'p-limit';
 import fsPromises from 'fs/promises';
 
@@ -3588,6 +3589,152 @@ async function generateExtensionFlow(input) {
   });
   return object;
 }
+function closeTruncatedJson(candidate) {
+  let inStr = false;
+  let esc = false;
+  const stack = [];
+  for (let i = 0; i < candidate.length; i++) {
+    const ch = candidate[i];
+    if (esc) {
+      esc = false;
+      continue;
+    }
+    if (inStr) {
+      if (ch === "\\") esc = 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 = stripLlmJsonWrappers(rawText);
+  if (!cleaned) return null;
+  try {
+    return JSON.parse(cleaned);
+  } catch {
+  }
+  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 (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(candidate));
+    } catch {
+    }
+  }
+  let depth = 0;
+  let inStr = false;
+  let esc = false;
+  for (let i = firstBrace; i < cleaned.length; i++) {
+    const ch = cleaned[i];
+    if (esc) {
+      esc = false;
+      continue;
+    }
+    if (inStr) {
+      if (ch === "\\") esc = 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 {
+          return JSON.parse(jsonrepair(candidate));
+        } catch {
+        }
+      }
+    }
+  }
+  try {
+    return JSON.parse(jsonrepair(cleaned));
+  } catch {
+  }
+  return null;
+}
+
+// src/ai/flows/auditCurriculumQualityFlow.ts
+var REPAIR_SUFFIX = "\n\nCRITICAL OUTPUT FORMAT FIX: your previous response could not be parsed as JSON. Output ONLY the raw JSON object \u2014 no markdown code fences, no prose before or after. Start with { and end with }.";
+function extractRawText(err) {
+  const e = err;
+  if (typeof e?.text === "string" && e.text) return e.text;
+  const cause = e?.cause;
+  if (typeof cause?.text === "string" && cause.text) return cause.text;
+  const msg = typeof e?.message === "string" ? e.message : typeof cause?.message === "string" ? cause.message : "";
+  const m = msg.match(/Text: ([\s\S]+)/);
+  return m ? m[1] : "";
+}
+async function generateJudgeObject(opts) {
+  let firstText = "";
+  try {
+    return (await generateObject({ model: opts.model, schema: opts.schema, prompt: opts.prompt })).object;
+  } catch (err) {
+    firstText = extractRawText(err);
+    if (firstText) {
+      const salvaged = safeParseJson(firstText);
+      if (salvaged) {
+        const check = opts.schema.safeParse(salvaged);
+        if (check.success) return check.data;
+      }
+    }
+    try {
+      return (await generateObject({ model: opts.model, schema: opts.schema, prompt: opts.prompt + REPAIR_SUFFIX })).object;
+    } catch (err2) {
+      const secondText = extractRawText(err2);
+      if (secondText && secondText !== firstText) {
+        const salvaged2 = safeParseJson(secondText);
+        if (salvaged2) {
+          const check2 = opts.schema.safeParse(salvaged2);
+          if (check2.success) return check2.data;
+        }
+      }
+      throw err2;
+    }
+  }
+}
 async function auditCurriculumQualityFlow(input) {
   const model = getAIModel(input.modelOptions);
   const prompt = buildJudgePrompt({
@@ -3598,7 +3745,7 @@ async function auditCurriculumQualityFlow(input) {
     generatedContentJson: input.generatedContentJson,
     standardStatements: input.standardStatements
   });
-  const { object } = await generateObject({
+  const object = await generateJudgeObject({
     model,
     schema: CurriculumQualityReportSchema,
     prompt: `${prompt}