mindee 3.16.0 → 3.17.0

data/docs/payslip_fra_v3.md

@@ -0,0 +1,319 @@
+---
+title: FR Payslip OCR Ruby
+category: 622b805aaec68102ea7fcbc2
+slug: ruby-fr-payslip-ocr
+parentDoc: 6294d97ee723f1008d2ab28e
+---
+The Ruby OCR SDK supports the [Payslip API](https://platform.mindee.com/mindee/payslip_fra).
+
+Using the [sample below](https://github.com/mindee/client-lib-test-data/blob/main/products/payslip_fra/default_sample.jpg), we are going to illustrate how to extract the data that we want using the OCR SDK.
+![Payslip sample](https://github.com/mindee/client-lib-test-data/blob/main/products/payslip_fra/default_sample.jpg?raw=true)
+
+# Quick-Start
+```rb
+require 'mindee'
+
+# Init a new client
+mindee_client = Mindee::Client.new(api_key: 'my-api-key')
+
+# Load a file from disk
+input_source = mindee_client.source_from_path('/path/to/the/file.ext')
+
+# Parse the file
+result = mindee_client.enqueue_and_parse(
+  input_source,
+  Mindee::Product::FR::Payslip::PayslipV3
+)
+
+# Print a full summary of the parsed data in RST format
+puts result.document
+
+# Print the document-level parsed data
+# puts result.document.inference.prediction
+
+```
+
+**Output (RST):**
+```rst
+########
+Document
+########
+:Mindee ID: a479e3e7-6838-4e82-9a7d-99289f34ec7f
+:Filename: default_sample.jpg
+
+Inference
+#########
+:Product: mindee/payslip_fra v3.0
+:Rotation applied: Yes
+
+Prediction
+==========
+:Pay Period:
+  :End Date: 2023-03-31
+  :Month: 03
+  :Payment Date: 2023-03-29
+  :Start Date: 2023-03-01
+  :Year: 2023
+:Employee:
+  :Address: 52 RUE DES FLEURS 33500 LIBOURNE FRANCE
+  :Date of Birth:
+  :First Name: Jean Luc
+  :Last Name: Picard
+  :Phone Number:
+  :Registration Number:
+  :Social Security Number: 123456789012345
+:Employer:
+  :Address: 1 RUE DU TONNOT 25210 DOUBS
+  :Company ID: 12345678901234
+  :Company Site:
+  :NAF Code: 1234A
+  :Name: DEMO COMPANY
+  :Phone Number:
+  :URSSAF Number:
+:Bank Account Details:
+  :Bank Name:
+  :IBAN:
+  :SWIFT:
+:Employment:
+  :Category: Cadre
+  :Coefficient: 600,000
+  :Collective Agreement: Construction -- Promotion
+  :Job Title: Directeur Régional du Développement
+  :Position Level: Niveau 5 Echelon 3
+  :Seniority Date:
+  :Start Date: 2022-05-01
+:Salary Details:
+  +--------------+-----------+--------------------------------------+--------+-----------+
+  | Amount       | Base      | Description                          | Number | Rate      |
+  +==============+===========+======================================+========+===========+
+  | 6666.67      |           | Salaire de base                      |        |           |
+  +--------------+-----------+--------------------------------------+--------+-----------+
+  | 9.30         |           | Part patronale Mutuelle NR           |        |           |
+  +--------------+-----------+--------------------------------------+--------+-----------+
+  | 508.30       |           | Avantages en nature voiture          |        |           |
+  +--------------+-----------+--------------------------------------+--------+-----------+
+:Pay Detail:
+  :Gross Salary: 7184.27
+  :Gross Salary YTD: 18074.81
+  :Income Tax Rate: 17.60
+  :Income Tax Withheld: 1030.99
+  :Net Paid: 3868.32
+  :Net Paid Before Tax: 4899.31
+  :Net Taxable: 5857.90
+  :Net Taxable YTD: 14752.73
+  :Total Cost Employer: 10486.94
+  :Total Taxes and Deductions: 1650.36
+:Paid Time Off:
+  +-----------+--------+-------------+-----------+-----------+
+  | Accrued   | Period | Type        | Remaining | Used      |
+  +===========+========+=============+===========+===========+
+  |           | N-1    | VACATION    |           |           |
+  +-----------+--------+-------------+-----------+-----------+
+  | 6.17      | N      | VACATION    | 6.17      |           |
+  +-----------+--------+-------------+-----------+-----------+
+  | 2.01      | N      | RTT         | 2.01      |           |
+  +-----------+--------+-------------+-----------+-----------+
+```
+
+# Field Types
+## Standard Fields
+These fields are generic and used in several products.
+
+### Basic Field
+Each prediction object contains a set of fields that inherit from the generic `Field` class.
+A typical `Field` object will have the following attributes:
+
+* **value** (`String`, `Float`, `Integer`, `Boolean`): corresponds to the field value. Can be `nil` if no value was extracted.
+* **confidence** (Float, nil): the confidence score of the field prediction.
+* **bounding_box** (`Mindee::Geometry::Quadrilateral`, `nil`): contains exactly 4 relative vertices (points) coordinates of a right rectangle containing the field in the document.
+* **polygon** (`Mindee::Geometry::Polygon`, `nil`): contains the relative vertices coordinates (`Point`) of a polygon containing the field in the image.
+* **page_id** (`Integer`, `nil`): the ID of the page, always `nil` when at document-level.
+* **reconstructed** (`Boolean`): indicates whether an object was reconstructed (not extracted as the API gave it).
+
+
+Aside from the previous attributes, all basic fields have access to a `to_s` method that can be used to print their value as a string.
+
+## Specific Fields
+Fields which are specific to this product; they are not used in any other product.
+
+### Bank Account Details Field
+Information about the employee's bank account.
+
+A `PayslipV3BankAccountDetail` implements the following attributes:
+
+* `bank_name` (String): The name of the bank.
+* `iban` (String): The IBAN of the bank account.
+* `swift` (String): The SWIFT code of the bank.
+Fields which are specific to this product; they are not used in any other product.
+
+### Employee Field
+Information about the employee.
+
+A `PayslipV3Employee` implements the following attributes:
+
+* `address` (String): The address of the employee.
+* `date_of_birth` (String): The date of birth of the employee.
+* `first_name` (String): The first name of the employee.
+* `last_name` (String): The last name of the employee.
+* `phone_number` (String): The phone number of the employee.
+* `registration_number` (String): The registration number of the employee.
+* `social_security_number` (String): The social security number of the employee.
+Fields which are specific to this product; they are not used in any other product.
+
+### Employer Field
+Information about the employer.
+
+A `PayslipV3Employer` implements the following attributes:
+
+* `address` (String): The address of the employer.
+* `company_id` (String): The company ID of the employer.
+* `company_site` (String): The site of the company.
+* `naf_code` (String): The NAF code of the employer.
+* `name` (String): The name of the employer.
+* `phone_number` (String): The phone number of the employer.
+* `urssaf_number` (String): The URSSAF number of the employer.
+Fields which are specific to this product; they are not used in any other product.
+
+### Employment Field
+Information about the employment.
+
+A `PayslipV3Employment` implements the following attributes:
+
+* `category` (String): The category of the employment.
+* `coefficient` (String): The coefficient of the employment.
+* `collective_agreement` (String): The collective agreement of the employment.
+* `job_title` (String): The job title of the employee.
+* `position_level` (String): The position level of the employment.
+* `seniority_date` (String): The seniority date of the employment.
+* `start_date` (String): The start date of the employment.
+Fields which are specific to this product; they are not used in any other product.
+
+### Paid Time Off Field
+Information about paid time off.
+
+A `PayslipV3PaidTimeOff` implements the following attributes:
+
+* `accrued` (Float): The amount of paid time off accrued in the period.
+* `period` (String): The paid time off period.
+
+#### Possible values include:
+ - N
+ - N-1
+ - N-2
+
+* `pto_type` (String): The type of paid time off.
+
+#### Possible values include:
+ - VACATION
+ - RTT
+ - COMPENSATORY
+
+* `remaining` (Float): The remaining amount of paid time off at the end of the period.
+* `used` (Float): The amount of paid time off used in the period.
+Fields which are specific to this product; they are not used in any other product.
+
+### Pay Detail Field
+Detailed information about the pay.
+
+A `PayslipV3PayDetail` implements the following attributes:
+
+* `gross_salary` (Float): The gross salary of the employee.
+* `gross_salary_ytd` (Float): The year-to-date gross salary of the employee.
+* `income_tax_rate` (Float): The income tax rate of the employee.
+* `income_tax_withheld` (Float): The income tax withheld from the employee's pay.
+* `net_paid` (Float): The net paid amount of the employee.
+* `net_paid_before_tax` (Float): The net paid amount before tax of the employee.
+* `net_taxable` (Float): The net taxable amount of the employee.
+* `net_taxable_ytd` (Float): The year-to-date net taxable amount of the employee.
+* `total_cost_employer` (Float): The total cost to the employer.
+* `total_taxes_and_deductions` (Float): The total taxes and deductions of the employee.
+Fields which are specific to this product; they are not used in any other product.
+
+### Pay Period Field
+Information about the pay period.
+
+A `PayslipV3PayPeriod` implements the following attributes:
+
+* `end_date` (String): The end date of the pay period.
+* `month` (String): The month of the pay period.
+* `payment_date` (String): The date of payment for the pay period.
+* `start_date` (String): The start date of the pay period.
+* `year` (String): The year of the pay period.
+Fields which are specific to this product; they are not used in any other product.
+
+### Salary Details Field
+Detailed information about the earnings.
+
+A `PayslipV3SalaryDetail` implements the following attributes:
+
+* `amount` (Float): The amount of the earning.
+* `base` (Float): The base rate value of the earning.
+* `description` (String): The description of the earnings.
+* `number` (Float): The number of units in the earning.
+* `rate` (Float): The rate of the earning.
+
+# Attributes
+The following fields are extracted for Payslip V3:
+
+## Bank Account Details
+**bank_account_details** ([PayslipV3BankAccountDetail](#bank-account-details-field)): Information about the employee's bank account.
+
+```rb
+puts result.document.inference.prediction.bank_account_details.value
+```
+
+## Employee
+**employee** ([PayslipV3Employee](#employee-field)): Information about the employee.
+
+```rb
+puts result.document.inference.prediction.employee.value
+```
+
+## Employer
+**employer** ([PayslipV3Employer](#employer-field)): Information about the employer.
+
+```rb
+puts result.document.inference.prediction.employer.value
+```
+
+## Employment
+**employment** ([PayslipV3Employment](#employment-field)): Information about the employment.
+
+```rb
+puts result.document.inference.prediction.employment.value
+```
+
+## Paid Time Off
+**paid_time_off** (Array<[PayslipV3PaidTimeOff](#paid-time-off-field)>): Information about paid time off.
+
+```rb
+for paid_time_off_elem in result.document.inference.prediction.paid_time_off do
+  puts paid_time_off_elem.value
+end
+```
+
+## Pay Detail
+**pay_detail** ([PayslipV3PayDetail](#pay-detail-field)): Detailed information about the pay.
+
+```rb
+puts result.document.inference.prediction.pay_detail.value
+```
+
+## Pay Period
+**pay_period** ([PayslipV3PayPeriod](#pay-period-field)): Information about the pay period.
+
+```rb
+puts result.document.inference.prediction.pay_period.value
+```
+
+## Salary Details
+**salary_details** (Array<[PayslipV3SalaryDetail](#salary-details-field)>): Detailed information about the earnings.
+
+```rb
+for salary_details_elem in result.document.inference.prediction.salary_details do
+  puts salary_details_elem.value
+end
+```
+
+# Questions?
+[Join our Slack](https://join.slack.com/t/mindee-community/shared_invite/zt-2d0ds7dtz-DPAF81ZqTy20chsYpQBW5g)

data/lib/mindee/client.rb

@@ -195,6 +195,46 @@ module Mindee
 
     # rubocop:enable Metrics/ParameterLists
 
+    # Sends a document to a workflow.
+    #
+    # @param input_source [Mindee::Input::Source::LocalInputSource, Mindee::Input::Source::UrlInputSource]
+    # @param document_alias [String, nil] Alias to give to the document.
+    # @param priority [Symbol, nil] Priority to give to the document.
+    # @param full_text [Boolean] Whether to include the full OCR text response in compatible APIs.
+    #  This performs a full OCR operation on the server and may increase response time.
+    #
+    # @param public_url [String, nil] A unique, encrypted URL for accessing the document validation interface without
+    # requiring authentication.
+    # @param page_options [Hash, nil] Page cutting/merge options:
+    #
+    #  * `:page_indexes` Zero-based list of page indexes.
+    #  * `:operation` Operation to apply on the document, given the `page_indexes specified:
+    #      * `:KEEP_ONLY` - keep only the specified pages, and remove all others.
+    #      * `:REMOVE` - remove the specified pages, and keep all others.
+    #  * `:on_min_pages` Apply the operation only if document has at least this many pages.
+    #
+    #
+    # @return [Mindee::Parsing::Common::WorkflowResponse]
+    def execute_workflow(
+      input_source,
+      workflow_id,
+      document_alias: nil,
+      priority: nil,
+      full_text: false,
+      public_url: nil,
+      page_options: nil
+    )
+      if input_source.is_a?(Mindee::Input::Source::LocalInputSource) && !page_options.nil? && input_source.pdf?
+        input_source.process_pdf(page_options)
+      end
+
+      workflow_endpoint = Mindee::HTTP::WorkflowEndpoint.new(workflow_id, api_key: @api_key)
+      prediction, raw_http = workflow_endpoint.execute_workflow(input_source, full_text, document_alias, priority,
+                                                                public_url)
+      Mindee::Parsing::Common::WorkflowResponse.new(Product::Generated::GeneratedV1,
+                                                    prediction, raw_http)
+    end
+
     # Load a prediction.
     #
     # @param product_class [Mindee::Inference] class of the product

data/lib/mindee/http/workflow_endpoint.rb

@@ -0,0 +1,90 @@
+# frozen_string_literal: true
+
+require 'json'
+require 'net/http'
+require_relative 'error'
+
+module Mindee
+  module HTTP
+    # Handles the routing for workflow calls.
+    class WorkflowEndpoint
+      # @return [String]
+      attr_reader :api_key
+      # @return [Integer]
+      attr_reader :request_timeout
+      # @return [String]
+      attr_reader :url
+
+      def initialize(workflow_id, api_key: '')
+        @request_timeout = ENV.fetch(REQUEST_TIMEOUT_ENV_NAME, TIMEOUT_DEFAULT).to_i
+        @api_key = api_key.nil? || api_key.empty? ? ENV.fetch(API_KEY_ENV_NAME, API_KEY_DEFAULT) : api_key
+        base_url = ENV.fetch(BASE_URL_ENV_NAME, BASE_URL_DEFAULT)
+        @url = "#{base_url.chomp('/')}/workflows/#{workflow_id}/executions"
+      end
+
+      # Sends a document to the workflow.
+      # @param input_source [Mindee::Input::Source::LocalInputSource, Mindee::Input::Source::UrlInputSource]
+      # @param document_alias [String, nil] Alias to give to the document.
+      # @param priority [Symbol, nil] Priority to give to the document.
+      # @param full_text [Boolean] Whether to include the full OCR text response in compatible APIs.
+      # @param public_url [String, nil] A unique, encrypted URL for accessing the document validation interface without
+      # requiring authentication.
+      # @return [Array]
+      def execute_workflow(input_source, full_text, document_alias, priority, public_url)
+        check_api_key
+        response = workflow_execution_req_post(input_source, document_alias, priority, full_text, public_url)
+        hashed_response = JSON.parse(response.body, object_class: Hash)
+        return [hashed_response, response.body] if ResponseValidation.valid_async_response?(response)
+
+        ResponseValidation.clean_request!(response)
+        error = Error.handle_error(@url_name, response)
+        raise error
+      end
+
+      # @param input_source [Mindee::Input::Source::LocalInputSource, Mindee::Input::Source::UrlInputSource]
+      # @param document_alias [String, nil] Alias to give to the document.
+      # @param priority [Symbol, nil] Priority to give to the document.
+      # @param full_text [Boolean] Whether to include the full OCR text response in compatible APIs.
+      # @param public_url [String, nil] A unique, encrypted URL for accessing the document validation interface without
+      # requiring authentication.
+      # @return [Net::HTTPResponse, nil]
+      def workflow_execution_req_post(input_source, document_alias, priority, full_text, public_url)
+        uri = URI(@url)
+        params = {}
+        params[:full_text_ocr] = 'true' if full_text
+        uri.query = URI.encode_www_form(params)
+
+        headers = {
+          'Authorization' => "Token #{@api_key}",
+          'User-Agent' => USER_AGENT,
+        }
+        req = Net::HTTP::Post.new(uri, headers)
+        form_data = if input_source.is_a?(Mindee::Input::Source::UrlInputSource)
+                      [['document', input_source.url]]
+                    else
+                      [input_source.read_document]
+                    end
+        form_data.push ['alias', document_alias] if document_alias
+        form_data.push ['public_url', public_url] if public_url
+        form_data.push ['priority', priority.to_s] if priority
+
+        req.set_form(form_data, 'multipart/form-data')
+
+        response = nil
+        Net::HTTP.start(uri.hostname, uri.port, use_ssl: true, read_timeout: @request_timeout) do |http|
+          response = http.request(req)
+        end
+        response
+      end
+
+      # Checks API key
+      def check_api_key
+        return unless @api_key.nil? || @api_key.empty?
+
+        raise "Missing API key. Check your Client Configuration.\n" \
+              'You can set this using the ' \
+              "'#{HTTP::API_KEY_ENV_NAME}' environment variable."
+      end
+    end
+  end
+end

data/lib/mindee/http.rb

@@ -2,3 +2,4 @@
 
 require_relative 'http/endpoint'
 require_relative 'http/error'
+require_relative 'http/workflow_endpoint'

data/lib/mindee/parsing/common/api_response.rb

@@ -32,7 +32,7 @@ module Mindee
         attr_reader :id
         # @return [Mindee::Parsing::Standard::DateField]
         attr_reader :issued_at
-        # @return [Mindee::Parsing::Standard::DateField, nil]
+        # @return [Time, nil]
         attr_reader :available_at
         # @return [JobStatus, Symbol]
         attr_reader :status
@@ -121,6 +121,27 @@ module Mindee
           @job = Mindee::Parsing::Common::Job.new(http_response['job']) if http_response.key?('job')
         end
       end
+
+      # Represents the server response after a document is sent to a workflow.
+      class WorkflowResponse
+        # Set the prediction model used to parse the document.
+        # The response object will be instantiated based on this parameter.
+        # @return [Mindee::Parsing::Common::Execution]
+        attr_reader :execution
+        # @return [Mindee::Parsing::Common::ApiRequest]
+        attr_reader :api_request
+        # @return [String]
+        attr_reader :raw_http
+
+        # @param http_response [Hash]
+        # @param product_class [Mindee::Inference]
+        def initialize(product_class, http_response, raw_http)
+          @raw_http = raw_http.to_s
+          @api_request = Mindee::Parsing::Common::ApiRequest.new(http_response['api_request'])
+          product_class = (product_class || Product::Generated::GeneratedV1)
+          @execution = Mindee::Parsing::Common::Execution.new(product_class, http_response['execution'])
+        end
+      end
     end
   end
 end

data/lib/mindee/parsing/common/execution.rb

@@ -0,0 +1,73 @@
+# frozen_string_literal: true
+
+module Mindee
+  module Parsing
+    module Common
+      # Identifier for the batch to which the execution belongs.
+      class Execution
+        # Identifier for the batch to which the execution belongs.
+        # @return [String]
+        attr_reader :batch_name
+        # The time at which the execution started.
+        # @return [Time, nil]
+        attr_reader :created_at
+        # File representation within a workflow execution.
+        # @return [ExecutionFile]
+        attr_reader :file
+        # Identifier for the execution.
+        # @return [String]
+        attr_reader :id
+        # Deserialized inference object.
+        # @return [Mindee::Inference]
+        attr_reader :inference
+        # Priority of the execution.
+        # @return [ExecutionPriority]
+        attr_reader :priority
+        # The time at which the file was tagged as reviewed.
+        # @return [Time, nil]
+        attr_reader :reviewed_at
+        # The time at which the file was uploaded to a workflow.
+        # @return [Time, nil]
+        attr_reader :available_at
+        # Reviewed fields and values.
+        # @return [Mindee::Product::Generated::GeneratedV1Document]
+        attr_reader :reviewed_prediction
+        # Execution Status.
+        # @return [String]
+        attr_reader :status
+        # Execution type.
+        # @return [String]
+        attr_reader :type
+        # The time at which the file was uploaded to a workflow.
+        # @return [Time, nil]
+        attr_reader :uploaded_at
+        # Identifier for the workflow.
+        # @return [String]
+        attr_reader :workflow_id
+
+        # rubocop:disable Metrics/CyclomaticComplexity
+
+        # @param product_class [Mindee::Inference]
+        # @param http_response [Hash]
+        def initialize(product_class, http_response)
+          @batch_name = http_response['batch_name']
+          @created_at = Time.iso8601(http_response['created_at']) if http_response['created_at']
+          @file = ExecutionFile.new(http_response['file']) if http_response['file']
+          @id = http_response['id']
+          @inference = product_class.new(http_response['inference']) if http_response['inference']
+          @priority = Mindee::Parsing::Common::ExecutionPriority.to_priority(http_response['priority'])
+          @reviewed_at = Time.iso8601(http_response['reviewed_at']) if http_response['reviewed_at']
+          @available_at = Time.iso8601(http_response['available_at']) if http_response['available_at']
+          if http_response['reviewed_prediction']
+            @reviewed_prediction = GeneratedV1Document.new(http_response['reviewed_prediction'])
+          end
+          @status = http_response['status']
+          @type = http_response['type']
+          @uploaded_at = Time.iso8601(http_response['uploaded_at']) if http_response['uploaded_at']
+          @workflow_id = http_response['workflow_id']
+        end
+        # rubocop:enable Metrics/CyclomaticComplexity
+      end
+    end
+  end
+end

data/lib/mindee/parsing/common/execution_file.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module Mindee
+  module Parsing
+    module Common
+      # Representation of a workflow execution's file data.
+      class ExecutionFile
+        # File name.
+        # @return [String]
+        attr_reader :name
+
+        # Optional alias for the file.
+        # @return [String]
+        attr_reader :alias
+
+        # @param http_response [Hash]
+        def initialize(http_response)
+          @name = http_response['name']
+          @alias = http_response['alias']
+        end
+      end
+    end
+  end
+end

data/lib/mindee/parsing/common/execution_priority.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+module Mindee
+  module Parsing
+    module Common
+      # Execution policy priority values.
+      module ExecutionPriority
+        LOW = :low
+        MEDIUM = :medium
+        HIGH = :high
+
+        # Sets the priority to one of its possibly values, defaults to nil otherwise.
+        # @param [String, nil] priority_str
+        # @return [Symbol, nil]
+        def self.to_priority(priority_str)
+          return nil if priority_str.nil?
+
+          case priority_str.downcase
+          when 'low'
+            :low
+          when 'high'
+            :high
+          else
+            :medium
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mindee/parsing/common.rb

@@ -2,6 +2,9 @@
 
 require_relative 'common/api_response'
 require_relative 'common/document'
+require_relative 'common/execution'
+require_relative 'common/execution_file'
+require_relative 'common/execution_priority'
 require_relative 'common/inference'
 require_relative 'common/ocr'
 require_relative 'common/prediction'

data/lib/mindee/product/{international_id/international_id_v1.rb → driver_license/driver_license_v1.rb}

@@ -1,26 +1,26 @@
 # frozen_string_literal: true
 
 require_relative '../../parsing'
-require_relative 'international_id_v1_document'
-require_relative 'international_id_v1_page'
+require_relative 'driver_license_v1_document'
+require_relative 'driver_license_v1_page'
 
 module Mindee
   module Product
-    # International ID module.
-    module InternationalId
-      # International ID V1 prediction inference.
-      class InternationalIdV1 < Mindee::Parsing::Common::Inference
-        @endpoint_name = 'international_id'
+    # Driver License module.
+    module DriverLicense
+      # Driver License API version 1 inference prediction.
+      class DriverLicenseV1 < Mindee::Parsing::Common::Inference
+        @endpoint_name = 'driver_license'
         @endpoint_version = '1'
 
         # @param prediction [Hash]
         def initialize(prediction)
           super
-          @prediction = InternationalIdV1Document.new(prediction['prediction'], nil)
+          @prediction = DriverLicenseV1Document.new(prediction['prediction'], nil)
           @pages = []
           prediction['pages'].each do |page|
             if page.key?('prediction') && !page['prediction'].nil? && !page['prediction'].empty?
-              @pages.push(InternationalIdV1Page.new(page))
+              @pages.push(DriverLicenseV1Page.new(page))
             end
           end
         end