turbopuffer 0.1.0 → 1.2.0

data/sig/turbopuffer/internal/type/file_input.rbs

@@ -0,0 +1,25 @@
+module Turbopuffer
+  module Internal
+    module Type
+      class FileInput
+        extend Turbopuffer::Internal::Type::Converter
+
+        def self.===: (top other) -> bool
+
+        def self.==: (top other) -> bool
+
+        def self.coerce: (
+          StringIO | String | top value,
+          state: Turbopuffer::Internal::Type::Converter::coerce_state
+        ) -> (StringIO | top)
+
+        def self.dump: (
+          Pathname | StringIO | IO | String | top value,
+          state: Turbopuffer::Internal::Type::Converter::dump_state
+        ) -> (Pathname | StringIO | IO | String | top)
+
+        def self.to_sorbet_type: -> top
+      end
+    end
+  end
+end

data/sig/turbopuffer/internal/type/hash_of.rbs

@@ -0,0 +1,48 @@
+module Turbopuffer
+  module Internal
+    module Type
+      class HashOf[Elem]
+        include Turbopuffer::Internal::Type::Converter
+        include Turbopuffer::Internal::Util::SorbetRuntimeSupport
+
+        def self.[]: (
+          ::Hash[Symbol, top]
+          | ^-> Turbopuffer::Internal::Type::Converter::input
+          | Turbopuffer::Internal::Type::Converter::input type_info,
+          ?::Hash[Symbol, top] spec
+        ) -> instance
+
+        def ===: (top other) -> bool
+
+        def ==: (top other) -> bool
+
+        def hash: -> Integer
+
+        def coerce: (
+          ::Hash[top, top] | top value,
+          state: Turbopuffer::Internal::Type::Converter::coerce_state
+        ) -> (::Hash[Symbol, top] | top)
+
+        def dump: (
+          ::Hash[top, top] | top value,
+          state: Turbopuffer::Internal::Type::Converter::dump_state
+        ) -> (::Hash[Symbol, top] | top)
+
+        def to_sorbet_type: -> top
+
+        def item_type: -> Elem
+
+        def nilable?: -> bool
+
+        def initialize: (
+          ::Hash[Symbol, top]
+          | ^-> Turbopuffer::Internal::Type::Converter::input
+          | Turbopuffer::Internal::Type::Converter::input type_info,
+          ?::Hash[Symbol, top] spec
+        ) -> void
+
+        def inspect: (?depth: Integer) -> String
+      end
+    end
+  end
+end

data/sig/turbopuffer/internal/type/request_parameters.rbs

@@ -0,0 +1,19 @@
+module Turbopuffer
+  module Internal
+    module Type
+      type request_parameters = { request_options: Turbopuffer::request_opts }
+
+      module RequestParameters
+        attr_reader request_options: Turbopuffer::request_opts
+
+        def request_options=: (
+          Turbopuffer::request_opts
+        ) -> Turbopuffer::request_opts
+
+        module Converter
+          def dump_request: (top params) -> [top, ::Hash[Symbol, top]]
+        end
+      end
+    end
+  end
+end

data/sig/turbopuffer/internal/type/union.rbs

@@ -0,0 +1,52 @@
+module Turbopuffer
+  module Internal
+    module Type
+      module Union
+        include Turbopuffer::Internal::Type::Converter
+        include Turbopuffer::Internal::Util::SorbetRuntimeSupport
+
+        private def self.known_variants: -> ::Array[[Symbol?, (^-> Turbopuffer::Internal::Type::Converter::input), ::Hash[Symbol, top]]]
+
+        def self.derefed_variants: -> ::Array[[Symbol?, top, ::Hash[Symbol, top]]]
+
+        def self.variants: -> ::Array[top]
+
+        private def self.discriminator: (Symbol property) -> void
+
+        private def self.variant: (
+          Symbol
+          | ::Hash[Symbol, top]
+          | ^-> Turbopuffer::Internal::Type::Converter::input
+          | Turbopuffer::Internal::Type::Converter::input key,
+          ?::Hash[Symbol, top]
+          | ^-> Turbopuffer::Internal::Type::Converter::input
+          | Turbopuffer::Internal::Type::Converter::input spec
+        ) -> void
+
+        private def self.resolve_variant: (
+          top value
+        ) -> Turbopuffer::Internal::Type::Converter::input?
+
+        def ===: (top other) -> bool
+
+        def ==: (top other) -> bool
+
+        def hash: -> Integer
+
+        def coerce: (
+          top value,
+          state: Turbopuffer::Internal::Type::Converter::coerce_state
+        ) -> top
+
+        def dump: (
+          top value,
+          state: Turbopuffer::Internal::Type::Converter::dump_state
+        ) -> top
+
+        def to_sorbet_type: -> top
+
+        def inspect: (?depth: Integer) -> String
+      end
+    end
+  end
+end

