turbopuffer-ruby 0.2.0.pre.alpha.1

data/sig/turbopuffer/models/client_namespaces_params.rbs

@@ -0,0 +1,38 @@
+module Turbopuffer
+  module Models
+    type client_namespaces_params =
+      { cursor: String, page_size: Integer, prefix: String }
+      & Turbopuffer::Internal::Type::request_parameters
+
+    class ClientNamespacesParams < Turbopuffer::Internal::Type::BaseModel
+      extend Turbopuffer::Internal::Type::RequestParameters::Converter
+      include Turbopuffer::Internal::Type::RequestParameters
+
+      attr_reader cursor: String?
+
+      def cursor=: (String) -> String
+
+      attr_reader page_size: Integer?
+
+      def page_size=: (Integer) -> Integer
+
+      attr_reader prefix: String?
+
+      def prefix=: (String) -> String
+
+      def initialize: (
+        ?cursor: String,
+        ?page_size: Integer,
+        ?prefix: String,
+        ?request_options: Turbopuffer::request_opts
+      ) -> void
+
+      def to_hash: -> {
+        cursor: String,
+        page_size: Integer,
+        prefix: String,
+        request_options: Turbopuffer::RequestOptions
+      }
+    end
+  end
+end

data/sig/turbopuffer/models/columns.rbs

@@ -0,0 +1,42 @@
+module Turbopuffer
+  module Models
+    type columns =
+      {
+        id: ::Array[Turbopuffer::Models::id],
+        vector: Turbopuffer::Models::Columns::vector
+      }
+
+    class Columns < Turbopuffer::Internal::Type::BaseModel
+      attr_accessor id: ::Array[Turbopuffer::Models::id]
+
+      attr_reader vector: Turbopuffer::Models::Columns::vector?
+
+      def vector=: (
+        Turbopuffer::Models::Columns::vector
+      ) -> Turbopuffer::Models::Columns::vector
+
+      def initialize: (
+        id: ::Array[Turbopuffer::Models::id],
+        ?vector: Turbopuffer::Models::Columns::vector
+      ) -> void
+
+      def to_hash: -> {
+        id: ::Array[Turbopuffer::Models::id],
+        vector: Turbopuffer::Models::Columns::vector
+      }
+
+      type vector =
+        ::Array[Turbopuffer::Models::vector] | ::Array[Float] | String
+
+      module Vector
+        extend Turbopuffer::Internal::Type::Union
+
+        def self?.variants: -> ::Array[Turbopuffer::Models::Columns::vector]
+
+        VectorArray: Turbopuffer::Internal::Type::Converter
+
+        FloatArray: Turbopuffer::Internal::Type::Converter
+      end
+    end
+  end
+end

data/sig/turbopuffer/models/distance_metric.rbs

@@ -0,0 +1,17 @@
+module Turbopuffer
+  module Models
+    type distance_metric = :cosine_distance | :euclidean_squared
+
+    module DistanceMetric
+      extend Turbopuffer::Internal::Type::Enum
+
+      # Defined as `1 - cosine_similarity` and ranges from 0 to 2. Lower is better.
+      COSINE_DISTANCE: :cosine_distance
+
+      # Defined as `sum((x - y)^2)`. Lower is better.
+      EUCLIDEAN_SQUARED: :euclidean_squared
+
+      def self?.values: -> ::Array[Turbopuffer::Models::distance_metric]
+    end
+  end
+end

data/sig/turbopuffer/models/full_text_search.rbs

@@ -0,0 +1,11 @@
+module Turbopuffer
+  module Models
+    type full_text_search = bool | Turbopuffer::FullTextSearchConfig
+
+    module FullTextSearch
+      extend Turbopuffer::Internal::Type::Union
+
+      def self?.variants: -> ::Array[Turbopuffer::Models::full_text_search]
+    end
+  end
+end

data/sig/turbopuffer/models/full_text_search_config.rbs

