retab 0.1.0 → 0.1.1

data/lib/retab/extractions.rb

@@ -2,7 +2,7 @@
 
 # This file is auto-generated by oagen. Do not edit.
 
-require 'json'
+require "json"
 
 module Retab
   class Extractions
@@ -23,12 +23,12 @@ module Retab
     # @param to_date [String, nil]
     # @param metadata [String, nil]
     # @param request_options [Hash] (see Retab::Types::RequestOptions)
-    # @return [Retab::Types::ListStruct<Retab::Extraction>]
+    # @return [Retab::PaginatedList<Retab::Extraction>]
     def list(
       before: nil,
       after: nil,
       limit: 10,
-      order: 'desc',
+      order: "desc",
       filename: nil,
       filename_regex: nil,
       filename_contains: nil,
@@ -39,26 +39,26 @@ module Retab
       request_options: {}
     )
       params = {
-        'before' => before,
-        'after' => after,
-        'limit' => limit,
-        'order' => order,
-        'filename' => filename,
-        'filename_regex' => filename_regex,
-        'filename_contains' => filename_contains,
-        'document_type' => document_type,
-        'from_date' => from_date,
-        'to_date' => to_date,
-        'metadata' => metadata
+        "before" => before,
+        "after" => after,
+        "limit" => limit,
+        "order" => order,
+        "filename" => filename,
+        "filename_regex" => filename_regex,
+        "filename_contains" => filename_contains,
+        "document_type" => document_type,
+        "from_date" => from_date,
+        "to_date" => to_date,
+        "metadata" => metadata
       }.compact
       response = @client.request(
         method: :get,
-        path: '/v1/extractions',
+        path: "/v1/extractions",
         auth: true,
         params: params,
         request_options: request_options
       )
-      fetch_next = ->(cursor) {
+      fetch_next = -> (cursor) {
         list(
           before: before,
           after: cursor,
@@ -74,10 +74,21 @@ module Retab
           request_options: request_options
         )
       }
-      Retab::Types::ListStruct.from_response(
+      Retab::PaginatedList.from_response(
         response,
         model: Retab::Extraction,
-        filters: { before: before, limit: limit, order: order, filename: filename, filename_regex: filename_regex, filename_contains: filename_contains, document_type: document_type, from_date: from_date, to_date: to_date, metadata: metadata },
+        filters: {
+          before: before,
+          limit: limit,
+          order: order,
+          filename: filename,
+          filename_regex: filename_regex,
+          filename_contains: filename_contains,
+          document_type: document_type,
+          from_date: from_date,
+          to_date: to_date,
+          metadata: metadata
+        },
         fetch_next: fetch_next
       )
     end
@@ -93,6 +104,7 @@ module Retab
     # @param additional_messages [Array<Hash{String => Object}>, nil] Additional chat messages forwarded to the extraction model.
     # @param bust_cache [Boolean, nil] If true, skip the LLM cache and force a fresh completion
     # @param stream [Boolean, nil]
+    # @param chunking_keys [Hash{String => String}, nil]
     # @param request_options [Hash] (see Retab::Types::RequestOptions)
     # @return [Retab::Extraction]
     def create(
@@ -106,30 +118,36 @@ module Retab
       additional_messages: nil,
       bust_cache: nil,
       stream: nil,
+      chunking_keys: nil,
       request_options: {}
     )
       document = Retab::MimeData.coerce(document) unless document.nil?
       body = {
-        'document' => document,
-        'json_schema' => json_schema,
-        'model' => model,
-        'image_resolution_dpi' => image_resolution_dpi,
-        'instructions' => instructions,
-        'n_consensus' => n_consensus,
-        'metadata' => metadata,
-        'additional_messages' => additional_messages,
-        'bust_cache' => bust_cache,
-        'stream' => stream
+        "document" => document,
+        "json_schema" => json_schema,
+        "model" => model,
+        "image_resolution_dpi" => image_resolution_dpi,
+        "instructions" => instructions,
+        "n_consensus" => n_consensus,
+        "metadata" => metadata,
+        "additional_messages" => additional_messages,
+        "bust_cache" => bust_cache,
+        "stream" => stream,
+        "chunking_keys" => chunking_keys
       }.compact
       response = @client.request(
         method: :post,
-        path: '/v1/extractions',
+        path: "/v1/extractions",
         auth: true,
         body: body,
         request_options: request_options
       )
       result = Retab::Extraction.new(response.body)
-      result.last_response = Retab::Types::ApiResponse.new(http_status: response.code.to_i, http_headers: response.each_header.to_h, request_id: response["x-request-id"])
+      result.last_response = Retab::Types::ApiResponse.new(
+        http_status: response.code.to_i,
+        http_headers: response.each_header.to_h,
+        request_id: response["x-request-id"]
+      )
       result
     end
 
@@ -148,7 +166,11 @@ module Retab
         request_options: request_options
       )
       result = Retab::Extraction.new(response.body)
-      result.last_response = Retab::Types::ApiResponse.new(http_status: response.code.to_i, http_headers: response.each_header.to_h, request_id: response["x-request-id"])
+      result.last_response = Retab::Types::ApiResponse.new(
+        http_status: response.code.to_i,
+        http_headers: response.each_header.to_h,
+        request_id: response["x-request-id"]
+      )
       result
     end
 
@@ -160,7 +182,7 @@ module Retab
       extraction_id:,
       request_options: {}
     )