data/sig/turbopuffer/internal/type/unknown.rbs

@@ -0,0 +1,26 @@
+module Turbopuffer
+  module Internal
+    module Type
+      class Unknown
+        extend Turbopuffer::Internal::Type::Converter
+        extend Turbopuffer::Internal::Util::SorbetRuntimeSupport
+
+        def self.===: (top other) -> bool
+
+        def self.==: (top other) -> bool
+
+        def self.coerce: (
+          top value,
+          state: Turbopuffer::Internal::Type::Converter::coerce_state
+        ) -> top
+
+        def self.dump: (
+          top value,
+          state: Turbopuffer::Internal::Type::Converter::dump_state
+        ) -> top
+
+        def self.to_sorbet_type: -> top
+      end
+    end
+  end
+end

data/sig/turbopuffer/internal/util.rbs

@@ -0,0 +1,185 @@
+module Turbopuffer
+  module Internal
+    module Util
+      extend Turbopuffer::Internal::Util::SorbetRuntimeSupport
+
+      def self?.monotonic_secs: -> Float
+
+      def self?.walk_namespaces: (
+        Module | Class ns
+      ) -> Enumerable[(Module | Class)]
+
+      def self?.arch: -> String
+
+      def self?.os: -> String
+
+      def self?.primitive?: (top input) -> bool
+
+      def self?.coerce_boolean: (String | bool input) -> (bool | top)
+
+      def self?.coerce_boolean!: (String | bool input) -> bool?
+
+      def self?.coerce_integer: (String | Integer input) -> (Integer | top)
+
+      def self?.coerce_float: (String | Integer | Float input) -> (Float | top)
+
+      def self?.coerce_hash: (top input) -> (::Hash[top, top] | top)
+
+      def self?.coerce_hash!: (top input) -> ::Hash[top, top]?
+
+      def self?.deep_merge_lr: (top lhs, top rhs, ?concat: bool) -> top
+
+      def self?.deep_merge: (
+        *::Array[top] values,
+        ?sentinel: top?,
+        ?concat: bool
+      ) -> top
+
+      def self?.dig: (
+        ::Hash[Symbol, top] | ::Array[top] | top data,
+        (Symbol
+        | Integer
+        | ::Array[(Symbol | Integer)]
+        | (^(top arg0) -> top))? pick
+      ) {
+        -> top?
+      } -> top?
+
+      def self?.uri_origin: (URI::Generic uri) -> String
+
+      def self?.interpolate_path: (String | ::Array[String] path) -> String
+
+      def self?.decode_query: (String? query) -> ::Hash[String, ::Array[String]]
+
+      def self?.encode_query: (
+        ::Hash[String, (::Array[String] | String)?]? query
+      ) -> String?
+
+      type parsed_uri =
+        {
+          scheme: String?,
+          host: String?,
+          port: Integer?,
+          path: String?,
+          query: ::Hash[String, ::Array[String]]
+        }
+
+      def self?.parse_uri: (
+        URI::Generic | String url
+      ) -> Turbopuffer::Internal::Util::parsed_uri
+
+      def self?.unparse_uri: (
+        Turbopuffer::Internal::Util::parsed_uri parsed
+      ) -> URI::Generic
+
+      def self?.join_parsed_uri: (
+        Turbopuffer::Internal::Util::parsed_uri lhs,
+        Turbopuffer::Internal::Util::parsed_uri rhs
+      ) -> URI::Generic
+
+      def self?.normalized_headers: (
+        *::Hash[String, (String
+        | Integer
+        | ::Array[(String | Integer)?])?] headers
+      ) -> ::Hash[String, String]
+
+      class ReadIOAdapter
+        def close?: -> bool?
+
+        def close: -> void
+
+        private def read_enum: (Integer? max_len) -> String
+
+        def read: (?Integer? max_len, ?String? out_string) -> String?
+
+        def initialize: (
+          String | Pathname | StringIO | Enumerable[String] src
+        ) {
+          (String arg0) -> void
+        } -> void
+      end
+
+      def self?.writable_enum: {
+        (Enumerator::Yielder y) -> void
+      } -> Enumerable[String]
+
+      JSON_CONTENT: Regexp
+      JSONL_CONTENT: Regexp
+
+      def self?.write_multipart_content: (
+        Enumerator::Yielder y,
+        val: top,
+        closing: ::Array[^-> void],
+        ?content_type: String?
+      ) -> void
+
+      def self?.write_multipart_chunk: (
+        Enumerator::Yielder y,
+        boundary: String,
+        key: Symbol | String,
+        val: top,
+        closing: ::Array[^-> void]
+      ) -> void
+
+      def self?.encode_multipart_streaming: (
+        top body
+      ) -> [String, Enumerable[String]]
+
+      def self?.encode_content: (
+        ::Hash[String, String] headers,
+        top body
+      ) -> top
+
+      def self?.force_charset!: (String content_type, text: String) -> void
+
+      def self?.decode_content: (
+        ::Hash[String, String] headers,
+        stream: Enumerable[String],
+        ?suppress_error: bool
+      ) -> top
+
+      def self?.fused_enum: (
+        Enumerable[top] enum,
+        ?external: bool
+      ) {
+        -> void
+      } -> Enumerable[top]
+
+      def self?.close_fused!: (Enumerable[top]? enum) -> void
+
+      def self?.chain_fused: (
+        Enumerable[top]? enum
+      ) {
+        (Enumerator::Yielder arg0) -> void
+      } -> Enumerable[top]
+
+      type server_sent_event =
+        { event: String?, data: String?, id: String?, retry: Integer? }
+
+      def self?.decode_lines: (Enumerable[String] enum) -> Enumerable[String]
+
+      def self?.decode_sse: (
+        Enumerable[String] lines
+      ) -> Enumerable[Turbopuffer::Internal::Util::server_sent_event]
+
+      module SorbetRuntimeSupport
+        class MissingSorbetRuntimeError < ::RuntimeError
+        end
+
+        private def sorbet_runtime_constants: -> ::Hash[Symbol, top]
+
+        def const_missing: (Symbol name) -> void
+
+        def sorbet_constant_defined?: (Symbol name) -> bool
+
+        def define_sorbet_constant!: (Symbol name) { -> top } -> void
+
+        def to_sorbet_type: -> top
+
+        def self.to_sorbet_type: (
+          Turbopuffer::Internal::Util::SorbetRuntimeSupport | top `type`
+        ) -> top
+      end
+    end
+  end
+end

