mindee 1.0.9 → 1.1.0

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # CHANGELOG
 
+## v1.1.0 (2022-01-03)
+
+### new
+
+- chg: :sparkles: New Mindee API V2 support
+- chg: :zap: Server side invoice reconstruction from multi-page pdfs
+- new: :sparkles: Added OS in User-Agent headers
+
+
 ## v1.0.9 (2021-12-09)
 
 ### Fixes

package/LICENSE

@@ -0,0 +1,202 @@
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright [yyyy] [name of copyright owner]
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+

package/lib/api/financialDocument.js

@@ -44,9 +44,9 @@ class APIFinancialDocument extends APIObject {
         filename: filename,
         allowCutPdf: cutPdf
       });
-      _this.apiToken = inputFile.fileExtension === "pdf" ? _this.invoiceToken : _this.receiptToken;
       yield inputFile.init();
-      var url = inputFile.fileExtension === "pdf" ? "/invoices/v1/predict" : "/expense_receipts/v3/predict";
+      _this.apiToken = inputFile.fileExtension === "application/pdf" ? _this.invoiceToken : _this.receiptToken;
+      var url = inputFile.fileExtension === "application/pdf" ? "/invoices/v2/predict" : "/expense_receipts/v3/predict";
 
       _superprop_getParse().call(_this);
 

package/lib/api/object.js

