turbopuffer 1.5.0 → 1.7.0

data/rbi/turbopuffer/models/attribute_schema_config.rbi

@@ -11,11 +11,35 @@ module Turbopuffer
           )
         end
 
-      # Whether to create an approximate nearest neighbor index for the attribute.
-      sig { returns(T.nilable(T::Boolean)) }
+      # The data type of the attribute. Valid values: string, int, uint, float, uuid,
+      # datetime, bool, []string, []int, []uint, []float, []uuid, []datetime, []bool,
+      # [DIMS]f16, [DIMS]f32.
+      sig { returns(String) }
+      attr_accessor :type
+
+      # Whether to create an approximate nearest neighbor index for the attribute. Can
+      # be a boolean or a detailed configuration object.
+      sig do
+        returns(
+          T.nilable(
+            T.any(
+              T::Boolean,
+              Turbopuffer::AttributeSchemaConfig::Ann::AnnConfig
+            )
+          )
+        )
+      end
       attr_reader :ann
 
-      sig { params(ann: T::Boolean).void }
+      sig do
+        params(
+          ann:
+            T.any(
+              T::Boolean,
+              Turbopuffer::AttributeSchemaConfig::Ann::AnnConfig::OrHash
+            )
+        ).void
+      end
       attr_writer :ann
 
       # Whether or not the attributes can be used in filters.
@@ -48,28 +72,28 @@ module Turbopuffer
       sig { params(regex: T::Boolean).void }
       attr_writer :regex
 
-      # The data type of the attribute. Valid values: string, int, uint, float, uuid,
-      # datetime, bool, []string, []int, []uint, []float, []uuid, []datetime, []bool,
-      # [DIMS]f16, [DIMS]f32.
-      sig { returns(T.nilable(String)) }
-      attr_reader :type
-
-      sig { params(type: String).void }
-      attr_writer :type
-
       # Detailed configuration for an attribute attached to a document.
       sig do
         params(
-          ann: T::Boolean,
+          type: String,
+          ann:
+            T.any(
+              T::Boolean,
+              Turbopuffer::AttributeSchemaConfig::Ann::AnnConfig::OrHash
+            ),
           filterable: T::Boolean,
           full_text_search:
             T.any(T::Boolean, Turbopuffer::FullTextSearchConfig::OrHash),
-          regex: T::Boolean,
-          type: String
+          regex: T::Boolean
         ).returns(T.attached_class)
       end
       def self.new(
-        # Whether to create an approximate nearest neighbor index for the attribute.
+        # The data type of the attribute. Valid values: string, int, uint, float, uuid,
+        # datetime, bool, []string, []int, []uint, []float, []uuid, []datetime, []bool,
+        # [DIMS]f16, [DIMS]f32.
+        type:,
+        # Whether to create an approximate nearest neighbor index for the attribute. Can
+        # be a boolean or a detailed configuration object.
         ann: nil,
         # Whether or not the attributes can be used in filters.
         filterable: nil,
@@ -78,28 +102,89 @@ module Turbopuffer
         # filterable. You can override this by setting `filterable: true`.
         full_text_search: nil,
         # Whether to enable Regex filters on this attribute.
-        regex: nil,
-        # The data type of the attribute. Valid values: string, int, uint, float, uuid,
-        # datetime, bool, []string, []int, []uint, []float, []uuid, []datetime, []bool,
-        # [DIMS]f16, [DIMS]f32.
-        type: nil
+        regex: nil
       )
       end
 
       sig do
         override.returns(
           {
-            ann: T::Boolean,
+            type: String,
+            ann:
+              T.any(
+                T::Boolean,
+                Turbopuffer::AttributeSchemaConfig::Ann::AnnConfig
+              ),
             filterable: T::Boolean,
             full_text_search:
               T.any(T::Boolean, Turbopuffer::FullTextSearchConfig),
-            regex: T::Boolean,
-            type: String
+            regex: T::Boolean
           }
         )
       end
       def to_hash
       end
