retab 0.1.0 → 0.1.1

data/lib/retab/experiment_run_metrics/metrics_stale_error_last_run.rb

@@ -12,13 +12,15 @@ module Retab
       created_at: :created_at
     }.freeze
 
-    attr_accessor \
+    attr_accessor(
       :run_id,
       :definition_fingerprint,
       :score,
       :created_at
+    )
 
     def initialize(json)
+      super()
       hash = self.class.normalize(json)
       @run_id = hash[:run_id]
       @definition_fingerprint = hash[:definition_fingerprint]

data/lib/retab/experiment_run_metrics.rb

@@ -2,7 +2,7 @@
 
 # This file is auto-generated by oagen. Do not edit.
 
-require 'json'
+require "json"
 
 module Retab
   class ExperimentRunMetrics
@@ -21,7 +21,7 @@ module Retab
     # @return [Retab::ExperimentSummaryMetricsResponse, Retab::ExperimentByDocumentMetricsResponse, Retab::ExperimentByTargetMetricsResponse, Retab::ExperimentVotesMetricsResponse, Retab::ExperimentMetricsStaleError, Retab::ExperimentMetricsMissingError]
     def get(
       run_id:,
-      view: 'summary',
+      view: "summary",
       document_id: nil,
       target_path: nil,
       include_prior: true,
@@ -29,16 +29,16 @@ module Retab
       request_options: {}
     )
       params = {
-        'run_id' => run_id,
-        'view' => view,
-        'document_id' => document_id,
-        'target_path' => target_path,
-        'include_prior' => include_prior,
-        'prior_run_id' => prior_run_id
+        "run_id" => run_id,
+        "view" => view,
+        "document_id" => document_id,
+        "target_path" => target_path,
+        "include_prior" => include_prior,
+        "prior_run_id" => prior_run_id
       }.compact
       response = @client.request(
         method: :get,
-        path: '/v1/workflows/experiments/metrics',
+        path: "/v1/workflows/experiments/metrics",
         auth: true,
         params: params,
         request_options: request_options

data/lib/retab/experiment_run_results/error_workflow_experiment_result.rb

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

data/lib/retab/experiment_run_results/experiment_result.rb

@@ -19,7 +19,7 @@ module Retab
       is_placeholder: :is_placeholder
     }.freeze
 
-    attr_accessor \
+    attr_accessor(
       :id,
       :run_id,
       :experiment_id,
@@ -31,14 +31,33 @@ module Retab
       :artifact,
       :attempt,
       :is_placeholder
+    )
 
     def initialize(json)
+      super()
       hash = self.class.normalize(json)
       @id = hash[:id]
       @run_id = hash[:run_id]
       @experiment_id = hash[:experiment_id]
       @document_id = hash[:document_id]
-      @lifecycle = hash[:lifecycle] ? (case hash[:lifecycle][:status] when "cancelled" then Retab::CancelledWorkflowExperimentResult.new(hash[:lifecycle]) when "completed" then Retab::CompletedWorkflowExperimentResult.new(hash[:lifecycle]) when "error" then Retab::ErrorWorkflowExperimentResult.new(hash[:lifecycle]) when "pending" then Retab::PendingWorkflowExperimentResult.new(hash[:lifecycle]) when "queued" then Retab::QueuedWorkflowExperimentResult.new(hash[:lifecycle]) when "running" then Retab::RunningWorkflowExperimentResult.new(hash[:lifecycle]) else hash[:lifecycle] end) : nil
+      @lifecycle = hash[:lifecycle] ? (
+        case hash[:lifecycle][:status]
+        when "cancelled"
+          Retab::CancelledWorkflowExperimentResult.new(hash[:lifecycle])
+        when "completed"
+          Retab::CompletedWorkflowExperimentResult.new(hash[:lifecycle])
+        when "error"
+          Retab::ErrorWorkflowExperimentResult.new(hash[:lifecycle])
+        when "pending"
+          Retab::PendingWorkflowExperimentResult.new(hash[:lifecycle])
+        when "queued"
+          Retab::QueuedWorkflowExperimentResult.new(hash[:lifecycle])
+        when "running"
+          Retab::RunningWorkflowExperimentResult.new(hash[:lifecycle])
+        else
+          hash[:lifecycle]
+        end
+      ) : nil
       @timing = hash[:timing] ? Retab::ExperimentResultTiming.new(hash[:timing]) : nil
       @block_type = hash[:block_type]
       @handle_inputs = hash[:handle_inputs] || {}

data/lib/retab/experiment_run_results/experiment_result_timing.rb

@@ -12,13 +12,15 @@ module Retab
       duration_ms: :duration_ms
     }.freeze
 
-    attr_accessor \
+    attr_accessor(
       :created_at,
       :started_at,
       :completed_at,
       :duration_ms
+    )
 
     def initialize(json)
