docground 0.1.0

package/dist/tests/compiler.test.js

@@ -0,0 +1,110 @@
+import { describe, it, expect } from "vitest";
+import { z } from "zod";
+import { compileZodToGeminiSchema } from "../src/compiler/zod-compiler.js";
+describe("compileZodToGeminiSchema", () => {
+    it("compiles a primitive schema", () => {
+        const schema = z.object({
+            name: z.string(),
+            age: z.number(),
+            active: z.boolean(),
+        });
+        const geminiSchema = compileZodToGeminiSchema(schema);
+        expect(geminiSchema.type).toBe("object");
+        expect(geminiSchema.properties).toMatchObject({
+            name: { type: "string" },
+            age: { type: "number" },
+            active: { type: "boolean" },
+        });
+        expect(geminiSchema.required).toContain("name");
+        expect(geminiSchema.required).toContain("age");
+        expect(geminiSchema.required).toContain("active");
+    });
+    it("compiles nested objects", () => {
+        const schema = z.object({
+            user: z.object({
+                id: z.number(),
+                email: z.string(),
+            }),
+        });
+        const geminiSchema = compileZodToGeminiSchema(schema);
+        expect(geminiSchema.type).toBe("object");
+        expect(geminiSchema.properties).toHaveProperty("user");
+        expect(geminiSchema.properties).toMatchObject({
+            user: {
+                type: "object",
+                properties: {
+                    id: { type: "number" },
+                    email: { type: "string" },
+                },
+                required: ["id", "email"],
+            },
+        });
+    });
+    it("compiles arrays", () => {
+        const schema = z.object({
+            tags: z.array(z.string()),
+        });
+        const geminiSchema = compileZodToGeminiSchema(schema);
+        expect(geminiSchema.properties).toMatchObject({
+            tags: {
+                type: "array",
+                items: { type: "string" },
+            },
+        });
+    });
+    it("compiles optional fields", () => {
+        const schema = z.object({
+            title: z.string(),
+            subtitle: z.string().optional(),
+        });
+        const geminiSchema = compileZodToGeminiSchema(schema);
+        expect(geminiSchema.properties).toHaveProperty("subtitle");
+        expect(geminiSchema.required).toContain("title");
+        expect(geminiSchema.required).not.toContain("subtitle");
+    });
+    it("compiles enums", () => {
+        const schema = z.object({
+            status: z.enum(["pending", "paid", "overdue"]),
+        });
+        const geminiSchema = compileZodToGeminiSchema(schema);
+        expect(geminiSchema.properties).toMatchObject({
+            status: {
+                type: "string",
+                enum: ["pending", "paid", "overdue"],
+            },
+        });
+    });
+    it("compiles complex nested schemas with arrays of objects", () => {
+        const schema = z.object({
+            invoice: z.object({
+                lines: z.array(z.object({
+                    description: z.string(),
+                    amount: z.number(),
+                })),
+                total: z.number(),
+            }),
+        });
+        const geminiSchema = compileZodToGeminiSchema(schema);
+        expect(geminiSchema.properties).toMatchObject({
+            invoice: {
+                type: "object",
+                properties: {
+                    lines: {
+                        type: "array",
+                        items: {
+                            type: "object",
+                            properties: {
+                                description: { type: "string" },
+                                amount: { type: "number" },
+                            },
+                            required: ["description", "amount"],
+                        },
+                    },
+                    total: { type: "number" },
+                },
+                required: ["lines", "total"],
+            },
+        });
+    });
+});
+//# sourceMappingURL=compiler.test.js.map

package/dist/tests/compiler.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"compiler.test.js","sourceRoot":"","sources":["../../tests/compiler.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,wBAAwB,EAAE,MAAM,iCAAiC,CAAC;AAE3E,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC;YACtB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;YAChB,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE;YACf,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE;SACpB,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,wBAAwB,CAAC,MAAM,CAAC,CAAC;QAEtD,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,aAAa,CAAC;YAC5C,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;YACxB,GAAG,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;YACvB,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;SAC5B,CAAC,CAAC;QACH,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAChD,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC/C,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACjC,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC;YACtB,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;gBACb,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;gBACd,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;aAClB,CAAC;SACH,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,wBAAwB,CAAC,MAAM,CAAC,CAAC;QAEtD,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QACvD,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,aAAa,CAAC;YAC5C,IAAI,EAAE;gBACJ,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACV,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;oBACtB,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;iBAC1B;gBACD,QAAQ,EAAE,CAAC,IAAI,EAAE,OAAO,CAAC;aAC1B;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iBAAiB,EAAE,GAAG,EAAE;QACzB,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC;YACtB,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;SAC1B,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,wBAAwB,CAAC,MAAM,CAAC,CAAC;QAEtD,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,aAAa,CAAC;YAC5C,IAAI,EAAE;gBACJ,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;aAC1B;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;QAClC,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC;YACtB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;YACjB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;SAChC,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,wBAAwB,CAAC,MAAM,CAAC,CAAC;QAEtD,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;QAC3D,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACjD,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gBAAgB,EAAE,GAAG,EAAE;QACxB,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC;YACtB,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;SAC/C,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,wBAAwB,CAAC,MAAM,CAAC,CAAC;QAEtD,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,aAAa,CAAC;YAC5C,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,CAAC;aACrC;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC;YACtB,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC;gBAChB,KAAK,EAAE,CAAC,CAAC,KAAK,CACZ,CAAC,CAAC,MAAM,CAAC;oBACP,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;oBACvB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;iBACnB,CAAC,CACH;gBACD,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;aAClB,CAAC;SACH,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,wBAAwB,CAAC,MAAM,CAAC,CAAC;QAEtD,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,aAAa,CAAC;YAC5C,OAAO,EAAE;gBACP,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACV,KAAK,EAAE;wBACL,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE;4BACL,IAAI,EAAE,QAAQ;4BACd,UAAU,EAAE;gCACV,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gCAC/B,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;6BAC3B;4BACD,QAAQ,EAAE,CAAC,aAAa,EAAE,QAAQ,CAAC;yBACpC;qBACF;oBACD,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;iBAC1B;gBACD,QAAQ,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC;aAC7B;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/tests/extractor.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=extractor.test.d.ts.map

package/dist/tests/extractor.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"extractor.test.d.ts","sourceRoot":"","sources":["../../tests/extractor.test.ts"],"names":[],"mappings":""}