+
+      # Whether to create an approximate nearest neighbor index for the attribute. Can
+      # be a boolean or a detailed configuration object.
+      module Ann
+        extend Turbopuffer::Internal::Type::Union
+
+        Variants =
+          T.type_alias do
+            T.any(
+              T::Boolean,
+              Turbopuffer::AttributeSchemaConfig::Ann::AnnConfig
+            )
+          end
+
+        class AnnConfig < Turbopuffer::Internal::Type::BaseModel
+          OrHash =
+            T.type_alias do
+              T.any(
+                Turbopuffer::AttributeSchemaConfig::Ann::AnnConfig,
+                Turbopuffer::Internal::AnyHash
+              )
+            end
+
+          # A function used to calculate vector similarity.
+          sig { returns(T.nilable(Turbopuffer::DistanceMetric::OrSymbol)) }
+          attr_reader :distance_metric
+
+          sig do
+            params(distance_metric: Turbopuffer::DistanceMetric::OrSymbol).void
+          end
+          attr_writer :distance_metric
+
+          # Configuration options for ANN (Approximate Nearest Neighbor) indexing.
+          sig do
+            params(
+              distance_metric: Turbopuffer::DistanceMetric::OrSymbol
+            ).returns(T.attached_class)
+          end
+          def self.new(
+            # A function used to calculate vector similarity.
+            distance_metric: nil
+          )
+          end
+
+          sig do
+            override.returns(
+              { distance_metric: Turbopuffer::DistanceMetric::OrSymbol }
+            )
+          end
+          def to_hash
+          end
+        end
+
+        sig do
+          override.returns(
+            T::Array[Turbopuffer::AttributeSchemaConfig::Ann::Variants]
+          )
+        end
+        def self.variants
+        end
+      end
     end
   end
 end

data/rbi/turbopuffer/models/namespace_metadata.rbi

@@ -20,6 +20,14 @@ module Turbopuffer
       sig { returns(Time) }
       attr_accessor :created_at
 
+      # Indicates that the namespace is encrypted with a customer-managed encryption key
+      # (CMEK).
+      sig { returns(Turbopuffer::NamespaceMetadata::Encryption::Variants) }
+      attr_accessor :encryption
+
+      sig { returns(Turbopuffer::NamespaceMetadata::Index::Variants) }
+      attr_accessor :index
+
       # The schema of the namespace.
       sig { returns(T::Hash[Symbol, Turbopuffer::AttributeSchemaConfig]) }
       attr_accessor :schema
@@ -28,58 +36,24 @@ module Turbopuffer
       sig { returns(Time) }
       attr_accessor :updated_at
 
-      # Indicates that the namespace is encrypted with a customer-managed encryption key
-      # (CMEK).
-      sig do
-        returns(T.nilable(Turbopuffer::NamespaceMetadata::Encryption::Variants))
-      end
-      attr_reader :encryption
-
-      sig do
-        params(
-          encryption:
-            T.any(
-              T::Boolean,
-              Turbopuffer::NamespaceMetadata::Encryption::Cmek::OrHash
-            )
-        ).void
-      end
-      attr_writer :encryption
-
-      sig do
-        returns(T.nilable(Turbopuffer::NamespaceMetadata::Index::Variants))
-      end
-      attr_reader :index
-
-      sig do
-        params(
-          index:
-            T.any(
-              Turbopuffer::NamespaceMetadata::Index::Status::OrHash,
-              Turbopuffer::NamespaceMetadata::Index::UnionMember1::OrHash
-            )
-        ).void
-      end
-      attr_writer :index
-
       # Metadata about a namespace.
       sig do
         params(
           approx_logical_bytes: Integer,
           approx_row_count: Integer,
           created_at: Time,
-          schema: T::Hash[Symbol, Turbopuffer::AttributeSchemaConfig::OrHash],
-          updated_at: Time,
           encryption:
             T.any(
-              T::Boolean,
+              Turbopuffer::NamespaceMetadata::Encryption::Sse::OrHash,
               Turbopuffer::NamespaceMetadata::Encryption::Cmek::OrHash
             ),
           index:
             T.any(
-              Turbopuffer::NamespaceMetadata::Index::Status::OrHash,
-              Turbopuffer::NamespaceMetadata::Index::UnionMember1::OrHash
-            )
+              Turbopuffer::NamespaceMetadata::Index::IndexUpToDate::OrHash,
+              Turbopuffer::NamespaceMetadata::Index::IndexUpdating::OrHash
+            ),
+          schema: T::Hash[Symbol, Turbopuffer::AttributeSchemaConfig::OrHash],
+          updated_at: Time
         ).returns(T.attached_class)
       end
       def self.new(
@@ -89,14 +63,14 @@ module Turbopuffer
         approx_row_count:,
         # The timestamp when the namespace was created.
         created_at:,
+        # Indicates that the namespace is encrypted with a customer-managed encryption key
+        # (CMEK).
+        encryption:,
+        index:,
         # The schema of the namespace.
         schema:,
         # The timestamp when the namespace was last modified by a write operation.
-        updated_at:,
-        # Indicates that the namespace is encrypted with a customer-managed encryption key
-        # (CMEK).
-        encryption: nil,
-        index: nil
+        updated_at:
       )
       end
 
