mindee 4.5.0 → 4.7.0.pre.rc1

data/lib/mindee/parsing/v2/job_webhook.rb

@@ -0,0 +1,60 @@
+# lib/mindee/parsing/v2/job_response_webhook.rb
+# frozen_string_literal: true
+
+require 'date'
+require_relative 'error_response'
+
+module Mindee
+  module Parsing
+    module V2
+      # Information about a webhook created for a job response.
+      class JobWebhook
+        # @return [String] Identifier of the webhook.
+        attr_reader :id
+        # @return [DateTime, nil] Creation timestamp (or +nil+ when absent).
+        attr_reader :created_at
+        # @return [String] Webhook status.
+        attr_reader :status
+        # @return [ErrorResponse, nil] Error information when something failed.
+        attr_reader :error
+
+        # @param server_response [Hash] Raw JSON parsed into a Hash.
+        def initialize(server_response)
+          @id = server_response['id']
+          @created_at = parse_date(server_response['created_at'])
+          @status     = server_response['status']
+          @error      = ErrorResponse.new(server_response['error']) if server_response.key?('error')
+        end
+
+        # RST display.
+        #
+        # @return [String]
+        def to_s
+          parts = [
+            'JobResponseWebhook',
+            '##################',
+            ":ID: #{@id}",
+            ":CreatedAt: #{@created_at}",
+            ":Status: #{@status}",
+          ]
+
+          parts << @error.to_s if @error
+          parts.join("\n")
+        end
+
+        private
+
+        # Safely parse an ISO-8601 date/time string to DateTime.
+        # @param str [String, nil] The date string.
+        # @return [DateTime, nil]
+        def parse_date(str)
+          return nil if str.to_s.empty?
+
+          DateTime.iso8601(str || '')
+        rescue ArgumentError
+          nil
+        end
+      end
+    end
+  end
+end

data/lib/mindee/parsing/v2/raw_text.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module Mindee
+  module Parsing
+    module V2
+      # Raw text extracted from a given page.
+      class RawText
+        # @return [Integer] Page number where the text was found.
+        attr_reader :page
+        # @return [String] Text content.
+        attr_reader :content
+
+        # @param server_response [Hash] Raw JSON parsed into a Hash.
+        def initialize(server_response)
+          @page = server_response['page']
+          @content = server_response['content']
+        end
+      end
+    end
+  end
+end

data/lib/mindee/parsing/v2.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+require_relative 'v2/common_response'
+require_relative 'v2/error_response'
+require_relative 'v2/field'
+require_relative 'v2/inference'
+require_relative 'v2/inference_file'
+require_relative 'v2/inference_model'
+require_relative 'v2/inference_response'
+require_relative 'v2/inference_result'
+require_relative 'v2/inference_result_options'
+require_relative 'v2/job'
+require_relative 'v2/job_response'
+require_relative 'v2/job_webhook'
+require_relative 'v2/raw_text'

data/lib/mindee/pdf/extracted_pdf.rb

@@ -7,14 +7,14 @@ module Mindee
       # An extracted sub-Pdf.
       class ExtractedPDF
         # Byte contents of the pdf
-        # @return [StreamIO]
+        # @return [StringIO]
         attr_reader :pdf_bytes
 
         # Name of the file.
         # @return [String]
         attr_reader :filename
 
-        # @param pdf_bytes [StreamIO]
+        # @param pdf_bytes [StringIO]
         # @param filename [String]
         def initialize(pdf_bytes, filename)
           @pdf_bytes = pdf_bytes
@@ -50,7 +50,9 @@ module Mindee
         # Returns the current PDF object as a usable BytesInputSource.
         # @return [Mindee::Input::Source::BytesInputSource]
         def as_input_source
-          Mindee::Input::Source::BytesInputSource.new(@pdf_bytes.read, @filename)
+          raise Errors::MindeePDFError, 'Bytes object is nil.' if @pdf_bytes.nil?
+
+          Mindee::Input::Source::BytesInputSource.new(@pdf_bytes.read || '', @filename)
         end
       end
     end

data/lib/mindee/pdf/pdf_extractor.rb

@@ -29,7 +29,7 @@ module Mindee
 
         # Creates a new Pdf from pages and save it into a buffer.
         # @param page_indexes [Array<Integer>] List of page number to use for merging in the original Pdf.
