@layer-ai/core 0.5.1 → 0.5.2

package/dist/services/providers/mistral-adapter.d.ts

@@ -8,5 +8,6 @@ export declare class MistralAdapter extends BaseProviderAdapter {
     call(request: LayerRequest): Promise<LayerResponse>;
     private handleChat;
     private handleEmbeddings;
+    private handleOCR;
 }
 //# sourceMappingURL=mistral-adapter.d.ts.map

package/dist/services/providers/mistral-adapter.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"mistral-adapter.d.ts","sourceRoot":"","sources":["../../../src/services/providers/mistral-adapter.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EACL,YAAY,EACZ,aAAa,EACb,IAAI,EACJ,YAAY,EAEb,MAAM,eAAe,CAAC;AAavB,qBAAa,cAAe,SAAQ,mBAAmB;IACrD,SAAS,CAAC,QAAQ,SAAa;IAE/B,SAAS,CAAC,YAAY,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAQ1C;IAGF,SAAS,CAAC,oBAAoB,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAM1D;IAEF,SAAS,CAAC,kBAAkB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAIlD;IAEI,IAAI,CAAC,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC,aAAa,CAAC;YAmB3C,UAAU;YA8LV,gBAAgB;CAkH/B"}
+{"version":3,"file":"mistral-adapter.d.ts","sourceRoot":"","sources":["../../../src/services/providers/mistral-adapter.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EACL,YAAY,EACZ,aAAa,EACb,IAAI,EACJ,YAAY,EAEb,MAAM,eAAe,CAAC;AAavB,qBAAa,cAAe,SAAQ,mBAAmB;IACrD,SAAS,CAAC,QAAQ,SAAa;IAE/B,SAAS,CAAC,YAAY,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAQ1C;IAGF,SAAS,CAAC,oBAAoB,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAM1D;IAEF,SAAS,CAAC,kBAAkB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAIlD;IAEI,IAAI,CAAC,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC,aAAa,CAAC;YAmB3C,UAAU;YA8LV,gBAAgB;YAuChB,SAAS;CA2ExB"}

package/dist/services/providers/mistral-adapter.js

@@ -36,73 +36,6 @@ export class MistralAdapter extends BaseProviderAdapter {
             none: 'none',
             required: 'any',
         };
-        // private async handleOCR(
-        //   request: Extract<LayerRequest, { type: 'ocr' }>
-        // ): Promise<LayerResponse> {
-        //   const startTime = Date.now();
-        //   const mistral = getMistralClient();
-        //   const { data: ocr, model } = request;
-        //   const ocrModel = model || 'mistral-ocr-latest';
-        //   let document: { type: string; documentUrl?: string; imageUrl?: string };
-        //   if (ocr.documentUrl) {
-        //     document = {
-        //       type: 'document_url',
-        //       documentUrl: ocr.documentUrl,
-        //     };
-        //   } else if (ocr.imageUrl) {
-        //     document = {
-        //       type: 'image_url',
-        //       imageUrl: ocr.imageUrl,
-        //     };
-        //   } else if (ocr.base64) {
-        //     const mimeType = ocr.mimeType || 'application/pdf';
-        //     const isImage = mimeType.startsWith('image/');
-        //     if (isImage) {
-        //       document = {
-        //         type: 'image_url',
-        //         imageUrl: `data:${mimeType};base64,${ocr.base64}`,
-        //       };
-        //     } else {
-        //       document = {
-        //         type: 'document_url',
-        //         documentUrl: `data:${mimeType};base64,${ocr.base64}`,
-        //       };
-        //     }
-        //   } else {
-        //     throw new Error('OCR requires either documentUrl, imageUrl, or base64 input');
-        //   }
-        //   const response = await (mistral as any).ocr.process({
-        //     model: ocrModel,
-        //     document,
-        //     ...(ocr.tableFormat && { tableFormat: ocr.tableFormat }),
-        //     ...(ocr.includeImageBase64 !== undefined && { includeImageBase64: ocr.includeImageBase64 }),
-        //     ...(ocr.extractHeader !== undefined && { extractHeader: ocr.extractHeader }),
-        //     ...(ocr.extractFooter !== undefined && { extractFooter: ocr.extractFooter }),
-        //   });
-        //   const pages = response.pages?.map((page: any) => ({
-        //     index: page.index,
-        //     markdown: page.markdown,
-        //     images: page.images,
-        //     tables: page.tables,
-        //     hyperlinks: page.hyperlinks,
-        //     header: page.header,
-        //     footer: page.footer,
-        //     dimensions: page.dimensions,
-        //   })) || [];
-        //   const combinedMarkdown = pages.map((p: any) => p.markdown).join('\n\n---\n\n');
-        //   return {
-        //     content: combinedMarkdown,
-        //     ocr: {
-        //       pages,
-        //       model: response.model || ocrModel,
-        //       documentAnnotation: response.documentAnnotation,
-        //       usageInfo: response.usageInfo,
-        //     },
-        //     model: response.model || ocrModel,
-        //     latencyMs: Date.now() - startTime,
-        //     raw: response,
-        //   };
-        // }
     }
     async call(request) {
         switch (request.type) {
@@ -110,8 +43,8 @@ export class MistralAdapter extends BaseProviderAdapter {
                 return this.handleChat(request);
             case 'embeddings':
                 return this.handleEmbeddings(request);
-            // case 'ocr':
-            //   return this.handleOCR(request);
+            case 'ocr':
+                return this.handleOCR(request);
             case 'image':
                 throw new Error('Image generation not supported by Mistral');
             case 'tts':
@@ -305,4 +238,73 @@ export class MistralAdapter extends BaseProviderAdapter {
             raw: response,
         };
     }
+    async handleOCR(request) {
+        const startTime = Date.now();
+        const mistral = getMistralClient();
+        const { data: ocr, model } = request;
+        const ocrModel = model || 'mistral-ocr-latest';
+        let document;
+        if (ocr.documentUrl) {
+            document = {
+                type: 'document_url',
+                documentUrl: ocr.documentUrl,
+            };
+        }
+        else if (ocr.imageUrl) {
+            document = {
+                type: 'image_url',
+                imageUrl: ocr.imageUrl,
+            };
+        }
+        else if (ocr.base64) {
+            const mimeType = ocr.mimeType || 'application/pdf';
+            const isImage = mimeType.startsWith('image/');
+            if (isImage) {
+                document = {
+                    type: 'image_url',
+                    imageUrl: `data:${mimeType};base64,${ocr.base64}`,
+                };
+            }
+            else {
+                document = {
+                    type: 'document_url',
+                    documentUrl: `data:${mimeType};base64,${ocr.base64}`,
+                };
+            }
+        }
+        else {
+            throw new Error('OCR requires either documentUrl, imageUrl, or base64 input');
+        }
+        const response = await mistral.ocr.process({
+            model: ocrModel,
+            document,
+            ...(ocr.tableFormat && { tableFormat: ocr.tableFormat }),
+            ...(ocr.includeImageBase64 !== undefined && { includeImageBase64: ocr.includeImageBase64 }),
+            ...(ocr.extractHeader !== undefined && { extractHeader: ocr.extractHeader }),
+            ...(ocr.extractFooter !== undefined && { extractFooter: ocr.extractFooter }),
+        });
+        const pages = response.pages?.map((page) => ({
+            index: page.index,
+            markdown: page.markdown,
+            images: page.images,
+            tables: page.tables,
+            hyperlinks: page.hyperlinks,
+            header: page.header,
+            footer: page.footer,
+            dimensions: page.dimensions,
+        })) || [];
+        const combinedMarkdown = pages.map((p) => p.markdown).join('\n\n---\n\n');
+        return {
+            content: combinedMarkdown,
+            ocr: {
+                pages,
+                model: response.model || ocrModel,
+                documentAnnotation: response.documentAnnotation,
+                usageInfo: response.usageInfo,
+            },
+            model: response.model || ocrModel,
+            latencyMs: Date.now() - startTime,
+            raw: response,
+        };
+    }
 }

package/dist/services/providers/tests/test-mistral-adapter.js

@@ -270,27 +270,28 @@ async function testMultiTurn() {
     console.log('Response:', response.content);
     console.log();
 }
-// async function testOCR() {
-//   console.log('--- Testing OCR Capability ---');
-//   const request: LayerRequest = {
-//     gate: 'test-gate',
-//     type: 'ocr',
-//     model: 'mistral-ocr-latest',
-//     data: {
-//       documentUrl: 'https://arxiv.org/pdf/2201.04234',
-//     },
-//   };
-//   try {
-//     const response = await adapter.call(request);
-//     console.log('OCR Response (first 500 chars):', response.content?.substring(0, 500));
-//     console.log('Model:', response.model);
-//     console.log('Pages extracted:', response.ocr?.pages?.length || 0);
-//     console.log('Latency:', response.latencyMs, 'ms');
-//   } catch (error) {
-//     console.log('Test failed:', (error as Error).message);
-//   }
-//   console.log();
-// }
+async function testOCR() {
+    console.log('--- Testing OCR Capability ---');
+    const request = {
+        gate: 'test-gate',
+        type: 'ocr',
+        model: 'mistral-ocr-latest',
+        data: {
+            documentUrl: 'https://arxiv.org/pdf/2201.04234',
+        },
+    };
+    try {
+        const response = await adapter.call(request);
+        console.log('OCR Response (first 500 chars):', response.content?.substring(0, 500));
+        console.log('Model:', response.model);
+        console.log('Pages extracted:', response.ocr?.pages?.length || 0);
+        console.log('Latency:', response.latencyMs, 'ms');
+    }
+    catch (error) {
+        console.log('Test failed:', error.message);
+    }
+    console.log();
+}
 async function testUnsupportedModality() {
     console.log('--- Testing Unsupported Modality (Image Generation) ---');
     const request = {
@@ -319,7 +320,7 @@ async function runAllTests() {
         await testToolResponse();
         await testEmbeddings();
         await testVisionCapability();
-        // await testOCR();
+        await testOCR();
         await testResponseFormat();
         await testMultiTurn();
         await testUnsupportedModality();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@layer-ai/core",
-  "version": "0.5.1",
+  "version": "0.5.2",
   "description": "Core API routes and services for Layer AI",
   "type": "module",
   "main": "./dist/index.js",