@@ -106,10 +80,10 @@ module Turbopuffer
             approx_logical_bytes: Integer,
             approx_row_count: Integer,
             created_at: Time,
-            schema: T::Hash[Symbol, Turbopuffer::AttributeSchemaConfig],
-            updated_at: Time,
             encryption: Turbopuffer::NamespaceMetadata::Encryption::Variants,
-            index: Turbopuffer::NamespaceMetadata::Index::Variants
+            index: Turbopuffer::NamespaceMetadata::Index::Variants,
+            schema: T::Hash[Symbol, Turbopuffer::AttributeSchemaConfig],
+            updated_at: Time
           }
         )
       end
@@ -123,9 +97,37 @@ module Turbopuffer
 
         Variants =
           T.type_alias do
-            T.any(T::Boolean, Turbopuffer::NamespaceMetadata::Encryption::Cmek)
+            T.any(
+              Turbopuffer::NamespaceMetadata::Encryption::Sse,
+              Turbopuffer::NamespaceMetadata::Encryption::Cmek
+            )
+          end
+
+        class Sse < Turbopuffer::Internal::Type::BaseModel
+          OrHash =
+            T.type_alias do
+              T.any(
+                Turbopuffer::NamespaceMetadata::Encryption::Sse,
+                Turbopuffer::Internal::AnyHash
+              )
+            end
+
+          # Always true. Indicates that the namespace is encrypted with SSE.
+          sig { returns(T::Boolean) }
+          attr_accessor :sse
+
+          sig { params(sse: T::Boolean).returns(T.attached_class) }
+          def self.new(
+            # Always true. Indicates that the namespace is encrypted with SSE.
+            sse:
+          )
           end
 
+          sig { override.returns({ sse: T::Boolean }) }
+          def to_hash
+          end
+        end
+
         class Cmek < Turbopuffer::Internal::Type::BaseModel
           OrHash =
             T.type_alias do
@@ -136,9 +138,7 @@ module Turbopuffer
             end
 
           sig do
-            returns(
-              T.nilable(Turbopuffer::NamespaceMetadata::Encryption::Cmek::Cmek)
-            )
+            returns(Turbopuffer::NamespaceMetadata::Encryption::Cmek::Cmek)
           end
           attr_reader :cmek
 
@@ -158,7 +158,7 @@ module Turbopuffer
                 Turbopuffer::NamespaceMetadata::Encryption::Cmek::Cmek::OrHash
             ).returns(T.attached_class)
           end
-          def self.new(cmek: nil)
+          def self.new(cmek:)
           end
 
           sig do
@@ -210,16 +210,16 @@ module Turbopuffer
         Variants =
           T.type_alias do
             T.any(
-              Turbopuffer::NamespaceMetadata::Index::Status,
-              Turbopuffer::NamespaceMetadata::Index::UnionMember1
+              Turbopuffer::NamespaceMetadata::Index::IndexUpToDate,
+              Turbopuffer::NamespaceMetadata::Index::IndexUpdating
             )
           end
 
-        class Status < Turbopuffer::Internal::Type::BaseModel
+        class IndexUpToDate < Turbopuffer::Internal::Type::BaseModel
           OrHash =
             T.type_alias do
               T.any(
-                Turbopuffer::NamespaceMetadata::Index::Status,
+                Turbopuffer::NamespaceMetadata::Index::IndexUpToDate,
                 Turbopuffer::Internal::AnyHash
               )
             end