-        # @return [StreamIO] The buffer containing the new Pdf.
+        # @return [StringIO] The buffer containing the new Pdf.
         def cut_pages(page_indexes)
           options = PageOptions.new(params: {
                                       page_indexes: page_indexes,
@@ -75,24 +75,28 @@ module Mindee
         # @return [Array<Mindee::PDF::PDFExtractor::ExtractedPDF>]
         def extract_invoices(page_indexes, strict: false)
           raise Errors::MindeePDFError, 'No indexes provided.' if page_indexes.empty?
+
           if page_indexes[0].is_a?(Array) && page_indexes[0].all? { |i| i.is_a?(Integer) }
-            return extract_sub_documents(page_indexes)
+            page_indexes_as_array = page_indexes # @type var page_indexes : Array[Array[Integer]]
+            return extract_sub_documents(page_indexes_as_array)
           end
-          return extract_sub_documents(page_indexes.map(&:page_indexes)) unless strict
+          p_ids = page_indexes # @type var page_indexes: Product::InvoiceSplitter::InvoiceSplitterV1InvoicePageGroups
+          return extract_sub_documents(p_ids.map(&:page_indexes)) unless strict
 
           correct_page_indexes = []
           current_list = []
           previous_confidence = nil
-          page_indexes.each_with_index do |page_index, i|
-            confidence = page_index.confidence
+          p_ids.each_with_index do |p_i, i|
+            page_index = p_i # @type var page_index: Product::InvoiceSplitter::InvoiceSplitterV1InvoicePageGroup
+            confidence = page_index.confidence.to_f
             page_list = page_index.page_indexes
 
             if confidence >= 0.5 && previous_confidence.nil?
               current_list = page_list
-            elsif confidence >= 0.5 && i < page_indexes.length - 1
+            elsif confidence >= 0.5 && i < p_ids.length - 1
               correct_page_indexes << current_list
               current_list = page_list
-            elsif confidence < 0.5 && i == page_indexes.length - 1
+            elsif confidence < 0.5 && i == p_ids.length - 1
               current_list.concat page_list
               correct_page_indexes << current_list
             else

data/lib/mindee/pdf/pdf_processor.rb

@@ -17,7 +17,7 @@ module Mindee
         pages_count = current_pdf.pages.size
         return current_pdf.to_io_stream if options.on_min_pages.to_i > pages_count
 
-        all_pages = (0..pages_count - 1).to_a
+        all_pages = (0..(pages_count - 1)).to_a
 
         if options.operation == :KEEP_ONLY
           pages_to_remove = indexes_from_keep(options.page_indexes, all_pages)
@@ -31,8 +31,8 @@ module Mindee
         current_pdf.to_io_stream
       end
 
-      # @param page_indexes [Array]
-      # @param all_pages [Array]
+      # @param page_indexes [Array<Integer>]
+      # @param all_pages [Array<Integer>]
       def self.indexes_from_keep(page_indexes, all_pages)
         pages_to_keep = Set.new
         page_indexes.each do |idx|
@@ -42,11 +42,11 @@ module Mindee
 
           pages_to_keep << page
         end
-        all_pages.to_set - pages_to_keep
+        (all_pages.to_set - pages_to_keep).to_a
       end
 
-      # @param page_indexes [Array[Integer]]
-      # @param all_pages [Array]
+      # @param page_indexes [Array<Integer>]
+      # @param all_pages [Array<Integer>]
       def self.indexes_from_remove(page_indexes, all_pages)
         pages_to_remove = Set.new
         page_indexes.each do |idx|

data/lib/mindee/pdf/pdf_tools.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require 'origami'
+
 module Mindee
   module PDF
     # Collection of miscellaneous PDF operations,as well as some monkey-patching for Origami.
@@ -84,7 +86,7 @@ module Mindee
       # Converts the given image to a binary stream using Mindee's image utilities, then creates
       # an Origami::Graphics::ImageXObject with a JPEG filter.
       #
-      # @param [Minimagick::Image] image An image object with the necessary data and structure.
+      # @param [MiniMagick::Image] image An image object with the necessary data and structure.
       # @return [Origami::Graphics::ImageXObject] The created image XObject.
       def self.create_xobject(image)
         image_io = Mindee::Image::ImageUtils.image_to_stringio(image)
@@ -153,7 +155,7 @@ module Mindee
 
       # Processes an image into an image XObject for PDF embedding.
       #
-      # @param [Hash] image_data The raw image data.
+      # @param [MiniMagick::Image, StringIO] image_data The raw image data.
       # @param [Integer] image_quality The quality setting for image compression.
       # @param [Numeric] width The desired width of the output image.
       # @param [Numeric] height The desired height of the output image.

data/lib/mindee/product/financial_document/financial_document_v1_document.rb

@@ -10,13 +10,13 @@ module Mindee
       class FinancialDocumentV1Document < Mindee::Parsing::Common::Prediction
         include Mindee::Parsing::Standard
         # The customer's address used for billing.
-        # @return [Mindee::Parsing::Standard::StringField]
+        # @return [Mindee::Parsing::Standard::AddressField]
         attr_reader :billing_address
         # The purchase category.
         # @return [Mindee::Parsing::Standard::ClassificationField]
         attr_reader :category
         # The address of the customer.
-        # @return [Mindee::Parsing::Standard::StringField]
+        # @return [Mindee::Parsing::Standard::AddressField]
         attr_reader :customer_address
         # List of company registration numbers associated to the customer.
         # @return [Array<Mindee::Parsing::Standard::CompanyRegistrationField>]
@@ -65,13 +65,13 @@ module Mindee
         # @return [Array<Mindee::Parsing::Standard::StringField>]
         attr_reader :reference_numbers
         # The customer's address used for shipping.
-        # @return [Mindee::Parsing::Standard::StringField]
+        # @return [Mindee::Parsing::Standard::AddressField]
         attr_reader :shipping_address
         # The purchase subcategory for transport, food and shooping.
         # @return [Mindee::Parsing::Standard::ClassificationField]
         attr_reader :subcategory
         # The address of the supplier or merchant.
-        # @return [Mindee::Parsing::Standard::StringField]
+        # @return [Mindee::Parsing::Standard::AddressField]
         attr_reader :supplier_address
         # List of company registration numbers associated to the supplier.
         # @return [Array<Mindee::Parsing::Standard::CompanyRegistrationField>]
@@ -114,7 +114,7 @@ module Mindee
         # @param page_id [Integer, nil]
         def initialize(prediction, page_id)
           super
-          @billing_address = Parsing::Standard::StringField.new(
+          @billing_address = Parsing::Standard::AddressField.new(
             prediction['billing_address'],
             page_id
           )
@@ -122,7 +122,7 @@ module Mindee
             prediction['category'],
             page_id
           )
-          @customer_address = Parsing::Standard::StringField.new(
+          @customer_address = Parsing::Standard::AddressField.new(
             prediction['customer_address'],
             page_id
           )
@@ -182,7 +182,7 @@ module Mindee
           prediction['reference_numbers'].each do |item|
             @reference_numbers.push(Parsing::Standard::StringField.new(item, page_id))
           end
-          @shipping_address = Parsing::Standard::StringField.new(
+          @shipping_address = Parsing::Standard::AddressField.new(
             prediction['shipping_address'],
             page_id
           )
@@ -190,7 +190,7 @@ module Mindee
             prediction['subcategory'],
             page_id
           )
-          @supplier_address = Parsing::Standard::StringField.new(
+          @supplier_address = Parsing::Standard::AddressField.new(
             prediction['supplier_address'],
             page_id
           )

data/lib/mindee/product/invoice/invoice_v4_document.rb

@@ -10,13 +10,13 @@ module Mindee
       class InvoiceV4Document < Mindee::Parsing::Common::Prediction
         include Mindee::Parsing::Standard
         # The customer billing address.
-        # @return [Mindee::Parsing::Standard::StringField]
+        # @return [Mindee::Parsing::Standard::AddressField]
         attr_reader :billing_address
         # The purchase category.
         # @return [Mindee::Parsing::Standard::ClassificationField]
         attr_reader :category
         # The address of the customer.
-        # @return [Mindee::Parsing::Standard::StringField]
+        # @return [Mindee::Parsing::Standard::AddressField]
         attr_reader :customer_address
         # List of company registration numbers associated to the customer.
         # @return [Array<Mindee::Parsing::Standard::CompanyRegistrationField>]
@@ -58,13 +58,13 @@ module Mindee
         # @return [Array<Mindee::Parsing::Standard::StringField>]
         attr_reader :reference_numbers
         # Customer's delivery address.
-        # @return [Mindee::Parsing::Standard::StringField]
+        # @return [Mindee::Parsing::Standard::AddressField]
         attr_reader :shipping_address
         # The purchase subcategory for transport, food and shopping.
         # @return [Mindee::Parsing::Standard::ClassificationField]
         attr_reader :subcategory
         # The address of the supplier or merchant.
-        # @return [Mindee::Parsing::Standard::StringField]
+        # @return [Mindee::Parsing::Standard::AddressField]
         attr_reader :supplier_address
         # List of company registration numbers associated to the supplier.
         # @return [Array<Mindee::Parsing::Standard::CompanyRegistrationField>]
@@ -101,7 +101,7 @@ module Mindee
         # @param page_id [Integer, nil]
         def initialize(prediction, page_id)
           super
-          @billing_address = Parsing::Standard::StringField.new(
+          @billing_address = Parsing::Standard::AddressField.new(
             prediction['billing_address'],
             page_id
           )
@@ -109,7 +109,7 @@ module Mindee
             prediction['category'],
             page_id
           )
-          @customer_address = Parsing::Standard::StringField.new(
+          @customer_address = Parsing::Standard::AddressField.new(
             prediction['customer_address'],
             page_id
           )
@@ -159,7 +159,7 @@ module Mindee
           prediction['reference_numbers'].each do |item|
             @reference_numbers.push(Parsing::Standard::StringField.new(item, page_id))
           end
-          @shipping_address = Parsing::Standard::StringField.new(
+          @shipping_address = Parsing::Standard::AddressField.new(
             prediction['shipping_address'],
             page_id
           )
@@ -167,7 +167,7 @@ module Mindee
             prediction['subcategory'],
             page_id
           )
-          @supplier_address = Parsing::Standard::StringField.new(
+          @supplier_address = Parsing::Standard::AddressField.new(
             prediction['supplier_address'],
             page_id
           )

data/lib/mindee/product/universal/universal_prediction.rb

@@ -16,7 +16,7 @@ module Mindee
 
         def initialize(_ = nil)
           super
-          @fields = {} # : Hash[Symbol | String, untyped]
+          @fields = {} # : Hash[String | Symbol, untyped]
         end
 
         # String representation.
@@ -81,7 +81,7 @@ module Mindee
         # Returns a hash of all fields that aren't a collection
         # @return [Hash<String, StringField>]
         def single_fields
-          single_fields = {} # : Hash[Symbol | String, untyped]
+          single_fields = {} # : Hash[String | Symbol, untyped]
           @fields.each do |field_name, field_value|
             single_fields[field_name] = field_value if field_value.is_a?(Mindee::Parsing::Standard::StringField)
           end
@@ -91,7 +91,7 @@ module Mindee
         # Returns a hash of all list-like fields
         # @return [Hash<String, UniversalListField>]
         def list_fields
-          list_fields = {} # : Hash[Symbol | String, Mindee::Parsing::Universal::UniversalListField]
+          list_fields = {} # : Hash[String | Symbol, Mindee::Parsing::Universal::UniversalListField]
           @fields.each do |field_name, field_value|
             list_fields[field_name] = field_value if field_value.is_a?(Mindee::Parsing::Universal::UniversalListField)
           end
@@ -101,7 +101,7 @@ module Mindee
         # Returns a hash of all object-like fields
         # @return [Hash<String, UniversalObjectField>]
         def object_fields
-          object_fields = {} # : Hash[Symbol | String, untyped]
+          object_fields = {} # : Hash[String | Symbol, untyped]
           @fields.each do |field_name, field_value|
             if field_value.is_a?(Mindee::Parsing::Universal::UniversalObjectField)
               object_fields[field_name] =

data/lib/mindee/version.rb

@@ -3,10 +3,10 @@
 # Mindee
 module Mindee
   # Current version.
-  VERSION = '4.5.0'
+  VERSION = '4.7.0-rc1'
 
   # Finds and return the current platform.
-  # @return [Symbol, Hash[Symbol | String, Regexp], Nil?]
+  # @return [Symbol, Hash[String | Symbol, Regexp], Nil?]
   def self.find_platform
     host = RbConfig::CONFIG['host_os']
     platforms = {

data/lib/mindee.rb

@@ -1,6 +1,8 @@
 # frozen_string_literal: true
 
 require 'mindee/client'
+require 'mindee/client_v2'
+require 'mindee/page_options'
 require 'mindee/logging'
 
 module Mindee
@@ -54,6 +56,10 @@ module Mindee
     # Universal fields and functions.
     module Universal
     end
+
+    # V2-specific module.
+    module V2
+    end
   end
 
   # Document input-related internals.

data/mindee.gemspec

@@ -37,7 +37,7 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency 'rake', '~> 13.2'
   spec.add_development_dependency 'rbs', '~> 3.6'
   spec.add_development_dependency 'rspec', '~> 3.13'
-  spec.add_development_dependency 'rubocop', '~> 1.75'
+  spec.add_development_dependency 'rubocop', '~> 1.76.0'
   spec.add_development_dependency 'steep', '~> 1.7'
   spec.add_development_dependency 'yard', '~> 0.9'
 end

data/sig/custom/mini_magick.rbs

@@ -1,24 +1,31 @@
 # Stub for the mini_magick library.
+# Note: though typing annotations for the MiniMagick library now exist, it seems that they aren't strict enough
+# to match the rules we have on the repo, hence the existence of this file and the overrides present below.
 module MiniMagick
   class Image
-    def format: (String, ?Integer?, ?Hash[Symbol, untyped]?) -> Net::BufferedIO
     def self.format: (String, ?Integer?, ?Hash[Symbol, untyped]?) -> Net::BufferedIO
+    def format: (String, ?Integer?, ?Hash[Symbol, untyped]?) -> Net::BufferedIO
     def self.quality: (String) -> Net::BufferedIO
     def self.crop: (String) -> Net::BufferedIO
+    def crop: (String) -> Net::BufferedIO
     def self.height: () -> Integer
-    def height: () -> Integer
     def self.width: () -> Integer
+    def height: () -> Integer
     def width: () -> Integer
-    def self.read: (untyped) -> self
+    def self.data: (?untyped?, ?untyped?) -> untyped
+    def data: (?untyped?, ?untyped) -> instance
+    def self.read: (?untyped?, ?untyped?) -> Image
+    def read: (?untyped?, ?untyped) -> Image
     def self.resolution: () -> Array[Float | Integer]
+    def resolution: () -> Array[Float | Integer]
     def self.write: (StringIO | String) -> Net::BufferedIO
     def write: (StringIO | String) -> Net::BufferedIO
-    def quality: (Integer) -> void
+    def quality: (String) -> void
     def self.resize: (String) -> void
     def resize: (String) -> void
     def self.to_blob: () -> String
     def to_blob: () -> String
-    def self.[]: (?untyped) -> untyped
-    def []: (?untyped) -> untyped
+    def self.[]: (*untyped) -> untyped
+    def []: (*untyped) -> untyped
   end
-end
+end

data/sig/custom/net_http.rbs

@@ -12,7 +12,7 @@ module Net
 
     # Stub for the HTTP POST request class.
     class Post
-      def initialize: (untyped, Hash[String, String]?) -> void
+      def initialize: (untyped, Hash[String, String?]?) -> void
       def set_form: (untyped, String?) -> void
       def new: (untyped, untyped) -> void
       def []=: (?untyped, ?untyped) -> bool
@@ -20,7 +20,7 @@ module Net
 
     # Stub for the HTTP GET request class.
     class Get
-      def initialize: (untyped, ?Hash[String, String]?) -> void
+      def initialize: (untyped, ?Hash[String, String?]?) -> void
       def new: (untyped, untyped) -> void
       def basic_auth: (untyped, untyped) -> void
       def []=: (?untyped, ?untyped) -> bool
@@ -28,8 +28,8 @@ module Net
   end
 
   class HTTPResponse
-    def self.body: -> untyped
-    def body: -> untyped
+    def self.body: -> String
+    def body: -> String
     def []: (untyped) -> untyped
     def key?: (untyped) -> bool
     def code: -> String

data/sig/custom/origami.rbs

@@ -1,21 +1,22 @@
 # Stubs for the origami library.
 # This one _should_ exist, but it would take too long, so this is a stub.
 module Origami
+  def compile: (Hash[Symbol, untyped]) -> void
   class Array
     def each: { (untyped) -> untyped } -> untyped
   end
   class PDF
-    def self.read: (StringIO) -> untyped
+    def self.read: (StringIO | File) -> untyped
     def initialize: () -> void
     def append_page: (Page) -> void
     def delete_pages_at: (::Array[Integer]) -> void
-    def pages: () -> untyped
-    def save: (StringIO) -> void
+    def pages: () -> ::Array[Page]
+    def save: (StringIO | File) -> void
     def to_io_stream: -> StringIO
     class LinearParser
       def initialize: (Hash[Symbol, untyped]) -> void
       def new: (Hash[Symbol, untyped]) -> void
-      def parse: (StringIO?) -> PDF
+      def parse: (StringIO | File?) -> PDF
     end
   end
 
@@ -23,6 +24,9 @@ module Origami
     def initialize: () -> void
     def add_xobject: (Graphics::ImageXObject, String) -> untyped
     def Contents: (?untyped?) -> untyped
+    def Contents=: (?untyped?, untyped?) -> untyped
+    def []: (untyped) -> untyped
+    def []=: (untyped, untyped) -> bool
   end
 
   module Graphics
@@ -31,6 +35,7 @@ module Origami
 
       def BitsPerComponent=: (Integer) -> untyped
       def ColorSpace=: (Symbol) -> untyped
+      def dictionary: () -> Hash[Symbol, untyped]
       def data=: (StringIO) -> untyped
       def self.from_image_file: (StringIO, String) -> untyped
       def Height=: (Integer | Float) -> untyped

data/sig/mindee/client.rbs

@@ -1,16 +1,8 @@
-OTS_OWNER: String
+# lib/mindee/client.rb
 
 module Mindee
   OTS_OWNER: String
 
-  class PageOptions
-    attr_accessor page_indexes: Array[Integer]
-    attr_accessor operation: :KEEP_ONLY | :REMOVE
-    attr_accessor on_min_pages: Integer?
-
-    def initialize: (params: Hash[Symbol | String, untyped]) -> void
-  end
-
   class ParseOptions
     attr_accessor all_words: bool
     attr_accessor full_text: bool
@@ -18,12 +10,12 @@ module Mindee
     attr_accessor page_options: PageOptions
     attr_accessor cropper: bool
     attr_accessor rag: bool
-    attr_accessor workflow_id: String | nil
+    attr_accessor workflow_id: String?
     attr_accessor initial_delay_sec: Integer | Float
     attr_accessor delay_sec: Integer | Float
     attr_accessor max_retries: Integer
 
-    def initialize: (params: Hash[Symbol | String, untyped]) -> void
+    def initialize: (params: Hash[String | Symbol, untyped]) -> void
   end
 
   class WorkflowOptions
@@ -33,19 +25,22 @@ module Mindee
     attr_accessor rag: bool?
     attr_accessor public_url: (String?)
     attr_accessor page_options: (PageOptions)
+    attr_accessor close_file: bool
 
-    def initialize: (params: Hash[Symbol | String, untyped]) -> void
+    def initialize: (params: Hash[String | Symbol, untyped]) -> void
   end
 
   class Client
+    @api_key: String?
+
     def initialize: (?api_key: String) -> void
-    def logger: () -> untyped
-    def parse: (Input::Source::LocalInputSource | Input::Source::URLInputSource, untyped, ?endpoint: (HTTP::Endpoint?), options: ParseOptions | Hash[Symbol | String, untyped]) -> Parsing::Common::ApiResponse
-    def parse_sync: (Input::Source::LocalInputSource | Input::Source::URLInputSource, untyped, HTTP::Endpoint, ParseOptions) -> Parsing::Common::ApiResponse
-    def enqueue: (Input::Source::LocalInputSource | Input::Source::URLInputSource, untyped, ?endpoint: (HTTP::Endpoint?), options: ParseOptions|Hash[Symbol | String, untyped]) -> Parsing::Common::ApiResponse
+    def logger: () -> Logging
+    def parse: (Input::Source::LocalInputSource | Input::Source::URLInputSource, singleton(Parsing::Common::Inference), ?endpoint: (HTTP::Endpoint?), options: ParseOptions | Hash[String | Symbol, untyped]) -> Parsing::Common::ApiResponse
+    def parse_sync: (Input::Source::LocalInputSource | Input::Source::URLInputSource, singleton(Parsing::Common::Inference), HTTP::Endpoint, ParseOptions) -> Parsing::Common::ApiResponse
+    def enqueue: (Input::Source::LocalInputSource | Input::Source::URLInputSource, singleton(Parsing::Common::Inference), ?endpoint: (HTTP::Endpoint?), options: ParseOptions|Hash[String | Symbol, untyped]) -> Parsing::Common::ApiResponse
     def parse_queued: (String, singleton(Parsing::Common::Inference), ?endpoint: HTTP::Endpoint?) -> Parsing::Common::ApiResponse
-    def enqueue_and_parse: (Input::Source::URLInputSource|Input::Source::LocalInputSource, untyped, HTTP::Endpoint, ParseOptions) -> Parsing::Common::ApiResponse
-    def execute_workflow: (Input::Source::URLInputSource|Input::Source::LocalInputSource, String, options: (Hash[Symbol | String, untyped] | WorkflowOptions)) -> Parsing::Common::WorkflowResponse
+    def enqueue_and_parse: (Input::Source::URLInputSource|Input::Source::LocalInputSource, singleton(Parsing::Common::Inference), HTTP::Endpoint, ParseOptions) -> Parsing::Common::ApiResponse
+    def execute_workflow: (Input::Source::URLInputSource|Input::Source::LocalInputSource, String, options: (Hash[String | Symbol, untyped] | WorkflowOptions)) -> Parsing::Common::WorkflowResponse
     def load_prediction: (singleton(Parsing::Common::Inference), Input::LocalResponse) -> Parsing::Common::ApiResponse
     def source_from_path: (String, ?repair_pdf: bool) -> Input::Source::PathInputSource
     def source_from_bytes: (String, String, ?repair_pdf: bool) -> Input::Source::BytesInputSource
@@ -55,7 +50,7 @@ module Mindee
     def create_endpoint: (endpoint_name: String, account_name: String, version: String) -> HTTP::Endpoint
 
     private
-    def normalize_parse_options: ((Hash[Symbol | String, untyped] | ParseOptions)) -> ParseOptions
+    def normalize_parse_options: ((Hash[String | Symbol, untyped] | ParseOptions)) -> ParseOptions
     def process_pdf_if_required: (Input::Source::LocalInputSource, ParseOptions | WorkflowOptions) -> void
     def initialize_endpoint: (singleton(Parsing::Common::Inference), ?endpoint_name: String, ?account_name: String, ?version: String) -> HTTP::Endpoint
     def validate_async_params: (Integer | Float, Integer | Float, Integer) -> void

data/sig/mindee/client_v2.rbs

@@ -0,0 +1,17 @@
+# lib/mindee/client_v2.rb
+
+OTS_OWNER: String
+module Mindee
+  class ClientV2
+    attr_reader mindee_api: HTTP::MindeeApiV2
+
+    def logger: () -> Logger
+    def initialize: (?api_key: String) -> void
+    def get_inference: (String) -> Parsing::V2::InferenceResponse
+    def get_job: (String) -> Parsing::V2::JobResponse
+    def enqueue_inference: (Input::Source::LocalInputSource | Input::Source::URLInputSource, Hash[String | Symbol, untyped] | Input::InferenceParameters) -> Parsing::V2::JobResponse
+    def enqueue_and_get_inference: (Input::Source::LocalInputSource | Input::Source::URLInputSource, Hash[String | Symbol, untyped] | Input::InferenceParameters) -> Parsing::V2::InferenceResponse
+    def validate_async_params: (Integer | Float, Integer | Float, Integer) -> void
+    def normalize_inference_parameters: (Hash[String | Symbol, untyped] | Input::InferenceParameters) -> Input::InferenceParameters
+  end
+end