@apart-tech/intelligence-core 1.12.0 → 1.13.0

package/dist/services/document-service.test.js

@@ -0,0 +1,289 @@
+import { describe, expect, it, vi, beforeEach } from "vitest";
+import { DocumentService } from "./document-service.js";
+// ── Mock factories ──────────────────────────────────────────────────────────
+function mockPrisma() {
+    return {
+        document: {
+            create: vi.fn(),
+            update: vi.fn(),
+            findFirst: vi.fn(),
+            findMany: vi.fn(),
+        },
+    };
+}
+function mockStorage() {
+    const signedUrl = "https://storage.googleapis.com/signed-url";
+    const fileObj = {
+        getSignedUrl: vi.fn().mockResolvedValue([signedUrl]),
+        exists: vi.fn().mockResolvedValue([true]),
+        download: vi.fn().mockResolvedValue([Buffer.from("file-content")]),
+    };
+    return {
+        bucket: vi.fn().mockReturnValue({
+            file: vi.fn().mockReturnValue(fileObj),
+        }),
+        _file: fileObj,
+        _signedUrl: signedUrl,
+    };
+}
+function mockOcrService() {
+    return {
+        extractText: vi.fn().mockResolvedValue({
+            text: "# Doc Title\n\nExtracted content\n\n---\n\nPage 2 content",
+            tokensIn: 100,
+            tokensOut: 50,
+            model: "mistral-ocr-latest",
+        }),
+    };
+}
+function mockNodeService() {
+    return {
+        create: vi.fn().mockResolvedValue({
+            id: "node-1",
+            type: "document",
+            title: "test.pdf",
+            content: "extracted text",
+        }),
+    };
+}
+function mockChunkService() {
+    return {
+        chunkAndEmbed: vi.fn().mockResolvedValue(undefined),
+    };
+}
+function mockTagService() {
+    return {
+        applyTags: vi.fn().mockResolvedValue([]),
+    };
+}
+const TENANT = { organizationId: "org-123" };
+const BUCKET_NAME = "test-bucket";
+describe("DocumentService", () => {
+    let db;
+    let storage;
+    let ocrService;
+    let nodeService;
+    let chunkService;
+    let tagService;
+    let service;
+    beforeEach(() => {
+        db = mockPrisma();
+        storage = mockStorage();
+        ocrService = mockOcrService();
+        nodeService = mockNodeService();
+        chunkService = mockChunkService();
+        tagService = mockTagService();
+        service = new DocumentService(db, storage, BUCKET_NAME, ocrService, nodeService, chunkService, tagService, undefined, undefined, TENANT);
+    });
+    // ── createUploadUrls ──────────────────────────────────────────────────────
+    describe("createUploadUrls", () => {
+        it("creates document records and returns signed URLs", async () => {
+            db.document.create.mockResolvedValue({
+                id: "doc-1",
+                fileName: "test.pdf",
+                mimeType: "application/pdf",
+                fileSizeBytes: 1024,
+                storagePath: "",
+                status: "uploading",
+            });
+            db.document.update.mockResolvedValue({});
+            const results = await service.createUploadUrls([
+                {
+                    fileName: "test.pdf",
+                    mimeType: "application/pdf",
+                    sizeBytes: 1024,
+                },
+            ], "user@example.com");
+            expect(results).toHaveLength(1);
+            expect(results[0].id).toBe("doc-1");
+            expect(results[0].uploadUrl).toBe(storage._signedUrl);
+            expect(results[0].fileName).toBe("test.pdf");
+            // Verify DB calls
+            expect(db.document.create).toHaveBeenCalledOnce();
+            expect(db.document.update).toHaveBeenCalledOnce();
+            // Verify storage path includes org and doc ID
+            const updateCall = db.document.update.mock.calls[0][0];
+            expect(updateCall.data.storagePath).toContain("org-123/doc-1/original.pdf");
+        });
+        it("maps mime types to correct extensions", async () => {
+            const mimeMap = {
+                "application/pdf": "pdf",
+                "image/png": "png",
+                "image/jpeg": "jpeg",
+                "image/avif": "avif",
+            };
+            for (const [mime, ext] of Object.entries(mimeMap)) {
+                db.document.create.mockResolvedValue({
+                    id: `doc-${ext}`,
+                    fileName: `test.${ext}`,
+                    mimeType: mime,
+                });
+                db.document.update.mockResolvedValue({});
+                await service.createUploadUrls([{ fileName: `test.${ext}`, mimeType: mime, sizeBytes: 100 }], "user");
+                const updateCall = db.document.update.mock.calls[db.document.update.mock.calls.length - 1][0];
+                expect(updateCall.data.storagePath).toContain(`original.${ext}`);
+            }
+        });
+    });
+    // ── processDocuments ──────────────────────────────────────────────────────
+    describe("processDocuments", () => {
+        it("sets status to processing and fires background work", async () => {
+            db.document.findFirst.mockResolvedValue({
+                id: "doc-1",
+                status: "uploading",
+                storagePath: "org-123/doc-1/original.pdf",
+                fileName: "test.pdf",
+                mimeType: "application/pdf",
+                createdBy: "user",
+            });
+            db.document.update.mockResolvedValue({});
+            const result = await service.processDocuments(["doc-1"]);
+            expect(result.documents).toHaveLength(1);
+            expect(result.documents[0].status).toBe("processing");
+            expect(result.warnings).toHaveLength(0);
+        });
+        it("warns on missing documents", async () => {
+            db.document.findFirst.mockResolvedValue(null);
+            const result = await service.processDocuments(["nonexistent"]);
+            expect(result.warnings).toContain("Document nonexistent not found");
+        });
+        it("warns on documents not in uploading status", async () => {
+            db.document.findFirst.mockResolvedValue({
+                id: "doc-1",
+                status: "ready",
+            });
+            const result = await service.processDocuments(["doc-1"]);
+            expect(result.warnings[0]).toContain("expected \"uploading\"");
+            expect(result.documents[0].status).toBe("ready");
+        });
+    });
+    // ── processDocument ───────────────────────────────────────────────────────
+    describe("processDocument", () => {
+        const mockDoc = {
+            id: "doc-1",
+            status: "processing",
+            storagePath: "org-123/doc-1/original.pdf",
+            fileName: "test.pdf",
+            mimeType: "application/pdf",
+            createdBy: "user",
+            fileSizeBytes: 1024,
+        };
+        it("runs full OCR → node → chunk → tag → update pipeline", async () => {
+            db.document.findFirst
+                .mockResolvedValueOnce(mockDoc) // initial lookup
+                .mockResolvedValueOnce(null); // dedup check
+            db.document.update.mockResolvedValue({});
+            await service.processDocument("doc-1");
+            // OCR was called
+            expect(ocrService.extractText).toHaveBeenCalledOnce();
+            // Node was created
+            expect(nodeService.create).toHaveBeenCalledWith(expect.objectContaining({
+                type: "document",
+                title: "test.pdf",
+                status: "approved",
+            }));
+            // Chunks were created
+            expect(chunkService.chunkAndEmbed).toHaveBeenCalledWith("node-1", "test.pdf", expect.any(String));
+            // Tags were applied
+            expect(tagService.applyTags).toHaveBeenCalledWith("node-1", expect.arrayContaining([
+                expect.objectContaining({ tagName: "source_document" }),
+                expect.objectContaining({ tagName: "source_file_format" }),
+            ]), "document-service");
+            // Document updated to ready
+            const lastUpdate = db.document.update.mock.calls[db.document.update.mock.calls.length - 1][0];
+            expect(lastUpdate.data.status).toBe("ready");
+            expect(lastUpdate.data.nodeId).toBe("node-1");
+            expect(lastUpdate.data.contentHash).toBeDefined();
+            expect(lastUpdate.data.ocrModel).toBe("mistral-ocr-latest");
+            expect(lastUpdate.data.pageCount).toBe(2); // two pages separated by ---
+        });
+        it("deduplicates by content hash", async () => {
+            db.document.findFirst
+                .mockResolvedValueOnce(mockDoc) // initial lookup
+                .mockResolvedValueOnce({ id: "existing-doc", nodeId: "existing-node", status: "ready" }); // dedup match
+            db.document.update.mockResolvedValue({});
+            await service.processDocument("doc-1");
+            // OCR should NOT have been called
+            expect(ocrService.extractText).not.toHaveBeenCalled();
+            // Document should link to existing node
+            const updateCall = db.document.update.mock.calls[db.document.update.mock.calls.length - 1][0];
+            expect(updateCall.data.nodeId).toBe("existing-node");
+            expect(updateCall.data.status).toBe("ready");
+        });
+        it("marks as failed when GCS file missing", async () => {
+            storage._file.exists.mockResolvedValue([false]);
+            db.document.findFirst
+                .mockResolvedValueOnce(mockDoc);
+            db.document.update.mockResolvedValue({});
+            await expect(service.processDocument("doc-1")).rejects.toThrow("File not found in GCS");
+            // Should have updated status to failed
+            const updateCall = db.document.update.mock.calls[0][0];
+            expect(updateCall.data.status).toBe("failed");
+            expect(updateCall.data.errorMessage).toContain("File not found in GCS");
+        });
+        it("marks as failed when OCR fails", async () => {
+            db.document.findFirst
+                .mockResolvedValueOnce(mockDoc)
+                .mockResolvedValueOnce(null); // no dedup
+            db.document.update.mockResolvedValue({});
+            ocrService.extractText.mockRejectedValueOnce(new Error("OCR timeout"));
+            await expect(service.processDocument("doc-1")).rejects.toThrow("OCR timeout");
+            const updateCall = db.document.update.mock.calls[db.document.update.mock.calls.length - 1][0];
+            expect(updateCall.data.status).toBe("failed");
+            expect(updateCall.data.errorMessage).toBe("OCR timeout");
+        });
+        it("throws when document not found", async () => {
+            db.document.findFirst.mockResolvedValue(null);
+            await expect(service.processDocument("nonexistent")).rejects.toThrow("Document nonexistent not found");
+        });
+    });
+    // ── listDocuments ─────────────────────────────────────────────────────────
+    describe("listDocuments", () => {
+        it("lists all documents without filters", async () => {
+            db.document.findMany.mockResolvedValue([{ id: "doc-1" }]);
+            const docs = await service.listDocuments();
+            expect(db.document.findMany).toHaveBeenCalledWith({
+                where: {},
+                orderBy: { createdAt: "desc" },
+            });
+            expect(docs).toHaveLength(1);
+        });
+        it("filters by IDs", async () => {
+            db.document.findMany.mockResolvedValue([]);
+            await service.listDocuments({ ids: ["doc-1", "doc-2"] });
+            expect(db.document.findMany).toHaveBeenCalledWith({
+                where: { id: { in: ["doc-1", "doc-2"] } },
+                orderBy: { createdAt: "desc" },
+            });
+        });
+        it("filters by status", async () => {
+            db.document.findMany.mockResolvedValue([]);
+            await service.listDocuments({ status: "ready" });
+            expect(db.document.findMany).toHaveBeenCalledWith({
+                where: { status: "ready" },
+                orderBy: { createdAt: "desc" },
+            });
+        });
+    });
+    // ── getDownloadUrl ────────────────────────────────────────────────────────
+    describe("getDownloadUrl", () => {
+        it("generates a signed read URL", async () => {
+            db.document.findFirst.mockResolvedValue({
+                id: "doc-1",
+                fileName: "test.pdf",
+                mimeType: "application/pdf",
+                storagePath: "org-123/doc-1/original.pdf",
+            });
+            const result = await service.getDownloadUrl("doc-1");
+            expect(result.downloadUrl).toBe(storage._signedUrl);
+            expect(result.fileName).toBe("test.pdf");
+            expect(result.mimeType).toBe("application/pdf");
+            expect(result.expiresIn).toBe(900); // 15 minutes
+        });
+        it("throws when document not found", async () => {
+            db.document.findFirst.mockResolvedValue(null);
+            await expect(service.getDownloadUrl("nonexistent")).rejects.toThrow("Document nonexistent not found");
+        });
+    });
+});
+//# sourceMappingURL=document-service.test.js.map

