agentic-json-repair 1.0.0

package/.vscode/extensions.json

@@ -0,0 +1,3 @@
+{
+  "recommendations": ["henrikdev.ag-quota"]
+}

package/dist/index.d.mts

@@ -0,0 +1,13 @@
+import { z } from 'zod';
+
+type AgenticParseResult<T> = {
+    success: true;
+    data: T;
+} | {
+    success: false;
+    error: string;
+    retryPrompt: string;
+};
+declare function safeParseAI<T>(aiOutput: string, schema: z.ZodType<T>): AgenticParseResult<T>;
+
+export { type AgenticParseResult, safeParseAI };

package/dist/index.d.ts

@@ -0,0 +1,13 @@
+import { z } from 'zod';
+
+type AgenticParseResult<T> = {
+    success: true;
+    data: T;
+} | {
+    success: false;
+    error: string;
+    retryPrompt: string;
+};
+declare function safeParseAI<T>(aiOutput: string, schema: z.ZodType<T>): AgenticParseResult<T>;
+
+export { type AgenticParseResult, safeParseAI };

package/dist/index.js

@@ -0,0 +1,115 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var index_exports = {};
+__export(index_exports, {
+  safeParseAI: () => safeParseAI
+});
+module.exports = __toCommonJS(index_exports);
+var import_jsonrepair = require("jsonrepair");
+function extractJsonContent(aiOutput) {
+  const codeBlockMatch = aiOutput.match(/```(?:json)?\s*([\s\S]*?)```/);
+  if (codeBlockMatch) {
+    return codeBlockMatch[1].trim();
+  }
+  const firstBrace = aiOutput.indexOf("{");
+  const firstBracket = aiOutput.indexOf("[");
+  let startChar;
+  let endChar;
+  let startIndex;
+  if (firstBrace === -1 && firstBracket === -1) {
+    return aiOutput.trim();
+  } else if (firstBrace === -1) {
+    startChar = "[";
+    endChar = "]";
+    startIndex = firstBracket;
+  } else if (firstBracket === -1) {
+    startChar = "{";
+    endChar = "}";
+    startIndex = firstBrace;
+  } else {
+    if (firstBrace < firstBracket) {
+      startChar = "{";
+      endChar = "}";
+      startIndex = firstBrace;
+    } else {
+      startChar = "[";
+      endChar = "]";
+      startIndex = firstBracket;
+    }
+  }
+  let depth = 0;
+  let inString = false;
+  let escapeNext = false;
+  for (let i = startIndex; i < aiOutput.length; i++) {
+    const char = aiOutput[i];
+    if (escapeNext) {
+      escapeNext = false;
+      continue;
+    }
+    if (char === "\\" && inString) {
+      escapeNext = true;
+      continue;
+    }
+    if (char === '"') {
+      inString = !inString;
+      continue;
+    }
+    if (!inString) {
+      if (char === startChar) {
+        depth++;
+      } else if (char === endChar) {
+        depth--;
+        if (depth === 0) {
+          return aiOutput.slice(startIndex, i + 1);
+        }
+      }
+    }
+  }
+  return aiOutput.slice(startIndex).trim();
+}
+function safeParseAI(aiOutput, schema) {
+  try {
+    const cleanOutput = extractJsonContent(aiOutput);
+    const repairedJson = (0, import_jsonrepair.jsonrepair)(cleanOutput);
+    const parsedObject = JSON.parse(repairedJson);
+    const validation = schema.safeParse(parsedObject);
+    if (validation.success) {
+      return { success: true, data: validation.data };
+    } else {
+      const issues = validation.error.issues.map((i) => `${i.path.join(".")}: ${i.message}`).join(", ");
+      return {
+        success: false,
+        error: "Validation Failed",
+        retryPrompt: `The JSON was parsed but failed validation. Fix these issues: [${issues}] and return ONLY the JSON.`
+      };
+    }
+  } catch (error) {
+    return {
+      success: false,
+      error: "JSON Syntax Error",
+      retryPrompt: `The output was not valid JSON. Error: ${error.message}. Return ONLY valid JSON.`
+    };
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  safeParseAI
+});

package/dist/index.mjs