data/sig/turbopuffer/internal.rbs

@@ -0,0 +1,9 @@
+module Turbopuffer
+  module Internal
+    extend Turbopuffer::Internal::Util::SorbetRuntimeSupport
+
+    type file_input = Pathname | StringIO | IO | String | Turbopuffer::FilePart
+
+    OMIT: Object
+  end
+end

data/sig/turbopuffer/models/aggregation_group.rbs

@@ -0,0 +1,7 @@
+module Turbopuffer
+  module Models
+    type aggregation_group = ::Hash[Symbol, top]
+
+    AggregationGroup: Turbopuffer::Internal::Type::Converter
+  end
+end

data/sig/turbopuffer/models/attribute_schema.rbs

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

data/sig/turbopuffer/models/attribute_schema_config.rbs

@@ -0,0 +1,52 @@
+module Turbopuffer
+  module Models
+    type attribute_schema_config =
+      {
+        ann: bool,
+        filterable: bool,
+        full_text_search: Turbopuffer::Models::full_text_search,
+        regex: bool,
+        type: String
+      }
+
+    class AttributeSchemaConfig < Turbopuffer::Internal::Type::BaseModel
+      attr_reader ann: bool?
+
+      def ann=: (bool) -> bool
+
+      attr_reader filterable: bool?
+
+      def filterable=: (bool) -> bool
+
+      attr_reader full_text_search: Turbopuffer::Models::full_text_search?
+
+      def full_text_search=: (
+        Turbopuffer::Models::full_text_search
+      ) -> Turbopuffer::Models::full_text_search
+
+      attr_reader regex: bool?
+
+      def regex=: (bool) -> bool
+
+      attr_reader type: String?
+
+      def type=: (String) -> String
+
+      def initialize: (
+        ?ann: bool,
+        ?filterable: bool,
+        ?full_text_search: Turbopuffer::Models::full_text_search,
+        ?regex: bool,
+        ?type: String
+      ) -> void
+
+      def to_hash: -> {
+        ann: bool,
+        filterable: bool,
+        full_text_search: Turbopuffer::Models::full_text_search,
+        regex: bool,
+        type: String
+      }
+    end
+  end
+end

data/sig/turbopuffer/models/attribute_type.rbs

@@ -0,0 +1,5 @@
+module Turbopuffer
+  module Models
+    class AttributeType = String
+  end
+end

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,75 @@
+module Turbopuffer
+  module Models
+    type full_text_search_config =
+      {
+        b: Float,
+        case_sensitive: bool,
+        :k1 => Float,
+        language: Turbopuffer::Models::language,
+        max_token_length: Integer,
+        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 max_token_length: Integer?
+
+      def max_token_length=: (Integer) -> Integer
+
+      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,
+        ?max_token_length: Integer,
+        ?remove_stopwords: bool,
+        ?stemming: bool,
+        ?tokenizer: Turbopuffer::Models::tokenizer
+      ) -> void
+
+      def to_hash: -> {
+        b: Float,
+        case_sensitive: bool,
+        :k1 => Float,
+        language: Turbopuffer::Models::language,
+        max_token_length: Integer,
+        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