retab 0.1.4 → 0.1.6

data/lib/retab/secrets.rb

@@ -0,0 +1,154 @@
+# frozen_string_literal: true
+
+# This file is auto-generated by oagen. Do not edit.
+
+require "json"
+
+module Retab
+  class Secrets
+    def initialize(client)
+      @client = client
+    end
+
+    # List Secrets
+    # @param request_options [Hash] (see Retab::Types::RequestOptions)
+    # @return [Retab::SecretListResponse]
+    def list_secrets(request_options: {})
+      response = @client.request(
+        method: :get,
+        path: "/v1/secrets",
+        auth: true,
+        request_options: request_options
+      )
+      result = Retab::SecretListResponse.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
+    end
+
+    # Create Secret
+    # @param name [String]
+    # @param value [String]
+    # @param request_options [Hash] (see Retab::Types::RequestOptions)
+    # @return [Retab::SecretResponse]
+    def create_secret(
+      name:,
+      value:,
+      request_options: {}
+    )
+      body = {
+        "name" => name,
+        "value" => value
+      }
+      response = @client.request(
+        method: :post,
+        path: "/v1/secrets",
+        auth: true,
+        body: body,
+        request_options: request_options
+      )
+      result = Retab::SecretResponse.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
+    end
+
+    # Get Secret
+    # @param name [String]
+    # @param request_options [Hash] (see Retab::Types::RequestOptions)
+    # @return [Retab::SecretResponse]
+    def get_secret(
+      name:,
+      request_options: {}
+    )
+      response = @client.request(
+        method: :get,
+        path: "/v1/secrets/#{Retab::Util.encode_path(name)}",
+        auth: true,
+        request_options: request_options
+      )
+      result = Retab::SecretResponse.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
+    end
+
+    # Set Secret
+    # @param name [String]
+    # @param value [String]
+    # @param request_options [Hash] (see Retab::Types::RequestOptions)
+    # @return [Retab::SecretResponse]
+    def update_secret(
+      name:,
+      value:,
+      request_options: {}
+    )
+      body = {
+        "value" => value
+      }
+      response = @client.request(
+        method: :put,
+        path: "/v1/secrets/#{Retab::Util.encode_path(name)}",
+        auth: true,
+        body: body,
+        request_options: request_options
+      )
+      result = Retab::SecretResponse.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
+    end
+
+    # Delete Secret
+    # @param name [String]
+    # @param request_options [Hash] (see Retab::Types::RequestOptions)
+    # @return [void]
+    def delete_secret(
+      name:,
+      request_options: {}
+    )
+      @client.request(
+        method: :delete,
+        path: "/v1/secrets/#{Retab::Util.encode_path(name)}",
+        auth: true,
+        request_options: request_options
+      )
+      nil
+    end
+
+    # Get Secret Value
+    # @param name [String]
+    # @param request_options [Hash] (see Retab::Types::RequestOptions)
+    # @return [Retab::SecretValueResponse]
+    def list_secret_value(
+      name:,
+      request_options: {}
+    )
+      response = @client.request(
+        method: :get,
+        path: "/v1/secrets/#{Retab::Util.encode_path(name)}/value",
+        auth: true,
+        request_options: request_options
+      )
+      result = Retab::SecretValueResponse.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
+    end
+  end
+end

data/lib/retab/splits/split.rb

@@ -13,6 +13,8 @@ module Retab
       n_consensus: :n_consensus,
       instructions: :instructions,
       output: :output,
+      status: :status,
+      error: :error,
       consensus: :consensus,
       usage: :usage,
       created_at: :created_at
@@ -26,6 +28,8 @@ module Retab
       :n_consensus,
       :instructions,
       :output,
+      :status,
+      :error,
       :consensus,
       :usage,
       :created_at
@@ -41,6 +45,8 @@ module Retab
       @n_consensus = hash[:n_consensus]
       @instructions = hash[:instructions]
       @output = (hash[:output] || []).map { |item| item ? Retab::SplitResult.new(item) : nil }
+      @status = hash[:status].nil? ? "pending" : hash[:status]
+      @error = hash[:error] ? Retab::PrimitiveError.new(hash[:error]) : nil
       @consensus = hash[:consensus] ? Retab::SplitConsensus.new(hash[:consensus]) : nil
       @usage = hash[:usage] ? Retab::RetabUsage.new(hash[:usage]) : nil
       @created_at = hash[:created_at]