package/dist/tests/extractor.test.js

@@ -0,0 +1,195 @@
+import { describe, it, expect, vi, beforeAll } from "vitest";
+import { z } from "zod";
+import { DocumentExtractor } from "../src/core/extractor.js";
+import { createBlankImage } from "./fixtures.js";
+const Invoice = z.object({
+    invoiceNumber: z.string(),
+    subtotal: z.number(),
+    tax: z.number(),
+    total: z.number(),
+    items: z.array(z.object({
+        description: z.string(),
+        amount: z.number(),
+    })),
+});
+describe("DocumentExtractor", () => {
+    let image;
+    beforeAll(async () => {
+        image = await createBlankImage(800, 600);
+    });
+    it("extracts typed data and groundings using a custom VLM", async () => {
+        const mockProvider = {
+            name: "mock",
+            extract: vi.fn().mockResolvedValue({
+                text: JSON.stringify({
+                    data: {
+                        invoiceNumber: "INV-001",
+                        subtotal: 100,
+                        tax: 21,
+                        total: 121,
+                        items: [{ description: "A", amount: 100 }],
+                    },
+                    groundings: [
+                        {
+                            pageIndex: 0,
+                            field: "invoiceNumber",
+                            value: "INV-001",
+                            bbox: [0.1, 0.1, 0.2, 0.4],
+                            confidence: 0.98,
+                        },
+                    ],
+                }),
+                raw: null,
+            }),
+        };
+        const extractor = new DocumentExtractor({ customVLM: mockProvider });
+        const result = await extractor.extract(image.buffer, Invoice, { tiling: "off" });
+        expect(result.data.invoiceNumber).toBe("INV-001");
+        expect(result.data.total).toBe(121);
+        expect(result.groundings).toHaveLength(1);
+        expect(result.fields.invoiceNumber.bbox).toEqual([0.1, 0.1, 0.2, 0.4]);
+        expect(result.confidence).toBeGreaterThan(0);
+        expect(result.provenance.provider).toBe("mock");
+        expect(result.provenance.tilesUsed).toBe(1);
+    });
+    it("runs invariants and returns a report", async () => {
+        const mockProvider = {
+            name: "mock",
+            extract: vi.fn().mockResolvedValue({
+                text: JSON.stringify({
+                    data: {
+                        invoiceNumber: "INV-002",
+                        subtotal: 100,
+                        tax: 21,
+                        total: 121,
+                        items: [{ description: "A", amount: 100 }],
+                    },
+                }),
+                raw: null,
+            }),
+        };
+        const extractor = new DocumentExtractor({ customVLM: mockProvider });
+        const result = await extractor.extract(image.buffer, Invoice, {
+            tiling: "off",
+            prove: true,
+            invariants: [
+                { id: "balance", expression: "subtotal + tax === total" },
+                { id: "items-total", expression: "sum(items.amount) === subtotal" },
+            ],
+        });
+        expect(result.invariants?.ok).toBe(true);
+        expect(result.invariants?.checks).toHaveLength(2);
+    });
+    it("auto-repairs when invariants fail", async () => {
+        const responses = [
+            {
+                text: JSON.stringify({
+                    data: {
+                        invoiceNumber: "INV-003",
+                        subtotal: 100,
+                        tax: 21,
+                        total: 120, // wrong
+                        items: [{ description: "A", amount: 100 }],
+                    },
+                }),
+                raw: null,
+            },
+            {
+                text: JSON.stringify({
+                    data: {
+                        invoiceNumber: "INV-003",
+                        subtotal: 100,
+                        tax: 21,
+                        total: 121, // corrected
+                        items: [{ description: "A", amount: 100 }],
+                    },
+                }),
+                raw: null,
+            },
+        ];
+        const mockProvider = {
+            name: "mock",
+            extract: vi
+                .fn()
+                .mockImplementation(() => Promise.resolve(responses.shift())),
+        };
+        const extractor = new DocumentExtractor({ customVLM: mockProvider });
+        const result = await extractor.extract(image.buffer, Invoice, {
+            tiling: "off",
+            prove: true,
+            autoRepair: true,
+            invariants: [{ id: "balance", expression: "subtotal + tax === total" }],
+        });
+        expect(mockProvider.extract).toHaveBeenCalledTimes(2);
+        expect(result.data.total).toBe(121);
+        expect(result.invariants?.ok).toBe(true);
+        expect(result.provenance.repairs).toBe(1);
+    });
+    it("stitches multi-page tables returned by the VLM", async () => {
+        const mockProvider = {
+            name: "mock",
+            extract: vi.fn().mockResolvedValue({
+                text: JSON.stringify({
+                    data: {
+                        invoiceNumber: "INV-004",
+                        subtotal: 200,
+                        tax: 42,
+                        total: 242,
+                        items: [{ description: "A", amount: 200 }],
+                    },
+                    tables: [
+                        {
+                            pageIndex: 1,
+                            header: ["Item", "Qty"],
+                            rows: [["A", 1]],
+                            bbox: [0, 0, 0.4, 0.9],
+                        },
+                        {
+                            pageIndex: 2,
+                            header: ["Item", "Qty"],
+                            rows: [
+                                ["Item", "Qty"],
+                                ["B", 2],
+                            ],
+                            bbox: [0, 0, 0.4, 0.9],
+                        },
+                    ],
+                }),
+                raw: null,
+            }),
+        };
+        const extractor = new DocumentExtractor({ customVLM: mockProvider });
+        const result = await extractor.extract(image.buffer, Invoice, { tiling: "off" });
+        expect(result.tables).toHaveLength(1);
+        expect(result.tables[0].rows).toEqual([
+            ["A", 1],
+            ["B", 2],
+        ]);
+    });
+    it("supports tiling when the image exceeds tile limits", async () => {
+        const largeImage = await createBlankImage(2000, 2000);
+        const mockProvider = {
+            name: "mock",
+            extract: vi.fn().mockResolvedValue({
+                text: JSON.stringify({
+                    data: {
+                        invoiceNumber: "INV-005",
+                        subtotal: 50,
+                        tax: 10.5,
+                        total: 60.5,
+                        items: [{ description: "A", amount: 50 }],
+                    },
+                }),
+                raw: null,
+            }),
+        };
+        const extractor = new DocumentExtractor({ customVLM: mockProvider });
+        const result = await extractor.extract(largeImage.buffer, Invoice, {
+            tiling: { maxTileWidth: 1024, maxTileHeight: 1024, overlap: 0.2 },
+        });
+        expect(result.provenance.tilesUsed).toBeGreaterThan(1);
+        const call = mockProvider.extract.mock.calls[0];
+        expect(call[0].images.length).toBeGreaterThan(1);
+    });
+});
+//# sourceMappingURL=extractor.test.js.map

