mindee 2.1.0 → 3.0.0

package/CHANGELOG.md

@@ -1,5 +1,21 @@
 # CHANGELOG
 
+## v3.0.0 - 2022-10-31
+### ¡Breaking Changes!
+* :sparkles: New PDF cut system, which allows specifying exactly which pages to keep or remove.
+* :recycle: PDF documents are no longer cut by default, use the `pageOptions` parameter in the `parse` method.
+* :sparkles: Document (endpoints) are now versioned, providing better backward-compatible support.
+* :recycle: Pass the document class instead of the response class to the `parse` method.
+* :recycle: Some methods and parameters renamed for better clarity.
+
+### Changes
+* :sparkles: New URL input source, `docFromUrl`.
+* :sparkles: Add support for expense receipts V4
+
+## v2.1.1 - 2022-10-28
+### Fixes
+* :bug: fix for sending base64 documents.
+
 ## v2.1.0 - 2022-10-19
 ### Changes
 * :sparkles: add cropper support to predict requests

package/README.md

@@ -24,8 +24,9 @@ const mindee = require("mindee");
 const mindeeClient = new mindee.Client({apiKey: "my-api-key"});
 
 // Load a file from disk and parse it
-const invoiceResponse = mindeeClient.docFromPath("/path/to/the/invoice.pdf")
-    .parse(mindee.InvoiceResponse);
+const invoiceResponse = mindeeClient
+    .docFromPath("/path/to/the/invoice.pdf")
+    .parse(mindee.InvoiceV3);
 
 // Print a brief summary of the parsed data
 invoiceResponse.then((resp) => {
@@ -42,18 +43,19 @@ const mindee = require("mindee");
 
 // Init a new client and add your document endpoint
 const mindeeClient = new mindee.Client({apiKey: "my-api-key"})
-  .addEndpoint({
-    accountName: "john",
-    documentType: "wsnine",
-});
+    .addEndpoint({
+        accountName: "john",
+        endpointName: "wsnine",
+    });
 
 // Load a file from disk and parse it
-const customResponse = mindeeClient.docFromPath("/path/to/the/wsnine.jpg")
-  .parse(mindee.CustomResponse, {docType: "wsnine"});
+const customResponse = mindeeClient
+    .docFromPath("/path/to/the/wsnine.jpg")
+    .parse(mindee.CustomV1, { endpointName: "wsnine" });
 
 // Print a brief summary of the parsed data
 customResponse.then((resp) => {
-  console.log(resp.document);
+    console.log(resp.document);
 });
 ```
 

package/package.json

@@ -1,12 +1,13 @@
 {
     "name": "mindee",
-    "version": "2.1.0",
+    "version": "3.0.0",
     "description": "Mindee Client Library for Node.js",
     "main": "src/index.js",
     "bin": "bin/mindee.js",
     "license": "MIT",
     "scripts": {
-        "build": "tsc --build && cp LICENSE README.md CHANGELOG.md ./dist",
+        "build": "tsc --build",
+        "build-for-publish": "tsc --build && cp LICENSE README.md CHANGELOG.md ./dist",
         "clean": "rm -rf ./dist",
         "test": "mocha 'tests/**/*.spec.ts'",
         "lint-fix": "eslint '**/*.ts' --fix",

package/src/api/endpoint.d.ts

@@ -1,32 +1,34 @@
-import { Input } from "../inputs";
+/// <reference types="node" />
+import { InputSource } from "../inputs";
+import { IncomingMessage } from "http";
 export declare const STANDARD_API_OWNER = "mindee";
 export declare const API_KEY_ENVVAR_NAME = "MINDEE_API_KEY";
+export interface predictResponse {
+    messageObj: IncomingMessage;
+    data: {
+        [key: string]: any;
+    };
+}
 export declare class Endpoint {
     apiKey: string;
     urlName: string;
-    keyName: string;
     owner: string;
     version: string;
     urlRoot: string;
-    constructor(owner: string, urlName: string, version: string, apiKey: string, keyName?: string);
+    private readonly baseHeaders;
+    constructor(owner: string, urlName: string, version: string, apiKey: string);
     /**
-     * Make a prediction request.
+     * Make a request to POST a document for prediction.
+     * @param input
+     * @param includeWords
+     * @param cropper
      */
-    predictRequest(input: Input, includeWords?: boolean, cropper?: boolean): Promise<unknown>;
+    predictReqPost(input: InputSource, includeWords?: boolean, cropper?: boolean): Promise<predictResponse>;
     protected apiKeyFromEnv(): string;
 }
-export declare class InvoiceEndpoint extends Endpoint {
-    constructor(apiKey: string);
-}
-export declare class ReceiptEndpoint extends Endpoint {
-    constructor(apiKey: string);
-}
-export declare class PassportEndpoint extends Endpoint {
-    constructor(apiKey: string);
-}
-export declare class CropperEndpoint extends Endpoint {
-    constructor(apiKey: string);
+export declare class StandardEndpoint extends Endpoint {
+    constructor(endpointName: string, version: string, apiKey: string);
 }
 export declare class CustomEndpoint extends Endpoint {
-    constructor(documentType: string, accountName: string, version: string, apiKey: string);
+    constructor(endpointName: string, accountName: string, version: string, apiKey: string);
 }

package/src/api/endpoint.js

@@ -26,7 +26,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.CustomEndpoint = exports.CropperEndpoint = exports.PassportEndpoint = exports.ReceiptEndpoint = exports.InvoiceEndpoint = exports.Endpoint = exports.API_KEY_ENVVAR_NAME = exports.STANDARD_API_OWNER = void 0;
+exports.CustomEndpoint = exports.StandardEndpoint = exports.Endpoint = exports.API_KEY_ENVVAR_NAME = exports.STANDARD_API_OWNER = void 0;
 const https = __importStar(require("https"));
 const os = __importStar(require("os"));
 const package_json_1 = require("../../package.json");
@@ -38,42 +38,47 @@ const USER_AGENT = `mindee-api-nodejs@v${package_json_1.version} nodejs-${proces
 exports.STANDARD_API_OWNER = "mindee";
 exports.API_KEY_ENVVAR_NAME = "MINDEE_API_KEY";
 class Endpoint {
-    constructor(owner, urlName, version, apiKey, keyName) {
+    constructor(owner, urlName, version, apiKey) {
         this.owner = owner;
         this.urlName = urlName;
         this.version = version;
-        this.keyName = keyName || urlName;
         this.apiKey = apiKey || this.apiKeyFromEnv();
         this.urlRoot = `${MINDEE_API_URL}/products/${owner}/${urlName}/v${version}`;
+        this.baseHeaders = {
+            "User-Agent": USER_AGENT,
+            Authorization: `Token ${this.apiKey}`,
+        };
     }
     /**
-     * Make a prediction request.
+     * Make a request to POST a document for prediction.
+     * @param input
+     * @param includeWords
+     * @param cropper
      */
-    predictRequest(input, includeWords = false, cropper = false) {
+    predictReqPost(input, includeWords = false, cropper = false) {
         return new Promise((resolve, reject) => {
-            const form = new form_data_1.default();
-            let body;
-            let headers = {
-                "User-Agent": USER_AGENT,
-                Authorization: `Token ${this.apiKey}`,
-            };
             const uri = new url_1.URL(`${this.urlRoot}/predict`);
-            const fileParams = { filename: input.filename };
-            form.append("document", input.fileObject, fileParams);
-            if (includeWords) {
-                form.append("include_mvision", "true");
-            }
             if (cropper) {
                 uri.searchParams.append("cropper", "true");
             }
-            headers = { ...headers, ...form.getHeaders() };
+            logger_1.logger.debug(`Request URI: ${uri}`);
+            const form = new form_data_1.default();
+            if (input.fileObject instanceof Buffer) {
+                form.append("document", input.fileObject, { filename: input.filename });
+            }
+            else {
+                form.append("document", input.fileObject);
+            }
+            if (includeWords) {
+                form.append("include_mvision", "true");
+            }
+            const headers = { ...this.baseHeaders, ...form.getHeaders() };
             const options = {
                 method: "POST",
                 headers: headers,
                 hostname: uri.hostname,
                 path: `${uri.pathname}${uri.search}`,
             };
-            logger_1.logger.debug(`Prediction request: ${uri}`);
             const req = https.request(options, function (res) {
                 // when the encoding is set, data chunks will be strings
                 res.setEncoding("utf-8");
@@ -84,7 +89,7 @@ class Endpoint {
                 res.on("end", function () {
                     try {
                         resolve({
-                            ...res,
+                            messageObj: res,
                             data: JSON.parse(responseBody),
                         });
                     }
@@ -93,55 +98,31 @@ class Endpoint {
                     }
                 });
             });
+            form.pipe(req);
             req.on("error", (err) => {
                 reject(err);
             });
-            if (["path", "stream"].includes(input.inputType)) {
-                form.pipe(req);
-            }
-            if (input.inputType === "base64") {
-                req.write(body);
-                req.end();
-            }
         });
     }
     apiKeyFromEnv() {
         const envVarValue = process.env[exports.API_KEY_ENVVAR_NAME];
         if (envVarValue) {
-            logger_1.logger.debug(`Set '${this.keyName}' API key from environment: ${exports.API_KEY_ENVVAR_NAME}`);
+            logger_1.logger.debug(`Set ${this.urlName} v${this.version} API key from environment: ${exports.API_KEY_ENVVAR_NAME}`);
             return envVarValue;
         }
         return "";
     }
 }
 exports.Endpoint = Endpoint;
-class InvoiceEndpoint extends Endpoint {
-    constructor(apiKey) {
-        super(exports.STANDARD_API_OWNER, "invoices", "3", apiKey, "invoice");
-    }
-}
-exports.InvoiceEndpoint = InvoiceEndpoint;
-class ReceiptEndpoint extends Endpoint {
-    constructor(apiKey) {
-        super(exports.STANDARD_API_OWNER, "expense_receipts", "3", apiKey, "receipt");
-    }
-}
-exports.ReceiptEndpoint = ReceiptEndpoint;
-class PassportEndpoint extends Endpoint {
-    constructor(apiKey) {
-        super(exports.STANDARD_API_OWNER, "passport", "1", apiKey);
-    }
-}
-exports.PassportEndpoint = PassportEndpoint;
-class CropperEndpoint extends Endpoint {
-    constructor(apiKey) {
-        super(exports.STANDARD_API_OWNER, "cropper", "1", apiKey);
+class StandardEndpoint extends Endpoint {
+    constructor(endpointName, version, apiKey) {
+        super(exports.STANDARD_API_OWNER, endpointName, version, apiKey);
     }
 }
-exports.CropperEndpoint = CropperEndpoint;
+exports.StandardEndpoint = StandardEndpoint;
 class CustomEndpoint extends Endpoint {
-    constructor(documentType, accountName, version, apiKey) {
-        super(accountName, documentType, version, apiKey);
+    constructor(endpointName, accountName, version, apiKey) {
+        super(accountName, endpointName, version, apiKey);
     }
 }
 exports.CustomEndpoint = CustomEndpoint;

package/src/api/index.d.ts

@@ -1,2 +1,2 @@
-export { Response, InvoiceResponse, ReceiptResponse, FinancialDocResponse, PassportResponse, CustomResponse, CropperResponse, } from "./response";
-export { Endpoint, InvoiceEndpoint, ReceiptEndpoint, PassportEndpoint, CustomEndpoint, CropperEndpoint, STANDARD_API_OWNER, API_KEY_ENVVAR_NAME, } from "./endpoint";
+export { Response } from "./response";
+export { Endpoint, CustomEndpoint, StandardEndpoint, predictResponse, STANDARD_API_OWNER, API_KEY_ENVVAR_NAME, } from "./endpoint";

package/src/api/index.js

@@ -1,20 +1,11 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.API_KEY_ENVVAR_NAME = exports.STANDARD_API_OWNER = exports.CropperEndpoint = exports.CustomEndpoint = exports.PassportEndpoint = exports.ReceiptEndpoint = exports.InvoiceEndpoint = exports.Endpoint = exports.CropperResponse = exports.CustomResponse = exports.PassportResponse = exports.FinancialDocResponse = exports.ReceiptResponse = exports.InvoiceResponse = exports.Response = void 0;
+exports.API_KEY_ENVVAR_NAME = exports.STANDARD_API_OWNER = exports.StandardEndpoint = exports.CustomEndpoint = exports.Endpoint = exports.Response = void 0;
 var response_1 = require("./response");
 Object.defineProperty(exports, "Response", { enumerable: true, get: function () { return response_1.Response; } });
-Object.defineProperty(exports, "InvoiceResponse", { enumerable: true, get: function () { return response_1.InvoiceResponse; } });
-Object.defineProperty(exports, "ReceiptResponse", { enumerable: true, get: function () { return response_1.ReceiptResponse; } });
-Object.defineProperty(exports, "FinancialDocResponse", { enumerable: true, get: function () { return response_1.FinancialDocResponse; } });
-Object.defineProperty(exports, "PassportResponse", { enumerable: true, get: function () { return response_1.PassportResponse; } });
-Object.defineProperty(exports, "CustomResponse", { enumerable: true, get: function () { return response_1.CustomResponse; } });
-Object.defineProperty(exports, "CropperResponse", { enumerable: true, get: function () { return response_1.CropperResponse; } });
 var endpoint_1 = require("./endpoint");
 Object.defineProperty(exports, "Endpoint", { enumerable: true, get: function () { return endpoint_1.Endpoint; } });
-Object.defineProperty(exports, "InvoiceEndpoint", { enumerable: true, get: function () { return endpoint_1.InvoiceEndpoint; } });
-Object.defineProperty(exports, "ReceiptEndpoint", { enumerable: true, get: function () { return endpoint_1.ReceiptEndpoint; } });
-Object.defineProperty(exports, "PassportEndpoint", { enumerable: true, get: function () { return endpoint_1.PassportEndpoint; } });
 Object.defineProperty(exports, "CustomEndpoint", { enumerable: true, get: function () { return endpoint_1.CustomEndpoint; } });
-Object.defineProperty(exports, "CropperEndpoint", { enumerable: true, get: function () { return endpoint_1.CropperEndpoint; } });
+Object.defineProperty(exports, "StandardEndpoint", { enumerable: true, get: function () { return endpoint_1.StandardEndpoint; } });
 Object.defineProperty(exports, "STANDARD_API_OWNER", { enumerable: true, get: function () { return endpoint_1.STANDARD_API_OWNER; } });
 Object.defineProperty(exports, "API_KEY_ENVVAR_NAME", { enumerable: true, get: function () { return endpoint_1.API_KEY_ENVVAR_NAME; } });

package/src/api/response.d.ts

@@ -1,10 +1,10 @@
-import { Document, Passport, Receipt, Invoice, FinancialDocument, CustomDocument, DocumentConstructorProps, Cropper } from "../documents";
+import { Document, DocumentSig } from "../documents";
 import { FullText } from "../fields";
-import { Input } from "../inputs";
+import { InputSource } from "../inputs";
 export interface ResponseProps {
     httpResponse: any;
-    documentType: string;
-    input: Input;
+    documentType?: string;
+    input: InputSource;
     error: boolean;
 }
 /**
@@ -12,44 +12,11 @@ export interface ResponseProps {
  */
 export declare class Response<DocType extends Document> {
     httpResponse: any;
-    readonly documentType: string;
-    inputFile: Input;
+    inputFile: InputSource;
     document?: DocType;
     pages: Array<DocType>;
-    constructor(params: ResponseProps);
+    readonly documentClass: DocumentSig<DocType>;
+    constructor(documentClass: DocumentSig<DocType>, params: ResponseProps);
+    protected formatResponse(documentType?: string): void;
     protected getPageText(httpDataDocument: any, pageId: number): FullText;
 }
-/**
- * Class for all constructed (API Builder) endpoint responses.
- */
-export declare class CustomResponse extends Response<CustomDocument> {
-    constructor(params: ResponseProps);
-    protected formatResponse(): void;
-}
-declare type StandardDocumentSig<DocType extends Document> = {
-    new ({ prediction, inputFile, pageId, fullText, }: DocumentConstructorProps): DocType;
-};
-/**
- * Generic class for all standard (Off-the-Shelf) endpoint responses.
- */
-export declare class StandardProductResponse<DocType extends Document> extends Response<DocType> {
-    documentClass: StandardDocumentSig<DocType>;
-    constructor(documentClass: StandardDocumentSig<DocType>, params: ResponseProps);
-    protected formatResponse(): void;
-}
-export declare class InvoiceResponse extends StandardProductResponse<Invoice> {
-    constructor(params: ResponseProps);
-}
-export declare class ReceiptResponse extends StandardProductResponse<Receipt> {
-    constructor(params: ResponseProps);
-}
-export declare class FinancialDocResponse extends StandardProductResponse<FinancialDocument> {
-    constructor(params: ResponseProps);
-}
-export declare class PassportResponse extends StandardProductResponse<Passport> {
-    constructor(params: ResponseProps);
-}
-export declare class CropperResponse extends StandardProductResponse<Cropper> {
-    constructor(params: ResponseProps);
-}
-export {};

package/src/api/response.js

@@ -1,79 +1,28 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.CropperResponse = exports.PassportResponse = exports.FinancialDocResponse = exports.ReceiptResponse = exports.InvoiceResponse = exports.StandardProductResponse = exports.CustomResponse = exports.Response = void 0;
-const documents_1 = require("../documents");
+exports.Response = void 0;
 const fields_1 = require("../fields");
 /**
  * Base class for all responses.
  */
 class Response {
-    constructor(params) {
+    constructor(documentClass, params) {
         this.pages = [];
+        this.documentClass = documentClass;
         this.httpResponse = params.httpResponse;
-        this.documentType = params.documentType;
         this.inputFile = params.input;
-    }
-    getPageText(httpDataDocument, pageId) {
-        const pageText = new fields_1.FullText();
-        if ("ocr" in httpDataDocument &&
-            Object.keys(httpDataDocument.ocr).length > 0) {
-            pageText.words =
-                httpDataDocument.ocr["mvision-v1"].pages[pageId].all_words;
-        }
-        return pageText;
-    }
-}
-exports.Response = Response;
-/**
- * Class for all constructed (API Builder) endpoint responses.
- */
-class CustomResponse extends Response {
-    constructor(params) {
-        super(params);
         if (!params.error) {
-            this.formatResponse();
+            this.formatResponse(params.documentType);
         }
     }
-    formatResponse() {
-        const httpDataDocument = this.httpResponse.data.document;
-        httpDataDocument.inference.pages.forEach((apiPage) => {
-            this.pages.push(new documents_1.CustomDocument({
-                prediction: apiPage.prediction,
-                orientation: apiPage.orientation,
-                extras: apiPage.extras,
-                inputFile: this.inputFile,
-                pageId: apiPage.id,
-                documentType: this.documentType,
-            }));
-        });
-        this.document = new documents_1.CustomDocument({
-            prediction: httpDataDocument.inference.prediction,
-            inputFile: this.inputFile,
-            documentType: this.documentType,
-            orientation: {},
-            extras: {},
-        });
-    }
-}
-exports.CustomResponse = CustomResponse;
-/**
- * Generic class for all standard (Off-the-Shelf) endpoint responses.
- */
-class StandardProductResponse extends Response {
-    constructor(documentClass, params) {
-        super(params);
-        this.documentClass = documentClass;
-        if (!params.error) {
-            this.formatResponse();
-        }
-    }
-    formatResponse() {
+    formatResponse(documentType) {
         const httpDataDocument = this.httpResponse.data.document;
         httpDataDocument.inference.pages.forEach((apiPage) => {
             const pageText = this.getPageText(httpDataDocument, apiPage.id);
             this.pages.push(new this.documentClass({
+                documentType: documentType,
                 prediction: apiPage.prediction,
-                inputFile: this.inputFile,
+                inputSource: this.inputFile,
                 pageId: apiPage.id,
                 orientation: apiPage.orientation,
                 extras: apiPage.extras,
@@ -81,41 +30,21 @@ class StandardProductResponse extends Response {
             }));
         });
         this.document = new this.documentClass({
+            documentType: documentType,
             prediction: httpDataDocument.inference.prediction,
-            inputFile: this.inputFile,
+            inputSource: this.inputFile,
             orientation: {},
             extras: {},
         });
     }
-}
-exports.StandardProductResponse = StandardProductResponse;
-class InvoiceResponse extends StandardProductResponse {
-    constructor(params) {
-        super(documents_1.Invoice, params);
-    }
-}
-exports.InvoiceResponse = InvoiceResponse;
-class ReceiptResponse extends StandardProductResponse {
-    constructor(params) {
-        super(documents_1.Receipt, params);
-    }
-}
-exports.ReceiptResponse = ReceiptResponse;
-class FinancialDocResponse extends StandardProductResponse {
-    constructor(params) {
-        super(documents_1.FinancialDocument, params);
-    }
-}
-exports.FinancialDocResponse = FinancialDocResponse;
-class PassportResponse extends StandardProductResponse {
-    constructor(params) {
-        super(documents_1.Passport, params);
-    }
-}
-exports.PassportResponse = PassportResponse;
-class CropperResponse extends StandardProductResponse {
-    constructor(params) {
-        super(documents_1.Cropper, params);
+    getPageText(httpDataDocument, pageId) {
+        const pageText = new fields_1.FullText();
+        if ("ocr" in httpDataDocument &&
+            Object.keys(httpDataDocument.ocr).length > 0) {
+            pageText.words =
+                httpDataDocument.ocr["mvision-v1"].pages[pageId].all_words;
+        }
+        return pageText;
     }
 }
-exports.CropperResponse = CropperResponse;
+exports.Response = Response;

package/src/cli.js

@@ -5,7 +5,6 @@ const commander_1 = require("commander");
 const documents_1 = require("./documents");
 const api_1 = require("./api");
 const client_1 = require("./client");
-const constants_1 = require("./constants");
 const program = new commander_1.Command();
 const COMMAND_INVOICE = "invoice";
 const COMMAND_RECEIPT = "receipt";
@@ -13,11 +12,46 @@ const COMMAND_PASSPORT = "passport";
 const COMMAND_FINANCIAL = "financial";
 const COMMAND_CUSTOM = "custom";
 const CLI_COMMAND_CONFIG = new Map([
-    [COMMAND_INVOICE, constants_1.ProductConfigs.getByDocType(documents_1.DOC_TYPE_INVOICE)],
-    [COMMAND_RECEIPT, constants_1.ProductConfigs.getByDocType(documents_1.DOC_TYPE_RECEIPT)],
-    [COMMAND_PASSPORT, constants_1.ProductConfigs.getByDocType(documents_1.DOC_TYPE_PASSPORT)],
-    [COMMAND_FINANCIAL, constants_1.ProductConfigs.getByDocType(documents_1.DOC_TYPE_FINANCIAL)],
-    [COMMAND_CUSTOM, constants_1.ProductConfigs.getByDocType(documents_1.DOC_TYPE_CUSTOM)],
+    [
+        COMMAND_INVOICE,
+        {
+            description: "Invoice V3",
+            docType: documents_1.DOC_TYPE_INVOICE_V3,
+            fullText: true,
+        },
+    ],
+    [
+        COMMAND_RECEIPT,
+        {
+            description: "Expense Receipt V4",
+            docType: documents_1.DOC_TYPE_RECEIPT_V4,
+            fullText: true,
+        },
+    ],
+    [
+        COMMAND_PASSPORT,
+        {
+            description: "Passport V1",
+            docType: documents_1.DOC_TYPE_PASSPORT_V1,
+            fullText: false,
+        },
+    ],
+    [
+        COMMAND_FINANCIAL,
+        {
+            description: "Financial Document V1 (receipt or invoice)",
+            docType: documents_1.DOC_TYPE_FINANCIAL_V1,
+            fullText: true,
+        },
+    ],
+    [
+        COMMAND_CUSTOM,
+        {
+            description: "A custom document",
+            docType: documents_1.DOC_TYPE_CUSTOM,
+            fullText: false,
+        },
+    ],
 ]);
 async function predictCall(command, inputPath, options) {
     const conf = CLI_COMMAND_CONFIG.get(command);
@@ -31,7 +65,7 @@ async function predictCall(command, inputPath, options) {
     if (command === COMMAND_CUSTOM) {
         mindeeClient.addEndpoint({
             accountName: options.user,
-            documentType: options.documentType,
+            endpointName: options.documentType,
         });
     }
     const doc = mindeeClient.docFromPath(inputPath);
@@ -50,19 +84,19 @@ async function predictCall(command, inputPath, options) {
     let response;
     switch (command) {
         case COMMAND_INVOICE:
-            response = await doc.parse(api_1.InvoiceResponse, predictParams);
+            response = await doc.parse(documents_1.InvoiceV3, predictParams);
             break;
         case COMMAND_RECEIPT:
-            response = await doc.parse(api_1.ReceiptResponse, predictParams);
+            response = await doc.parse(documents_1.ReceiptV4, predictParams);
             break;
         case COMMAND_FINANCIAL:
-            response = await doc.parse(api_1.FinancialDocResponse, predictParams);
+            response = await doc.parse(documents_1.FinancialDocumentV1, predictParams);
             break;
         case COMMAND_PASSPORT:
-            response = await doc.parse(api_1.PassportResponse, predictParams);
+            response = await doc.parse(documents_1.PassportV1, predictParams);
             break;
         case COMMAND_CUSTOM:
-            response = await doc.parse(api_1.CustomResponse, predictParams);
+            response = await doc.parse(documents_1.CustomV1, predictParams);
             break;
         default:
             throw `Unhandled command: ${command}`;