@@ -236,11 +236,11 @@ module Turbopuffer
           end
         end
 
-        class UnionMember1 < Turbopuffer::Internal::Type::BaseModel
+        class IndexUpdating < Turbopuffer::Internal::Type::BaseModel
           OrHash =
             T.type_alias do
               T.any(
-                Turbopuffer::NamespaceMetadata::Index::UnionMember1,
+                Turbopuffer::NamespaceMetadata::Index::IndexUpdating,
                 Turbopuffer::Internal::AnyHash
               )
             end

data/rbi/turbopuffer/models/namespace_write_params.rbi

@@ -21,10 +21,27 @@ module Turbopuffer
       attr_writer :namespace
 
       # The namespace to copy documents from.
-      sig { returns(T.nilable(String)) }
+      sig do
+        returns(
+          T.nilable(
+            T.any(
+              String,
+              Turbopuffer::NamespaceWriteParams::CopyFromNamespace::CopyFromNamespaceConfig
+            )
+          )
+        )
+      end
       attr_reader :copy_from_namespace
 
-      sig { params(copy_from_namespace: String).void }
+      sig do
+        params(
+          copy_from_namespace:
+            T.any(
+              String,
+              Turbopuffer::NamespaceWriteParams::CopyFromNamespace::CopyFromNamespaceConfig::OrHash
+            )
+        ).void
+      end
       attr_writer :copy_from_namespace
 
       # The filter specifying which documents to delete.