package/dist/tests/extractor.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"extractor.test.js","sourceRoot":"","sources":["../../tests/extractor.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AAC7D,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAE7D,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AAEjD,MAAM,OAAO,GAAG,CAAC,CAAC,MAAM,CAAC;IACvB,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE;IACzB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;IACpB,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE;IACf,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;IACjB,KAAK,EAAE,CAAC,CAAC,KAAK,CACZ,CAAC,CAAC,MAAM,CAAC;QACP,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;QACvB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;KACnB,CAAC,CACH;CACF,CAAC,CAAC;AAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,IAAI,KAAmD,CAAC;IAExD,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,KAAK,GAAG,MAAM,gBAAgB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,YAAY,GAAgB;YAChC,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBACjC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,IAAI,EAAE;wBACJ,aAAa,EAAE,SAAS;wBACxB,QAAQ,EAAE,GAAG;wBACb,GAAG,EAAE,EAAE;wBACP,KAAK,EAAE,GAAG;wBACV,KAAK,EAAE,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC;qBAC3C;oBACD,UAAU,EAAE;wBACV;4BACE,SAAS,EAAE,CAAC;4BACZ,KAAK,EAAE,eAAe;4BACtB,KAAK,EAAE,SAAS;4BAChB,IAAI,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;4BAC1B,UAAU,EAAE,IAAI;yBACjB;qBACF;iBACF,CAAC;gBACF,GAAG,EAAE,IAAI;aACQ,CAAC;SACrB,CAAC;QAEF,MAAM,SAAS,GAAG,IAAI,iBAAiB,CAAC,EAAE,SAAS,EAAE,YAAY,EAAE,CAAC,CAAC;QACrE,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;QAEjF,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAClD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC7C,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAChD,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,YAAY,GAAgB;YAChC,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBACjC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,IAAI,EAAE;wBACJ,aAAa,EAAE,SAAS;wBACxB,QAAQ,EAAE,GAAG;wBACb,GAAG,EAAE,EAAE;wBACP,KAAK,EAAE,GAAG;wBACV,KAAK,EAAE,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC;qBAC3C;iBACF,CAAC;gBACF,GAAG,EAAE,IAAI;aACQ,CAAC;SACrB,CAAC;QAEF,MAAM,SAAS,GAAG,IAAI,iBAAiB,CAAC,EAAE,SAAS,EAAE,YAAY,EAAE,CAAC,CAAC;QACrE,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,OAAO,EAAE;YAC5D,MAAM,EAAE,KAAK;YACb,KAAK,EAAE,IAAI;YACX,UAAU,EAAE;gBACV,EAAE,EAAE,EAAE,SAAS,EAAE,UAAU,EAAE,0BAA0B,EAAE;gBACzD,EAAE,EAAE,EAAE,aAAa,EAAE,UAAU,EAAE,gCAAgC,EAAE;aACpE;SACF,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,SAAS,GAAqB;YAClC;gBACE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,IAAI,EAAE;wBACJ,aAAa,EAAE,SAAS;wBACxB,QAAQ,EAAE,GAAG;wBACb,GAAG,EAAE,EAAE;wBACP,KAAK,EAAE,GAAG,EAAE,QAAQ;wBACpB,KAAK,EAAE,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC;qBAC3C;iBACF,CAAC;gBACF,GAAG,EAAE,IAAI;aACV;YACD;gBACE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,IAAI,EAAE;wBACJ,aAAa,EAAE,SAAS;wBACxB,QAAQ,EAAE,GAAG;wBACb,GAAG,EAAE,EAAE;wBACP,KAAK,EAAE,GAAG,EAAE,YAAY;wBACxB,KAAK,EAAE,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC;qBAC3C;iBACF,CAAC;gBACF,GAAG,EAAE,IAAI;aACV;SACF,CAAC;QAEF,MAAM,YAAY,GAAgB;YAChC,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,EAAE;iBACR,EAAE,EAAE;iBACJ,kBAAkB,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,KAAK,EAAoB,CAAC,CAAC;SAClF,CAAC;QAEF,MAAM,SAAS,GAAG,IAAI,iBAAiB,CAAC,EAAE,SAAS,EAAE,YAAY,EAAE,CAAC,CAAC;QACrE,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,OAAO,EAAE;YAC5D,MAAM,EAAE,KAAK;YACb,KAAK,EAAE,IAAI;YACX,UAAU,EAAE,IAAI;YAChB,UAAU,EAAE,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,UAAU,EAAE,0BAA0B,EAAE,CAAC;SACxE,CAAC,CAAC;QAEH,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,YAAY,GAAgB;YAChC,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBACjC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,IAAI,EAAE;wBACJ,aAAa,EAAE,SAAS;wBACxB,QAAQ,EAAE,GAAG;wBACb,GAAG,EAAE,EAAE;wBACP,KAAK,EAAE,GAAG;wBACV,KAAK,EAAE,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC;qBAC3C;oBACD,MAAM,EAAE;wBACN;4BACE,SAAS,EAAE,CAAC;4BACZ,MAAM,EAAE,CAAC,MAAM,EAAE,KAAK,CAAC;4BACvB,IAAI,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;4BAChB,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC;yBACvB;wBACD;4BACE,SAAS,EAAE,CAAC;4BACZ,MAAM,EAAE,CAAC,MAAM,EAAE,KAAK,CAAC;4BACvB,IAAI,EAAE;gCACJ,CAAC,MAAM,EAAE,KAAK,CAAC;gCACf,CAAC,GAAG,EAAE,CAAC,CAAC;6BACT;4BACD,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC;yBACvB;qBACF;iBACF,CAAC;gBACF,GAAG,EAAE,IAAI;aACQ,CAAC;SACrB,CAAC;QAEF,MAAM,SAAS,GAAG,IAAI,iBAAiB,CAAC,EAAE,SAAS,EAAE,YAAY,EAAE,CAAC,CAAC;QACrE,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;QAEjF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC;YACpC,CAAC,GAAG,EAAE,CAAC,CAAC;YACR,CAAC,GAAG,EAAE,CAAC,CAAC;SACT,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,UAAU,GAAG,MAAM,gBAAgB,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QACtD,MAAM,YAAY,GAAgB;YAChC,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBACjC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,IAAI,EAAE;wBACJ,aAAa,EAAE,SAAS;wBACxB,QAAQ,EAAE,EAAE;wBACZ,GAAG,EAAE,IAAI;wBACT,KAAK,EAAE,IAAI;wBACX,KAAK,EAAE,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;qBAC1C;iBACF,CAAC;gBACF,GAAG,EAAE,IAAI;aACQ,CAAC;SACrB,CAAC;QAEF,MAAM,SAAS,GAAG,IAAI,iBAAiB,CAAC,EAAE,SAAS,EAAE,YAAY,EAAE,CAAC,CAAC;QACrE,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,EAAE,OAAO,EAAE;YACjE,MAAM,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE;SAClE,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACvD,MAAM,IAAI,GAAI,YAAY,CAAC,OAAoC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAuB,CAAC;QACpG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/tests/fixtures.d.ts

@@ -0,0 +1,9 @@
+export interface TestImage {
+    mimeType: string;
+    base64: string;
+    width: number;
+    height: number;
+    buffer: Buffer;
+}
+export declare function createBlankImage(width: number, height: number, mimeType?: "image/png" | "image/jpeg"): Promise<TestImage>;
+//# sourceMappingURL=fixtures.d.ts.map

package/dist/tests/fixtures.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"fixtures.d.ts","sourceRoot":"","sources":["../../tests/fixtures.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,SAAS;IACxB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;CAChB;AAED,wBAAsB,gBAAgB,CACpC,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,MAAM,EACd,QAAQ,GAAE,WAAW,GAAG,YAA0B,GACjD,OAAO,CAAC,SAAS,CAAC,CAiBpB"}

package/dist/tests/fixtures.js

@@ -0,0 +1,20 @@
+import sharp from "sharp";
+export async function createBlankImage(width, height, mimeType = "image/png") {
+    const pipeline = sharp({
+        create: {
+            width,
+            height,
+            channels: 3,
+            background: "white",
+        },
+    });
+    const buf = mimeType === "image/jpeg" ? await pipeline.jpeg().toBuffer() : await pipeline.png().toBuffer();
+    return {
+        mimeType,
+        base64: buf.toString("base64"),
+        width,
+        height,
+        buffer: buf,
+    };
+}
+//# sourceMappingURL=fixtures.js.map

package/dist/tests/fixtures.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"fixtures.js","sourceRoot":"","sources":["../../tests/fixtures.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAU1B,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,KAAa,EACb,MAAc,EACd,WAAuC,WAAW;IAElD,MAAM,QAAQ,GAAG,KAAK,CAAC;QACrB,MAAM,EAAE;YACN,KAAK;YACL,MAAM;YACN,QAAQ,EAAE,CAAC;YACX,UAAU,EAAE,OAAO;SACpB;KACF,CAAC,CAAC;IACH,MAAM,GAAG,GAAG,QAAQ,KAAK,YAAY,CAAC,CAAC,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,MAAM,QAAQ,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC;IAC3G,OAAO;QACL,QAAQ;QACR,MAAM,EAAE,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAC9B,KAAK;QACL,MAAM;QACN,MAAM,EAAE,GAAG;KACZ,CAAC;AACJ,CAAC"}

package/dist/tests/grounding.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=grounding.test.d.ts.map

package/dist/tests/grounding.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"grounding.test.d.ts","sourceRoot":"","sources":["../../tests/grounding.test.ts"],"names":[],"mappings":""}

