retab 0.1.0

data/lib/retab/workflow_artifacts/extraction_workflow_artifact.rb

@@ -0,0 +1,56 @@
+# frozen_string_literal: true
+
+# This file is auto-generated by oagen. Do not edit.
+
+module Retab
+  class ExtractionWorkflowArtifact < Retab::Types::BaseModel
+
+    HASH_ATTRS = {
+      id: :id,
+      file: :file,
+      model: :model,
+      json_schema: :json_schema,
+      n_consensus: :n_consensus,
+      image_resolution_dpi: :image_resolution_dpi,
+      instructions: :instructions,
+      output: :output,
+      consensus: :consensus,
+      metadata: :metadata,
+      usage: :usage,
+      created_at: :created_at,
+      operation: :operation
+    }.freeze
+
+    attr_accessor \
+      :id,
+      :file,
+      :model,
+      :json_schema,
+      :n_consensus,
+      :image_resolution_dpi,
+      :instructions,
+      :output,
+      :consensus,
+      :metadata,
+      :usage,
+      :created_at,
+      :operation
+
+    def initialize(json)
+      hash = self.class.normalize(json)
+      @id = hash[:id]
+      @file = hash[:file] ? Retab::FileRef.new(hash[:file]) : nil
+      @model = hash[:model]
+      @json_schema = hash[:json_schema] || {}
+      @n_consensus = hash[:n_consensus]
+      @image_resolution_dpi = hash[:image_resolution_dpi]
+      @instructions = hash[:instructions]
+      @output = hash[:output] || {}
+      @consensus = hash[:consensus] ? Retab::ExtractionConsensus.new(hash[:consensus]) : nil
+      @metadata = hash[:metadata] || {}
+      @usage = hash[:usage] ? Retab::RetabUsage.new(hash[:usage]) : nil
+      @created_at = hash[:created_at]
+      @operation = hash[:operation]
+    end
+  end
+end

data/lib/retab/workflow_artifacts/form_field.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+# This file is auto-generated by oagen. Do not edit.
+
+module Retab
+  class FormField < Retab::Types::BaseModel
+
+    HASH_ATTRS = {
+      bbox: :bbox,
+      description: :description,
+      type: :type,
+      key: :key,
+      value: :value
+    }.freeze
+
+    attr_accessor \
+      :bbox,
+      :description,
+      :type,
+      :key,
+      :value
+
+    def initialize(json)
+      hash = self.class.normalize(json)
+      @bbox = hash[:bbox] ? Retab::BBox.new(hash[:bbox]) : nil
+      @description = hash[:description]
+      @type = hash[:type]
+      @key = hash[:key]
+      @value = hash[:value]
+    end
+  end
+end

data/lib/retab/workflow_artifacts/function_invocation.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+# This file is auto-generated by oagen. Do not edit.
+
+module Retab
+  class FunctionInvocation < Retab::Types::BaseModel
+
+    HASH_ATTRS = {
+      operation: :operation,
+      id: :id,
+      workflow_run_id: :workflow_run_id,
+      step_id: :step_id,
+      inputs: :inputs,
+      output: :output,
+      duration_ms: :duration_ms,
+      error: :error,
+      created_at: :created_at
+    }.freeze
+
+    attr_accessor \
+      :operation,
+      :id,
+      :workflow_run_id,
+      :step_id,
+      :inputs,
+      :output,
+      :duration_ms,
+      :error,
+      :created_at
+
+    def initialize(json)
+      hash = self.class.normalize(json)
+      @operation = hash[:operation]
+      @id = hash[:id]
+      @workflow_run_id = hash[:workflow_run_id]
+      @step_id = hash[:step_id]
+      @inputs = hash[:inputs] || {}
+      @output = hash[:output]
+      @duration_ms = hash[:duration_ms]
+      @error = hash[:error] ? Retab::ErrorDetails.new(hash[:error]) : nil
+      @created_at = hash[:created_at]
+    end
+  end
+end

data/lib/retab/workflow_artifacts/parse_workflow_artifact.rb

