@apart-tech/intelligence-core 1.12.1 → 1.13.0

package/dist/services/document-service.js

@@ -0,0 +1,273 @@
+import { createHash } from "node:crypto";
+import { SINGLE_TENANT_ORG_ID } from "../db/tenant.js";
+const SIGNED_URL_EXPIRY_MS = 15 * 60 * 1000; // 15 minutes
+const MIME_TO_EXT = {
+    "application/pdf": "pdf",
+    "application/vnd.openxmlformats-officedocument.wordprocessingml.document": "docx",
+    "application/vnd.openxmlformats-officedocument.presentationml.presentation": "pptx",
+    "image/png": "png",
+    "image/jpeg": "jpeg",
+    "image/avif": "avif",
+};
+export class DocumentService {
+    db;
+    storage;
+    bucketName;
+    ocrService;
+    nodeService;
+    chunkService;
+    tagService;
+    piiEncryption;
+    piiOptions;
+    tenantCtx;
+    constructor(db, storage, bucketName, ocrService, nodeService, chunkService, tagService, piiEncryption, piiOptions, tenantCtx) {
+        this.db = db;
+        this.storage = storage;
+        this.bucketName = bucketName;
+        this.ocrService = ocrService;
+        this.nodeService = nodeService;
+        this.chunkService = chunkService;
+        this.tagService = tagService;
+        this.piiEncryption = piiEncryption;
+        this.piiOptions = piiOptions;
+        this.tenantCtx = tenantCtx ?? { organizationId: SINGLE_TENANT_ORG_ID };
+    }
+    /**
+     * Create Document records and generate signed PUT URLs for direct upload.
+     */
+    async createUploadUrls(files, createdBy) {
+        const results = [];
+        for (const file of files) {
+            const ext = MIME_TO_EXT[file.mimeType] ?? "bin";
+            // Create Document record first to get the ID
+            const doc = await this.db.document.create({
+                data: {
+                    fileName: file.fileName,
+                    mimeType: file.mimeType,
+                    fileSizeBytes: file.sizeBytes,
+                    storagePath: "", // will be set below
+                    status: "uploading",
+                    createdBy,
+                    organizationId: this.tenantCtx.organizationId,
+                },
+            });
+            const storagePath = `${this.tenantCtx.organizationId}/${doc.id}/original.${ext}`;
+            // Update with the real storage path
+            await this.db.document.update({
+                where: { id: doc.id },
+                data: { storagePath },
+            });
+            const bucket = this.storage.bucket(this.bucketName);
+            const gcsFile = bucket.file(storagePath);
+            const [uploadUrl] = await gcsFile.getSignedUrl({
+                version: "v4",
+                action: "write",
+                expires: Date.now() + SIGNED_URL_EXPIRY_MS,
+                contentType: file.mimeType,
+            });
+            results.push({
+                id: doc.id,
+                uploadUrl,
+                fileName: file.fileName,
+            });
+        }
+        return results;
+    }
+    /**
+     * Kick off async processing for uploaded documents.
+     * Returns immediately with current statuses; processing happens in background.
+     */
+    async processDocuments(documentIds) {
+        const warnings = [];
+        const statuses = [];
+        for (const id of documentIds) {
+            const doc = await this.db.document.findFirst({
+                where: { id },
+            });
+            if (!doc) {
+                warnings.push(`Document ${id} not found`);
+                continue;
+            }
+            if (doc.status !== "uploading") {
+                warnings.push(`Document ${id} is in status "${doc.status}", expected "uploading"`);
+                statuses.push({ id: doc.id, status: doc.status });
+                continue;
+            }
+            // Set status to processing
+            await this.db.document.update({
+                where: { id: doc.id },
+                data: { status: "processing" },
+            });
+            statuses.push({ id: doc.id, status: "processing" });
+            // Fire-and-forget: process in background
+            this.processDocument(doc.id).catch((err) => {
+                console.error(`Document processing failed for ${doc.id}:`, err);
+            });
+        }
+        return { documents: statuses, warnings };
+    }
+    /**
+     * Process a single document: verify GCS → hash → dedup → OCR → create node → PII → chunk → tag.
+     */
+    async processDocument(documentId) {
+        const doc = await this.db.document.findFirst({
+            where: { id: documentId },
+        });
+        if (!doc) {
+            throw new Error(`Document ${documentId} not found`);
+        }
+        try {
+            // 1. Verify file exists in GCS
+            const bucket = this.storage.bucket(this.bucketName);
+            const gcsFile = bucket.file(doc.storagePath);
+            const [exists] = await gcsFile.exists();
+            if (!exists) {
+                throw new Error(`File not found in GCS: ${doc.storagePath}`);
+            }
+            // 2. Compute SHA-256 content hash
+            const [fileBuffer] = await gcsFile.download();
+            const contentHash = createHash("sha256")
+                .update(fileBuffer)
+                .digest("hex");
+            // 3. Dedup check — same hash within org
+            const existing = await this.db.document.findFirst({
+                where: {
+                    contentHash,
+                    id: { not: doc.id },
+                    status: "ready",
+                },
+            });
+            if (existing) {
+                await this.db.document.update({
+                    where: { id: doc.id },
+                    data: {
+                        contentHash,
+                        status: "ready",
+                        nodeId: existing.nodeId,
+                    },
+                });
+                return;
+            }
+            // 4. Generate signed read URL for OCR
+            const [readUrl] = await gcsFile.getSignedUrl({
+                version: "v4",
+                action: "read",
+                expires: Date.now() + SIGNED_URL_EXPIRY_MS,
+            });
+            // 5. OCR
+            const ocrResult = await this.ocrService.extractText(readUrl, this.tenantCtx.organizationId);
+            // Count pages from --- separators
+            const pageCount = ocrResult.text.split("\n---\n").length;
+            // 6. Create node with extracted text
+            const node = await this.nodeService.create({
+                type: "document",
+                title: doc.fileName,
+                content: ocrResult.text,
+                createdBy: doc.createdBy,
+                status: "approved",
+                metadata: {
+                    sourceDocumentId: doc.id,
+                    ocrModel: ocrResult.model,
+                    mimeType: doc.mimeType,
+                },
+            });
+            // 7. Chunk the node
+            try {
+                await this.chunkService.chunkAndEmbed(node.id, doc.fileName, ocrResult.text);
+            }
+            catch (err) {
+                console.error(`Chunking failed for document ${doc.id}:`, err);
+                // Non-fatal — the node still exists
+            }
+            // 8. Apply provenance tags
+            try {
+                await this.tagService.applyTags(node.id, [
+                    {
+                        tagName: "source_document",
+                        valueText: doc.id,
+                        attrs: {
+                            fileName: doc.fileName,
+                            mimeType: doc.mimeType,
+                            storagePath: doc.storagePath,
+                        },
+                    },
+                    {
+                        tagName: "source_file_format",
+                        valueText: MIME_TO_EXT[doc.mimeType] ?? "unknown",
+                    },
+                ], "document-service");
+            }
+            catch (err) {
+                console.error(`Tagging failed for document ${doc.id}:`, err);
+                // Non-fatal — the node still exists
+            }
+            // 9. Update document record
+            await this.db.document.update({
+                where: { id: doc.id },
+                data: {
+                    contentHash,
+                    status: "ready",
+                    ocrModel: ocrResult.model,
+                    ocrTokensIn: ocrResult.tokensIn,
+                    ocrTokensOut: ocrResult.tokensOut,
+                    pageCount,
+                    nodeId: node.id,
+                },
+            });
+        }
+        catch (err) {
+            // Mark as failed
+            const errorMessage = err instanceof Error ? err.message : String(err);
+            await this.db.document.update({
+                where: { id: doc.id },
+                data: {
+                    status: "failed",
+                    errorMessage,
+                },
+            });
+            throw err;
+        }
+    }
+    /**
+     * List documents, optionally filtered by IDs or status.
+     */
+    async listDocuments(opts) {
+        const where = {};
+        if (opts?.ids?.length) {
+            where.id = { in: opts.ids };
+        }
+        if (opts?.status) {
+            where.status = opts.status;
+        }
+        return this.db.document.findMany({
+            where,
+            orderBy: { createdAt: "desc" },
+        });
+    }
+    /**
+     * Generate a signed download URL for a document's original file.
+     */
+    async getDownloadUrl(documentId) {
+        const doc = await this.db.document.findFirst({
+            where: { id: documentId },
+        });
+        if (!doc) {
+            throw new Error(`Document ${documentId} not found`);
+        }
+        const bucket = this.storage.bucket(this.bucketName);
+        const gcsFile = bucket.file(doc.storagePath);
+        const [downloadUrl] = await gcsFile.getSignedUrl({
+            version: "v4",
+            action: "read",
+            expires: Date.now() + SIGNED_URL_EXPIRY_MS,
+            responseDisposition: `attachment; filename="${doc.fileName}"`,
+        });
+        return {
+            downloadUrl,
+            fileName: doc.fileName,
+            mimeType: doc.mimeType,
+            expiresIn: SIGNED_URL_EXPIRY_MS / 1000,
+        };
+    }
+}
+//# sourceMappingURL=document-service.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"document-service.js","sourceRoot":"","sources":["../../src/services/document-service.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAGzC,OAAO,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAwCvD,MAAM,oBAAoB,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,aAAa;AAC1D,MAAM,WAAW,GAA2B;IAC1C,iBAAiB,EAAE,KAAK;IACxB,yEAAyE,EAAE,MAAM;IACjF,2EAA2E,EAAE,MAAM;IACnF,WAAW,EAAE,KAAK;IAClB,YAAY,EAAE,MAAM;IACpB,YAAY,EAAE,MAAM;CACrB,CAAC;AAEF,MAAM,OAAO,eAAe;IAIhB;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IAXF,SAAS,CAAgB;IAEjC,YACU,EAAgB,EAChB,OAAmB,EACnB,UAAkB,EAClB,UAAsB,EACtB,WAAwB,EACxB,YAA0B,EAC1B,UAAsB,EACtB,aAAoC,EACpC,UAAiC,EACzC,SAAyB;QATjB,OAAE,GAAF,EAAE,CAAc;QAChB,YAAO,GAAP,OAAO,CAAY;QACnB,eAAU,GAAV,UAAU,CAAQ;QAClB,eAAU,GAAV,UAAU,CAAY;QACtB,gBAAW,GAAX,WAAW,CAAa;QACxB,iBAAY,GAAZ,YAAY,CAAc;QAC1B,eAAU,GAAV,UAAU,CAAY;QACtB,kBAAa,GAAb,aAAa,CAAuB;QACpC,eAAU,GAAV,UAAU,CAAuB;QAGzC,IAAI,CAAC,SAAS,GAAG,SAAS,IAAI,EAAE,cAAc,EAAE,oBAAoB,EAAE,CAAC;IACzE,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,gBAAgB,CACpB,KAA8B,EAC9B,SAAiB;QAEjB,MAAM,OAAO,GAA2B,EAAE,CAAC;QAE3C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,GAAG,GAAG,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,KAAK,CAAC;YAChD,6CAA6C;YAC7C,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC;gBACxC,IAAI,EAAE;oBACJ,QAAQ,EAAE,IAAI,CAAC,QAAQ;oBACvB,QAAQ,EAAE,IAAI,CAAC,QAAQ;oBACvB,aAAa,EAAE,IAAI,CAAC,SAAS;oBAC7B,WAAW,EAAE,EAAE,EAAE,oBAAoB;oBACrC,MAAM,EAAE,WAAW;oBACnB,SAAS;oBACT,cAAc,EAAE,IAAI,CAAC,SAAS,CAAC,cAAc;iBAC9C;aACF,CAAC,CAAC;YAEH,MAAM,WAAW,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,cAAc,IAAI,GAAG,CAAC,EAAE,aAAa,GAAG,EAAE,CAAC;YAEjF,oCAAoC;YACpC,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC;gBAC5B,KAAK,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE;gBACrB,IAAI,EAAE,EAAE,WAAW,EAAE;aACtB,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACpD,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACzC,MAAM,CAAC,SAAS,CAAC,GAAG,MAAM,OAAO,CAAC,YAAY,CAAC;gBAC7C,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,OAAO;gBACf,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,oBAAoB;gBAC1C,WAAW,EAAE,IAAI,CAAC,QAAQ;aAC3B,CAAC,CAAC;YAEH,OAAO,CAAC,IAAI,CAAC;gBACX,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,SAAS;gBACT,QAAQ,EAAE,IAAI,CAAC,QAAQ;aACxB,CAAC,CAAC;QACL,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,gBAAgB,CACpB,WAAqB;QAErB,MAAM,QAAQ,GAAa,EAAE,CAAC;QAC9B,MAAM,QAAQ,GAAqC,EAAE,CAAC;QAEtD,KAAK,MAAM,EAAE,IAAI,WAAW,EAAE,CAAC;YAC7B,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;gBAC3C,KAAK,EAAE,EAAE,EAAE,EAAE;aACd,CAAC,CAAC;YAEH,IAAI,CAAC,GAAG,EAAE,CAAC;gBACT,QAAQ,CAAC,IAAI,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;gBAC1C,SAAS;YACX,CAAC;YAED,IAAI,GAAG,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;gBAC/B,QAAQ,CAAC,IAAI,CAAC,YAAY,EAAE,kBAAkB,GAAG,CAAC,MAAM,yBAAyB,CAAC,CAAC;gBACnF,QAAQ,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;gBAClD,SAAS;YACX,CAAC;YAED,2BAA2B;YAC3B,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC;gBAC5B,KAAK,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE;gBACrB,IAAI,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE;aAC/B,CAAC,CAAC;YAEH,QAAQ,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC,CAAC;YAEpD,yCAAyC;YACzC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBACzC,OAAO,CAAC,KAAK,CAAC,kCAAkC,GAAG,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YAClE,CAAC,CAAC,CAAC;QACL,CAAC;QAED,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC;IAC3C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe,CAAC,UAAkB;QACtC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;YAC3C,KAAK,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE;SAC1B,CAAC,CAAC;QAEH,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,MAAM,IAAI,KAAK,CAAC,YAAY,UAAU,YAAY,CAAC,CAAC;QACtD,CAAC;QAED,IAAI,CAAC;YACH,+BAA+B;YAC/B,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACpD,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,GAAG,MAAM,OAAO,CAAC,MAAM,EAAE,CAAC;YACxC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,IAAI,KAAK,CAAC,0BAA0B,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC;YAC/D,CAAC;YAED,kCAAkC;YAClC,MAAM,CAAC,UAAU,CAAC,GAAG,MAAM,OAAO,CAAC,QAAQ,EAAE,CAAC;YAC9C,MAAM,WAAW,GAAG,UAAU,CAAC,QAAQ,CAAC;iBACrC,MAAM,CAAC,UAAU,CAAC;iBAClB,MAAM,CAAC,KAAK,CAAC,CAAC;YAEjB,wCAAwC;YACxC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;gBAChD,KAAK,EAAE;oBACL,WAAW;oBACX,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,EAAE;oBACnB,MAAM,EAAE,OAAO;iBAChB;aACF,CAAC,CAAC;YAEH,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC;oBAC5B,KAAK,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE;oBACrB,IAAI,EAAE;wBACJ,WAAW;wBACX,MAAM,EAAE,OAAO;wBACf,MAAM,EAAE,QAAQ,CAAC,MAAM;qBACxB;iBACF,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;YAED,sCAAsC;YACtC,MAAM,CAAC,OAAO,CAAC,GAAG,MAAM,OAAO,CAAC,YAAY,CAAC;gBAC3C,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,oBAAoB;aAC3C,CAAC,CAAC;YAEH,SAAS;YACT,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,WAAW,CACjD,OAAO,EACP,IAAI,CAAC,SAAS,CAAC,cAAc,CAC9B,CAAC;YAEF,kCAAkC;YAClC,MAAM,SAAS,GAAG,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC;YAEzD,qCAAqC;YACrC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;gBACzC,IAAI,EAAE,UAAU;gBAChB,KAAK,EAAE,GAAG,CAAC,QAAQ;gBACnB,OAAO,EAAE,SAAS,CAAC,IAAI;gBACvB,SAAS,EAAE,GAAG,CAAC,SAAS;gBACxB,MAAM,EAAE,UAAU;gBAClB,QAAQ,EAAE;oBACR,gBAAgB,EAAE,GAAG,CAAC,EAAE;oBACxB,QAAQ,EAAE,SAAS,CAAC,KAAK;oBACzB,QAAQ,EAAE,GAAG,CAAC,QAAQ;iBACvB;aACF,CAAC,CAAC;YAEH,oBAAoB;YACpB,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC;YAC/E,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,CAAC,KAAK,CAAC,gCAAgC,GAAG,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;gBAC9D,oCAAoC;YACtC,CAAC;YAED,2BAA2B;YAC3B,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,UAAU,CAAC,SAAS,CAC7B,IAAI,CAAC,EAAE,EACP;oBACE;wBACE,OAAO,EAAE,iBAAiB;wBAC1B,SAAS,EAAE,GAAG,CAAC,EAAE;wBACjB,KAAK,EAAE;4BACL,QAAQ,EAAE,GAAG,CAAC,QAAQ;4BACtB,QAAQ,EAAE,GAAG,CAAC,QAAQ;4BACtB,WAAW,EAAE,GAAG,CAAC,WAAW;yBAC7B;qBACF;oBACD;wBACE,OAAO,EAAE,oBAAoB;wBAC7B,SAAS,EAAE,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,SAAS;qBAClD;iBACF,EACD,kBAAkB,CACnB,CAAC;YACJ,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,CAAC,KAAK,CAAC,+BAA+B,GAAG,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;gBAC7D,oCAAoC;YACtC,CAAC;YAED,4BAA4B;YAC5B,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC;gBAC5B,KAAK,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE;gBACrB,IAAI,EAAE;oBACJ,WAAW;oBACX,MAAM,EAAE,OAAO;oBACf,QAAQ,EAAE,SAAS,CAAC,KAAK;oBACzB,WAAW,EAAE,SAAS,CAAC,QAAQ;oBAC/B,YAAY,EAAE,SAAS,CAAC,SAAS;oBACjC,SAAS;oBACT,MAAM,EAAE,IAAI,CAAC,EAAE;iBAChB;aACF,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,iBAAiB;YACjB,MAAM,YAAY,GAChB,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACnD,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC;gBAC5B,KAAK,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE;gBACrB,IAAI,EAAE;oBACJ,MAAM,EAAE,QAAQ;oBAChB,YAAY;iBACb;aACF,CAAC,CAAC;YACH,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,aAAa,CAAC,IAGnB;QACC,MAAM,KAAK,GAA4B,EAAE,CAAC;QAC1C,IAAI,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC;YACtB,KAAK,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC;QAC9B,CAAC;QACD,IAAI,IAAI,EAAE,MAAM,EAAE,CAAC;YACjB,KAAK,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC7B,CAAC;QACD,OAAO,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAC/B,KAAK;YACL,OAAO,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE;SAC/B,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAClB,UAAkB;QAOlB,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;YAC3C,KAAK,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE;SAC1B,CAAC,CAAC;QAEH,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,MAAM,IAAI,KAAK,CAAC,YAAY,UAAU,YAAY,CAAC,CAAC;QACtD,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACpD,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QAE7C,MAAM,CAAC,WAAW,CAAC,GAAG,MAAM,OAAO,CAAC,YAAY,CAAC;YAC/C,OAAO,EAAE,IAAI;YACb,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,oBAAoB;YAC1C,mBAAmB,EAAE,yBAAyB,GAAG,CAAC,QAAQ,GAAG;SAC9D,CAAC,CAAC;QAEH,OAAO;YACL,WAAW;YACX,QAAQ,EAAE,GAAG,CAAC,QAAQ;YACtB,QAAQ,EAAE,GAAG,CAAC,QAAQ;YACtB,SAAS,EAAE,oBAAoB,GAAG,IAAI;SACvC,CAAC;IACJ,CAAC;CACF"}

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

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

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

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

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"}