package/dist/tests/grounding.test.js

@@ -0,0 +1,82 @@
+import { describe, it, expect } from "vitest";
+import { normalizeBBox, denormalizeBBox, iou, mergeBoxes, } from "../src/grounding/bbox.js";
+describe("bbox grounding", () => {
+    const pageSize = { width: 800, height: 1200 };
+    describe("normalizeBBox / denormalizeBBox", () => {
+        it("normalizes absolute pixel coordinates to [ymin, xmin, ymax, xmax]", () => {
+            const box = { x1: 200, y1: 300, x2: 400, y2: 600 };
+            const normalized = normalizeBBox(box, pageSize);
+            expect(normalized).toEqual([
+                300 / 1200,
+                200 / 800,
+                600 / 1200,
+                400 / 800,
+            ]);
+        });
+        it("round-trips through denormalize", () => {
+            const box = { x1: 100, y1: 150, x2: 300, y2: 450 };
+            const normalized = normalizeBBox(box, pageSize);
+            const denormalized = denormalizeBBox(normalized, pageSize);
+            expect(denormalized.x1).toBeCloseTo(box.x1);
+            expect(denormalized.y1).toBeCloseTo(box.y1);
+            expect(denormalized.x2).toBeCloseTo(box.x2);
+            expect(denormalized.y2).toBeCloseTo(box.y2);
+        });
+        it("normalizes inverted absolute boxes", () => {
+            const box = { x1: 400, y1: 600, x2: 200, y2: 300 };
+            const normalized = normalizeBBox(box, pageSize);
+            expect(normalized).toEqual([
+                300 / 1200,
+                200 / 800,
+                600 / 1200,
+                400 / 800,
+            ]);
+        });
+    });
+    describe("iou", () => {
+        it("returns 1.0 for identical boxes", () => {
+            const a = [0.1, 0.1, 0.5, 0.5];
+            const b = [0.1, 0.1, 0.5, 0.5];
+            expect(iou(a, b)).toBe(1);
+        });
+        it("returns 0.0 for non-overlapping boxes", () => {
+            const a = [0, 0, 0.2, 0.2];
+            const b = [0.3, 0.3, 0.5, 0.5];
+            expect(iou(a, b)).toBe(0);
+        });
+        it("returns the correct overlap ratio for partial overlap", () => {
+            const a = [0, 0, 0.5, 0.5];
+            const b = [0.25, 0.25, 0.75, 0.75];
+            // Intersection area: 0.25 * 0.25 = 0.0625
+            // Union area: 0.25 + 0.25 - 0.0625 = 0.4375
+            // IoU: 0.0625 / 0.4375 ≈ 0.1428
+            expect(iou(a, b)).toBeCloseTo(0.0625 / 0.4375, 4);
+        });
+    });
+    describe("mergeBoxes", () => {
+        it("merges two overlapping boxes", () => {
+            // Box b is almost fully inside a, producing a high IoU (>0.5).
+            const a = [0, 0, 0.5, 0.5];
+            const b = [0.1, 0.1, 0.5, 0.5];
+            const merged = mergeBoxes([a, b], 0.5);
+            expect(merged).toHaveLength(1);
+            expect(merged[0]).toEqual([0, 0, 0.5, 0.5]);
+        });
+        it("does not merge boxes below the IoU threshold", () => {
+            const a = [0, 0, 0.1, 0.1];
+            const b = [0.2, 0.2, 0.3, 0.3];
+            const merged = mergeBoxes([a, b], 0.5);
+            expect(merged).toHaveLength(2);
+        });
+        it("chains merges across multiple boxes", () => {
+            // Three boxes with high mutual overlap (>0.5 IoU) that should collapse into one.
+            const a = [0, 0, 0.5, 0.5];
+            const b = [0, 0, 0.45, 0.45];
+            const c = [0.05, 0.05, 0.5, 0.5];
+            const merged = mergeBoxes([a, b, c], 0.5);
+            expect(merged).toHaveLength(1);
+            expect(merged[0]).toEqual([0, 0, 0.5, 0.5]);
+        });
+    });
+});
+//# sourceMappingURL=grounding.test.js.map