@@ -0,0 +1,47 @@
+# frozen_string_literal: true
+
+# This file is auto-generated by oagen. Do not edit.
+
+module Retab
+  class ParseWorkflowArtifact < Retab::Types::BaseModel
+
+    HASH_ATTRS = {
+      id: :id,
+      file: :file,
+      model: :model,
+      table_parsing_format: :table_parsing_format,
+      image_resolution_dpi: :image_resolution_dpi,
+      instructions: :instructions,
+      output: :output,
+      usage: :usage,
+      created_at: :created_at,
+      operation: :operation
+    }.freeze
+
+    attr_accessor \
+      :id,
+      :file,
+      :model,
+      :table_parsing_format,
+      :image_resolution_dpi,
+      :instructions,
+      :output,
+      :usage,
+      :created_at,
+      :operation
+
+    def initialize(json)
+      hash = self.class.normalize(json)
+      @id = hash[:id]
+      @file = hash[:file] ? Retab::FileRef.new(hash[:file]) : nil
+      @model = hash[:model]
+      @table_parsing_format = hash[:table_parsing_format]
+      @image_resolution_dpi = hash[:image_resolution_dpi]
+      @instructions = hash[:instructions]
+      @output = hash[:output] ? Retab::ParseOutput.new(hash[:output]) : nil
+      @usage = hash[:usage] ? Retab::RetabUsage.new(hash[:usage]) : nil
+      @created_at = hash[:created_at]
+      @operation = hash[:operation]
+    end
+  end
+end

data/lib/retab/workflow_artifacts/partition_workflow_artifact.rb

@@ -0,0 +1,53 @@
+# frozen_string_literal: true
+
+# This file is auto-generated by oagen. Do not edit.
+
+module Retab
+  class PartitionWorkflowArtifact < Retab::Types::BaseModel
+
+    HASH_ATTRS = {
+      id: :id,
+      file: :file,
+      model: :model,
+      key: :key,
+      instructions: :instructions,
+      n_consensus: :n_consensus,
+      allow_overlap: :allow_overlap,
+      output: :output,
+      consensus: :consensus,
+      usage: :usage,
+      created_at: :created_at,
+      operation: :operation
+    }.freeze
+
+    attr_accessor \
+      :id,
+      :file,
+      :model,
+      :key,
+      :instructions,
+      :n_consensus,
+      :allow_overlap,
+      :output,
+      :consensus,
+      :usage,
+      :created_at,
+      :operation
+
+    def initialize(json)
+      hash = self.class.normalize(json)
+      @id = hash[:id]
+      @file = hash[:file] ? Retab::FileRef.new(hash[:file]) : nil
+      @model = hash[:model]
+      @key = hash[:key]
+      @instructions = hash[:instructions]
+      @n_consensus = hash[:n_consensus]
+      @allow_overlap = hash[:allow_overlap]
+      @output = (hash[:output] || []).map { |item| item ? Retab::PartitionChunk.new(item) : nil }
+      @consensus = hash[:consensus] ? Retab::PartitionConsensus.new(hash[:consensus]) : nil
+      @usage = hash[:usage] ? Retab::RetabUsage.new(hash[:usage]) : nil
+      @created_at = hash[:created_at]
+      @operation = hash[:operation]
+    end
+  end
+end

data/lib/retab/workflow_artifacts/review_evaluation.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+# This file is auto-generated by oagen. Do not edit.
+
+module Retab
+  class ReviewEvaluation < Retab::Types::BaseModel
+
+    HASH_ATTRS = {
+      operation: :operation,
+      id: :id,
+      workflow_run_id: :workflow_run_id,
+      step_id: :step_id,
+      evaluations: :evaluations,
+      selected_handles: :selected_handles,
+      matched_branch_id: :matched_branch_id,
+      matched_condition_ids: :matched_condition_ids,
+      requires_human_review: :requires_human_review,
+      reviewer_id: :reviewer_id,
+      review_decision: :review_decision,
+      review_notes: :review_notes,
+      requested_revision: :requested_revision,
+      reviewed_at: :reviewed_at,
+      created_at: :created_at
+    }.freeze
+
+    attr_accessor \
+      :operation,
+      :id,
+      :workflow_run_id,
+      :step_id,
+      :evaluations,
+      :selected_handles,
+      :matched_branch_id,
+      :matched_condition_ids,
+      :requires_human_review,
+      :reviewer_id,
+      :review_decision,
+      :review_notes,
+      :requested_revision,
+      :reviewed_at,
+      :created_at
+
+    def initialize(json)
+      hash = self.class.normalize(json)
+      @operation = hash[:operation]
+      @id = hash[:id]
+      @workflow_run_id = hash[:workflow_run_id]
+      @step_id = hash[:step_id]
+      @evaluations = (hash[:evaluations] || []).map { |item| item ? Retab::ConditionEvaluationResult.new(item) : nil }
+      @selected_handles = (hash[:selected_handles] || [])
+      @matched_branch_id = hash[:matched_branch_id]
+      @matched_condition_ids = (hash[:matched_condition_ids] || [])
+      @requires_human_review = hash[:requires_human_review]
+      @reviewer_id = hash[:reviewer_id]
+      @review_decision = hash[:review_decision]
+      @review_notes = hash[:review_notes]
+      @requested_revision = hash[:requested_revision]
+      @reviewed_at = hash[:reviewed_at]
+      @created_at = hash[:created_at]
+    end
+  end
+end