@@ -0,0 +1,68 @@
+module Turbopuffer
+  module Models
+    type full_text_search_config =
+      {
+        b: Float,
+        case_sensitive: bool,
+        :k1 => Float,
+        language: Turbopuffer::Models::language,
+        remove_stopwords: bool,
+        stemming: bool,
+        tokenizer: Turbopuffer::Models::tokenizer
+      }
+
+    class FullTextSearchConfig < Turbopuffer::Internal::Type::BaseModel
+      attr_reader b: Float?
+
+      def b=: (Float) -> Float
+
+      attr_reader case_sensitive: bool?
+
+      def case_sensitive=: (bool) -> bool
+
+      attr_reader k1: Float?
+
+      def k1=: (Float) -> Float
+
+      attr_reader language: Turbopuffer::Models::language?
+
+      def language=: (
+        Turbopuffer::Models::language
+      ) -> Turbopuffer::Models::language
+
+      attr_reader remove_stopwords: bool?
+
+      def remove_stopwords=: (bool) -> bool
+
+      attr_reader stemming: bool?
+
+      def stemming=: (bool) -> bool
+
+      attr_reader tokenizer: Turbopuffer::Models::tokenizer?
+
+      def tokenizer=: (
+        Turbopuffer::Models::tokenizer
+      ) -> Turbopuffer::Models::tokenizer
+
+      def initialize: (
+        ?b: Float,
+        ?case_sensitive: bool,
+        ?k1: Float,
+        ?language: Turbopuffer::Models::language,
+        ?remove_stopwords: bool,
+        ?stemming: bool,
+        ?tokenizer: Turbopuffer::Models::tokenizer
+      ) -> void
+
+      def to_hash: -> {
+        b: Float,
+        case_sensitive: bool,
+        :k1 => Float,
+        language: Turbopuffer::Models::language,
+        remove_stopwords: bool,
+        stemming: bool,
+        tokenizer: Turbopuffer::Models::tokenizer
+      }
+    end
+  end
+end

data/sig/turbopuffer/models/id.rbs

@@ -0,0 +1,11 @@
+module Turbopuffer
+  module Models
+    type id = String | Integer
+
+    module ID
+      extend Turbopuffer::Internal::Type::Union
+
+      def self?.variants: -> ::Array[Turbopuffer::Models::id]
+    end
+  end
+end

data/sig/turbopuffer/models/include_attributes.rbs

@@ -0,0 +1,13 @@
+module Turbopuffer
+  module Models
+    type include_attributes = bool | ::Array[String]
+
+    module IncludeAttributes
+      extend Turbopuffer::Internal::Type::Union
+
+      def self?.variants: -> ::Array[Turbopuffer::Models::include_attributes]
+
+      StringArray: Turbopuffer::Internal::Type::Converter
+    end
+  end
+end

data/sig/turbopuffer/models/language.rbs

@@ -0,0 +1,48 @@
+module Turbopuffer
+  module Models
+    type language =
+      :arabic
+      | :danish
+      | :dutch
+      | :english
+      | :finnish
+      | :french
+      | :german
+      | :greek
+      | :hungarian
+      | :italian
+      | :norwegian
+      | :portuguese
+      | :romanian
+      | :russian
+      | :spanish
+      | :swedish
+      | :tamil
+      | :turkish
+
+    module Language
+      extend Turbopuffer::Internal::Type::Enum
+
+      ARABIC: :arabic
+      DANISH: :danish
+      DUTCH: :dutch
+      ENGLISH: :english
+      FINNISH: :finnish
+      FRENCH: :french
+      GERMAN: :german
+      GREEK: :greek
+      HUNGARIAN: :hungarian
+      ITALIAN: :italian
+      NORWEGIAN: :norwegian
+      PORTUGUESE: :portuguese
+      ROMANIAN: :romanian
+      RUSSIAN: :russian
+      SPANISH: :spanish
+      SWEDISH: :swedish
+      TAMIL: :tamil
+      TURKISH: :turkish
+
+      def self?.values: -> ::Array[Turbopuffer::Models::language]
+    end
+  end
+end