package/dist/tests/grounding.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"grounding.test.js","sourceRoot":"","sources":["../../tests/grounding.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EACL,aAAa,EACb,eAAe,EACf,GAAG,EACH,UAAU,GAIX,MAAM,0BAA0B,CAAC;AAElC,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,MAAM,QAAQ,GAAa,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IAExD,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;QAC/C,EAAE,CAAC,mEAAmE,EAAE,GAAG,EAAE;YAC3E,MAAM,GAAG,GAAa,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC;YAC7D,MAAM,UAAU,GAAG,aAAa,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;YAEhD,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC;gBACzB,GAAG,GAAG,IAAI;gBACV,GAAG,GAAG,GAAG;gBACT,GAAG,GAAG,IAAI;gBACV,GAAG,GAAG,GAAG;aACV,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,GAAG,GAAa,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC;YAC7D,MAAM,UAAU,GAAG,aAAa,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;YAChD,MAAM,YAAY,GAAG,eAAe,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;YAE3D,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC5C,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC5C,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC5C,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,GAAG,GAAa,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC;YAC7D,MAAM,UAAU,GAAG,aAAa,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;YAEhD,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC;gBACzB,GAAG,GAAG,IAAI;gBACV,GAAG,GAAG,GAAG;gBACT,GAAG,GAAG,IAAI;gBACV,GAAG,GAAG,GAAG;aACV,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,KAAK,EAAE,GAAG,EAAE;QACnB,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,CAAC,GAAmB,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YAC/C,MAAM,CAAC,GAAmB,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YAC/C,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,CAAC,GAAmB,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YAC3C,MAAM,CAAC,GAAmB,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YAC/C,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;YAC/D,MAAM,CAAC,GAAmB,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YAC3C,MAAM,CAAC,GAAmB,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;YACnD,0CAA0C;YAC1C,4CAA4C;YAC5C,gCAAgC;YAChC,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,MAAM,GAAG,MAAM,EAAE,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,+DAA+D;YAC/D,MAAM,CAAC,GAAmB,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YAC3C,MAAM,CAAC,GAAmB,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YAC/C,MAAM,MAAM,GAAG,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YAEvC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,CAAC,GAAmB,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YAC3C,MAAM,CAAC,GAAmB,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YAC/C,MAAM,MAAM,GAAG,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YAEvC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,iFAAiF;YACjF,MAAM,CAAC,GAAmB,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YAC3C,MAAM,CAAC,GAAmB,CAAC,CAAC,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;YAC7C,MAAM,CAAC,GAAmB,CAAC,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YAEjD,MAAM,MAAM,GAAG,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YAE1C,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/tests/multi-domain.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=multi-domain.test.d.ts.map

package/dist/tests/multi-domain.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"multi-domain.test.d.ts","sourceRoot":"","sources":["../../tests/multi-domain.test.ts"],"names":[],"mappings":""}