data/lib/retab/splits/split_request.rb

@@ -11,7 +11,8 @@ module Retab
       model: :model,
       instructions: :instructions,
       n_consensus: :n_consensus,
-      bust_cache: :bust_cache
+      bust_cache: :bust_cache,
+      background: :background
     }.freeze
 
     attr_accessor(
@@ -20,7 +21,8 @@ module Retab
       :model,
       :instructions,
       :n_consensus,
-      :bust_cache
+      :bust_cache,
+      :background
     )
 
     def initialize(json)
@@ -32,6 +34,7 @@ module Retab
       @instructions = hash[:instructions]
       @n_consensus = hash[:n_consensus]
       @bust_cache = hash[:bust_cache].nil? ? false : hash[:bust_cache]
+      @background = hash[:background].nil? ? false : hash[:background]
     end
   end
 end

data/lib/retab/splits.rb

@@ -16,6 +16,7 @@ module Retab
     # @param limit [Integer, nil]
     # @param order [Retab::Types::SplitsOrder, nil]
     # @param filename [String, nil]
+    # @param status [Retab::Types::SplitsStatus, nil]
     # @param from_date [String, nil]
     # @param to_date [String, nil]
     # @param request_options [Hash] (see Retab::Types::RequestOptions)
@@ -26,6 +27,7 @@ module Retab
       limit: 10,
       order: "desc",
       filename: nil,
+      status: nil,
       from_date: nil,
       to_date: nil,
       request_options: {}
@@ -36,6 +38,7 @@ module Retab
         "limit" => limit,
         "order" => order,
         "filename" => filename,
+        "status" => status,
         "from_date" => from_date,
         "to_date" => to_date
       }.compact
@@ -53,6 +56,7 @@ module Retab
           limit: limit,
           order: order,
           filename: filename,
+          status: status,
           from_date: from_date,
           to_date: to_date,
           request_options: request_options
@@ -66,6 +70,7 @@ module Retab
           limit: limit,
           order: order,
           filename: filename,
+          status: status,
           from_date: from_date,
           to_date: to_date
         },
@@ -80,6 +85,7 @@ module Retab
     # @param instructions [String, nil] Free-form instructions appended to the system prompt to steer the split.
     # @param n_consensus [Integer, nil] Number of consensus split runs to perform. Uses deterministic single-pass when set to 1.
     # @param bust_cache [Boolean, nil] If true, skip the LLM cache and force a fresh completion
+    # @param background [Boolean, nil] If true, run asynchronously: returns immediately with status 'queued' and an empty output. Poll GET /v1/<primitive>/{id} until status is terminal. Mutually exclusive with stream.
     # @param request_options [Hash] (see Retab::Types::RequestOptions)
     # @return [Retab::Split]
     def create(
@@ -89,6 +95,7 @@ module Retab
       instructions: nil,
       n_consensus: nil,
       bust_cache: nil,
+      background: nil,
       request_options: {}
     )
       document = Retab::MimeData.coerce(document) unless document.nil?
@@ -98,7 +105,8 @@ module Retab
         "model" => model,
         "instructions" => instructions,
         "n_consensus" => n_consensus,
-        "bust_cache" => bust_cache
+        "bust_cache" => bust_cache,
+        "background" => background
       }.compact
       response = @client.request(
         method: :post,
@@ -118,16 +126,22 @@ module Retab
 
     # Get Split
     # @param split_id [String]
+    # @param include_output [Boolean, nil] When false, returns a cheap status-only projection (no output), served from cache for in-flight background runs.
     # @param request_options [Hash] (see Retab::Types::RequestOptions)
     # @return [Retab::Split]
     def get(
       split_id:,
+      include_output: true,
       request_options: {}
     )
+      params = {
+        "include_output" => include_output
+      }.compact
       response = @client.request(
         method: :get,
         path: "/v1/splits/#{Retab::Util.encode_path(split_id)}",
         auth: true,
+        params: params,
         request_options: request_options
       )
       result = Retab::Split.new(response.body)
@@ -155,5 +169,28 @@ module Retab
       )
       nil
     end
+
+    # Cancel Split
+    # @param split_id [String]
+    # @param request_options [Hash] (see Retab::Types::RequestOptions)
+    # @return [Retab::Split]
+    def create_split_cancel(
+      split_id:,
+      request_options: {}
+    )
+      response = @client.request(
+        method: :post,
+        path: "/v1/splits/#{Retab::Util.encode_path(split_id)}/cancel",
+        auth: true,
+        request_options: request_options
+      )
+      result = Retab::Split.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
+    end
   end
 end