data/sig/turbopuffer/models/namespace_delete_all_params.rbs

@@ -0,0 +1,25 @@
+module Turbopuffer
+  module Models
+    type namespace_delete_all_params =
+      { namespace: String } & Turbopuffer::Internal::Type::request_parameters
+
+    class NamespaceDeleteAllParams < Turbopuffer::Internal::Type::BaseModel
+      extend Turbopuffer::Internal::Type::RequestParameters::Converter
+      include Turbopuffer::Internal::Type::RequestParameters
+
+      attr_reader namespace: String?
+
+      def namespace=: (String) -> String
+
+      def initialize: (
+        ?namespace: String,
+        ?request_options: Turbopuffer::request_opts
+      ) -> void
+
+      def to_hash: -> {
+        namespace: String,
+        request_options: Turbopuffer::RequestOptions
+      }
+    end
+  end
+end

data/sig/turbopuffer/models/namespace_delete_all_response.rbs

@@ -0,0 +1,13 @@
+module Turbopuffer
+  module Models
+    type namespace_delete_all_response = { status: :OK }
+
+    class NamespaceDeleteAllResponse < Turbopuffer::Internal::Type::BaseModel
+      attr_accessor status: :OK
+
+      def initialize: (?status: :OK) -> void
+
+      def to_hash: -> { status: :OK }
+    end
+  end
+end

data/sig/turbopuffer/models/namespace_hint_cache_warm_params.rbs

@@ -0,0 +1,25 @@
+module Turbopuffer
+  module Models
+    type namespace_hint_cache_warm_params =
+      { namespace: String } & Turbopuffer::Internal::Type::request_parameters
+
+    class NamespaceHintCacheWarmParams < Turbopuffer::Internal::Type::BaseModel
+      extend Turbopuffer::Internal::Type::RequestParameters::Converter
+      include Turbopuffer::Internal::Type::RequestParameters
+
+      attr_reader namespace: String?
+
+      def namespace=: (String) -> String
+
+      def initialize: (
+        ?namespace: String,
+        ?request_options: Turbopuffer::request_opts
+      ) -> void
+
+      def to_hash: -> {
+        namespace: String,
+        request_options: Turbopuffer::RequestOptions
+      }
+    end
+  end
+end

data/sig/turbopuffer/models/namespace_hint_cache_warm_response.rbs

@@ -0,0 +1,17 @@
+module Turbopuffer
+  module Models
+    type namespace_hint_cache_warm_response = { status: :OK, message: String }
+
+    class NamespaceHintCacheWarmResponse < Turbopuffer::Internal::Type::BaseModel
+      attr_accessor status: :OK
+
+      attr_reader message: String?
+
+      def message=: (String) -> String
+
+      def initialize: (?message: String, ?status: :OK) -> void
+
+      def to_hash: -> { status: :OK, message: String }
+    end
+  end
+end

data/sig/turbopuffer/models/namespace_metadata.rbs

@@ -0,0 +1,30 @@
+module Turbopuffer
+  module Models
+    type namespace_metadata =
+      {
+        approx_logical_bytes: Integer,
+        created_at: Time,
+        schema: ::Hash[Symbol, Turbopuffer::AttributeSchemaConfig]
+      }
+
+    class NamespaceMetadata < Turbopuffer::Internal::Type::BaseModel
+      attr_accessor approx_logical_bytes: Integer
+
+      attr_accessor created_at: Time
+
+      attr_accessor schema: ::Hash[Symbol, Turbopuffer::AttributeSchemaConfig]
+
+      def initialize: (
+        approx_logical_bytes: Integer,
+        created_at: Time,
+        schema: ::Hash[Symbol, Turbopuffer::AttributeSchemaConfig]
+      ) -> void
+
+      def to_hash: -> {
+        approx_logical_bytes: Integer,
+        created_at: Time,
+        schema: ::Hash[Symbol, Turbopuffer::AttributeSchemaConfig]
+      }
+    end
+  end
+end