+      super()
       hash = self.class.normalize(json)
       @created_at = hash[:created_at]
       @started_at = hash[:started_at]

data/lib/retab/experiment_run_results.rb

@@ -2,7 +2,7 @@
 
 # This file is auto-generated by oagen. Do not edit.
 
-require 'json'
+require "json"
 
 module Retab
   class ExperimentRunResults
@@ -12,29 +12,49 @@ module Retab
 
     # List Experiment Results
     # @param run_id [String]
+    # @param before [String, nil]
+    # @param after [String, nil]
     # @param limit [Integer, nil]
+    # @param order [Retab::Types::ExperimentRunResultsOrder, nil]
     # @param request_options [Hash] (see Retab::Types::RequestOptions)
-    # @return [Retab::Types::ListStruct<Retab::ExperimentResult>]
+    # @return [Retab::PaginatedList<Retab::ExperimentResult>]
     def list(
       run_id:,
+      before: nil,
+      after: nil,
       limit: 100,
+      order: "desc",
       request_options: {}
     )
       params = {
-        'run_id' => run_id,
-        'limit' => limit
+        "run_id" => run_id,
+        "before" => before,
+        "after" => after,
+        "limit" => limit,
+        "order" => order
       }.compact
       response = @client.request(
         method: :get,
-        path: '/v1/workflows/experiments/results',
+        path: "/v1/workflows/experiments/results",
         auth: true,
         params: params,
         request_options: request_options
       )
-      Retab::Types::ListStruct.from_response(
+      fetch_next = -> (cursor) {
+        list(
+          run_id: run_id,
+          before: before,
+          after: cursor,
+          limit: limit,
+          order: order,
+          request_options: request_options
+        )
+      }
+      Retab::PaginatedList.from_response(
         response,
         model: Retab::ExperimentResult,
-        filters: { run_id: run_id, limit: limit }
+        filters: {run_id: run_id, before: before, limit: limit, order: order},
+        fetch_next: fetch_next
       )
     end
 
@@ -53,7 +73,11 @@ module Retab
         request_options: request_options
       )
       result = Retab::ExperimentResult.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/experiment_runs/cancel_workflow_experiment_run_response.rb

@@ -10,14 +10,33 @@ module Retab
       lifecycle: :lifecycle
     }.freeze
 
-    attr_accessor \
+    attr_accessor(
       :id,
       :lifecycle
+    )
 
     def initialize(json)
+      super()
       hash = self.class.normalize(json)
       @id = hash[:id]
-      @lifecycle = hash[:lifecycle] ? (case hash[:lifecycle][:status] when "cancelled" then Retab::CancelledWorkflowExperimentRun.new(hash[:lifecycle]) when "completed" then Retab::CompletedWorkflowExperimentRun.new(hash[:lifecycle]) when "error" then Retab::ErrorWorkflowExperimentRun.new(hash[:lifecycle]) when "pending" then Retab::PendingWorkflowExperimentRun.new(hash[:lifecycle]) when "queued" then Retab::QueuedWorkflowExperimentRun.new(hash[:lifecycle]) when "running" then Retab::RunningWorkflowExperimentRun.new(hash[:lifecycle]) else hash[:lifecycle] end) : nil
+      @lifecycle = hash[:lifecycle] ? (
+        case hash[:lifecycle][:status]
+        when "cancelled"
+          Retab::CancelledWorkflowExperimentRun.new(hash[:lifecycle])
+        when "completed"
+          Retab::CompletedWorkflowExperimentRun.new(hash[:lifecycle])
+        when "error"
+          Retab::ErrorWorkflowExperimentRun.new(hash[:lifecycle])
+        when "pending"
+          Retab::PendingWorkflowExperimentRun.new(hash[:lifecycle])
+        when "queued"
+          Retab::QueuedWorkflowExperimentRun.new(hash[:lifecycle])
+        when "running"
+          Retab::RunningWorkflowExperimentRun.new(hash[:lifecycle])
+        else
+          hash[:lifecycle]
+        end
+      ) : nil
     end
   end
 end

data/lib/retab/experiment_runs/create_experiment_run_request.rb

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

data/lib/retab/experiment_runs/experiment_run.rb

@@ -25,7 +25,7 @@ module Retab
       error_count: :error_count
     }.freeze
 
-    attr_accessor \
+    attr_accessor(
       :id,
       :workflow,
       :trigger,
@@ -43,8 +43,10 @@ module Retab
       :completed_document_count,
       :document_count,
       :error_count
+    )
 
     def initialize(json)
+      super()
       hash = self.class.normalize(json)
       @id = hash[:id]
       @workflow = hash[:workflow] ? Retab::WorkflowSnapshotRef.new(hash[:workflow]) : nil