package/dist/tests/multi-domain.test.js

@@ -0,0 +1,135 @@
+import { describe, it, expect, vi, beforeAll } from "vitest";
+import { z } from "zod";
+import { DocumentExtractor } from "../src/core/extractor.js";
+import { compileZodToJsonSchema } from "../src/compiler/zod-compiler.js";
+import { createBlankImage } from "./fixtures.js";
+const LegalContractSchema = z.object({
+    contractNumber: z.string(),
+    parties: z.object({
+        client: z.string(),
+        contractor: z.string(),
+    }),
+    effectiveDate: z.string(),
+    termYears: z.number(),
+    clauseCount: z.number(),
+    penaltyAmount: z.number(),
+});
+const EngineeringSpecSchema = z.object({
+    project: z.string(),
+    revision: z.string(),
+    material: z.string(),
+    dimensions: z.object({
+        length: z.number(),
+        width: z.number(),
+        height: z.number(),
+    }),
+    tolerances: z.object({
+        linear: z.number(),
+        angular: z.number(),
+    }),
+    testResults: z.array(z.object({
+        test: z.string(),
+        passed: z.boolean(),
+        value: z.number(),
+    })),
+});
+const ClinicalReportSchema = z.object({
+    patient: z.object({
+        name: z.string(),
+        id: z.string(),
+        dob: z.string(),
+    }),
+    reportDate: z.string(),
+    biomarkers: z.array(z.object({
+        marker: z.string(),
+        value: z.number(),
+        unit: z.string(),
+        referenceRange: z.string(),
+    })),
+});
+function mockProvider(responseData) {
+    return {
+        name: "mock",
+        extract: vi.fn().mockResolvedValue({
+            text: JSON.stringify({ data: responseData, groundings: [] }),
+            raw: null,
+        }),
+    };
+}
+describe("Multi-Domain Extraction", () => {
+    let image;
+    beforeAll(async () => {
+        image = await createBlankImage(800, 600);
+    });
+    it("compiles heterogeneous Zod schemas to valid JSON Schema", () => {
+        expect(compileZodToJsonSchema(LegalContractSchema)).toMatchObject({ type: "object" });
+        expect(compileZodToJsonSchema(EngineeringSpecSchema)).toMatchObject({ type: "object" });
+        expect(compileZodToJsonSchema(ClinicalReportSchema)).toMatchObject({ type: "object" });
+    });
+    it("extracts a Legal Contract", async () => {
+        const data = {
+            contractNumber: "CNT-2026-001",
+            parties: { client: "Acme Corp", contractor: "BuildRight LLC" },
+            effectiveDate: "2026-01-15",
+            termYears: 3,
+            clauseCount: 12,
+            penaltyAmount: 50000,
+        };
+        const extractor = new DocumentExtractor({ customVLM: mockProvider(data) });
+        const result = await extractor.extract(image.buffer, LegalContractSchema, { tiling: "off" });
+        expect(result.data).toEqual(data);
+        expect(result.data.parties.client).toBe("Acme Corp");
+        expect(result.confidence).toBeGreaterThan(0);
+    });
+    it("extracts an Engineering Spec", async () => {
+        const data = {
+            project: "Brigde-Alpha",
+            revision: "Rev-C",
+            material: "Steel S355",
+            dimensions: { length: 1200, width: 400, height: 80 },
+            tolerances: { linear: 0.5, angular: 0.25 },
+            testResults: [
+                { test: "Tensile", passed: true, value: 520 },
+                { test: "Compression", passed: true, value: 410 },
+            ],
+        };
+        const extractor = new DocumentExtractor({ customVLM: mockProvider(data) });
+        const result = await extractor.extract(image.buffer, EngineeringSpecSchema, { tiling: "off" });
+        expect(result.data.project).toBe("Brigde-Alpha");
+        expect(result.data.dimensions.height).toBe(80);
+        expect(result.data.testResults).toHaveLength(2);
+    });
+    it("extracts a Clinical Report with biomarkers", async () => {
+        const data = {
+            patient: { name: "John Doe", id: "PAT-987654", dob: "1985-03-15" },
+            reportDate: "2026-08-30",
+            biomarkers: [
+                { marker: "Glucose", value: 95, unit: "mg/dL", referenceRange: "70 - 100" },
+                { marker: "LDL", value: 110, unit: "mg/dL", referenceRange: "< 130" },
+            ],
+        };
+        const extractor = new DocumentExtractor({ customVLM: mockProvider(data) });
+        const result = await extractor.extract(image.buffer, ClinicalReportSchema, { tiling: "off" });
+        expect(result.data.patient.name).toBe("John Doe");
+        expect(result.data.biomarkers[0].marker).toBe("Glucose");
+        expect(result.data.biomarkers).toHaveLength(2);
+    });
+    it("runs invariants across domains", async () => {
+        const data = {
+            contractNumber: "CNT-2026-002",
+            parties: { client: "Client A", contractor: "Contractor B" },
+            effectiveDate: "2026-02-01",
+            termYears: 2,
+            clauseCount: 8,
+            penaltyAmount: 10000,
+        };
+        const extractor = new DocumentExtractor({ customVLM: mockProvider(data) });
+        const result = await extractor.extract(image.buffer, LegalContractSchema, {
+            tiling: "off",
+            prove: true,
+            invariants: [{ id: "min-clauses", expression: "clauseCount >= 5" }],
+        });
+        expect(result.invariants?.ok).toBe(true);
+    });
+});
+//# sourceMappingURL=multi-domain.test.js.map