data/sig/turbopuffer/models/namespace_metadata_params.rbs

@@ -0,0 +1,25 @@
+module Turbopuffer
+  module Models
+    type namespace_metadata_params =
+      { namespace: String } & Turbopuffer::Internal::Type::request_parameters
+
+    class NamespaceMetadataParams < Turbopuffer::Internal::Type::BaseModel
+      extend Turbopuffer::Internal::Type::RequestParameters::Converter
+      include Turbopuffer::Internal::Type::RequestParameters
+
+      attr_reader namespace: String?
+
+      def namespace=: (String) -> String
+
+      def initialize: (
+        ?namespace: String,
+        ?request_options: Turbopuffer::request_opts
+      ) -> void
+
+      def to_hash: -> {
+        namespace: String,
+        request_options: Turbopuffer::RequestOptions
+      }
+    end
+  end
+end

data/sig/turbopuffer/models/namespace_multi_query_params.rbs

@@ -0,0 +1,86 @@
+module Turbopuffer
+  module Models
+    type namespace_multi_query_params =
+      {
+        namespace: String,
+        queries: ::Array[Turbopuffer::Query],
+        consistency: Turbopuffer::NamespaceMultiQueryParams::Consistency,
+        vector_encoding: Turbopuffer::Models::vector_encoding
+      }
+      & Turbopuffer::Internal::Type::request_parameters
+
+    class NamespaceMultiQueryParams < Turbopuffer::Internal::Type::BaseModel
+      extend Turbopuffer::Internal::Type::RequestParameters::Converter
+      include Turbopuffer::Internal::Type::RequestParameters
+
+      attr_reader namespace: String?
+
+      def namespace=: (String) -> String
+
+      attr_accessor queries: ::Array[Turbopuffer::Query]
+
+      attr_reader consistency: Turbopuffer::NamespaceMultiQueryParams::Consistency?
+
+      def consistency=: (
+        Turbopuffer::NamespaceMultiQueryParams::Consistency
+      ) -> Turbopuffer::NamespaceMultiQueryParams::Consistency
+
+      attr_reader vector_encoding: Turbopuffer::Models::vector_encoding?
+
+      def vector_encoding=: (
+        Turbopuffer::Models::vector_encoding
+      ) -> Turbopuffer::Models::vector_encoding
+
+      def initialize: (
+        queries: ::Array[Turbopuffer::Query],
+        ?namespace: String,
+        ?consistency: Turbopuffer::NamespaceMultiQueryParams::Consistency,
+        ?vector_encoding: Turbopuffer::Models::vector_encoding,
+        ?request_options: Turbopuffer::request_opts
+      ) -> void
+
+      def to_hash: -> {
+        namespace: String,
+        queries: ::Array[Turbopuffer::Query],
+        consistency: Turbopuffer::NamespaceMultiQueryParams::Consistency,
+        vector_encoding: Turbopuffer::Models::vector_encoding,
+        request_options: Turbopuffer::RequestOptions
+      }
+
+      type consistency =
+        {
+          level: Turbopuffer::Models::NamespaceMultiQueryParams::Consistency::level
+        }
+
+      class Consistency < Turbopuffer::Internal::Type::BaseModel
+        attr_reader level: Turbopuffer::Models::NamespaceMultiQueryParams::Consistency::level?
+
+        def level=: (
+          Turbopuffer::Models::NamespaceMultiQueryParams::Consistency::level
+        ) -> Turbopuffer::Models::NamespaceMultiQueryParams::Consistency::level
+
+        def initialize: (
+          ?level: Turbopuffer::Models::NamespaceMultiQueryParams::Consistency::level
+        ) -> void
+
+        def to_hash: -> {
+          level: Turbopuffer::Models::NamespaceMultiQueryParams::Consistency::level
+        }
+
+        type level = :strong | :eventual
+
+        module Level
+          extend Turbopuffer::Internal::Type::Enum
+
+          # Strong consistency. Requires a round-trip to object storage to fetch the latest writes.
+          STRONG: :strong
+
+          # Eventual consistency. Does not require a round-trip to object storage, but may not see the latest writes.
+          EVENTUAL: :eventual
+
+          def self?.values: -> ::Array[Turbopuffer::Models::NamespaceMultiQueryParams::Consistency::level]
+        end
+      end
+    end
+  end
+end

