mindee 1.0.9 → 1.2.0

package/lib/documents/financialDocument.js

@@ -45,14 +45,19 @@ class FinancialDocument extends Document {
    *  @param {Object} Date - date value for creating FinancialDocument object from scratch
    *  @param {Object} InvoiceNumber - Invoice number value for creating FinancialDocument object from scratch
    *  @param {Object} taxes - taxes value for creating FinancialDocument object from scratch
-   *  @param {Object} merchantName - merchant name value for creating FinancialDocument object from scratch
+   *  @param {Object} supplier - supplier value for creating FinancialDocument object from scratch
+   *  @param {Object} supplierAddress - supplier address value for creating FinancialDocument object from scratch
    *  @param {Object} paymentDetails - payment details value for creating FinancialDocument object from scratch
    *  @param {Object} companyNumber - company number value for creating FinancialDocument object from scratch
    *  @param {Object} vatNumber - vat number value for creating FinancialDocument object from scratch
    *  @param {Object} orientation - orientation value for creating FinancialDocument object from scratch
    *  @param {Object} totalTax - total tax value for creating FinancialDocument object from scratch
    *  @param {Object} time - time value for creating FinancialDocument object from scratch
+   *  @param {Object} customerName - customer name value for creating FinancialDocument object from scratch
+   *  @param {Object} customerAddress - customer address value for creating FinancialDocument object from scratch
+   *  @param {Object} customerCompanyRegistration - customer company registration value for creating FinancialDocument object from scratch
    *  @param {Object} pageNumber - pageNumber for multi pages pdf input
+   *  @param {String} level - specify whether object is built from "page" level or "document" level prediction
    */
   constructor(_ref) {
     var {
@@ -65,14 +70,19 @@ class FinancialDocument extends Document {
       invoiceNumber: _invoiceNumber = undefined,
       dueDate: _dueDate = undefined,
       taxes: _taxes = undefined,
-      merchantName: _merchantName = undefined,
+      supplier: _supplier = undefined,
+      supplierAddress: _supplierAddress = undefined,
       paymentDetails: _paymentDetails = undefined,
       companyNumber: _companyNumber = undefined,
       vatNumber: _vatNumber = undefined,
       orientation: _orientation = undefined,
       totalTax: _totalTax = undefined,
       time: _time = undefined,
-      pageNumber: _pageNumber = 0
+      customerName: _customerName = undefined,
+      customerAddress: _customerAddress = undefined,
+      customerCompanyRegistration: _customerCompanyRegistration = undefined,
+      pageNumber: _pageNumber = 0,
+      level = "page"
     } = _ref;
     super(_inputFile);
 
@@ -84,6 +94,8 @@ class FinancialDocument extends Document {
 
     _initFromScratch.add(this);
 
+    this.level = level;
+
     if (_apiPrediction === undefined) {
       _classPrivateMethodGet(this, _initFromScratch, _initFromScratch2).call(this, {
         locale: _locale,
@@ -93,14 +105,18 @@ class FinancialDocument extends Document {
         invoiceNumber: _invoiceNumber,
         dueDate: _dueDate,
         taxes: _taxes,
-        merchantName: _merchantName,
+        supplier: _supplier,
+        supplierAddress: _supplierAddress,
         paymentDetails: _paymentDetails,
         companyNumber: _companyNumber,
         vatNumber: _vatNumber,
         orientation: _orientation,
         pageNumber: _pageNumber,
         totalTax: _totalTax,
-        time: _time
+        time: _time,
+        customerName: _customerName,
+        customerAddress: _customerAddress,
+        customerCompanyRegistration: _customerCompanyRegistration
       });
     } else {
       _classPrivateMethodGet(this, _initFromApiPrediction, _initFromApiPrediction2).call(this, _apiPrediction, _inputFile, _pageNumber);
@@ -110,7 +126,7 @@ class FinancialDocument extends Document {
   }
 
   toString() {
-    return "\n    -----Financial document-----\n    Filename: ".concat(this.filename, "\n    Total amount: ").concat(this.totalIncl.value, "\n    Date: ").concat(this.date.value, "\n    Merchant name: ").concat(this.merchantName.value, "\n    Total taxes: ").concat(this.totalTax.value, "\n    ");
+    return "\n    -----Financial document-----\n    Filename: ".concat(this.filename, "\n    Total amount: ").concat(this.totalIncl.value, "\n    Date: ").concat(this.date.value, "\n    Supplier: ").concat(this.supplier.value, "\n    Total taxes: ").concat(this.totalTax.value, "\n    ");
   }
 
 }
@@ -130,8 +146,12 @@ var _initFromScratch2 = function _initFromScratch2(_ref2) {
     vatNumber,
     orientation,
     pageNumber,
-    merchantName,
-    time
+    supplier,
+    supplierAddress,
+    time,
+    customerName,
+    customerAddress,
+    customerCompanyRegistration
   } = _ref2;
 
   var constructPrediction = function constructPrediction(item) {
@@ -150,13 +170,17 @@ var _initFromScratch2 = function _initFromScratch2(_ref2) {
   this.totalTax = new Amount(constructPrediction(totalTax));
   this.date = new Date(constructPrediction(date));
   this.dueDate = new Date(constructPrediction(dueDate));
-  this.merchantName = new Field(constructPrediction(merchantName));
+  this.supplier = new Field(constructPrediction(supplier));
+  this.supplierAddress = new Field(this.constructPrediction(supplierAddress));
   this.time = new Field(constructPrediction(time));
   this.orientation = new Orientation(constructPrediction(orientation));
   this.invoiceNumber = new Field(constructPrediction(invoiceNumber));
   this.paymentDetails = new Field(constructPrediction(paymentDetails));
   this.companyNumber = new Field(constructPrediction(companyNumber));
   this.vatNumber = new Field(constructPrediction(vatNumber));
+  this.customerName = new Field(this.constructPrediction(customerName));
+  this.customerAddress = new Field(this.constructPrediction(customerAddress));
+  this.customerCompanyRegistration = new Field(this.constructPrediction(customerCompanyRegistration));
 
   if (taxes !== undefined) {
     this.taxes = [];
@@ -180,58 +204,85 @@ var _initFromApiPrediction2 = function _initFromApiPrediction2(apiPrediction, in
     var invoice = new Invoice({
       apiPrediction,
       inputFile,
-      pageNumber
+      pageNumber,
+      level: this.level
     });
-    Object.assign(this, invoice);
+    this.locale = invoice.locale;
+    this.totalIncl = invoice.totalIncl;
+    this.totalExcl = invoice.totalExcl;
+    this.date = invoice.invoiceDate;
+    this.invoiceNumber = invoice.invoiceNumber;
+    this.dueDate = invoice.dueDate;
+    this.taxes = invoice.taxes;
+    this.supplier = invoice.supplier;
+    this.supplierAddress = invoice.supplierAddress;
+    this.paymentDetails = invoice.paymentDetails;
+    this.companyNumber = invoice.companyNumber;
+    this.orientation = invoice.orientation;
+    this.totalTax = invoice.totalTax;
     this.time = new Field({
       prediction: {
         value: undefined,
-        probability: 0.0
-      }
-    });
-    this.merchantName = new Field({
-      prediction: {
-        value: undefined,
-        probability: 0.0
+        confidence: 0.0
       }
     });
+    this.customerName = invoice.customerName;
+    this.customerAddress = invoice.customerAddress;
+    this.customerCompanyRegistration = invoice.customerCompanyRegistration;
   } else {
     var receipt = new Receipt({
       apiPrediction,
       inputFile,
-      pageNumber
+      pageNumber,
+      level: this.level
     });
-    Object.assign(this, receipt);
-    this.invoiceDate = new Field({
+    this.orientation = receipt.orientation;
+    this.date = receipt.date;
+    this.dueDate = receipt.date;
+    this.taxes = receipt.taxes;
+    this.locale = receipt.locale;
+    this.totalIncl = receipt.totalIncl;
+    this.totalExcl = receipt.totalExcl;
+    this.supplier = receipt.merchantName;
+    this.supplierAddress = new Field({
       prediction: {
         value: undefined,
-        probability: 0.0
+        confidence: 0.0
       }
     });
+    this.time = receipt.time;
+    this.totalTax = receipt.totalTax;
     this.invoiceNumber = new Field({
       prediction: {
         value: undefined,
-        probability: 0.0
+        confidence: 0.0
       }
     });
-    this.dueDate = new Field({
+    this.paymentDetails = new Field({
       prediction: {
         value: undefined,
-        probability: 0.0
+        confidence: 0.0
       }
     });
-    this.paymentDetails = new Field({
+    this.companyNumber = new Field({
       prediction: {
         value: undefined,
-        probability: 0.0
+        confidence: 0.0
       }
     });
-    this.companyNumber = new Field({
+    this.customerName = new Field({
+      prediction: {
+        value: undefined,
+        confidence: 0.0
+      }
+    });
+    this.customerAddress = new Field({
       prediction: {
         value: undefined,
-        probability: 0.0
+        confidence: 0.0
       }
     });
+    this.customerCompanyRegistration = [];
   }
 };
 
@@ -259,7 +310,7 @@ var _taxesMatchTotalIncl2 = function _taxesMatchTotalIncl2() {
 
   if (this.totalIncl.value * (1 - eps) - 0.02 <= reconstructedTotal && reconstructedTotal <= this.totalIncl.value * (1 + eps) + 0.02) {
     this.taxes = this.taxes.map(tax => _objectSpread(_objectSpread({}, tax), {}, {
-      probability: 1.0
+      confidence: 1.0
     }));
     this.totalTax.probability = 1.0;
     this.totalIncl.probability = 1.0;

package/lib/documents/invoice.js

@@ -50,7 +50,6 @@ class Invoice extends Document {
   /**
    *  @param {Object} apiPrediction - Json parsed prediction from HTTP response
    *  @param {Input} input - Input object
-   *  @param {Integer} pageNumber - Page number for multi pages pdf input
    *  @param {Object} locale - locale value for creating Invoice object from scratch
    *  @param {Object} totalIncl - total tax included value for creating Invoice object from scratch
    *  @param {Object} totalExcl - total tax excluded value for creating Invoice object from scratch
@@ -58,12 +57,17 @@ class Invoice extends Document {
    *  @param {Object} invoiceNumber - invoice number value for creating Invoice object from scratch
    *  @param {Object} taxes - taxes value for creating Invoice object from scratch
    *  @param {Object} supplier - supplier value for creating Invoice object from scratch
+   *  @param {Object} supplierAddress - supplier address value for creating Invoice object from scratch
    *  @param {Object} paymentDetails - payment details value for creating Invoice object from scratch
    *  @param {Object} companyNumber - company number value for creating Invoice object from scratch
    *  @param {Object} vatNumber - vat number value for creating Invoice object from scratch
    *  @param {Object} orientation - orientation value for creating Invoice object from scratch
    *  @param {Object} totalTax - total tax value for creating Invoice object from scratch
-   *  @param {Object} pageNumber - pageNumber for multi pages pdf input
+   *  @param {Object} customerName - customer name value for creating Invoice object from scratch
+   *  @param {Object} customerAddress - customer address value for creating Invoice object from scratch
+   *  @param {Object} customerCompanyRegistration - customer company registration value for creating Invoice object from scratch
+   *  @param {Number} pageNumber - pageNumber for multi pages pdf input
+   *  @param {String} level - specify whether object is built from "page" level or "document" level prediction
    */
   constructor(_ref) {
     var {
@@ -77,12 +81,17 @@ class Invoice extends Document {
       dueDate: _dueDate = undefined,
       taxes: _taxes = undefined,
       supplier: _supplier = undefined,
+      supplierAddress: _supplierAddress = undefined,
       paymentDetails: _paymentDetails = undefined,
       companyNumber: _companyNumber = undefined,
       vatNumber: _vatNumber = undefined,
       orientation: _orientation = undefined,
       totalTax: _totalTax = undefined,
-      pageNumber: _pageNumber = 0
+      customerName: _customerName = undefined,
+      customerAddress: _customerAddress = undefined,
+      customerCompanyRegistration: _customerCompanyRegistration = undefined,
+      pageNumber: _pageNumber = 0,
+      level = "page"
     } = _ref;
     super(inputFile);
 
@@ -108,6 +117,18 @@ class Invoice extends Document {
 
     _initFromScratch.add(this);
 
+    this.level = level;
+
+    this.constructPrediction = function (item) {
+      return {
+        prediction: {
+          value: item
+        },
+        valueKey: "value",
+        pageNumber: _pageNumber
+      };
+    };
+
     if (_apiPrediction === undefined) {
       _classPrivateMethodGet(this, _initFromScratch, _initFromScratch2).call(this, {
         locale: _locale,
@@ -118,12 +139,16 @@ class Invoice extends Document {
         dueDate: _dueDate,
         taxes: _taxes,
         supplier: _supplier,
+        supplierAddress: _supplierAddress,
         paymentDetails: _paymentDetails,
         companyNumber: _companyNumber,
         vatNumber: _vatNumber,
         orientation: _orientation,
         pageNumber: _pageNumber,
-        totalTax: _totalTax
+        totalTax: _totalTax,
+        customerName: _customerName,
+        customerAddress: _customerAddress,
+        customerCompanyRegistration: _customerCompanyRegistration
       });
     } else {
       _classPrivateMethodGet(this, _initFromApiPrediction, _initFromApiPrediction2).call(this, _apiPrediction, _pageNumber);
@@ -135,7 +160,7 @@ class Invoice extends Document {
   }
 
   toString() {
-    return "\n    -----Invoice data-----\n    Filename: ".concat(this.filename, "\n    Invoice number: ").concat(this.invoiceNumber.value, "\n    Total amount including taxes: ").concat(this.totalIncl.value, "\n    Total amount excluding taxes: ").concat(this.totalExcl.value, "\n    Invoice Date: ").concat(this.invoiceDate.value, "\n    Supplier name: ").concat(this.supplier.value, "\n    Taxes: ").concat(this.taxes.map(tax => tax.toString()).join(" - "), "\n    Total taxes: ").concat(this.totalTax.value, "\n    ");
+    return "\n    -----Invoice data-----\n    Filename: ".concat(this.filename, "\n    Invoice number: ").concat(this.invoiceNumber.value, "\n    Total amount including taxes: ").concat(this.totalIncl.value, "\n    Total amount excluding taxes: ").concat(this.totalExcl.value, "\n    Invoice Date: ").concat(this.invoiceDate.value, "\n    Supplier name: ").concat(this.supplier.value, "\n    Supplier address: ").concat(this.supplierAddress.value, "\n    Customer name: ").concat(this.customerName.value, "\n    Customer address: ").concat(this.customerAddress.value, "\n    Taxes: ").concat(this.taxes.map(tax => tax.toString()).join(" - "), "\n    Total taxes: ").concat(this.totalTax.value, "\n    ");
   }
 
 }
@@ -151,36 +176,32 @@ var _initFromScratch2 = function _initFromScratch2(_ref2) {
     dueDate,
     taxes,
     supplier,
+    supplierAddress,
     paymentDetails,
     companyNumber,
     vatNumber,
     orientation,
-    pageNumber
+    pageNumber,
+    customerName,
+    customerAddress,
+    customerCompanyRegistration
   } = _ref2;
-
-  var constructPrediction = function constructPrediction(item) {
-    return {
-      prediction: {
-        value: item
-      },
-      valueKey: "value",
-      pageNumber
-    };
-  };
-
-  this.locale = new Locale(constructPrediction(locale));
-  this.totalIncl = new Amount(constructPrediction(totalIncl));
-  this.totalExcl = new Amount(constructPrediction(totalExcl));
-  this.totalTax = new Amount(constructPrediction(totalTax));
-  this.date = new Date(constructPrediction(invoiceDate));
-  this.invoiceDate = new Date(constructPrediction(invoiceDate));
-  this.dueDate = new Date(constructPrediction(dueDate));
-  this.supplier = new Field(constructPrediction(supplier));
-  this.orientation = new Orientation(constructPrediction(orientation));
-  this.invoiceNumber = new Field(constructPrediction(invoiceNumber));
-  this.paymentDetails = new Field(constructPrediction(paymentDetails));
-  this.companyNumber = new Field(constructPrediction(companyNumber));
-  this.vatNumber = new Field(constructPrediction(vatNumber));
+  this.locale = new Locale(this.constructPrediction(locale));
+  this.totalIncl = new Amount(this.constructPrediction(totalIncl));
+  this.totalExcl = new Amount(this.constructPrediction(totalExcl));
+  this.totalTax = new Amount(this.constructPrediction(totalTax));
+  this.date = new Date(this.constructPrediction(invoiceDate));
+  this.invoiceDate = new Date(this.constructPrediction(invoiceDate));
+  this.dueDate = new Date(this.constructPrediction(dueDate));
+  this.supplier = new Field(this.constructPrediction(supplier));
+  this.supplierAddress = new Field(this.constructPrediction(supplierAddress));
+  this.invoiceNumber = new Field(this.constructPrediction(invoiceNumber));
+  this.paymentDetails = new Field(this.constructPrediction(paymentDetails));
+  this.companyNumber = new Field(this.constructPrediction(companyNumber));
+  this.vatNumber = new Field(this.constructPrediction(vatNumber));
+  this.customerName = new Field(this.constructPrediction(customerName));
+  this.customerAddress = new Field(this.constructPrediction(customerAddress));
+  this.customerCompanyRegistration = new Field(this.constructPrediction(customerCompanyRegistration));
 
   if (taxes !== undefined) {
     this.taxes = [];
@@ -197,6 +218,10 @@ var _initFromScratch2 = function _initFromScratch2(_ref2) {
       }));
     }
   }
+
+  if (this.level === "page") {
+    this.orientation = new Orientation(this.constructPrediction(orientation));
+  }
 };
 
 var _initFromApiPrediction2 = function _initFromApiPrediction2(apiPrediction, pageNumber) {
@@ -213,7 +238,7 @@ var _initFromApiPrediction2 = function _initFromApiPrediction2(apiPrediction, pa
   this.totalTax = new Amount({
     prediction: {
       value: undefined,
-      probability: 0.0
+      confidence: 0.0
     },
     valueKey: "value",
     pageNumber
@@ -242,10 +267,6 @@ var _initFromApiPrediction2 = function _initFromApiPrediction2(apiPrediction, pa
       codeKey: "code"
     });
   });
-  this.orientation = new Orientation({
-    prediction: apiPrediction.orientation,
-    pageNumber
-  });
   this.companyNumber = apiPrediction.company_registration.map(function (companyNumber) {
     return new Field({
       prediction: companyNumber,
@@ -266,12 +287,48 @@ var _initFromApiPrediction2 = function _initFromApiPrediction2(apiPrediction, pa
     prediction: apiPrediction.supplier,
     pageNumber
   });
+  this.supplierAddress = new Field({
+    prediction: apiPrediction.supplier_address,
+    pageNumber
+  });
+  this.customerName = new Field({
+    prediction: apiPrediction.customer,
+    pageNumber
+  });
+  this.customerAddress = new Field({
+    prediction: apiPrediction.customer_address,
+    pageNumber
+  });
+  this.customerCompanyRegistration = apiPrediction.customer_company_registration.map(function (customerCompanyRegistration) {
+    return new Field({
+      prediction: customerCompanyRegistration,
+      pageNumber,
+      extraFields: ["type"]
+    });
+  });
   this.paymentDetails = apiPrediction.payment_details.map(function (paymentDetail) {
     return new PaymentDetails({
       prediction: paymentDetail,
       pageNumber
     });
   });
+
+  if (this.level === "page") {
+    this.orientation = new Orientation({
+      prediction: apiPrediction.orientation,
+      pageNumber
+    });
+  } else {
+    this.orientation = new Orientation(this.constructPrediction({
+      prediction: {
+        value: undefined,
+        confidence: 0.0,
+        degrees: undefined
+      }
+    }));
+  } // document.inference.ocr
+
+
   if ("mvision" in apiPrediction) this.words = apiPrediction.mvision;
 };
 
@@ -374,7 +431,7 @@ var _reconstructTotalTax2 = function _reconstructTotalTax2() {
       value: this.taxes.reduce((acc, tax) => {
         return tax.value !== undefined ? acc + tax.value : acc;
       }, 0),
-      probability: Field.arrayProbability(this.taxes)
+      confidence: Field.arrayProbability(this.taxes)
     };
     if (totalTax.value > 0) this.totalTax = new Amount({
       prediction: totalTax,
@@ -388,7 +445,7 @@ var _reconstructTotalTaxFromTotals2 = function _reconstructTotalTaxFromTotals2()
   if (this.totalTax.value === undefined && this.totalIncl.value > 0 && this.totalExcl.value > 0 && this.totalExcl.value <= this.totalIncl.value) {
     var totalTax = {
       value: this.totalIncl.value - this.totalExcl.value,
-      probability: this.totalIncl.probability * this.totalExcl.probability
+      confidence: this.totalIncl.probability * this.totalExcl.probability
     };
     if (totalTax.value > 0) this.totalTax = new Amount({
       prediction: totalTax,
@@ -404,7 +461,7 @@ var _reconstructTotalExcl2 = function _reconstructTotalExcl2() {
       value: this.totalIncl.value - this.taxes.reduce((acc, tax) => {
         return tax.value !== undefined ? acc + tax.value : acc;
       }, 0),
-      probability: Field.arrayProbability(this.taxes) * this.totalIncl.probability
+      confidence: Field.arrayProbability(this.taxes) * this.totalIncl.probability
     };
     this.totalExcl = new Amount({
       prediction: totalExcl,
@@ -420,7 +477,7 @@ var _reconstructTotalIncl2 = function _reconstructTotalIncl2() {
       value: this.totalExcl.value + this.taxes.reduce((acc, tax) => {
         return tax.value ? acc + tax.value : acc;
       }, 0.0),
-      probability: Field.arrayProbability(this.taxes) * this.totalExcl.probability
+      confidence: Field.arrayProbability(this.taxes) * this.totalExcl.probability
     };
     this.totalIncl = new Amount({
       prediction: totalIncl,

package/lib/documents/receipt.js

@@ -57,6 +57,7 @@ class Receipt extends Document {
    *  @param {Object} orientation - orientation value for creating Receipt object from scratch
    *  @param {Object} totalTax - total taxes value for creating Receipt object from scratch
    *  @param {Object} totalExcl - total taxes excluded value for creating Receipt object from scratch
+   *  @param {String} level - specify whether object is built from "page" level or "document" level prediction
    */
   constructor(_ref) {
     var {
@@ -72,7 +73,8 @@ class Receipt extends Document {
       orientation: _orientation = undefined,
       totalTax: _totalTax = undefined,
       totalExcl: _totalExcl = undefined,
-      pageNumber: _pageNumber = 0
+      pageNumber: _pageNumber = 0,
+      level = "page"
     } = _ref;
     super(inputFile);
 
@@ -90,6 +92,18 @@ class Receipt extends Document {
 
     _initFromScratch.add(this);
 
+    this.level = level;
+
+    this.constructPrediction = function (item) {
+      return {
+        prediction: {
+          value: item
+        },
+        valueKey: "value",
+        pageNumber: _pageNumber
+      };
+    };
+
     if (_apiPrediction === undefined) {
       _classPrivateMethodGet(this, _initFromScratch, _initFromScratch2).call(this, {
         locale: _locale,
@@ -147,23 +161,12 @@ var _initFromScratch2 = function _initFromScratch2(_ref2) {
     totalTax,
     pageNumber
   } = _ref2;
-
-  var constructPrediction = function constructPrediction(item) {
-    return {
-      prediction: {
-        value: item
-      },
-      valueKey: "value",
-      pageNumber
-    };
-  };
-
-  this.locale = new Locale(constructPrediction(locale));
-  this.totalIncl = new Amount(constructPrediction(totalIncl));
-  this.date = new Date(constructPrediction(date));
-  this.category = new Field(constructPrediction(category));
-  this.merchantName = new Field(constructPrediction(merchantName));
-  this.time = new Field(constructPrediction(time));
+  this.locale = new Locale(this.constructPrediction(locale));
+  this.totalIncl = new Amount(this.constructPrediction(totalIncl));
+  this.date = new Date(this.constructPrediction(date));
+  this.category = new Field(this.constructPrediction(category));
+  this.merchantName = new Field(this.constructPrediction(merchantName));
+  this.time = new Field(this.constructPrediction(time));
 
   if (taxes !== undefined) {
     this.taxes = [];
@@ -181,9 +184,9 @@ var _initFromScratch2 = function _initFromScratch2(_ref2) {
     }
   }
 
-  this.orientation = new Orientation(constructPrediction(orientation));
-  this.totalTax = new Amount(constructPrediction(totalTax));
-  this.totalExcl = new Amount(constructPrediction(totalExcl));
+  this.orientation = new Orientation(this.constructPrediction(orientation));
+  this.totalTax = new Amount(this.constructPrediction(totalTax));
+  this.totalExcl = new Amount(this.constructPrediction(totalExcl));
 };
 
 var _initFromApiPrediction2 = function _initFromApiPrediction2(apiPrediction, pageNumber) {
@@ -223,14 +226,10 @@ var _initFromApiPrediction2 = function _initFromApiPrediction2(apiPrediction, pa
     rateKey: "rate",
     codeKey: "code"
   }));
-  this.orientation = new Orientation({
-    prediction: apiPrediction.orientation,
-    pageNumber
-  });
   this.totalTax = new Amount({
     prediction: {
       value: undefined,
-      probability: 0
+      confidence: 0
     },
     valueKey: "value",
     pageNumber
@@ -238,11 +237,27 @@ var _initFromApiPrediction2 = function _initFromApiPrediction2(apiPrediction, pa
   this.totalExcl = new Amount({
     prediction: {
       value: undefined,
-      probability: 0
+      confidence: 0
     },
     valueKey: "value",
     pageNumber
   });
+
+  if (this.level === "page") {
+    this.orientation = new Orientation({
+      prediction: apiPrediction.orientation,
+      pageNumber
+    });
+  } else {
+    this.orientation = new Orientation(this.constructPrediction({
+      prediction: {
+        value: undefined,
+        confidence: 0.0,
+        degrees: undefined
+      }
+    }));
+  }
+
   if ("mvision" in apiPrediction) this.words = apiPrediction.mvision;
 };
 
@@ -290,7 +305,7 @@ var _reconstructTotalExclFromTCCAndTaxes2 = function _reconstructTotalExclFromTC
   if (this.taxes.length && this.totalIncl.value != null) {
     var totalExcl = {
       value: this.totalIncl.value - Field.arraySum(this.taxes),
-      probability: Field.arrayProbability(this.taxes) * this.totalIncl.probability
+      confidence: Field.arrayProbability(this.taxes) * this.totalIncl.probability
     };
     this.totalExcl = new Amount({
       prediction: totalExcl,
@@ -304,7 +319,7 @@ var _reconstructTotalTax2 = function _reconstructTotalTax2() {
   if (this.taxes.length && this.totalTax.value == null) {
     var totalTax = {
       value: this.taxes.map(tax => tax.value || 0).reduce((a, b) => a + b, 0),
-      probability: Field.arrayProbability(this.taxes)
+      confidence: Field.arrayProbability(this.taxes)
     };
     if (totalTax.value > 0) this.totalTax = new Amount({
       prediction: totalTax,

package/mindee/api/financialDocument.js

@@ -30,15 +30,17 @@ class APIFinancialDocument extends APIObject {
       filename: filename,
       allowCutPdf: cutPdf,
     });
-    this.apiToken =
-      inputFile.fileExtension === "pdf" ? this.invoiceToken : this.receiptToken;
     await inputFile.init();
+    this.apiToken =
+      inputFile.fileExtension === "application/pdf"
+        ? this.invoiceToken
+        : this.receiptToken;
     const url =
-      inputFile.fileExtension === "pdf"
-        ? "/invoices/v1/predict"
-        : "/expense_receipts/v3/predict";
+      inputFile.fileExtension === "application/pdf"
+        ? `/invoices/v${version}/predict`
+        : `/expense_receipts/v${version}/predict`;
     super.parse();
-    return await super._request(url, inputFile, version, includeWords);
+    return super._request(url, inputFile, includeWords);
   }
 }
 

package/mindee/api/invoice.js

@@ -19,7 +19,7 @@ class APIInvoice extends APIObject {
     input,
     inputType = "path",
     filename = undefined,
-    version = "2",
+    version = "3",
     cutPdf = true,
     includeWords = false,
   }) {