@axlsdk/axl 0.13.1 → 0.13.3

package/dist/index.cjs

@@ -141,6 +141,7 @@ __export(index_exports, {
   agent: () => agent,
   createSpanManager: () => createSpanManager,
   defineConfig: () => defineConfig,
+  extractJson: () => extractJson,
   resolveThinkingOptions: () => resolveThinkingOptions,
   tool: () => tool,
   workflow: () => workflow,
@@ -2255,15 +2256,57 @@ function zodToJsonSchema(schema) {
 function estimateTokens(text) {
   return Math.ceil(text.length / 4);
 }
-function stripMarkdownFences(text) {
-  const trimmed = text.trim();
-  if (trimmed.startsWith("```")) {
-    const withoutOpening = trimmed.replace(/^```\w*\s*\n?/, "");
-    const withoutClosing = withoutOpening.replace(/\n?```\s*$/, "");
-    return withoutClosing.trim();
+function extractJson(content) {
+  const trimmed = content.trim();
+  if (trimmed.startsWith("{")) {
+    return extractBalanced(trimmed, 0, "{", "}") ?? trimmed;
+  }
+  if (trimmed.startsWith("[")) {
+    return extractBalanced(trimmed, 0, "[", "]") ?? trimmed;
+  }
+  const fenceMatch = trimmed.match(/```(?:json)?\s*\n?([\s\S]*?)\n?\s*```/);
+  if (fenceMatch) {
+    return fenceMatch[1].trim();
+  }
+  const open = trimmed.indexOf("{");
+  if (open >= 0) {
+    const extracted = extractBalanced(trimmed, open, "{", "}");
+    if (extracted) return extracted;
+  }
+  const openBracket = trimmed.indexOf("[");
+  if (openBracket >= 0) {
+    const extracted = extractBalanced(trimmed, openBracket, "[", "]");
+    if (extracted) return extracted;
   }
   return trimmed;
 }
+function extractBalanced(str, start, openChar, closeChar) {
+  let depth = 0;
+  let inString = false;
+  let escape = false;
+  for (let i = start; i < str.length; i++) {
+    const ch = str[i];
+    if (escape) {
+      escape = false;
+      continue;
+    }
+    if (ch === "\\" && inString) {
+      escape = true;
+      continue;
+    }
+    if (ch === '"') {
+      inString = !inString;
+      continue;
+    }
+    if (inString) continue;
+    if (ch === openChar) depth++;
+    else if (ch === closeChar) {
+      depth--;
+      if (depth === 0) return str.slice(start, i + 1);
+    }
+  }
+  return null;
+}
 function estimateMessagesTokens(messages) {
   let total = 0;
   for (const msg of messages) {
@@ -3044,7 +3087,7 @@ ${JSON.stringify(jsonSchema, null, 2)}`;
       let validated = void 0;
       if (options?.schema) {
         try {
-          const parsed = JSON.parse(stripMarkdownFences(content));
+          const parsed = JSON.parse(extractJson(content));
           validated = options.schema.parse(parsed);
         } catch (err) {
           const maxSchemaRetries = options.retries ?? 3;
@@ -6595,6 +6638,7 @@ function cosineSimilarity2(a, b) {
   agent,
   createSpanManager,
   defineConfig,
+  extractJson,
   resolveThinkingOptions,
   tool,
   workflow,