data/sig/turbopuffer/models/namespace_multi_query_response.rbs

@@ -0,0 +1,53 @@
+module Turbopuffer
+  module Models
+    type namespace_multi_query_response =
+      {
+        billing: Turbopuffer::QueryBilling,
+        performance: Turbopuffer::QueryPerformance,
+        results: ::Array[Turbopuffer::Models::NamespaceMultiQueryResponse::Result]
+      }
+
+    class NamespaceMultiQueryResponse < Turbopuffer::Internal::Type::BaseModel
+      attr_accessor billing: Turbopuffer::QueryBilling
+
+      attr_accessor performance: Turbopuffer::QueryPerformance
+
+      attr_accessor results: ::Array[Turbopuffer::Models::NamespaceMultiQueryResponse::Result]
+
+      def initialize: (
+        billing: Turbopuffer::QueryBilling,
+        performance: Turbopuffer::QueryPerformance,
+        results: ::Array[Turbopuffer::Models::NamespaceMultiQueryResponse::Result]
+      ) -> void
+
+      def to_hash: -> {
+        billing: Turbopuffer::QueryBilling,
+        performance: Turbopuffer::QueryPerformance,
+        results: ::Array[Turbopuffer::Models::NamespaceMultiQueryResponse::Result]
+      }
+
+      type result =
+        { aggregations: ::Hash[Symbol, top], rows: ::Array[Turbopuffer::Row] }
+
+      class Result < Turbopuffer::Internal::Type::BaseModel
+        attr_reader aggregations: ::Hash[Symbol, top]?
+
+        def aggregations=: (::Hash[Symbol, top]) -> ::Hash[Symbol, top]
+
+        attr_reader rows: ::Array[Turbopuffer::Row]?
+
+        def rows=: (::Array[Turbopuffer::Row]) -> ::Array[Turbopuffer::Row]
+
+        def initialize: (
+          ?aggregations: ::Hash[Symbol, top],
+          ?rows: ::Array[Turbopuffer::Row]
+        ) -> void
+
+        def to_hash: -> {
+          aggregations: ::Hash[Symbol, top],
+          rows: ::Array[Turbopuffer::Row]
+        }
+      end
+    end
+  end
+end

data/sig/turbopuffer/models/namespace_query_params.rbs