package/dist/services/document-service.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"document-service.test.js","sourceRoot":"","sources":["../../src/services/document-service.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAMxD,+EAA+E;AAE/E,SAAS,UAAU;IACjB,OAAO;QACL,QAAQ,EAAE;YACR,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YACf,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YACf,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;YAClB,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;SAClB;KACK,CAAC;AACX,CAAC;AAED,SAAS,WAAW;IAClB,MAAM,SAAS,GAAG,2CAA2C,CAAC;IAC9D,MAAM,OAAO,GAAG;QACd,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,SAAS,CAAC,CAAC;QACpD,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,CAAC;QACzC,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC;KACnE,CAAC;IACF,OAAO;QACL,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;YAC9B,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,OAAO,CAAC;SACvC,CAAC;QACF,KAAK,EAAE,OAAO;QACd,UAAU,EAAE,SAAS;KACtB,CAAC;AACJ,CAAC;AAED,SAAS,cAAc;IACrB,OAAO;QACL,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YACrC,IAAI,EAAE,2DAA2D;YACjE,QAAQ,EAAE,GAAG;YACb,SAAS,EAAE,EAAE;YACb,KAAK,EAAE,oBAAoB;SAC5B,CAAC;KACI,CAAC;AACX,CAAC;AAED,SAAS,eAAe;IACtB,OAAO;QACL,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YAChC,EAAE,EAAE,QAAQ;YACZ,IAAI,EAAE,UAAU;YAChB,KAAK,EAAE,UAAU;YACjB,OAAO,EAAE,gBAAgB;SAC1B,CAAC;KACI,CAAC;AACX,CAAC;AAED,SAAS,gBAAgB;IACvB,OAAO;QACL,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;KAC7C,CAAC;AACX,CAAC;AAED,SAAS,cAAc;IACrB,OAAO;QACL,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;KAClC,CAAC;AACX,CAAC;AAED,MAAM,MAAM,GAAG,EAAE,cAAc,EAAE,SAAS,EAAE,CAAC;AAC7C,MAAM,WAAW,GAAG,aAAa,CAAC;AAElC,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,IAAI,EAAiC,CAAC;IACtC,IAAI,OAAuC,CAAC;IAC5C,IAAI,UAA6C,CAAC;IAClD,IAAI,WAA+C,CAAC;IACpD,IAAI,YAAiD,CAAC;IACtD,IAAI,UAA6C,CAAC;IAClD,IAAI,OAAwB,CAAC;IAE7B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,GAAG,UAAU,EAAE,CAAC;QAClB,OAAO,GAAG,WAAW,EAAE,CAAC;QACxB,UAAU,GAAG,cAAc,EAAE,CAAC;QAC9B,WAAW,GAAG,eAAe,EAAE,CAAC;QAChC,YAAY,GAAG,gBAAgB,EAAE,CAAC;QAClC,UAAU,GAAG,cAAc,EAAE,CAAC;QAC9B,OAAO,GAAG,IAAI,eAAe,CAC3B,EAAE,EACF,OAAc,EACd,WAAW,EACX,UAAU,EACV,WAAW,EACX,YAAY,EACZ,UAAU,EACV,SAAS,EACT,SAAS,EACT,MAAM,CACP,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,6EAA6E;IAE7E,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,iBAAiB,CAAC;gBACnC,EAAE,EAAE,OAAO;gBACX,QAAQ,EAAE,UAAU;gBACpB,QAAQ,EAAE,iBAAiB;gBAC3B,aAAa,EAAE,IAAI;gBACnB,WAAW,EAAE,EAAE;gBACf,MAAM,EAAE,WAAW;aACpB,CAAC,CAAC;YACH,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAEzC,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,gBAAgB,CAC5C;gBACE;oBACE,QAAQ,EAAE,UAAU;oBACpB,QAAQ,EAAE,iBAAiB;oBAC3B,SAAS,EAAE,IAAI;iBAChB;aACF,EACD,kBAAkB,CACnB,CAAC;YAEF,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACpC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YACtD,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAE7C,kBAAkB;YAClB,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,oBAAoB,EAAE,CAAC;YAClD,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,oBAAoB,EAAE,CAAC;YAElD,8CAA8C;YAC9C,MAAM,UAAU,GAAG,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACvD,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,4BAA4B,CAAC,CAAC;QAC9E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,OAAO,GAAG;gBACd,iBAAiB,EAAE,KAAK;gBACxB,WAAW,EAAE,KAAK;gBAClB,YAAY,EAAE,MAAM;gBACpB,YAAY,EAAE,MAAM;aACrB,CAAC;YAEF,KAAK,MAAM,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;gBAClD,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,iBAAiB,CAAC;oBACnC,EAAE,EAAE,OAAO,GAAG,EAAE;oBAChB,QAAQ,EAAE,QAAQ,GAAG,EAAE;oBACvB,QAAQ,EAAE,IAAI;iBACf,CAAC,CAAC;gBACH,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;gBAEzC,MAAM,OAAO,CAAC,gBAAgB,CAC5B,CAAC,EAAE,QAAQ,EAAE,QAAQ,GAAG,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,EAC7D,MAAM,CACP,CAAC;gBAEF,MAAM,UAAU,GACd,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAC3B,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CACzC,CAAC,CAAC,CAAC,CAAC;gBACP,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,YAAY,GAAG,EAAE,CAAC,CAAC;YACnE,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,6EAA6E;IAE7E,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,iBAAiB,CAAC;gBACtC,EAAE,EAAE,OAAO;gBACX,MAAM,EAAE,WAAW;gBACnB,WAAW,EAAE,4BAA4B;gBACzC,QAAQ,EAAE,UAAU;gBACpB,QAAQ,EAAE,iBAAiB;gBAC3B,SAAS,EAAE,MAAM;aAClB,CAAC,CAAC;YACH,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAEzC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,gBAAgB,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;YAEzD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACtD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAE9C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,gBAAgB,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC;YAE/D,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC,gCAAgC,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,iBAAiB,CAAC;gBACtC,EAAE,EAAE,OAAO;gBACX,MAAM,EAAE,OAAO;aAChB,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,gBAAgB,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;YAEzD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,wBAAwB,CAAC,CAAC;YAC/D,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,6EAA6E;IAE7E,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,MAAM,OAAO,GAAG;YACd,EAAE,EAAE,OAAO;YACX,MAAM,EAAE,YAAY;YACpB,WAAW,EAAE,4BAA4B;YACzC,QAAQ,EAAE,UAAU;YACpB,QAAQ,EAAE,iBAAiB;YAC3B,SAAS,EAAE,MAAM;YACjB,aAAa,EAAE,IAAI;SACpB,CAAC;QAEF,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,EAAE,CAAC,QAAQ,CAAC,SAAS;iBAClB,qBAAqB,CAAC,OAAO,CAAC,CAAC,iBAAiB;iBAChD,qBAAqB,CAAC,IAAI,CAAC,CAAC,CAAC,cAAc;YAC9C,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAEzC,MAAM,OAAO,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;YAEvC,iBAAiB;YACjB,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,oBAAoB,EAAE,CAAC;YAEtD,mBAAmB;YACnB,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAC7C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,UAAU;gBAChB,KAAK,EAAE,UAAU;gBACjB,MAAM,EAAE,UAAU;aACnB,CAAC,CACH,CAAC;YAEF,sBAAsB;YACtB,MAAM,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACrD,QAAQ,EACR,UAAU,EACV,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CACnB,CAAC;YAEF,oBAAoB;YACpB,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAC/C,QAAQ,EACR,MAAM,CAAC,eAAe,CAAC;gBACrB,MAAM,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,iBAAiB,EAAE,CAAC;gBACvD,MAAM,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,oBAAoB,EAAE,CAAC;aAC3D,CAAC,EACF,kBAAkB,CACnB,CAAC;YAEF,4BAA4B;YAC5B,MAAM,UAAU,GAAG,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC9F,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC7C,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC9C,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC;YAClD,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YAC5D,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,6BAA6B;QAC1E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC5C,EAAE,CAAC,QAAQ,CAAC,SAAS;iBAClB,qBAAqB,CAAC,OAAO,CAAC,CAAC,iBAAiB;iBAChD,qBAAqB,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,MAAM,EAAE,eAAe,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,cAAc;YAC1G,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAEzC,MAAM,OAAO,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;YAEvC,kCAAkC;YAClC,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAEtD,wCAAwC;YACxC,MAAM,UAAU,GAAG,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC9F,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YACrD,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;YAChD,EAAE,CAAC,QAAQ,CAAC,SAAS;iBAClB,qBAAqB,CAAC,OAAO,CAAC,CAAC;YAClC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAEzC,MAAM,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC5D,uBAAuB,CACxB,CAAC;YAEF,uCAAuC;YACvC,MAAM,UAAU,GAAG,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACvD,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC9C,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,EAAE,CAAC,QAAQ,CAAC,SAAS;iBAClB,qBAAqB,CAAC,OAAO,CAAC;iBAC9B,qBAAqB,CAAC,IAAI,CAAC,CAAC,CAAC,WAAW;YAC3C,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACxC,UAAU,CAAC,WAAmB,CAAC,qBAAqB,CACnD,IAAI,KAAK,CAAC,aAAa,CAAC,CACzB,CAAC;YAEF,MAAM,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC5D,aAAa,CACd,CAAC;YAEF,MAAM,UAAU,GAAG,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC9F,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC9C,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAE9C,MAAM,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAClE,gCAAgC,CACjC,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,6EAA6E;IAE7E,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;YAE1D,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,aAAa,EAAE,CAAC;YAE3C,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC;gBAChD,KAAK,EAAE,EAAE;gBACT,OAAO,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE;aAC/B,CAAC,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gBAAgB,EAAE,KAAK,IAAI,EAAE;YAC9B,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAE3C,MAAM,OAAO,CAAC,aAAa,CAAC,EAAE,GAAG,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC;YAEzD,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC;gBAChD,KAAK,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC,EAAE,EAAE;gBACzC,OAAO,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE;aAC/B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mBAAmB,EAAE,KAAK,IAAI,EAAE;YACjC,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAE3C,MAAM,OAAO,CAAC,aAAa,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;YAEjD,MAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC;gBAChD,KAAK,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE;gBAC1B,OAAO,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE;aAC/B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,6EAA6E;IAE7E,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,iBAAiB,CAAC;gBACtC,EAAE,EAAE,OAAO;gBACX,QAAQ,EAAE,UAAU;gBACpB,QAAQ,EAAE,iBAAiB;gBAC3B,WAAW,EAAE,4BAA4B;aAC1C,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;YAErD,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YACpD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,aAAa;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAE9C,MAAM,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACjE,gCAAgC,CACjC,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/services/ocr-service.d.ts

@@ -0,0 +1,21 @@
+/**
+ * OCR text extraction via the AIShield proxy (Mistral OCR model).
+ *
+ * Calls the OpenAI-compatible chat completions endpoint at AIShield,
+ * passing a signed GCS document URL as a `document_url` content part.
+ * The model returns extracted text as structured markdown.
+ */
+export interface OcrResult {
+    text: string;
+    tokensIn: number;
+    tokensOut: number;
+    model: string;
+}
+export declare class OcrService {
+    private baseUrl;
+    private apiKey;
+    private model;
+    constructor(baseUrl: string, apiKey: string, model?: string);
+    extractText(documentUrl: string, organizationId: string): Promise<OcrResult>;
+}
+//# sourceMappingURL=ocr-service.d.ts.map

package/dist/services/ocr-service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ocr-service.d.ts","sourceRoot":"","sources":["../../src/services/ocr-service.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,MAAM,WAAW,SAAS;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;CACf;AAQD,qBAAa,UAAU;IAEnB,OAAO,CAAC,OAAO;IACf,OAAO,CAAC,MAAM;IACd,OAAO,CAAC,KAAK;gBAFL,OAAO,EAAE,MAAM,EACf,MAAM,EAAE,MAAM,EACd,KAAK,GAAE,MAA6B;IAGxC,WAAW,CACf,WAAW,EAAE,MAAM,EACnB,cAAc,EAAE,MAAM,GACrB,OAAO,CAAC,SAAS,CAAC;CAiDtB"}

package/dist/services/ocr-service.js

@@ -0,0 +1,61 @@
+/**
+ * OCR text extraction via the AIShield proxy (Mistral OCR model).
+ *
+ * Calls the OpenAI-compatible chat completions endpoint at AIShield,
+ * passing a signed GCS document URL as a `document_url` content part.
+ * The model returns extracted text as structured markdown.
+ */
+const OCR_PROMPT = `Extract all text content from this document. Format the output as clean markdown:
+- Preserve document structure with headings (#, ##, ###).
+- Render tables as markdown tables.
+- Separate pages with --- (horizontal rule).
+- Do not add any commentary — output only the extracted text.`;
+export class OcrService {
+    baseUrl;
+    apiKey;
+    model;
+    constructor(baseUrl, apiKey, model = "mistral-ocr-latest") {
+        this.baseUrl = baseUrl;
+        this.apiKey = apiKey;
+        this.model = model;
+    }
+    async extractText(documentUrl, organizationId) {
+        const body = {
+            model: this.model,
+            messages: [
+                {
+                    role: "user",
+                    content: [
+                        { type: "text", text: OCR_PROMPT },
+                        { type: "document_url", document_url: documentUrl },
+                    ],
+                },
+            ],
+        };
+        const response = await fetch(`${this.baseUrl}/v1/chat/completions`, {
+            method: "POST",
+            headers: {
+                "Content-Type": "application/json",
+                Authorization: `Bearer ${this.apiKey}`,
+                Organisation: organizationId,
+            },
+            body: JSON.stringify(body),
+        });
+        if (!response.ok) {
+            const errorBody = await response.text().catch(() => "");
+            throw new Error(`OCR request failed (${response.status}): ${errorBody}`);
+        }
+        const result = await response.json();
+        const text = result.choices?.[0]?.message?.content;
+        if (text === undefined || text === null) {
+            throw new Error("OCR response missing choices[0].message.content");
+        }
+        return {
+            text,
+            tokensIn: result.usage?.prompt_tokens ?? 0,
+            tokensOut: result.usage?.completion_tokens ?? 0,
+            model: result.model ?? this.model,
+        };
+    }
+}
+//# sourceMappingURL=ocr-service.js.map

package/dist/services/ocr-service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ocr-service.js","sourceRoot":"","sources":["../../src/services/ocr-service.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AASH,MAAM,UAAU,GAAG;;;;8DAI2C,CAAC;AAE/D,MAAM,OAAO,UAAU;IAEX;IACA;IACA;IAHV,YACU,OAAe,EACf,MAAc,EACd,QAAgB,oBAAoB;QAFpC,YAAO,GAAP,OAAO,CAAQ;QACf,WAAM,GAAN,MAAM,CAAQ;QACd,UAAK,GAAL,KAAK,CAA+B;IAC3C,CAAC;IAEJ,KAAK,CAAC,WAAW,CACf,WAAmB,EACnB,cAAsB;QAEtB,MAAM,IAAI,GAAG;YACX,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,QAAQ,EAAE;gBACR;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE;wBACP,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE;wBAClC,EAAE,IAAI,EAAE,cAAc,EAAE,YAAY,EAAE,WAAW,EAAE;qBACpD;iBACF;aACF;SACF,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,sBAAsB,EAAE;YAClE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;gBACtC,YAAY,EAAE,cAAc;aAC7B;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;YACxD,MAAM,IAAI,KAAK,CACb,uBAAuB,QAAQ,CAAC,MAAM,MAAM,SAAS,EAAE,CACxD,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,EAIjC,CAAC;QAEF,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC;QACnD,IAAI,IAAI,KAAK,SAAS,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YACxC,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;QACrE,CAAC;QAED,OAAO;YACL,IAAI;YACJ,QAAQ,EAAE,MAAM,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC;YAC1C,SAAS,EAAE,MAAM,CAAC,KAAK,EAAE,iBAAiB,IAAI,CAAC;YAC/C,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK;SAClC,CAAC;IACJ,CAAC;CACF"}

package/dist/services/ocr-service.test.d.ts

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

package/dist/services/ocr-service.test.d.ts.map

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

package/dist/services/ocr-service.test.js

@@ -0,0 +1,66 @@
+import { describe, expect, it, vi, afterEach, beforeEach } from "vitest";
+import { OcrService } from "./ocr-service.js";
+describe("OcrService", () => {
+    const BASE_URL = "https://aishield.apart.tech";
+    const API_KEY = "test-api-key";
+    const MODEL = "mistral-ocr-latest";
+    const ORG_ID = "org-123";
+    const DOC_URL = "https://storage.googleapis.com/bucket/doc.pdf?X-Goog-Signature=...";
+    let service;
+    const mockFetch = vi.fn();
+    beforeEach(() => {
+        service = new OcrService(BASE_URL, API_KEY, MODEL);
+        mockFetch.mockReset();
+        vi.stubGlobal("fetch", mockFetch);
+    });
+    afterEach(() => {
+        vi.restoreAllMocks();
+    });
+    it("sends correct request format and parses response", async () => {
+        mockFetch.mockResolvedValueOnce(new Response(JSON.stringify({
+            choices: [{ message: { content: "# Title\n\nExtracted text" } }],
+            usage: { prompt_tokens: 100, completion_tokens: 50 },
+            model: "mistral-ocr-latest",
+        }), { status: 200 }));
+        const result = await service.extractText(DOC_URL, ORG_ID);
+        expect(result.text).toBe("# Title\n\nExtracted text");
+        expect(result.tokensIn).toBe(100);
+        expect(result.tokensOut).toBe(50);
+        expect(result.model).toBe("mistral-ocr-latest");
+        expect(mockFetch).toHaveBeenCalledOnce();
+        const [url, opts] = mockFetch.mock.calls[0];
+        expect(url).toBe(`${BASE_URL}/v1/chat/completions`);
+        expect(opts.method).toBe("POST");
+        const headers = opts.headers;
+        expect(headers["Authorization"]).toBe(`Bearer ${API_KEY}`);
+        expect(headers["Organisation"]).toBe(ORG_ID);
+        expect(headers["Content-Type"]).toBe("application/json");
+        const body = JSON.parse(opts.body);
+        expect(body.model).toBe(MODEL);
+        expect(body.messages[0].content).toHaveLength(2);
+        expect(body.messages[0].content[0].type).toBe("text");
+        expect(body.messages[0].content[1].type).toBe("document_url");
+        expect(body.messages[0].content[1].document_url).toBe(DOC_URL);
+    });
+    it("throws on non-200 response", async () => {
+        mockFetch.mockResolvedValueOnce(new Response("Rate limited", { status: 429 }));
+        await expect(service.extractText(DOC_URL, ORG_ID)).rejects.toThrow("OCR request failed (429): Rate limited");
+    });
+    it("throws when response missing content", async () => {
+        mockFetch.mockResolvedValueOnce(new Response(JSON.stringify({ choices: [{ message: {} }], usage: {} }), { status: 200 }));
+        await expect(service.extractText(DOC_URL, ORG_ID)).rejects.toThrow("OCR response missing choices[0].message.content");
+    });
+    it("handles missing usage gracefully", async () => {
+        mockFetch.mockResolvedValueOnce(new Response(JSON.stringify({
+            choices: [{ message: { content: "text" } }],
+        }), { status: 200 }));
+        const result = await service.extractText(DOC_URL, ORG_ID);
+        expect(result.tokensIn).toBe(0);
+        expect(result.tokensOut).toBe(0);
+    });
+    it("handles network error", async () => {
+        mockFetch.mockRejectedValueOnce(new Error("Network error"));
+        await expect(service.extractText(DOC_URL, ORG_ID)).rejects.toThrow("Network error");
+    });
+});
+//# sourceMappingURL=ocr-service.test.js.map

package/dist/services/ocr-service.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ocr-service.test.js","sourceRoot":"","sources":["../../src/services/ocr-service.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAE9C,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,MAAM,QAAQ,GAAG,6BAA6B,CAAC;IAC/C,MAAM,OAAO,GAAG,cAAc,CAAC;IAC/B,MAAM,KAAK,GAAG,oBAAoB,CAAC;IACnC,MAAM,MAAM,GAAG,SAAS,CAAC;IACzB,MAAM,OAAO,GAAG,oEAAoE,CAAC;IAErF,IAAI,OAAmB,CAAC;IACxB,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAE1B,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,GAAG,IAAI,UAAU,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;QACnD,SAAS,CAAC,SAAS,EAAE,CAAC;QACtB,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,SAAS,CAAC,qBAAqB,CAC7B,IAAI,QAAQ,CACV,IAAI,CAAC,SAAS,CAAC;YACb,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,2BAA2B,EAAE,EAAE,CAAC;YAChE,KAAK,EAAE,EAAE,aAAa,EAAE,GAAG,EAAE,iBAAiB,EAAE,EAAE,EAAE;YACpD,KAAK,EAAE,oBAAoB;SAC5B,CAAC,EACF,EAAE,MAAM,EAAE,GAAG,EAAE,CAChB,CACF,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAE1D,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAEhD,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,EAAE,CAAC;QACzC,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAA0B,CAAC;QACrE,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,QAAQ,sBAAsB,CAAC,CAAC;QACpD,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEjC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAiC,CAAC;QACvD,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,OAAO,EAAE,CAAC,CAAC;QAC3D,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7C,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAEzD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAc,CAAC,CAAC;QAC7C,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/B,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACjD,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACtD,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC9D,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;QAC1C,SAAS,CAAC,qBAAqB,CAC7B,IAAI,QAAQ,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAC9C,CAAC;QAEF,MAAM,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAChE,wCAAwC,CACzC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,SAAS,CAAC,qBAAqB,CAC7B,IAAI,QAAQ,CACV,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,EACzD,EAAE,MAAM,EAAE,GAAG,EAAE,CAChB,CACF,CAAC;QAEF,MAAM,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAChE,iDAAiD,CAClD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAChD,SAAS,CAAC,qBAAqB,CAC7B,IAAI,QAAQ,CACV,IAAI,CAAC,SAAS,CAAC;YACb,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,CAAC;SAC5C,CAAC,EACF,EAAE,MAAM,EAAE,GAAG,EAAE,CAChB,CACF,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uBAAuB,EAAE,KAAK,IAAI,EAAE;QACrC,SAAS,CAAC,qBAAqB,CAAC,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC,CAAC;QAE5D,MAAM,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAChE,eAAe,CAChB,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@apart-tech/intelligence-core",
-  "version": "1.12.0",
+  "version": "1.13.0",
   "description": "Core library: database, services, and providers for Apart Intelligence",
   "type": "module",
   "main": "dist/index.js",

package/prisma/schema.prisma

@@ -39,6 +39,7 @@ model Organization {
   nodeTags        NodeTag[]
   nodeChunks      NodeChunk[]
   piiQueryLogs    PiiQueryLog[]
+  documents       Document[]
 
   @@map("organizations")
 }
@@ -608,3 +609,31 @@ model NodeTagAudit {
   @@index([orgId, createdAt], map: "idx_tag_audit_org_time")
   @@map("node_tag_audit")
 }
+
+// ── Document Archive ────────────────────────────────────────────────────────
+
+model Document {
+  id              String    @id @default(dbgenerated("gen_random_uuid()")) @db.Uuid
+  organizationId  String    @map("organization_id") @db.Uuid
+  fileName        String    @map("file_name") @db.VarChar(500)
+  mimeType        String    @map("mime_type") @db.VarChar(100)
+  fileSizeBytes   Int       @map("file_size_bytes")
+  storagePath     String    @map("storage_path") @db.VarChar(1000)
+  contentHash     String?   @map("content_hash") @db.VarChar(64)
+  status          String    @default("uploading") @db.VarChar(20)
+  ocrModel        String?   @map("ocr_model") @db.VarChar(100)
+  ocrTokensIn     Int?      @map("ocr_tokens_in")
+  ocrTokensOut    Int?      @map("ocr_tokens_out")
+  pageCount       Int?      @map("page_count")
+  nodeId          String?   @map("node_id") @db.Uuid
+  errorMessage    String?   @map("error_message")
+  createdBy       String    @map("created_by") @db.VarChar(255)
+  createdAt       DateTime  @default(now()) @map("created_at") @db.Timestamptz
+  updatedAt       DateTime  @updatedAt @map("updated_at") @db.Timestamptz
+
+  organization    Organization @relation(fields: [organizationId], references: [id], onDelete: Cascade)
+
+  @@index([organizationId, contentHash], map: "idx_documents_org_hash")
+  @@index([organizationId, status], map: "idx_documents_org_status")
+  @@map("documents")
+}