@superatomai/sdk-node 0.0.34 → 0.0.35

package/dist/index.mjs

@@ -4076,19 +4076,44 @@ var LLM = class {
     }
     const firstBrace = jsonText.indexOf("{");
     const firstBracket = jsonText.indexOf("[");
-    const lastBrace = jsonText.lastIndexOf("}");
-    const lastBracket = jsonText.lastIndexOf("]");
     let startIdx = -1;
-    let endIdx = -1;
+    let openChar = "";
+    let closeChar = "";
     if (firstBrace !== -1 && (firstBracket === -1 || firstBrace < firstBracket)) {
       startIdx = firstBrace;
-      endIdx = lastBrace;
+      openChar = "{";
+      closeChar = "}";
     } else if (firstBracket !== -1) {
       startIdx = firstBracket;
-      endIdx = lastBracket;
-    }
-    if (startIdx !== -1 && endIdx !== -1 && startIdx < endIdx) {
-      jsonText = jsonText.substring(startIdx, endIdx + 1);
+      openChar = "[";
+      closeChar = "]";
+    }
+    if (startIdx !== -1) {
+      let depth = 0;
+      let inString = false;
+      let endIdx = -1;
+      for (let i = startIdx; i < jsonText.length; i++) {
+        const char = jsonText[i];
+        const prevChar = i > 0 ? jsonText[i - 1] : "";
+        if (char === '"' && prevChar !== "\\") {
+          inString = !inString;
+          continue;
+        }
+        if (!inString) {
+          if (char === openChar) {
+            depth++;
+          } else if (char === closeChar) {
+            depth--;
+            if (depth === 0) {
+              endIdx = i;
+              break;
+            }
+          }
+        }
+      }
+      if (endIdx !== -1) {
+        jsonText = jsonText.substring(startIdx, endIdx + 1);
+      }
     }
     try {
       const repairedJson = jsonrepair(jsonText);