data/lib/retab/tables/body_create_table_v_1_tables_post.rb

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

data/lib/retab/tables/body_replace_table_v_1_tables_table_id_put.rb

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

data/lib/retab/tables/query_workflow_table_request.rb

@@ -0,0 +1,72 @@
+# frozen_string_literal: true
+
+# This file is auto-generated by oagen. Do not edit.
+
+module Retab
+  class QueryWorkflowTableRequest < Retab::Types::BaseModel
+
+    HASH_ATTRS = {
+      filters: :filters,
+      search: :search,
+      case_sensitive: :case_sensitive,
+      select: :select,
+      distinct: :distinct,
+      group_by: :group_by,
+      aggregations: :aggregations,
+      sort: :sort,
+      sample: :sample,
+      tail: :tail,
+      count_only: :count_only,
+      include_explain: :include_explain,
+      sort_column: :sort_column,
+      sort_direction: :sort_direction,
+      viewer_mode: :viewer_mode,
+      offset: :offset,
+      limit: :limit
+    }.freeze
+
+    attr_accessor(
+      :filters,
+      :search,
+      :case_sensitive,
+      :select,
+      :distinct,
+      :group_by,
+      :aggregations,
+      :sort,
+      :sample,
+      :tail,
+      :count_only,
+      :include_explain,
+      :sort_column,
+      :sort_direction,
+      :viewer_mode,
+      :offset,
+      :limit
+    )
+
+    def initialize(json)
+      super()
+      hash = self.class.normalize(json)
+      @filters = (hash[:filters] || []).map { |item| item ? Retab::WorkflowTableFilterRule.new(item) : nil }
+      @search = hash[:search] ? Retab::WorkflowTableSearchRequest.new(hash[:search]) : nil
+      @case_sensitive = hash[:case_sensitive].nil? ? false : hash[:case_sensitive]
+      @select = (hash[:select] || [])
+      @distinct = hash[:distinct] ? Retab::WorkflowTableDistinctRequest.new(hash[:distinct]) : nil
+      @group_by = (hash[:group_by] || [])
+      @aggregations = (hash[:aggregations] || []).map { |item|
+        item ? Retab::WorkflowTableAggregationRequest.new(item) : nil
+      }
+      @sort = (hash[:sort] || []).map { |item| item ? Retab::WorkflowTableSortRule.new(item) : nil }
+      @sample = hash[:sample] ? Retab::WorkflowTableSampleRequest.new(hash[:sample]) : nil
+      @tail = hash[:tail] ? Retab::WorkflowTableTailRequest.new(hash[:tail]) : nil
+      @count_only = hash[:count_only].nil? ? false : hash[:count_only]
+      @include_explain = hash[:include_explain].nil? ? false : hash[:include_explain]
+      @sort_column = hash[:sort_column]
+      @sort_direction = hash[:sort_direction]
+      @viewer_mode = hash[:viewer_mode]
+      @offset = hash[:offset]
+      @limit = hash[:limit]
+    end
+  end
+end

data/lib/retab/{jobs/create_job_request.rb → tables/update_workflow_table_request.rb}

@@ -3,25 +3,22 @@
 # This file is auto-generated by oagen. Do not edit.
 
 module Retab
-  class CreateJobRequest < Retab::Types::BaseModel
+  class UpdateWorkflowTableRequest < Retab::Types::BaseModel
 
     HASH_ATTRS = {
-      endpoint: :endpoint,
-      request: :request,
+      name: :name,
       metadata: :metadata
     }.freeze
 
     attr_accessor(
-      :endpoint,
-      :request,
+      :name,
       :metadata
     )
 
     def initialize(json)
       super()
       hash = self.class.normalize(json)
-      @endpoint = hash[:endpoint]
-      @request = hash[:request] || {}
+      @name = hash[:name]
       @metadata = hash[:metadata] || {}
     end
   end