@@ -21,7 +21,7 @@ class APIObject {
     var apiToken = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : undefined;
     var apiName = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : "";
     this.apiToken = apiToken;
-    this.baseUrl = "https://api.mindee.net/products";
+    this.baseUrl = "https://api.mindee.net/v1/products/mindee/";
     this.apiName = apiName;
   }
   /**
@@ -65,7 +65,7 @@ class APIObject {
 
 
   wrapResponse(inputFile, response, documentType) {
-    if (response.statusCode != 200) {
+    if (response.statusCode > 201) {
       var errorMessage = JSON.stringify(response.data, null, 4);
       errorHandler.throw(new Error("".concat(this.apiName, " API ").concat(response.statusCode, " HTTP error: ").concat(errorMessage)), false);
       return new Response({

package/lib/api/request.js

@@ -18,23 +18,25 @@ var {
 
 var FormData = require("form-data");
 
+var os = require("os");
+
 var request = function request(url, method, headers, input) {
   var includeWords = arguments.length > 4 && arguments[4] !== undefined ? arguments[4] : false;
   return new Promise(function (resolve, reject) {
     var form = new FormData();
     var body;
-    headers["User-Agent"] = "mindee-node/".concat(sdkVersion, " node/").concat(process.version);
+    headers["User-Agent"] = "mindee-api-nodejs@v".concat(sdkVersion, " nodejs-").concat(process.version, " ").concat(os.type().toLowerCase());
 
     if (["path", "stream"].includes(input.inputType)) {
       var fileParams = {
         filename: input.filename
       };
-      form.append("file", input.fileObject, fileParams);
+      form.append("document", input.fileObject, fileParams);
       if (includeWords) form.append("include_mvision", "true");
       headers = _objectSpread(_objectSpread({}, headers), form.getHeaders());
     } else if (input.inputType === "base64") {
       var body_obj = {
-        file: input.fileObject
+        document: input.fileObject
       };
       if (includeWords) body_obj["include_mvision"] = "true";
       body = JSON.stringify(body_obj);
@@ -60,7 +62,7 @@ var request = function request(url, method, headers, input) {
             data: JSON.parse(responseBody)
           }));
         } catch (error) {
-          console.log(responseBody);
+          console.log("error");
         }
       });
     });

package/lib/api/response.js

@@ -69,19 +69,24 @@ class Response {
       invoice: params => new Invoice(params),
       financialDocument: params => new FinancialDocument(params)
     };
-    var predictions = this.httpResponse.data.predictions.entries();
+    var predictions = this.httpResponse.data.document.inference.pages.entries();
     this["".concat(this.documentType, "s")] = []; // Create a list of Document (Receipt, Invoice...) for each page of the input document
 
     for (var [pageNumber, prediction] of predictions) {
       this["".concat(this.documentType, "s")].push(constructors[this.documentType]({
-        apiPrediction: prediction,
+        apiPrediction: prediction.prediction,
         inputFile: this.input,
         pageNumber: pageNumber
       }));
     } // Merge the list of Document into a unique Document
 
 
-    this[this.documentType] = Document.mergePages(this["".concat(this.documentType, "s")]);
+    this[this.documentType] = constructors[this.documentType]({
+      apiPrediction: this.httpResponse.data.document.inference.prediction,
+      inputFile: this.input,
+      pageNumber: -1,
+      level: "document"
+    });
   }
 
 }

package/lib/documents/document.js

@@ -79,7 +79,11 @@ class Document {
       for (var attribute of attributes) {
         var _document$attribute;
 
-        if ((document === null || document === void 0 ? void 0 : (_document$attribute = document[attribute]) === null || _document$attribute === void 0 ? void 0 : _document$attribute.probability) > finalDocument[attribute].probability) {
+        if (Array.isArray(document === null || document === void 0 ? void 0 : document[attribute])) {
+          var _finalDocument$attrib;
+
+          finalDocument[attribute] = (_finalDocument$attrib = finalDocument[attribute]) !== null && _finalDocument$attrib !== void 0 && _finalDocument$attrib.length ? finalDocument[attribute] : document === null || document === void 0 ? void 0 : document[attribute];
+        } else if ((document === null || document === void 0 ? void 0 : (_document$attribute = document[attribute]) === null || _document$attribute === void 0 ? void 0 : _document$attribute.probability) > finalDocument[attribute].probability) {
           finalDocument[attribute] = document === null || document === void 0 ? void 0 : document[attribute];
         }
       }

package/lib/documents/fields/field.js

@@ -22,7 +22,7 @@ class Field {
     this.probability = 0.0;
     this.bbox = [];
 
-    if (valueKey in prediction && prediction[valueKey] !== "N/A") {
+    if (valueKey in prediction && prediction[valueKey] !== null) {
       this.value = prediction[valueKey];
       if (prediction.probability) this.probability = prediction.probability;
       if (prediction.segmentation) this.bbox = prediction.segmentation.bounding_box || [];

package/lib/documents/financialDocument.js

@@ -53,6 +53,7 @@ class FinancialDocument extends Document {
    *  @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} 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 +66,15 @@ class FinancialDocument extends Document {
       invoiceNumber: _invoiceNumber = undefined,
       dueDate: _dueDate = undefined,
       taxes: _taxes = undefined,
-      merchantName: _merchantName = undefined,
+      merchantName = undefined,
       paymentDetails: _paymentDetails = undefined,
       companyNumber: _companyNumber = undefined,
       vatNumber: _vatNumber = undefined,
       orientation: _orientation = undefined,
       totalTax: _totalTax = undefined,
       time: _time = undefined,
-      pageNumber: _pageNumber = 0
+      pageNumber: _pageNumber = 0,
+      level = "page"
     } = _ref;
     super(_inputFile);
 
@@ -84,6 +86,8 @@ class FinancialDocument extends Document {
 
     _initFromScratch.add(this);
 
+    this.level = level;
+
     if (_apiPrediction === undefined) {
       _classPrivateMethodGet(this, _initFromScratch, _initFromScratch2).call(this, {
         locale: _locale,
@@ -93,7 +97,7 @@ class FinancialDocument extends Document {
         invoiceNumber: _invoiceNumber,
         dueDate: _dueDate,
         taxes: _taxes,
-        merchantName: _merchantName,
+        supplier,
         paymentDetails: _paymentDetails,
         companyNumber: _companyNumber,
         vatNumber: _vatNumber,
@@ -110,7 +114,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,7 +134,7 @@ var _initFromScratch2 = function _initFromScratch2(_ref2) {
     vatNumber,
     orientation,
     pageNumber,
-    merchantName,
+    supplier,
     time
   } = _ref2;
 
@@ -150,7 +154,7 @@ 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.time = new Field(constructPrediction(time));
   this.orientation = new Orientation(constructPrediction(orientation));
   this.invoiceNumber = new Field(constructPrediction(invoiceNumber));
@@ -180,46 +184,50 @@ 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.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
-      }
-    });
   } else {
     var receipt = new Receipt({
       apiPrediction,
       inputFile,
-      pageNumber
-    });
-    Object.assign(this, receipt);
-    this.invoiceDate = new Field({
-      prediction: {
-        value: undefined,
-        probability: 0.0
-      }
+      pageNumber,
+      level: this.level
     });
+    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.time = receipt.time;
+    this.totalTax = receipt.totalTax;
     this.invoiceNumber = new Field({
       prediction: {
         value: undefined,
         probability: 0.0
       }
     });
-    this.dueDate = new Field({
-      prediction: {
-        value: undefined,
-        probability: 0.0
-      }
-    });
     this.paymentDetails = new Field({
       prediction: {
         value: undefined,

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
@@ -63,7 +62,8 @@ class Invoice extends Document {
    *  @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 {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 {
@@ -82,7 +82,8 @@ class Invoice extends Document {
       vatNumber: _vatNumber = undefined,
       orientation: _orientation = undefined,
       totalTax: _totalTax = undefined,
-      pageNumber: _pageNumber = 0
+      pageNumber: _pageNumber = 0,
+      level = "page"
     } = _ref;
     super(inputFile);
 
@@ -108,6 +109,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,
@@ -157,30 +170,18 @@ var _initFromScratch2 = function _initFromScratch2(_ref2) {
     orientation,
     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.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.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));
 
   if (taxes !== undefined) {
     this.taxes = [];
@@ -197,6 +198,10 @@ var _initFromScratch2 = function _initFromScratch2(_ref2) {
       }));
     }
   }
+
+  if (this.level === "page") {
+    this.orientation = new Orientation(this.constructPrediction(orientation));
+  }
 };
 
 var _initFromApiPrediction2 = function _initFromApiPrediction2(apiPrediction, pageNumber) {
@@ -242,10 +247,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,
@@ -272,6 +273,23 @@ var _initFromApiPrediction2 = function _initFromApiPrediction2(apiPrediction, pa
       pageNumber
     });
   });
+
+  if (this.level === "page") {
+    this.orientation = new Orientation({
+      prediction: apiPrediction.orientation,
+      pageNumber
+    });
+  } else {
+    this.orientation = new Orientation(this.constructPrediction({
+      prediction: {
+        value: undefined,
+        probability: 0.0,
+        degrees: undefined
+      }
+    }));
+  } // document.inference.ocr
+
+
   if ("mvision" in apiPrediction) this.words = apiPrediction.mvision;
 };