@@ -0,0 +1,90 @@
+// src/index.ts
+import { jsonrepair } from "jsonrepair";
+function extractJsonContent(aiOutput) {
+  const codeBlockMatch = aiOutput.match(/```(?:json)?\s*([\s\S]*?)```/);
+  if (codeBlockMatch) {
+    return codeBlockMatch[1].trim();
+  }
+  const firstBrace = aiOutput.indexOf("{");
+  const firstBracket = aiOutput.indexOf("[");
+  let startChar;
+  let endChar;
+  let startIndex;
+  if (firstBrace === -1 && firstBracket === -1) {
+    return aiOutput.trim();
+  } else if (firstBrace === -1) {
+    startChar = "[";
+    endChar = "]";
+    startIndex = firstBracket;
+  } else if (firstBracket === -1) {
+    startChar = "{";
+    endChar = "}";
+    startIndex = firstBrace;
+  } else {
+    if (firstBrace < firstBracket) {
+      startChar = "{";
+      endChar = "}";
+      startIndex = firstBrace;
+    } else {
+      startChar = "[";
+      endChar = "]";
+      startIndex = firstBracket;
+    }
+  }
+  let depth = 0;
+  let inString = false;
+  let escapeNext = false;
+  for (let i = startIndex; i < aiOutput.length; i++) {
+    const char = aiOutput[i];
+    if (escapeNext) {
+      escapeNext = false;
+      continue;
+    }
+    if (char === "\\" && inString) {
+      escapeNext = true;
+      continue;
+    }
+    if (char === '"') {
+      inString = !inString;
+      continue;
+    }
+    if (!inString) {
+      if (char === startChar) {
+        depth++;
+      } else if (char === endChar) {
+        depth--;
+        if (depth === 0) {
+          return aiOutput.slice(startIndex, i + 1);
+        }
+      }
+    }
+  }
+  return aiOutput.slice(startIndex).trim();
+}
+function safeParseAI(aiOutput, schema) {
+  try {
+    const cleanOutput = extractJsonContent(aiOutput);
+    const repairedJson = jsonrepair(cleanOutput);
+    const parsedObject = JSON.parse(repairedJson);
+    const validation = schema.safeParse(parsedObject);
+    if (validation.success) {
+      return { success: true, data: validation.data };
+    } else {
+      const issues = validation.error.issues.map((i) => `${i.path.join(".")}: ${i.message}`).join(", ");
+      return {
+        success: false,
+        error: "Validation Failed",
+        retryPrompt: `The JSON was parsed but failed validation. Fix these issues: [${issues}] and return ONLY the JSON.`
+      };
+    }
+  } catch (error) {
+    return {
+      success: false,
+      error: "JSON Syntax Error",
+      retryPrompt: `The output was not valid JSON. Error: ${error.message}. Return ONLY valid JSON.`
+    };
+  }
+}
+export {
+  safeParseAI
+};

package/package.json

@@ -0,0 +1,40 @@
+{
+  "name": "agentic-json-repair",
+  "version": "1.0.0",
+  "description": "Repair and validate AI-generated JSON with auto-generated retry prompts.",
+  "main": "./dist/index.js",
+  "module": "./dist/index.mjs",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.mjs",
+      "require": "./dist/index.js"
+    }
+  },
+  "scripts": {
+    "build": "tsup src/index.ts --format cjs,esm --dts --clean",
+    "test": "vitest run",
+    "prepublishOnly": "npm run build"
+  },
+  "keywords": [
+    "ai",
+    "llm",
+    "json",
+    "repair",
+    "zod",
+    "agent"
+  ],
+  "author": "Your Name",
+  "license": "MIT",
+  "dependencies": {
+    "jsonrepair": "^3.1.0",
+    "zod": "^3.22.0"
+  },
+  "devDependencies": {
+    "tsup": "^8.0.0",
+    "typescript": "^5.0.0",
+    "vitest": "^1.0.0",
+    "@types/node": "^20.0.0"
+  }
+}

package/src/index.test.ts

@@ -0,0 +1,35 @@
+import { expect, test } from 'vitest';
+import { safeParseAI } from './index';
+import { z } from 'zod';
+
+test('fixes broken AI output', () => {
+    const brokenAiOutput = `
+    Sure! Here is the user:
+    \`\`\`json
+    {
+      name: "Alice",
+      age: 30,
+    }
+    \`\`\`
+  `;
+
+    const UserSchema = z.object({ name: z.string(), age: z.number() });
+    const result = safeParseAI(brokenAiOutput, UserSchema);
+
+    expect(result.success).toBe(true);
+    if (result.success) {
+        expect(result.data.name).toBe("Alice");
+    }
+});
+
+test('generates retry prompt on wrong schema', () => {
+    const wrongTypeOutput = `{"age": "thirty"}`; // string instead of number
+    const Schema = z.object({ age: z.number() });
+
+    const result = safeParseAI(wrongTypeOutput, Schema);
+
+    expect(result.success).toBe(false);
+    if (!result.success) {
+        expect(result.retryPrompt).toContain("expected number, received string");
+    }
+});

package/src/index.ts