@@ -157,7 +174,11 @@ module Turbopuffer
       sig do
         params(
           namespace: String,
-          copy_from_namespace: String,
+          copy_from_namespace:
+            T.any(
+              String,
+              Turbopuffer::NamespaceWriteParams::CopyFromNamespace::CopyFromNamespaceConfig::OrHash
+            ),
           delete_by_filter: T.anything,
           delete_condition: T.anything,
           deletes: T::Array[Turbopuffer::ID::Variants],
@@ -222,7 +243,11 @@ module Turbopuffer
         override.returns(
           {
             namespace: String,
-            copy_from_namespace: String,
+            copy_from_namespace:
+              T.any(
+                String,
+                Turbopuffer::NamespaceWriteParams::CopyFromNamespace::CopyFromNamespaceConfig
+              ),
             delete_by_filter: T.anything,
             delete_condition: T.anything,
             deletes: T::Array[Turbopuffer::ID::Variants],
@@ -248,6 +273,68 @@ module Turbopuffer
       def to_hash
       end
 
+      # The namespace to copy documents from.
+      module CopyFromNamespace
+        extend Turbopuffer::Internal::Type::Union
+
+        Variants =
+          T.type_alias do
+            T.any(
+              String,
+              Turbopuffer::NamespaceWriteParams::CopyFromNamespace::CopyFromNamespaceConfig
+            )
+          end
+
+        class CopyFromNamespaceConfig < Turbopuffer::Internal::Type::BaseModel
+          OrHash =
+            T.type_alias do
+              T.any(
+                Turbopuffer::NamespaceWriteParams::CopyFromNamespace::CopyFromNamespaceConfig,
+                Turbopuffer::Internal::AnyHash
+              )
+            end
+
+          # An API key for the organization containing the source namespace
+          sig { returns(String) }
+          attr_accessor :source_api_key
+
+          # The namespace to copy documents from.
+          sig { returns(String) }
+          attr_accessor :source_namespace
+
+          sig do
+            params(source_api_key: String, source_namespace: String).returns(
+              T.attached_class
+            )
+          end
+          def self.new(
+            # An API key for the organization containing the source namespace
+            source_api_key:,
+            # The namespace to copy documents from.
+            source_namespace:
+          )
+          end
+
+          sig do
+            override.returns(
+              { source_api_key: String, source_namespace: String }
+            )
+          end
+          def to_hash
+          end
+        end
+
+        sig do
+          override.returns(
+            T::Array[
+              Turbopuffer::NamespaceWriteParams::CopyFromNamespace::Variants
+            ]
+          )
+        end
+        def self.variants
+        end
+      end
+
       class Encryption < Turbopuffer::Internal::Type::BaseModel
         OrHash =
           T.type_alias do

data/rbi/turbopuffer/resources/namespaces.rbi

@@ -235,8 +235,12 @@ module Turbopuffer
       # Create, update, or delete documents.
       sig do
         params(
-          namespace: T.nilable(String),
-          copy_from_namespace: String,
+          namespace: String,
+          copy_from_namespace:
+            T.any(
+              String,
+              Turbopuffer::NamespaceWriteParams::CopyFromNamespace::CopyFromNamespaceConfig::OrHash
+            ),
           delete_by_filter: T.anything,
           delete_condition: T.anything,
           deletes: T::Array[Turbopuffer::ID::Variants],

data/sig/turbopuffer/internal/transport/base_client.rbs

@@ -87,6 +87,8 @@ module Turbopuffer
 
         private def auth_headers: -> ::Hash[String, String]
 
+        private def user_agent: -> String
+
         private def generate_idempotency_key: -> String
 
         private def build_request: (

data/sig/turbopuffer/internal/transport/pooled_net_requester.rbs

@@ -17,7 +17,10 @@ module Turbopuffer
 
         DEFAULT_MAX_CONNECTIONS: Integer
 
-        def self.connect: (URI::Generic url) -> top
+        def self.connect: (
+          cert_store: OpenSSL::X509::Store,
+          url: URI::Generic
+        ) -> top
 
         def self.calibrate_socket_timeout: (top conn, Float deadline) -> void
 

data/sig/turbopuffer/models/attribute_schema_config.rbs

@@ -2,17 +2,21 @@ module Turbopuffer
   module Models
     type attribute_schema_config =
       {
-        ann: bool,
+        type: String,
+        ann: Turbopuffer::Models::AttributeSchemaConfig::ann,
         filterable: bool,
         full_text_search: Turbopuffer::Models::full_text_search,
-        regex: bool,
-        type: String
+        regex: bool
       }
 
     class AttributeSchemaConfig < Turbopuffer::Internal::Type::BaseModel
-      attr_reader ann: bool?
+      attr_accessor type: String
 
-      def ann=: (bool) -> bool
+      attr_reader ann: Turbopuffer::Models::AttributeSchemaConfig::ann?
+
+      def ann=: (
+        Turbopuffer::Models::AttributeSchemaConfig::ann
+      ) -> Turbopuffer::Models::AttributeSchemaConfig::ann
 
       attr_reader filterable: bool?
 
@@ -28,25 +32,48 @@ module Turbopuffer
 
       def regex=: (bool) -> bool
 
-      attr_reader type: String?
-
-      def type=: (String) -> String
-
       def initialize: (
-        ?ann: bool,
+        type: String,
+        ?ann: Turbopuffer::Models::AttributeSchemaConfig::ann,
         ?filterable: bool,
         ?full_text_search: Turbopuffer::Models::full_text_search,
-        ?regex: bool,
-        ?type: String
+        ?regex: bool
       ) -> void
 
       def to_hash: -> {
-        ann: bool,
+        type: String,
+        ann: Turbopuffer::Models::AttributeSchemaConfig::ann,
         filterable: bool,
         full_text_search: Turbopuffer::Models::full_text_search,
-        regex: bool,
-        type: String
+        regex: bool
       }
+
+      type ann = bool | Turbopuffer::AttributeSchemaConfig::Ann::AnnConfig
+
+      module Ann
+        extend Turbopuffer::Internal::Type::Union
+
+        type ann_config =
+          { distance_metric: Turbopuffer::Models::distance_metric }
+
+        class AnnConfig < Turbopuffer::Internal::Type::BaseModel
+          attr_reader distance_metric: Turbopuffer::Models::distance_metric?
+
+          def distance_metric=: (
+            Turbopuffer::Models::distance_metric
+          ) -> Turbopuffer::Models::distance_metric
+
+          def initialize: (
+            ?distance_metric: Turbopuffer::Models::distance_metric
+          ) -> void
+
+          def to_hash: -> {
+            distance_metric: Turbopuffer::Models::distance_metric
+          }
+        end
+
+        def self?.variants: -> ::Array[Turbopuffer::Models::AttributeSchemaConfig::ann]
+      end
     end
   end
 end