data/lib/retab/workflow_artifacts/split_workflow_artifact.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+# This file is auto-generated by oagen. Do not edit.
+
+module Retab
+  class SplitWorkflowArtifact < Retab::Types::BaseModel
+
+    HASH_ATTRS = {
+      id: :id,
+      file: :file,
+      model: :model,
+      subdocuments: :subdocuments,
+      n_consensus: :n_consensus,
+      instructions: :instructions,
+      output: :output,
+      consensus: :consensus,
+      usage: :usage,
+      created_at: :created_at,
+      operation: :operation
+    }.freeze
+
+    attr_accessor \
+      :id,
+      :file,
+      :model,
+      :subdocuments,
+      :n_consensus,
+      :instructions,
+      :output,
+      :consensus,
+      :usage,
+      :created_at,
+      :operation
+
+    def initialize(json)
+      hash = self.class.normalize(json)
+      @id = hash[:id]
+      @file = hash[:file] ? Retab::FileRef.new(hash[:file]) : nil
+      @model = hash[:model]
+      @subdocuments = (hash[:subdocuments] || []).map { |item| item ? Retab::Subdocument.new(item) : nil }
+      @n_consensus = hash[:n_consensus]
+      @instructions = hash[:instructions]
+      @output = (hash[:output] || []).map { |item| item ? Retab::SplitResult.new(item) : nil }
+      @consensus = hash[:consensus] ? Retab::SplitConsensus.new(hash[:consensus]) : nil
+      @usage = hash[:usage] ? Retab::RetabUsage.new(hash[:usage]) : nil
+      @created_at = hash[:created_at]
+      @operation = hash[:operation]
+    end
+  end
+end

data/lib/retab/workflow_artifacts/while_loop_termination.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+# This file is auto-generated by oagen. Do not edit.
+
+module Retab
+  class WhileLoopTermination < Retab::Types::BaseModel
+
+    HASH_ATTRS = {
+      operation: :operation,
+      id: :id,
+      workflow_run_id: :workflow_run_id,
+      step_id: :step_id,
+      termination_reason: :termination_reason,
+      evaluations: :evaluations,
+      created_at: :created_at
+    }.freeze
+
+    attr_accessor \
+      :operation,
+      :id,
+      :workflow_run_id,
+      :step_id,
+      :termination_reason,
+      :evaluations,
+      :created_at
+
+    def initialize(json)
+      hash = self.class.normalize(json)
+      @operation = hash[:operation]
+      @id = hash[:id]
+      @workflow_run_id = hash[:workflow_run_id]
+      @step_id = hash[:step_id]
+      @termination_reason = hash[:termination_reason]
+      @evaluations = (hash[:evaluations] || []).map { |item| item ? Retab::ConditionEvaluationResult.new(item) : nil }
+      @created_at = hash[:created_at]
+    end
+  end
+end

data/lib/retab/workflow_artifacts/workflow_artifact.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+# This file is auto-generated by oagen. Do not edit.
+
+module Retab
+  class WorkflowArtifact < Retab::Types::BaseModel
+
+    HASH_ATTRS = {
+      operation: :operation,
+      id: :id
+    }.freeze
+
+    attr_accessor \
+      :operation,
+      :id
+
+    def initialize(json)
+      hash = self.class.normalize(json)
+      @operation = hash[:operation]
+      @id = hash[:id]
+    end
+  end
+end