@@ -0,0 +1,128 @@
+import { jsonrepair } from 'jsonrepair';
+import { z } from 'zod';
+
+export type AgenticParseResult<T> =
+    | { success: true; data: T }
+    | { success: false; error: string; retryPrompt: string };
+
+/**
+ * Attempts to extract, repair, and validate JSON from an AI's text output.
+ * * @param aiOutput The raw string from the LLM (e.g., "Here is the data: ```json ...")
+ * @param schema The Zod schema to validate against
+ */
+/**
+ * Extracts JSON content from AI output that may contain markdown code blocks or prose.
+ */
+function extractJsonContent(aiOutput: string): string {
+    // First, try to extract from markdown code blocks
+    const codeBlockMatch = aiOutput.match(/```(?:json)?\s*([\s\S]*?)```/);
+    if (codeBlockMatch) {
+        return codeBlockMatch[1].trim();
+    }
+
+    // Otherwise, try to find the first JSON object or array
+    const firstBrace = aiOutput.indexOf('{');
+    const firstBracket = aiOutput.indexOf('[');
+
+    let startChar: '{' | '[';
+    let endChar: '}' | ']';
+    let startIndex: number;
+
+    if (firstBrace === -1 && firstBracket === -1) {
+        // No JSON structure found, return as-is for jsonrepair to handle
+        return aiOutput.trim();
+    } else if (firstBrace === -1) {
+        startChar = '[';
+        endChar = ']';
+        startIndex = firstBracket;
+    } else if (firstBracket === -1) {
+        startChar = '{';
+        endChar = '}';
+        startIndex = firstBrace;
+    } else {
+        // Both found, use whichever comes first
+        if (firstBrace < firstBracket) {
+            startChar = '{';
+            endChar = '}';
+            startIndex = firstBrace;
+        } else {
+            startChar = '[';
+            endChar = ']';
+            startIndex = firstBracket;
+        }
+    }
+
+    // Find the matching closing bracket/brace
+    let depth = 0;
+    let inString = false;
+    let escapeNext = false;
+
+    for (let i = startIndex; i < aiOutput.length; i++) {
+        const char = aiOutput[i];
+
+        if (escapeNext) {
+            escapeNext = false;
+            continue;
+        }
+
+        if (char === '\\' && inString) {
+            escapeNext = true;
+            continue;
+        }
+
+        if (char === '"') {
+            inString = !inString;
+            continue;
+        }
+
+        if (!inString) {
+            if (char === startChar) {
+                depth++;
+            } else if (char === endChar) {
+                depth--;
+                if (depth === 0) {
+                    return aiOutput.slice(startIndex, i + 1);
+                }
+            }
+        }
+    }
+
+    // If we couldn't find matching brackets, return from startIndex to end
+    return aiOutput.slice(startIndex).trim();
+}
+
+export function safeParseAI<T>(aiOutput: string, schema: z.ZodType<T>): AgenticParseResult<T> {
+    try {
+        // 1. Extract JSON: Find the JSON content within the AI output
+        const cleanOutput = extractJsonContent(aiOutput);
+
+        // 2. Repair JSON: Fixes missing quotes, trailing commas, etc.
+        const repairedJson = jsonrepair(cleanOutput);
+
+        // 3. Parse JSON
+        const parsedObject = JSON.parse(repairedJson);
+
+        // 4. Validate Schema
+        const validation = schema.safeParse(parsedObject);
+
+        if (validation.success) {
+            return { success: true, data: validation.data };
+        } else {
+            // 5. Generate a "Self-Correction" prompt for the AI
+            const issues = validation.error.issues.map(i => `${i.path.join('.')}: ${i.message}`).join(', ');
+            return {
+                success: false,
+                error: "Validation Failed",
+                retryPrompt: `The JSON was parsed but failed validation. Fix these issues: [${issues}] and return ONLY the JSON.`
+            };
+        }
+
+    } catch (error) {
+        // 6. Handle Syntax Errors that jsonrepair couldn't fix
+        return {
+            success: false,
+            error: "JSON Syntax Error",
+            retryPrompt: `The output was not valid JSON. Error: ${(error as Error).message}. Return ONLY valid JSON.`
+        };
+    }
+}

package/tsconfig.json

@@ -0,0 +1,14 @@
+{
+  "compilerOptions": {
+    "target": "ES2020",
+    "module": "ESNext",
+    "moduleResolution": "bundler",
+    "esModuleInterop": true,
+    "strict": true,
+    "declaration": true,
+    "outDir": "dist",
+    "skipLibCheck": true
+  },
+  "include": ["src"],
+  "exclude": ["node_modules", "dist"]
+}