@@ -53,7 +55,24 @@ module Retab
       @block_id = hash[:block_id]
       @block_type = hash[:block_type]
       @n_consensus = hash[:n_consensus]
-      @lifecycle = hash[:lifecycle] ? (case hash[:lifecycle][:status] when "cancelled" then Retab::CancelledWorkflowExperimentRun.new(hash[:lifecycle]) when "completed" then Retab::CompletedWorkflowExperimentRun.new(hash[:lifecycle]) when "error" then Retab::ErrorWorkflowExperimentRun.new(hash[:lifecycle]) when "pending" then Retab::PendingWorkflowExperimentRun.new(hash[:lifecycle]) when "queued" then Retab::QueuedWorkflowExperimentRun.new(hash[:lifecycle]) when "running" then Retab::RunningWorkflowExperimentRun.new(hash[:lifecycle]) else hash[:lifecycle] end) : nil
+      @lifecycle = hash[:lifecycle] ? (
+        case hash[:lifecycle][:status]
+        when "cancelled"
+          Retab::CancelledWorkflowExperimentRun.new(hash[:lifecycle])
+        when "completed"
+          Retab::CompletedWorkflowExperimentRun.new(hash[:lifecycle])
+        when "error"
+          Retab::ErrorWorkflowExperimentRun.new(hash[:lifecycle])
+        when "pending"
+          Retab::PendingWorkflowExperimentRun.new(hash[:lifecycle])
+        when "queued"
+          Retab::QueuedWorkflowExperimentRun.new(hash[:lifecycle])
+        when "running"
+          Retab::RunningWorkflowExperimentRun.new(hash[:lifecycle])
+        else
+          hash[:lifecycle]
+        end
+      ) : nil
       @timing = hash[:timing] ? Retab::ExperimentRunTiming.new(hash[:timing]) : nil
       @parent_run_id = hash[:parent_run_id]
       @definition_fingerprint = hash[:definition_fingerprint]

data/lib/retab/experiment_runs/experiment_run_timing.rb

@@ -12,13 +12,15 @@ module Retab
       duration_ms: :duration_ms
     }.freeze
 
-    attr_accessor \
+    attr_accessor(
       :created_at,
       :started_at,
       :completed_at,
       :duration_ms
+    )
 
     def initialize(json)
+      super()
       hash = self.class.normalize(json)
       @created_at = hash[:created_at]
       @started_at = hash[:started_at]

data/lib/retab/experiment_runs/experiment_run_trigger.rb

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

data/lib/retab/experiment_runs.rb

@@ -2,7 +2,7 @@
 
 # This file is auto-generated by oagen. Do not edit.
 
-require 'json'
+require "json"
 
 module Retab
   class ExperimentRuns
@@ -22,13 +22,12 @@ module Retab
     # @param from_date [String, nil]
     # @param to_date [String, nil]
     # @param sort_by [String, nil]
-    # @param fields [String, nil]
     # @param before [String, nil]
     # @param after [String, nil]
     # @param limit [Integer, nil]
     # @param order [Retab::Types::ExperimentRunsOrder, nil]
     # @param request_options [Hash] (see Retab::Types::RequestOptions)
-    # @return [Retab::Types::ListStruct<Retab::ExperimentRun>]
+    # @return [Retab::PaginatedList<Retab::ExperimentRun>]
     def list(
       workflow_id: nil,
       experiment_id: nil,
@@ -40,40 +39,38 @@ module Retab
       trigger_types: nil,
       from_date: nil,
       to_date: nil,
-      sort_by: 'created_at',
-      fields: nil,
+      sort_by: "created_at",
       before: nil,
       after: nil,
       limit: 20,
-      order: 'desc',
+      order: "desc",
       request_options: {}
     )
       params = {
-        'workflow_id' => workflow_id,
-        'experiment_id' => experiment_id,
-        'block_id' => block_id,
-        'status' => status,
-        'statuses' => statuses,
-        'exclude_status' => exclude_status,
-        'trigger_type' => trigger_type,
-        'trigger_types' => trigger_types,
-        'from_date' => from_date,
-        'to_date' => to_date,
-        'sort_by' => sort_by,
-        'fields' => fields,
-        'before' => before,
-        'after' => after,
-        'limit' => limit,
-        'order' => order
+        "workflow_id" => workflow_id,
+        "experiment_id" => experiment_id,
+        "block_id" => block_id,
+        "status" => status,
+        "statuses" => statuses,
+        "exclude_status" => exclude_status,
+        "trigger_type" => trigger_type,
+        "trigger_types" => trigger_types,
+        "from_date" => from_date,
+        "to_date" => to_date,
+        "sort_by" => sort_by,
+        "before" => before,
+        "after" => after,
+        "limit" => limit,
+        "order" => order
       }.compact
       response = @client.request(
         method: :get,
-        path: '/v1/workflows/experiments/runs',
+        path: "/v1/workflows/experiments/runs",
         auth: true,
         params: params,
         request_options: request_options
       )