-      response = @client.request(
+      @client.request(
         method: :delete,
         path: "/v1/extractions/#{Retab::Util.encode_path(extraction_id)}",
         auth: true,
@@ -184,7 +206,11 @@ module Retab
         request_options: request_options
       )
       result = Retab::SourcesResponse.new(response.body)
-      result.last_response = Retab::Types::ApiResponse.new(http_status: response.code.to_i, http_headers: response.each_header.to_h, request_id: response["x-request-id"])
+      result.last_response = Retab::Types::ApiResponse.new(
+        http_status: response.code.to_i,
+        http_headers: response.each_header.to_h,
+        request_id: response["x-request-id"]
+      )
       result
     end
   end

data/lib/retab/files/complete_file_upload_request.rb

@@ -12,6 +12,7 @@ module Retab
     attr_accessor :sha_256
 
     def initialize(json)
+      super()
       hash = self.class.normalize(json)
       @sha_256 = hash[:sha256]
     end

data/lib/retab/files/create_upload_response.rb

@@ -14,15 +14,17 @@ module Retab
       expiresAt: :expires_at
     }.freeze
 
-    attr_accessor \
+    attr_accessor(
       :file_id,
       :upload_url,
       :upload_method,
       :upload_headers,
       :mime_data,
       :expires_at
+    )
 
     def initialize(json)
+      super()
       hash = self.class.normalize(json)
       @file_id = hash[:fileId]
       @upload_url = hash[:uploadUrl]

data/lib/retab/files/file.rb

@@ -9,24 +9,29 @@ module Retab
       object: :object,
       id: :id,
       filename: :filename,
+      mime_type: :mime_type,
       created_at: :created_at,
       updated_at: :updated_at,
       page_count: :page_count
     }.freeze
 
-    attr_accessor \
+    attr_accessor(
       :object,
       :id,
       :filename,
+      :mime_type,
       :created_at,
       :updated_at,
       :page_count
+    )
 
     def initialize(json)
+      super()
       hash = self.class.normalize(json)
       @object = hash[:object]
       @id = hash[:id]
       @filename = hash[:filename]
+      @mime_type = hash[:mime_type]
       @created_at = hash[:created_at]
       @updated_at = hash[:updated_at]
       @page_count = hash[:page_count]

data/lib/retab/files/file_link.rb

@@ -12,13 +12,15 @@ module Retab
       mime_data: :mime_data
     }.freeze
 
-    attr_accessor \
+    attr_accessor(
       :download_url,
       :expires_in,
       :filename,
       :mime_data
+    )
 
     def initialize(json)