@@ -0,0 +1,125 @@
+module Turbopuffer
+  module Models
+    type namespace_query_params =
+      {
+        namespace: String,
+        aggregate_by: ::Hash[Symbol, top],
+        consistency: Turbopuffer::NamespaceQueryParams::Consistency,
+        distance_metric: Turbopuffer::Models::distance_metric,
+        filters: top,
+        include_attributes: Turbopuffer::Models::include_attributes,
+        rank_by: top,
+        top_k: Integer,
+        vector_encoding: Turbopuffer::Models::vector_encoding
+      }
+      & Turbopuffer::Internal::Type::request_parameters
+
+    class NamespaceQueryParams < Turbopuffer::Internal::Type::BaseModel
+      extend Turbopuffer::Internal::Type::RequestParameters::Converter
+      include Turbopuffer::Internal::Type::RequestParameters
+
+      attr_reader namespace: String?
+
+      def namespace=: (String) -> String
+
+      attr_reader aggregate_by: ::Hash[Symbol, top]?
+
+      def aggregate_by=: (::Hash[Symbol, top]) -> ::Hash[Symbol, top]
+
+      attr_reader consistency: Turbopuffer::NamespaceQueryParams::Consistency?
+
+      def consistency=: (
+        Turbopuffer::NamespaceQueryParams::Consistency
+      ) -> Turbopuffer::NamespaceQueryParams::Consistency
+
+      attr_reader distance_metric: Turbopuffer::Models::distance_metric?
+
+      def distance_metric=: (
+        Turbopuffer::Models::distance_metric
+      ) -> Turbopuffer::Models::distance_metric
+
+      attr_reader filters: top?
+
+      def filters=: (top) -> top
+
+      attr_reader include_attributes: Turbopuffer::Models::include_attributes?
+
+      def include_attributes=: (
+        Turbopuffer::Models::include_attributes
+      ) -> Turbopuffer::Models::include_attributes
+
+      attr_reader rank_by: top?
+
+      def rank_by=: (top) -> top
+
+      attr_reader top_k: Integer?
+
+      def top_k=: (Integer) -> Integer
+
+      attr_reader vector_encoding: Turbopuffer::Models::vector_encoding?
+
+      def vector_encoding=: (
+        Turbopuffer::Models::vector_encoding
+      ) -> Turbopuffer::Models::vector_encoding
+
+      def initialize: (
+        ?namespace: String,
+        ?aggregate_by: ::Hash[Symbol, top],
+        ?consistency: Turbopuffer::NamespaceQueryParams::Consistency,
+        ?distance_metric: Turbopuffer::Models::distance_metric,
+        ?filters: top,
+        ?include_attributes: Turbopuffer::Models::include_attributes,
+        ?rank_by: top,
+        ?top_k: Integer,
+        ?vector_encoding: Turbopuffer::Models::vector_encoding,
+        ?request_options: Turbopuffer::request_opts
+      ) -> void
+
+      def to_hash: -> {
+        namespace: String,
+        aggregate_by: ::Hash[Symbol, top],
+        consistency: Turbopuffer::NamespaceQueryParams::Consistency,
+        distance_metric: Turbopuffer::Models::distance_metric,
+        filters: top,
+        include_attributes: Turbopuffer::Models::include_attributes,
+        rank_by: top,
+        top_k: Integer,
+        vector_encoding: Turbopuffer::Models::vector_encoding,
+        request_options: Turbopuffer::RequestOptions
+      }
+
+      type consistency =
+        { level: Turbopuffer::Models::NamespaceQueryParams::Consistency::level }
+
+      class Consistency < Turbopuffer::Internal::Type::BaseModel
+        attr_reader level: Turbopuffer::Models::NamespaceQueryParams::Consistency::level?
+
+        def level=: (
+          Turbopuffer::Models::NamespaceQueryParams::Consistency::level
+        ) -> Turbopuffer::Models::NamespaceQueryParams::Consistency::level
+
+        def initialize: (
+          ?level: Turbopuffer::Models::NamespaceQueryParams::Consistency::level
+        ) -> void
+
+        def to_hash: -> {
+          level: Turbopuffer::Models::NamespaceQueryParams::Consistency::level
+        }
+
+        type level = :strong | :eventual
+
+        module Level
+          extend Turbopuffer::Internal::Type::Enum
+
+          # Strong consistency. Requires a round-trip to object storage to fetch the latest writes.
+          STRONG: :strong
+
+          # Eventual consistency. Does not require a round-trip to object storage, but may not see the latest writes.
+          EVENTUAL: :eventual
+
+          def self?.values: -> ::Array[Turbopuffer::Models::NamespaceQueryParams::Consistency::level]
+        end
+      end
+    end
+  end
+end