data/lib/retab/workflow_artifacts.rb

@@ -0,0 +1,86 @@
+# frozen_string_literal: true
+
+# This file is auto-generated by oagen. Do not edit.
+
+require 'json'
+
+module Retab
+  class WorkflowArtifacts
+    def initialize(client)
+      @client = client
+    end
+
+    # Get Workflow Artifact By Id
+    # @param artifact_id [String]
+    # @param request_options [Hash] (see Retab::Types::RequestOptions)
+    # @return [Retab::ExtractionWorkflowArtifact, Retab::SplitWorkflowArtifact, Retab::ClassificationWorkflowArtifact, Retab::ParseWorkflowArtifact, Retab::EditWorkflowArtifact, Retab::PartitionWorkflowArtifact, Retab::ConditionalEvaluation, Retab::ReviewEvaluation, Retab::WhileLoopTermination, Retab::ApiCallInvocation, Retab::FunctionInvocation]
+    def get(
+      artifact_id:,
+      request_options: {}
+    )
+      response = @client.request(
+        method: :get,
+        path: "/v1/workflows/artifacts/#{Retab::Util.encode_path(artifact_id)}",
+        auth: true,
+        request_options: request_options
+      )
+      JSON.parse(response.body)
+    end
+
+    # List Workflow Artifacts
+    # @param run_id [String, nil] Workflow run ID whose artifacts should be listed. Required unless ``step_id`` is provided.
+    # @param operation [Retab::Types::WorkflowArtifactsOperation, nil] Optional artifact operation filter
+    # @param block_id [String, nil] Optional block_id or step_id filter
+    # @param step_id [String, nil] Optional step id filter. When provided, returns the single artifact attached to that step (or an empty list if the step has no artifact). ``run_id`` is not required when ``step_id`` is set — it is resolved from the step record.
+    # @param before [String, nil] Step id cursor: return the page before this step (mutually exclusive with ``after``). Ignored when ``step_id`` is set.
+    # @param after [String, nil] Step id cursor: return the page after this step (mutually exclusive with ``before``). Ignored when ``step_id`` is set.
+    # @param limit [Integer, nil] Maximum number of artifacts to return per page (1-200). Ignored when ``step_id`` is set (that path returns the single attached artifact).
+    # @param request_options [Hash] (see Retab::Types::RequestOptions)
+    # @return [Retab::Types::ListStruct<Retab::WorkflowArtifact>]
+    def list(
+      run_id: nil,
+      operation: nil,
+      block_id: nil,
+      step_id: nil,
+      before: nil,
+      after: nil,
+      limit: 100,
+      request_options: {}
+    )
+      params = {
+        'run_id' => run_id,
+        'operation' => operation,
+        'block_id' => block_id,
+        'step_id' => step_id,
+        'before' => before,
+        'after' => after,
+        'limit' => limit
+      }.compact
+      response = @client.request(
+        method: :get,
+        path: '/v1/workflows/artifacts',
+        auth: true,
+        params: params,
+        request_options: request_options
+      )
+      fetch_next = ->(cursor) {
+        list(
+          run_id: run_id,
+          operation: operation,
+          block_id: block_id,
+          step_id: step_id,
+          before: before,
+          after: cursor,
+          limit: limit,
+          request_options: request_options
+        )
+      }
+      Retab::Types::ListStruct.from_response(
+        response,
+        model: Retab::WorkflowArtifact,
+        filters: { run_id: run_id, operation: operation, block_id: block_id, step_id: step_id, before: before, limit: limit },
+        fetch_next: fetch_next
+      )
+    end
+  end
+end

data/lib/retab/workflow_block_executions/completed_block_execution_lifecycle.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+# This file is auto-generated by oagen. Do not edit.
+
+module Retab
+  class CompletedBlockExecutionLifecycle < Retab::Types::BaseModel
+
+    HASH_ATTRS = {
+      status: :status
+    }.freeze
+
+    attr_accessor :status
+
+    def initialize(json)
+      hash = self.class.normalize(json)
+      @status = hash[:status]
+    end
+  end
+end