+      super()
       hash = self.class.normalize(json)
       @download_url = hash[:download_url]
       @expires_in = hash[:expires_in]

data/lib/retab/files/upload_file_request.rb

@@ -12,13 +12,15 @@ module Retab
       sha256: :sha_256
     }.freeze
 
-    attr_accessor \
+    attr_accessor(
       :filename,
       :content_type,
       :size_bytes,
       :sha_256
+    )
 
     def initialize(json)
+      super()
       hash = self.class.normalize(json)
       @filename = hash[:filename]
       @content_type = hash[:content_type]

data/lib/retab/files.rb

@@ -2,7 +2,7 @@
 
 # This file is auto-generated by oagen. Do not edit.
 
-require 'json'
+require "json"
 
 module Retab
   class Files
@@ -17,7 +17,7 @@ module Retab
     # @param sha_256 [String, nil] Optional SHA-256 checksum
     # @param request_options [Hash] (see Retab::Types::RequestOptions)
     # @return [Retab::CreateUploadResponse]
-    def upload(
+    def create_upload(
       filename:,
       size_bytes:,
       content_type: nil,
@@ -25,20 +25,24 @@ module Retab
       request_options: {}
     )
       body = {
-        'filename' => filename,
-        'content_type' => content_type,
-        'size_bytes' => size_bytes,
-        'sha256' => sha_256
+        "filename" => filename,
+        "content_type" => content_type,
+        "size_bytes" => size_bytes,
+        "sha256" => sha_256
       }.compact
       response = @client.request(
         method: :post,
-        path: '/v1/files/upload',
+        path: "/v1/files/upload",
         auth: true,
         body: body,
         request_options: request_options
       )
       result = Retab::CreateUploadResponse.new(response.body)
-      result.last_response = Retab::Types::ApiResponse.new(http_status: response.code.to_i, http_headers: response.each_header.to_h, request_id: response["x-request-id"])
+      result.last_response = Retab::Types::ApiResponse.new(
+        http_status: response.code.to_i,
+        http_headers: response.each_header.to_h,
+        request_id: response["x-request-id"]
+      )
       result
     end
 
@@ -53,7 +57,7 @@ module Retab
       request_options: {}
     )
       body = {
-        'sha256' => sha_256
+        "sha256" => sha_256
       }.compact
       response = @client.request(
         method: :post,
@@ -63,7 +67,11 @@ module Retab
         request_options: request_options
       )
       result = Retab::MimeData.new(response.body)
-      result.last_response = Retab::Types::ApiResponse.new(http_status: response.code.to_i, http_headers: response.each_header.to_h, request_id: response["x-request-id"])
+      result.last_response = Retab::Types::ApiResponse.new(
+        http_status: response.code.to_i,
+        http_headers: response.each_header.to_h,
+        request_id: response["x-request-id"]
+      )
       result
     end
 
@@ -79,40 +87,40 @@ module Retab
     # @param include_embeddings [Boolean, nil] Include embeddings in the response
     # @param sort_by [String, nil]
     # @param request_options [Hash] (see Retab::Types::RequestOptions)
-    # @return [Retab::Types::ListStruct<Retab::FileModel>]
+    # @return [Retab::PaginatedList<Retab::FileModel>]
     def list(
       before: nil,
       after: nil,
       limit: 10,
-      order: 'desc',
+      order: "desc",
       filename: nil,
       mime_type: nil,
       from_date: nil,
       to_date: nil,
       include_embeddings: false,
-      sort_by: 'created_at',
+      sort_by: "created_at",
       request_options: {}
     )
       params = {
-        'before' => before,
-        'after' => after,
-        'limit' => limit,
-        'order' => order,
-        'filename' => filename,
-        'mime_type' => mime_type,
-        'from_date' => from_date,
-        'to_date' => to_date,
-        'include_embeddings' => include_embeddings,
-        'sort_by' => sort_by
+        "before" => before,
+        "after" => after,
+        "limit" => limit,
+        "order" => order,
+        "filename" => filename,
+        "mime_type" => mime_type,
+        "from_date" => from_date,
+        "to_date" => to_date,
+        "include_embeddings" => include_embeddings,
+        "sort_by" => sort_by
       }.compact
       response = @client.request(
         method: :get,
-        path: '/v1/files',
+        path: "/v1/files",
         auth: true,
         params: params,
         request_options: request_options
       )