-      fetch_next = ->(cursor) {
+      fetch_next = -> (cursor) {
         list(
           workflow_id: workflow_id,
           experiment_id: experiment_id,
@@ -86,7 +83,6 @@ module Retab
           from_date: from_date,
           to_date: to_date,
           sort_by: sort_by,
-          fields: fields,
           before: before,
           after: cursor,
           limit: limit,
@@ -94,10 +90,25 @@ module Retab
           request_options: request_options
         )
       }
-      Retab::Types::ListStruct.from_response(
+      Retab::PaginatedList.from_response(
         response,
         model: Retab::ExperimentRun,
-        filters: { workflow_id: workflow_id, experiment_id: experiment_id, block_id: block_id, status: status, statuses: statuses, exclude_status: exclude_status, trigger_type: trigger_type, trigger_types: trigger_types, from_date: from_date, to_date: to_date, sort_by: sort_by, fields: fields, before: before, limit: limit, order: order },
+        filters: {
+          workflow_id: workflow_id,
+          experiment_id: experiment_id,
+          block_id: block_id,
+          status: status,
+          statuses: statuses,
+          exclude_status: exclude_status,
+          trigger_type: trigger_type,
+          trigger_types: trigger_types,
+          from_date: from_date,
+          to_date: to_date,
+          sort_by: sort_by,
+          before: before,
+          limit: limit,
+          order: order
+        },
         fetch_next: fetch_next
       )
     end
@@ -113,18 +124,22 @@ module Retab
       request_options: {}
     )
       body = {
-        'experiment_id' => experiment_id,
-        'workflow_id' => workflow_id
+        "experiment_id" => experiment_id,
+        "workflow_id" => workflow_id
       }.compact
       response = @client.request(
         method: :post,
-        path: '/v1/workflows/experiments/runs',
+        path: "/v1/workflows/experiments/runs",
         auth: true,
         body: body,
         request_options: request_options
       )
       result = Retab::ExperimentRun.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
 
@@ -143,7 +158,11 @@ module Retab
         request_options: request_options
       )
       result = Retab::ExperimentRun.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
 
@@ -162,7 +181,11 @@ module Retab
         request_options: request_options
       )
       result = Retab::CancelWorkflowExperimentRunResponse.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/extractions/extraction.rb

@@ -20,7 +20,7 @@ module Retab
       created_at: :created_at
     }.freeze
 
-    attr_accessor \
+    attr_accessor(
       :id,
       :file,
       :model,
@@ -33,8 +33,10 @@ module Retab
       :metadata,
       :usage,
       :created_at
+    )
 
     def initialize(json)
+      super()
       hash = self.class.normalize(json)
       @id = hash[:id]
       @file = hash[:file] ? Retab::FileRef.new(hash[:file]) : nil

data/lib/retab/extractions/extraction_consensus.rb

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

data/lib/retab/extractions/extraction_request.rb

@@ -15,10 +15,11 @@ module Retab
       metadata: :metadata,
       additional_messages: :additional_messages,
       bust_cache: :bust_cache,
-      stream: :stream
+      stream: :stream,
+      chunking_keys: :chunking_keys
     }.freeze
 
-    attr_accessor \
+    attr_accessor(
       :document,
       :json_schema,
       :model,
@@ -28,9 +29,12 @@ module Retab
       :metadata,
       :additional_messages,
       :bust_cache,
-      :stream
+      :stream,
+      :chunking_keys
+    )
 
     def initialize(json)
+      super()
       hash = self.class.normalize(json)
       @document = hash[:document] ? Retab::MimeData.new(hash[:document]) : nil
       @json_schema = hash[:json_schema] || {}
@@ -42,6 +46,7 @@ module Retab
       @additional_messages = (hash[:additional_messages] || []).map { |item| item || {} }
       @bust_cache = hash[:bust_cache]
       @stream = hash[:stream]
+      @chunking_keys = hash[:chunking_keys] || {}
     end
   end
 end

data/lib/retab/extractions/file_ref.rb

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

data/lib/retab/extractions/retab_usage.rb

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

data/lib/retab/extractions/sources_response.rb

@@ -14,15 +14,17 @@ module Retab
       sources: :sources
     }.freeze
 
-    attr_accessor \
+    attr_accessor(
       :object,
       :extraction_id,
       :document_type,
       :file,
       :extraction,
       :sources
+    )
 
     def initialize(json)
+      super()
       hash = self.class.normalize(json)
       @object = hash[:object]
       @extraction_id = hash[:extraction_id]