data/lib/retab/workflow_block_executions/create_block_execution_request.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+# This file is auto-generated by oagen. Do not edit.
+
+module Retab
+  class CreateBlockExecutionRequest < Retab::Types::BaseModel
+
+    HASH_ATTRS = {
+      run_id: :run_id,
+      block_id: :block_id,
+      step_id: :step_id,
+      n_consensus: :n_consensus,
+      check_eligibility: :check_eligibility
+    }.freeze
+
+    attr_accessor \
+      :run_id,
+      :block_id,
+      :step_id,
+      :n_consensus,
+      :check_eligibility
+
+    def initialize(json)
+      hash = self.class.normalize(json)
+      @run_id = hash[:run_id]
+      @block_id = hash[:block_id]
+      @step_id = hash[:step_id]
+      @n_consensus = hash[:n_consensus]
+      @check_eligibility = hash[:check_eligibility]
+    end
+  end
+end

data/lib/retab/workflow_block_executions/error_block_execution_lifecycle.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+# This file is auto-generated by oagen. Do not edit.
+
+module Retab
+  class ErrorBlockExecutionLifecycle < Retab::Types::BaseModel
+
+    HASH_ATTRS = {
+      status: :status,
+      message: :message
+    }.freeze
+
+    attr_accessor \
+      :status,
+      :message
+
+    def initialize(json)
+      hash = self.class.normalize(json)
+      @status = hash[:status]
+      @message = hash[:message]
+    end
+  end
+end

data/lib/retab/workflow_block_executions/skipped_block_execution_lifecycle.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+# This file is auto-generated by oagen. Do not edit.
+
+module Retab
+  class SkippedBlockExecutionLifecycle < Retab::Types::BaseModel
+
+    HASH_ATTRS = {
+      status: :status,
+      reason: :reason
+    }.freeze
+
+    attr_accessor \
+      :status,
+      :reason
+
+    def initialize(json)
+      hash = self.class.normalize(json)
+      @status = hash[:status]
+      @reason = hash[:reason]
+    end
+  end
+end

data/lib/retab/workflow_block_executions/stored_block_execution.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+# This file is auto-generated by oagen. Do not edit.
+
+module Retab
+  class StoredBlockExecution < Retab::Types::BaseModel
+
+    HASH_ATTRS = {
+      id: :id,
+      workflow_id: :workflow_id,
+      run_id: :run_id,
+      block_id: :block_id,
+      block_type: :block_type,
+      lifecycle: :lifecycle,
+      handle_inputs: :handle_inputs,
+      artifact: :artifact,
+      handle_outputs: :handle_outputs,
+      routing_decision: :routing_decision,
+      duration_ms: :duration_ms,
+      created_at: :created_at,
+      block_config: :block_config,
+      step_id: :step_id,
+      available_iterations: :available_iterations
+    }.freeze
+
+    attr_accessor \
+      :id,
+      :workflow_id,
+      :run_id,
+      :block_id,
+      :block_type,
+      :lifecycle,
+      :handle_inputs,
+      :artifact,
+      :handle_outputs,
+      :routing_decision,
+      :duration_ms,
+      :created_at,
+      :block_config,
+      :step_id,
+      :available_iterations
+
+    def initialize(json)
+      hash = self.class.normalize(json)
+      @id = hash[:id]
+      @workflow_id = hash[:workflow_id]
+      @run_id = hash[:run_id]
+      @block_id = hash[:block_id]
+      @block_type = hash[:block_type]
+      @lifecycle = hash[:lifecycle] ? (case hash[:lifecycle][:status] when "completed" then Retab::CompletedBlockExecutionLifecycle.new(hash[:lifecycle]) when "error" then Retab::ErrorBlockExecutionLifecycle.new(hash[:lifecycle]) when "skipped" then Retab::SkippedBlockExecutionLifecycle.new(hash[:lifecycle]) else hash[:lifecycle] end) : nil
+      @handle_inputs = hash[:handle_inputs] || {}
+      @artifact = hash[:artifact] ? Retab::StepArtifactRef.new(hash[:artifact]) : nil
+      @handle_outputs = hash[:handle_outputs] || {}
+      @routing_decision = (hash[:routing_decision] || [])
+      @duration_ms = hash[:duration_ms]
+      @created_at = hash[:created_at]
+      @block_config = hash[:block_config] || {}
+      @step_id = hash[:step_id]
+      @available_iterations = (hash[:available_iterations] || []).map { |item| item || {} }
+    end
+  end
+end