-      fetch_next = ->(cursor) {
+      fetch_next = -> (cursor) {
         list(
           before: before,
           after: cursor,
@@ -127,10 +135,20 @@ module Retab
           request_options: request_options
         )
       }
-      Retab::Types::ListStruct.from_response(
+      Retab::PaginatedList.from_response(
         response,
         model: Retab::FileModel,
-        filters: { before: before, limit: limit, order: order, filename: filename, mime_type: mime_type, from_date: from_date, to_date: to_date, include_embeddings: include_embeddings, sort_by: sort_by },
+        filters: {
+          before: before,
+          limit: limit,
+          order: order,
+          filename: filename,
+          mime_type: mime_type,
+          from_date: from_date,
+          to_date: to_date,
+          include_embeddings: include_embeddings,
+          sort_by: sort_by
+        },
         fetch_next: fetch_next
       )
     end
@@ -150,7 +168,11 @@ module Retab
         request_options: request_options
       )
       result = Retab::FileModel.new(response.body)
-      result.last_response = Retab::Types::ApiResponse.new(http_status: response.code.to_i, http_headers: response.each_header.to_h, request_id: response["x-request-id"])
+      result.last_response = Retab::Types::ApiResponse.new(
+        http_status: response.code.to_i,
+        http_headers: response.each_header.to_h,
+        request_id: response["x-request-id"]
+      )
       result
     end
 
@@ -169,7 +191,11 @@ module Retab
         request_options: request_options
       )
       result = Retab::FileLink.new(response.body)
-      result.last_response = Retab::Types::ApiResponse.new(http_status: response.code.to_i, http_headers: response.each_header.to_h, request_id: response["x-request-id"])
+      result.last_response = Retab::Types::ApiResponse.new(
+        http_status: response.code.to_i,
+        http_headers: response.each_header.to_h,
+        request_id: response["x-request-id"]
+      )
       result
     end
   end

data/lib/retab/jobs/create_job_request.rb

@@ -11,12 +11,14 @@ module Retab
       metadata: :metadata
     }.freeze
 
-    attr_accessor \
+    attr_accessor(
       :endpoint,
       :request,
       :metadata
+    )
 
     def initialize(json)
+      super()
       hash = self.class.normalize(json)
       @endpoint = hash[:endpoint]
       @request = hash[:request] || {}

data/lib/retab/jobs/job.rb

@@ -25,7 +25,7 @@ module Retab
       response: :response
     }.freeze
 
-    attr_accessor \
+    attr_accessor(
       :id,
       :object,
       :status,
@@ -43,8 +43,10 @@ module Retab
       :last_failure_code,
       :request,
       :response
+    )
 
     def initialize(json)
+      super()
       hash = self.class.normalize(json)
       @id = hash[:id]
       @object = hash[:object]

data/lib/retab/jobs/job_error.rb

@@ -11,12 +11,14 @@ module Retab
       details: :details
     }.freeze
 
-    attr_accessor \
+    attr_accessor(
       :code,
       :message,
       :details
+    )
 
     def initialize(json)
+      super()
       hash = self.class.normalize(json)
       @code = hash[:code]
       @message = hash[:message]

data/lib/retab/jobs/job_response.rb

@@ -10,11 +10,13 @@ module Retab
       body: :body
     }.freeze
 
-    attr_accessor \
+    attr_accessor(
       :status_code,
       :body
+    )
 
     def initialize(json)
+      super()
       hash = self.class.normalize(json)
       @status_code = hash[:status_code]
       @body = hash[:body] || {}