package/dist/tests/multi-domain.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"multi-domain.test.js","sourceRoot":"","sources":["../../tests/multi-domain.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AAC7D,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAC7D,OAAO,EAAE,sBAAsB,EAAE,MAAM,iCAAiC,CAAC;AAEzE,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AAEjD,MAAM,mBAAmB,GAAG,CAAC,CAAC,MAAM,CAAC;IACnC,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE;IAC1B,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC;QAChB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;QAClB,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;KACvB,CAAC;IACF,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE;IACzB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;IACrB,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;IACvB,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE;CAC1B,CAAC,CAAC;AAEH,MAAM,qBAAqB,GAAG,CAAC,CAAC,MAAM,CAAC;IACrC,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;IACnB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;IACpB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;IACpB,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC;QACnB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;QAClB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;QACjB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;KACnB,CAAC;IACF,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC;QACnB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;QAClB,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;KACpB,CAAC;IACF,WAAW,EAAE,CAAC,CAAC,KAAK,CAClB,CAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;QAChB,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE;QACnB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;KAClB,CAAC,CACH;CACF,CAAC,CAAC;AAEH,MAAM,oBAAoB,GAAG,CAAC,CAAC,MAAM,CAAC;IACpC,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC;QAChB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;QAChB,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;QACd,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE;KAChB,CAAC;IACF,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;IACtB,UAAU,EAAE,CAAC,CAAC,KAAK,CACjB,CAAC,CAAC,MAAM,CAAC;QACP,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;QAClB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;QACjB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;QAChB,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE;KAC3B,CAAC,CACH;CACF,CAAC,CAAC;AAEH,SAAS,YAAY,CAAC,YAAqB;IACzC,OAAO;QACL,IAAI,EAAE,MAAM;QACZ,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YACjC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;YAC5D,GAAG,EAAE,IAAI;SACQ,CAAC;KACrB,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,IAAI,KAAmD,CAAC;IAExD,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,KAAK,GAAG,MAAM,gBAAgB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,CAAC,sBAAsB,CAAC,mBAAmB,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;QACtF,MAAM,CAAC,sBAAsB,CAAC,qBAAqB,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;QACxF,MAAM,CAAC,sBAAsB,CAAC,oBAAoB,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC;IACzF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QACzC,MAAM,IAAI,GAAG;YACX,cAAc,EAAE,cAAc;YAC9B,OAAO,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,UAAU,EAAE,gBAAgB,EAAE;YAC9D,aAAa,EAAE,YAAY;YAC3B,SAAS,EAAE,CAAC;YACZ,WAAW,EAAE,EAAE;YACf,aAAa,EAAE,KAAK;SACrB,CAAC;QACF,MAAM,SAAS,GAAG,IAAI,iBAAiB,CAAC,EAAE,SAAS,EAAE,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC3E,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,mBAAmB,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;QAE7F,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,IAAI,GAAG;YACX,OAAO,EAAE,cAAc;YACvB,QAAQ,EAAE,OAAO;YACjB,QAAQ,EAAE,YAAY;YACtB,UAAU,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE;YACpD,UAAU,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE;YAC1C,WAAW,EAAE;gBACX,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE;gBAC7C,EAAE,IAAI,EAAE,aAAa,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE;aAClD;SACF,CAAC;QACF,MAAM,SAAS,GAAG,IAAI,iBAAiB,CAAC,EAAE,SAAS,EAAE,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC3E,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,qBAAqB,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;QAE/F,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,IAAI,GAAG;YACX,OAAO,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,YAAY,EAAE,GAAG,EAAE,YAAY,EAAE;YAClE,UAAU,EAAE,YAAY;YACxB,UAAU,EAAE;gBACV,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE;gBAC3E,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,cAAc,EAAE,OAAO,EAAE;aACtE;SACF,CAAC;QACF,MAAM,SAAS,GAAG,IAAI,iBAAiB,CAAC,EAAE,SAAS,EAAE,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC3E,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,oBAAoB,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;QAE9F,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACzD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC9C,MAAM,IAAI,GAAG;YACX,cAAc,EAAE,cAAc;YAC9B,OAAO,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,cAAc,EAAE;YAC3D,aAAa,EAAE,YAAY;YAC3B,SAAS,EAAE,CAAC;YACZ,WAAW,EAAE,CAAC;YACd,aAAa,EAAE,KAAK;SACrB,CAAC;QACF,MAAM,SAAS,GAAG,IAAI,iBAAiB,CAAC,EAAE,SAAS,EAAE,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC3E,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,mBAAmB,EAAE;YACxE,MAAM,EAAE,KAAK;YACb,KAAK,EAAE,IAAI;YACX,UAAU,EAAE,CAAC,EAAE,EAAE,EAAE,aAAa,EAAE,UAAU,EAAE,kBAAkB,EAAE,CAAC;SACpE,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/tests/normalizer.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=normalizer.test.d.ts.map

package/dist/tests/normalizer.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"normalizer.test.d.ts","sourceRoot":"","sources":["../../tests/normalizer.test.ts"],"names":[],"mappings":""}