data/lib/retab/tables/workflow_table.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+# This file is auto-generated by oagen. Do not edit.
+
+module Retab
+  class WorkflowTable < Retab::Types::BaseModel
+
+    HASH_ATTRS = {
+      id: :id,
+      name: :name,
+      filename: :filename,
+      source_file_id: :source_file_id,
+      snapshot_file_id: :snapshot_file_id,
+      row_count: :row_count,
+      columns: :columns,
+      sample_rows: :sample_rows,
+      metadata: :metadata,
+      uploaded_by_user_id: :uploaded_by_user_id,
+      created_at: :created_at,
+      updated_at: :updated_at
+    }.freeze
+
+    attr_accessor(
+      :id,
+      :name,
+      :filename,
+      :source_file_id,
+      :snapshot_file_id,
+      :row_count,
+      :columns,
+      :sample_rows,
+      :metadata,
+      :uploaded_by_user_id,
+      :created_at,
+      :updated_at
+    )
+
+    def initialize(json)
+      super()
+      hash = self.class.normalize(json)
+      @id = hash[:id]
+      @name = hash[:name]
+      @filename = hash[:filename]
+      @source_file_id = hash[:source_file_id].nil? ? "" : hash[:source_file_id]
+      @snapshot_file_id = hash[:snapshot_file_id].nil? ? "" : hash[:snapshot_file_id]
+      @row_count = hash[:row_count]
+      @columns = (hash[:columns] || []).map { |item| item ? Retab::WorkflowTableColumn.new(item) : nil }
+      @sample_rows = (hash[:sample_rows] || []).map { |item| item || {} }
+      @metadata = hash[:metadata] || {}
+      @uploaded_by_user_id = hash[:uploaded_by_user_id]
+      @created_at = hash[:created_at]
+      @updated_at = hash[:updated_at]
+    end
+  end
+end

data/lib/retab/tables/workflow_table_aggregation_request.rb

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

data/lib/retab/tables/workflow_table_column.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+# This file is auto-generated by oagen. Do not edit.
+
+module Retab
+  class WorkflowTableColumn < Retab::Types::BaseModel
+
+    HASH_ATTRS = {
+      name: :name,
+      json_schema: :json_schema,
+      sample_values: :sample_values,
+      required: :required,
+      unique: :unique
+    }.freeze
+
+    attr_accessor(
+      :name,
+      :json_schema,
+      :sample_values,
+      :required,
+      :unique
+    )
+
+    def initialize(json)
+      super()
+      hash = self.class.normalize(json)
+      @name = hash[:name]
+      @json_schema = hash[:json_schema] || {}
+      @sample_values = (hash[:sample_values] || [])
+      @required = hash[:required].nil? ? false : hash[:required]
+      @unique = hash[:unique].nil? ? false : hash[:unique]
+    end
+  end
+end

data/lib/retab/tables/workflow_table_distinct_request.rb

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

data/lib/retab/tables/workflow_table_explain.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+# This file is auto-generated by oagen. Do not edit.
+
+module Retab
+  class WorkflowTableExplain < Retab::Types::BaseModel
+
+    HASH_ATTRS = {
+      table_id: :table_id,
+      snapshot_file_id: :snapshot_file_id,
+      selected_columns: :selected_columns,
+      filters: :filters,
+      search: :search,
+      sort: :sort,
+      offset: :offset,
+      limit: :limit,
+      distinct: :distinct,
+      group_by: :group_by,
+      aggregations: :aggregations
+    }.freeze
+
+    attr_accessor(
+      :table_id,
+      :snapshot_file_id,
+      :selected_columns,
+      :filters,
+      :search,
+      :sort,
+      :offset,
+      :limit,
+      :distinct,
+      :group_by,
+      :aggregations
+    )
+
+    def initialize(json)
+      super()
+      hash = self.class.normalize(json)
+      @table_id = hash[:table_id]
+      @snapshot_file_id = hash[:snapshot_file_id].nil? ? "" : hash[:snapshot_file_id]
+      @selected_columns = (hash[:selected_columns] || [])
+      @filters = (hash[:filters] || []).map { |item| item ? Retab::WorkflowTableFilterRule.new(item) : nil }
+      @search = hash[:search] ? Retab::WorkflowTableSearchRequest.new(hash[:search]) : nil
+      @sort = (hash[:sort] || []).map { |item| item ? Retab::WorkflowTableSortRule.new(item) : nil }
+      @offset = hash[:offset]
+      @limit = hash[:limit]
+      @distinct = hash[:distinct] ? Retab::WorkflowTableDistinctRequest.new(hash[:distinct]) : nil
+      @group_by = (hash[:group_by] || [])
+      @aggregations = (hash[:aggregations] || []).map { |item|
+        item ? Retab::WorkflowTableAggregationRequest.new(item) : nil
+      }
+    end
+  end
+end