turbopuffer 1.5.0 → 1.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a2f5759ef6a54d8d20fd914a7afb761b2f63e9383fe76094336a59d2dd77c887
-  data.tar.gz: 6186eaa83038a13743abdc63737933c49666249bf4435067551ac1271c47202e
+  metadata.gz: 07fda2e1e9249b4c16e37504d02f6b65e5cff40d0d8b448b9d58c1ad7432391a
+  data.tar.gz: 73ee6163f398dff241b51040941899ee65027ded8bff6ef8ee474e63e8d0cf13
 SHA512:
-  metadata.gz: 6c1dc39006702ca0bdba4684355c7c4436522fd796c33031b553523808b2ae6486bbc5865a1e55d5986ccafa837c6ac914319f9171914e1391ff9c9e527f116d
-  data.tar.gz: 5b77e8508f24ffbac98423426cc270e812f6ef434d50ceab6397cb8b905e4e4e4a8a2f52f62b84c4051afc2955899f74c39cec35ef6039386eae837567ed5908
+  metadata.gz: 7019c102f0665d6912974007b9ed3d3837e19f210596b4386faeacb028a62c594c60ce39791568721f7873ae56c30062cc28c360efe7be94204db4683d589229
+  data.tar.gz: 34ca700c304917dd2e096c660a2db731394dd364bc879d5d76fc2d60ca22e552f568be0dfb5eb9b4f5206035edd580d1c4330fe8fcfa1ae89e7846bc72b1a6fb

data/CHANGELOG.md

@@ -1,5 +1,28 @@
 # Changelog
 
+## 1.6.0 (2025-11-06)
+
+Full Changelog: [v1.5.0...v1.6.0](https://github.com/turbopuffer/turbopuffer-ruby/compare/v1.5.0...v1.6.0)
+
+### Features
+
+* Add vector attribute schema to metadata endpoint ([a509e47](https://github.com/turbopuffer/turbopuffer-ruby/commit/a509e479bc8fdeb6a72b39703636d1ccc331274e))
+* handle thread interrupts in the core HTTP client ([ad531ac](https://github.com/turbopuffer/turbopuffer-ruby/commit/ad531ac4776d6445a23033601d7e46900f8824bc))
+* Make `type` required on `AttributeSchemaConfig` ([c9300f6](https://github.com/turbopuffer/turbopuffer-ruby/commit/c9300f621999c6937d16c363ba7bdb423605a825))
+* openapi: Fix stainless warnings ([1b7685f](https://github.com/turbopuffer/turbopuffer-ruby/commit/1b7685f30fd448eff4d17b809714ee01aa864ffd))
+* openapi: name variants of `NamespaceMetadata.index` ([3b7cd24](https://github.com/turbopuffer/turbopuffer-ruby/commit/3b7cd24aa4f0fbb0c7f805d28f3c0e20726697de))
+
+
+### Bug Fixes
+
+* better thread safety via early initializing SSL store during HTTP client creation ([588faf7](https://github.com/turbopuffer/turbopuffer-ruby/commit/588faf74de4e21652afadd3eb557062818639cd6))
+
+
+### Chores
+
+* bump dependency version and update sorbet types ([b8c6c9a](https://github.com/turbopuffer/turbopuffer-ruby/commit/b8c6c9a56cf98142d6736e21b25fa3a541fafd47))
+* **client:** send user-agent header ([dda4633](https://github.com/turbopuffer/turbopuffer-ruby/commit/dda4633d4fe3cfb16b90d5d4b32aa887ee810d52))
+
 ## 1.5.0 (2025-10-21)
 
 Full Changelog: [v1.4.0...v1.5.0](https://github.com/turbopuffer/turbopuffer-ruby/compare/v1.4.0...v1.5.0)

data/README.md

@@ -17,7 +17,7 @@ To use this gem, install via Bundler by adding the following to your application
 <!-- x-release-please-start-version -->
 
 ```ruby
-gem "turbopuffer", "~> 1.5.0"
+gem "turbopuffer", "~> 1.6.0"
 ```
 
 <!-- x-release-please-end -->

data/lib/turbopuffer/internal/transport/base_client.rb

@@ -201,7 +201,8 @@ module Turbopuffer
             self.class::PLATFORM_HEADERS,
             {
               "accept" => "application/json",
-              "content-type" => "application/json"
+              "content-type" => "application/json",
+              "user-agent" => user_agent
             },
             headers
           )
@@ -219,6 +220,11 @@ module Turbopuffer
         # @return [Hash{String=>String}]
         private def auth_headers = {}
 
+        # @api private
+        #
+        # @return [String]
+        private def user_agent = "#{self.class.name}/Ruby #{Turbopuffer::VERSION}"
+
         # @api private
         #
         # @return [String]

data/lib/turbopuffer/internal/transport/pooled_net_requester.rb

@@ -16,10 +16,11 @@ module Turbopuffer
         class << self
           # @api private
           #
+          # @param cert_store [OpenSSL::X509::Store]
           # @param url [URI::Generic]
           #
           # @return [Net::HTTP]
-          def connect(url)
+          def connect(cert_store:, url:)
             port =
               case [url.port, url.scheme]
               in [Integer, _]
@@ -33,6 +34,8 @@ module Turbopuffer
             Net::HTTP.new(url.host, port).tap do
               _1.use_ssl = %w[https wss].include?(url.scheme)
               _1.max_retries = 0
+
+              (_1.cert_store = cert_store) if _1.use_ssl?
             end
           end
 
@@ -102,7 +105,7 @@ module Turbopuffer
           pool =
             @mutex.synchronize do
               @pools[origin] ||= ConnectionPool.new(size: @size) do
-                self.class.connect(url)
+                self.class.connect(cert_store: @cert_store, url: url)
               end
             end
 
@@ -128,40 +131,48 @@ module Turbopuffer
           url, deadline = request.fetch_values(:url, :deadline)
 
           req = nil
-          eof = false
           finished = false
-          closing = nil
 
           # rubocop:disable Metrics/BlockLength
           enum = Enumerator.new do |y|
             next if finished
 
             with_pool(url, deadline: deadline) do |conn|
-              req, closing = self.class.build_request(request) do
-                self.class.calibrate_socket_timeout(conn, deadline)
-              end
-
-              self.class.calibrate_socket_timeout(conn, deadline)
-              unless conn.started?
-                conn.keep_alive_timeout = self.class::KEEP_ALIVE_TIMEOUT
-                conn.start
-              end
+              eof = false
+              closing = nil
+              ::Thread.handle_interrupt(Object => :never) do
+                ::Thread.handle_interrupt(Object => :immediate) do
+                  req, closing = self.class.build_request(request) do
+                    self.class.calibrate_socket_timeout(conn, deadline)
+                  end
 
-              self.class.calibrate_socket_timeout(conn, deadline)
-              conn.request(req) do |rsp|
-                y << [req, rsp]
-                break if finished
-
-                rsp.read_body do |bytes|
-                  y << bytes.force_encoding(Encoding::BINARY)
-                  break if finished
+                  self.class.calibrate_socket_timeout(conn, deadline)
+                  unless conn.started?
+                    conn.keep_alive_timeout = self.class::KEEP_ALIVE_TIMEOUT
+                    conn.start
+                  end
 
                   self.class.calibrate_socket_timeout(conn, deadline)
+                  conn.request(req) do |rsp|
+                    y << [req, rsp]
+                    break if finished
+
+                    rsp.read_body do |bytes|
+                      y << bytes.force_encoding(Encoding::BINARY)
+                      break if finished
+
+                      self.class.calibrate_socket_timeout(conn, deadline)
+                    end
+                    eof = true
+                  end
+                end
+              ensure
+                begin
+                  conn.finish if !eof && conn&.started?
+                ensure
+                  closing&.call
                 end
-                eof = true
               end
-            ensure
-              conn.finish if !eof && conn&.started?
             end
           rescue Timeout::Error
             raise Turbopuffer::Errors::APITimeoutError.new(url: url, request: req)
@@ -174,8 +185,6 @@ module Turbopuffer
           body = Turbopuffer::Internal::Util.fused_enum(enum, external: true) do
             finished = true
             loop { enum.next }
-          ensure
-            closing&.call
           end
           [Integer(response.code), response, body]
         end
@@ -186,6 +195,7 @@ module Turbopuffer
         def initialize(size: self.class::DEFAULT_MAX_CONNECTIONS)
           @mutex = Mutex.new
           @size = size
+          @cert_store = OpenSSL::X509::Store.new.tap(&:set_default_paths)
           @pools = {}
         end
 

data/lib/turbopuffer/internal/type/union.rb

@@ -11,7 +11,7 @@ module Turbopuffer
       #   when Turbopuffer::AttributeType
       #     # ...
       #   when Turbopuffer::AttributeSchemaConfig
-      #     puts(attribute_schema.ann)
+      #     puts(attribute_schema.type)
       #   else
       #     puts(attribute_schema)
       #   end

data/lib/turbopuffer/models/attribute_schema_config.rb

@@ -3,11 +3,20 @@
 module Turbopuffer
   module Models
     class AttributeSchemaConfig < Turbopuffer::Internal::Type::BaseModel
+      # @!attribute type
+      #   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.
+      #
+      #   @return [String]
+      required :type, String
+
       # @!attribute ann
-      #   Whether to create an approximate nearest neighbor index for the attribute.
+      #   Whether to create an approximate nearest neighbor index for the attribute. Can
+      #   be a boolean or a detailed configuration object.
       #
-      #   @return [Boolean, nil]
-      optional :ann, Turbopuffer::Internal::Type::Boolean
+      #   @return [Boolean, Turbopuffer::Models::AttributeSchemaConfig::Ann::AnnConfig, nil]
+      optional :ann, union: -> { Turbopuffer::AttributeSchemaConfig::Ann }
 
       # @!attribute filterable
       #   Whether or not the attributes can be used in filters.
@@ -29,29 +38,50 @@ module Turbopuffer
       #   @return [Boolean, nil]
       optional :regex, Turbopuffer::Internal::Type::Boolean
 
-      # @!attribute type
-      #   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.
-      #
-      #   @return [String, nil]
-      optional :type, String
-
-      # @!method initialize(ann: nil, filterable: nil, full_text_search: nil, regex: nil, type: nil)
+      # @!method initialize(type:, ann: nil, filterable: nil, full_text_search: nil, regex: nil)
       #   Some parameter documentations has been truncated, see
       #   {Turbopuffer::Models::AttributeSchemaConfig} for more details.
       #
       #   Detailed configuration for an attribute attached to a document.
       #
-      #   @param ann [Boolean] Whether to create an approximate nearest neighbor index for the attribute.
+      #   @param type [String] The data type of the attribute. Valid values: string, int, uint, float, uuid, da
+      #
+      #   @param ann [Boolean, Turbopuffer::Models::AttributeSchemaConfig::Ann::AnnConfig] Whether to create an approximate nearest neighbor index for the attribute. Can b
       #
       #   @param filterable [Boolean] Whether or not the attributes can be used in filters.
       #
       #   @param full_text_search [Boolean, Turbopuffer::Models::FullTextSearchConfig] Whether this attribute can be used as part of a BM25 full-text search. Requires
       #
       #   @param regex [Boolean] Whether to enable Regex filters on this attribute.
+
+      # Whether to create an approximate nearest neighbor index for the attribute. Can
+      # be a boolean or a detailed configuration object.
       #
-      #   @param type [String] The data type of the attribute. Valid values: string, int, uint, float, uuid, da
+      # @see Turbopuffer::Models::AttributeSchemaConfig#ann
+      module Ann
+        extend Turbopuffer::Internal::Type::Union
+
+        variant Turbopuffer::Internal::Type::Boolean
+
+        # Configuration options for ANN (Approximate Nearest Neighbor) indexing.
+        variant -> { Turbopuffer::AttributeSchemaConfig::Ann::AnnConfig }
+
+        class AnnConfig < Turbopuffer::Internal::Type::BaseModel
+          # @!attribute distance_metric
+          #   A function used to calculate vector similarity.
+          #
+          #   @return [Symbol, Turbopuffer::Models::DistanceMetric, nil]
+          optional :distance_metric, enum: -> { Turbopuffer::DistanceMetric }
+
+          # @!method initialize(distance_metric: nil)
+          #   Configuration options for ANN (Approximate Nearest Neighbor) indexing.
+          #
+          #   @param distance_metric [Symbol, Turbopuffer::Models::DistanceMetric] A function used to calculate vector similarity.
+        end
+
+        # @!method self.variants
+        #   @return [Array(Boolean, Turbopuffer::Models::AttributeSchemaConfig::Ann::AnnConfig)]
+      end
     end
   end
 end

data/lib/turbopuffer/models/namespace_metadata.rb

@@ -22,6 +22,18 @@ module Turbopuffer
       #   @return [Time]
       required :created_at, Time
 
+      # @!attribute encryption
+      #   Indicates that the namespace is encrypted with a customer-managed encryption key
+      #   (CMEK).
+      #
+      #   @return [Turbopuffer::Models::NamespaceMetadata::Encryption::Sse, Turbopuffer::Models::NamespaceMetadata::Encryption::Cmek]
+      required :encryption, union: -> { Turbopuffer::NamespaceMetadata::Encryption }
+
+      # @!attribute index
+      #
+      #   @return [Turbopuffer::Models::NamespaceMetadata::Index::IndexUpToDate, Turbopuffer::Models::NamespaceMetadata::Index::IndexUpdating]
+      required :index, union: -> { Turbopuffer::NamespaceMetadata::Index }
+
       # @!attribute schema
       #   The schema of the namespace.
       #
@@ -34,19 +46,7 @@ module Turbopuffer
       #   @return [Time]
       required :updated_at, Time
 
-      # @!attribute encryption
-      #   Indicates that the namespace is encrypted with a customer-managed encryption key
-      #   (CMEK).
-      #
-      #   @return [Boolean, Turbopuffer::Models::NamespaceMetadata::Encryption::Cmek, nil]
-      optional :encryption, union: -> { Turbopuffer::NamespaceMetadata::Encryption }
-
-      # @!attribute index
-      #
-      #   @return [Turbopuffer::Models::NamespaceMetadata::Index::Status, Turbopuffer::Models::NamespaceMetadata::Index::UnionMember1, nil]
-      optional :index, union: -> { Turbopuffer::NamespaceMetadata::Index }
-
-      # @!method initialize(approx_logical_bytes:, approx_row_count:, created_at:, schema:, updated_at:, encryption: nil, index: nil)
+      # @!method initialize(approx_logical_bytes:, approx_row_count:, created_at:, encryption:, index:, schema:, updated_at:)
       #   Some parameter documentations has been truncated, see
       #   {Turbopuffer::Models::NamespaceMetadata} for more details.
       #
@@ -58,13 +58,13 @@ module Turbopuffer
       #
       #   @param created_at [Time] The timestamp when the namespace was created.
       #
-      #   @param schema [Hash{Symbol=>Turbopuffer::Models::AttributeSchemaConfig}] The schema of the namespace.
+      #   @param encryption [Turbopuffer::Models::NamespaceMetadata::Encryption::Sse, Turbopuffer::Models::NamespaceMetadata::Encryption::Cmek] Indicates that the namespace is encrypted with a customer-managed encryption key
       #
-      #   @param updated_at [Time] The timestamp when the namespace was last modified by a write operation.
+      #   @param index [Turbopuffer::Models::NamespaceMetadata::Index::IndexUpToDate, Turbopuffer::Models::NamespaceMetadata::Index::IndexUpdating]
       #
-      #   @param encryption [Boolean, Turbopuffer::Models::NamespaceMetadata::Encryption::Cmek] Indicates that the namespace is encrypted with a customer-managed encryption key
+      #   @param schema [Hash{Symbol=>Turbopuffer::Models::AttributeSchemaConfig}] The schema of the namespace.
       #
-      #   @param index [Turbopuffer::Models::NamespaceMetadata::Index::Status, Turbopuffer::Models::NamespaceMetadata::Index::UnionMember1]
+      #   @param updated_at [Time] The timestamp when the namespace was last modified by a write operation.
 
       # Indicates that the namespace is encrypted with a customer-managed encryption key
       # (CMEK).
@@ -73,18 +73,29 @@ module Turbopuffer
       module Encryption
         extend Turbopuffer::Internal::Type::Union
 
-        variant Turbopuffer::Internal::Type::Boolean
+        variant -> { Turbopuffer::NamespaceMetadata::Encryption::Sse }
 
         # Indicates that the namespace is encrypted with a customer-managed encryption key (CMEK).
         variant -> { Turbopuffer::NamespaceMetadata::Encryption::Cmek }
 
+        class Sse < Turbopuffer::Internal::Type::BaseModel
+          # @!attribute sse
+          #   Always true. Indicates that the namespace is encrypted with SSE.
+          #
+          #   @return [Boolean]
+          required :sse, Turbopuffer::Internal::Type::Boolean
+
+          # @!method initialize(sse:)
+          #   @param sse [Boolean] Always true. Indicates that the namespace is encrypted with SSE.
+        end
+
         class Cmek < Turbopuffer::Internal::Type::BaseModel
           # @!attribute cmek
           #
-          #   @return [Turbopuffer::Models::NamespaceMetadata::Encryption::Cmek::Cmek, nil]
-          optional :cmek, -> { Turbopuffer::NamespaceMetadata::Encryption::Cmek::Cmek }
+          #   @return [Turbopuffer::Models::NamespaceMetadata::Encryption::Cmek::Cmek]
+          required :cmek, -> { Turbopuffer::NamespaceMetadata::Encryption::Cmek::Cmek }
 
-          # @!method initialize(cmek: nil)
+          # @!method initialize(cmek:)
           #   Indicates that the namespace is encrypted with a customer-managed encryption key
           #   (CMEK).
           #
@@ -104,18 +115,18 @@ module Turbopuffer
         end
 
         # @!method self.variants
-        #   @return [Array(Boolean, Turbopuffer::Models::NamespaceMetadata::Encryption::Cmek)]
+        #   @return [Array(Turbopuffer::Models::NamespaceMetadata::Encryption::Sse, Turbopuffer::Models::NamespaceMetadata::Encryption::Cmek)]
       end
 
       # @see Turbopuffer::Models::NamespaceMetadata#index
       module Index
         extend Turbopuffer::Internal::Type::Union
 
-        variant -> { Turbopuffer::NamespaceMetadata::Index::Status }
+        variant -> { Turbopuffer::NamespaceMetadata::Index::IndexUpToDate }
 
-        variant -> { Turbopuffer::NamespaceMetadata::Index::UnionMember1 }
+        variant -> { Turbopuffer::NamespaceMetadata::Index::IndexUpdating }
 
-        class Status < Turbopuffer::Internal::Type::BaseModel
+        class IndexUpToDate < Turbopuffer::Internal::Type::BaseModel
           # @!attribute status
           #
           #   @return [Symbol, :"up-to-date"]
@@ -125,7 +136,7 @@ module Turbopuffer
           #   @param status [Symbol, :"up-to-date"]
         end
 
-        class UnionMember1 < Turbopuffer::Internal::Type::BaseModel
+        class IndexUpdating < Turbopuffer::Internal::Type::BaseModel
           # @!attribute status
           #
           #   @return [Symbol, :updating]
@@ -140,7 +151,7 @@ module Turbopuffer
 
           # @!method initialize(unindexed_bytes:, status: :updating)
           #   Some parameter documentations has been truncated, see
-          #   {Turbopuffer::Models::NamespaceMetadata::Index::UnionMember1} for more details.
+          #   {Turbopuffer::Models::NamespaceMetadata::Index::IndexUpdating} for more details.
           #
           #   @param unindexed_bytes [Integer] The number of bytes in the namespace that are in the write-ahead log but have no
           #
@@ -148,7 +159,7 @@ module Turbopuffer
         end
 
         # @!method self.variants
-        #   @return [Array(Turbopuffer::Models::NamespaceMetadata::Index::Status, Turbopuffer::Models::NamespaceMetadata::Index::UnionMember1)]
+        #   @return [Array(Turbopuffer::Models::NamespaceMetadata::Index::IndexUpToDate, Turbopuffer::Models::NamespaceMetadata::Index::IndexUpdating)]
       end
     end
   end

data/lib/turbopuffer/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Turbopuffer
-  VERSION = "1.5.0"
+  VERSION = "1.6.0"
 end

data/lib/turbopuffer.rb

@@ -9,6 +9,7 @@ require "erb"
 require "etc"
 require "json"
 require "net/http"
+require "openssl"
 require "pathname"
 require "rbconfig"
 require "securerandom"

data/manifest.yaml

@@ -6,6 +6,7 @@ dependencies:
   - etc
   - json
   - net/http
+  - openssl
   - pathname
   - rbconfig
   - securerandom

data/rbi/turbopuffer/internal/transport/base_client.rbi

@@ -180,6 +180,11 @@ module Turbopuffer
         private def auth_headers
         end
 
+        # @api private
+        sig { returns(String) }
+        private def user_agent
+        end
+
         # @api private
         sig { returns(String) }
         private def generate_idempotency_key

data/rbi/turbopuffer/internal/transport/pooled_net_requester.rbi

@@ -26,8 +26,12 @@ module Turbopuffer
 
         class << self
           # @api private
-          sig { params(url: URI::Generic).returns(Net::HTTP) }
-          def connect(url)
+          sig do
+            params(cert_store: OpenSSL::X509::Store, url: URI::Generic).returns(
+              Net::HTTP
+            )
+          end
+          def connect(cert_store:, url:)
           end
 
           # @api private

data/rbi/turbopuffer/internal/type/base_model.rbi

@@ -31,7 +31,7 @@ module Turbopuffer
           #
           # Assumes superclass fields are totally defined before fields are accessed /
           # defined on subclasses.
-          sig { params(child: T.self_type).void }
+          sig { params(child: Turbopuffer::Internal::Type::BaseModel).void }
           def inherited(child)
           end
 
@@ -276,9 +276,13 @@ module Turbopuffer
 
         # Create a new instance of a model.
         sig do
-          params(data: T.any(T::Hash[Symbol, T.anything], T.self_type)).returns(
-            T.attached_class
-          )
+          params(
+            data:
+              T.any(
+                T::Hash[Symbol, T.anything],
+                Turbopuffer::Internal::Type::BaseModel
+              )
+          ).returns(T.attached_class)
         end
         def self.new(data = {})
         end

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/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

data/sig/turbopuffer/models/namespace_metadata.rbs

@@ -5,10 +5,10 @@ module Turbopuffer
         approx_logical_bytes: Integer,
         approx_row_count: Integer,
         created_at: Time,
-        schema: ::Hash[Symbol, Turbopuffer::AttributeSchemaConfig],
-        updated_at: Time,
         encryption: Turbopuffer::Models::NamespaceMetadata::encryption,
-        index: Turbopuffer::Models::NamespaceMetadata::index
+        index: Turbopuffer::Models::NamespaceMetadata::index,
+        schema: ::Hash[Symbol, Turbopuffer::AttributeSchemaConfig],
+        updated_at: Time
       }
 
     class NamespaceMetadata < Turbopuffer::Internal::Type::BaseModel
@@ -18,59 +18,59 @@ module Turbopuffer
 
       attr_accessor created_at: Time
 
-      attr_accessor schema: ::Hash[Symbol, Turbopuffer::AttributeSchemaConfig]
-
-      attr_accessor updated_at: Time
+      attr_accessor encryption: Turbopuffer::Models::NamespaceMetadata::encryption
 
-      attr_reader encryption: Turbopuffer::Models::NamespaceMetadata::encryption?
+      attr_accessor index: Turbopuffer::Models::NamespaceMetadata::index
 
-      def encryption=: (
-        Turbopuffer::Models::NamespaceMetadata::encryption
-      ) -> Turbopuffer::Models::NamespaceMetadata::encryption
-
-      attr_reader index: Turbopuffer::Models::NamespaceMetadata::index?
+      attr_accessor schema: ::Hash[Symbol, Turbopuffer::AttributeSchemaConfig]
 
-      def index=: (
-        Turbopuffer::Models::NamespaceMetadata::index
-      ) -> Turbopuffer::Models::NamespaceMetadata::index
+      attr_accessor updated_at: Time
 
       def initialize: (
         approx_logical_bytes: Integer,
         approx_row_count: Integer,
         created_at: Time,
+        encryption: Turbopuffer::Models::NamespaceMetadata::encryption,
+        index: Turbopuffer::Models::NamespaceMetadata::index,
         schema: ::Hash[Symbol, Turbopuffer::AttributeSchemaConfig],
-        updated_at: Time,
-        ?encryption: Turbopuffer::Models::NamespaceMetadata::encryption,
-        ?index: Turbopuffer::Models::NamespaceMetadata::index
+        updated_at: Time
       ) -> void
 
       def to_hash: -> {
         approx_logical_bytes: Integer,
         approx_row_count: Integer,
         created_at: Time,
-        schema: ::Hash[Symbol, Turbopuffer::AttributeSchemaConfig],
-        updated_at: Time,
         encryption: Turbopuffer::Models::NamespaceMetadata::encryption,
-        index: Turbopuffer::Models::NamespaceMetadata::index
+        index: Turbopuffer::Models::NamespaceMetadata::index,
+        schema: ::Hash[Symbol, Turbopuffer::AttributeSchemaConfig],
+        updated_at: Time
       }
 
-      type encryption = bool | Turbopuffer::NamespaceMetadata::Encryption::Cmek
+      type encryption =
+        Turbopuffer::NamespaceMetadata::Encryption::Sse
+        | Turbopuffer::NamespaceMetadata::Encryption::Cmek
 
       module Encryption
         extend Turbopuffer::Internal::Type::Union
 
+        type sse = { sse: bool }
+
+        class Sse < Turbopuffer::Internal::Type::BaseModel
+          attr_accessor sse: bool
+
+          def initialize: (sse: bool) -> void
+
+          def to_hash: -> { sse: bool }
+        end
+
         type cmek =
           { cmek: Turbopuffer::NamespaceMetadata::Encryption::Cmek::Cmek }
 
         class Cmek < Turbopuffer::Internal::Type::BaseModel
-          attr_reader cmek: Turbopuffer::NamespaceMetadata::Encryption::Cmek::Cmek?
-
-          def cmek=: (
-            Turbopuffer::NamespaceMetadata::Encryption::Cmek::Cmek
-          ) -> Turbopuffer::NamespaceMetadata::Encryption::Cmek::Cmek
+          attr_accessor cmek: Turbopuffer::NamespaceMetadata::Encryption::Cmek::Cmek
 
           def initialize: (
-            ?cmek: Turbopuffer::NamespaceMetadata::Encryption::Cmek::Cmek
+            cmek: Turbopuffer::NamespaceMetadata::Encryption::Cmek::Cmek
           ) -> void
 
           def to_hash: -> {
@@ -92,15 +92,15 @@ module Turbopuffer
       end
 
       type index =
-        Turbopuffer::NamespaceMetadata::Index::Status
-        | Turbopuffer::NamespaceMetadata::Index::UnionMember1
+        Turbopuffer::NamespaceMetadata::Index::IndexUpToDate
+        | Turbopuffer::NamespaceMetadata::Index::IndexUpdating
 
       module Index
         extend Turbopuffer::Internal::Type::Union
 
-        type status = { status: :"up-to-date" }
+        type index_up_to_date = { status: :"up-to-date" }
 
-        class Status < Turbopuffer::Internal::Type::BaseModel
+        class IndexUpToDate < Turbopuffer::Internal::Type::BaseModel
           attr_accessor status: :"up-to-date"
 
           def initialize: (?status: :"up-to-date") -> void
@@ -108,9 +108,9 @@ module Turbopuffer
           def to_hash: -> { status: :"up-to-date" }
         end
 
-        type union_member1 = { status: :updating, unindexed_bytes: Integer }
+        type index_updating = { status: :updating, unindexed_bytes: Integer }
 
-        class UnionMember1 < Turbopuffer::Internal::Type::BaseModel
+        class IndexUpdating < Turbopuffer::Internal::Type::BaseModel
           attr_accessor status: :updating
 
           attr_accessor unindexed_bytes: Integer

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: turbopuffer
 version: !ruby/object:Gem::Version
-  version: 1.5.0
+  version: 1.6.0
 platform: ruby
 authors:
 - Turbopuffer
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-10-22 00:00:00.000000000 Z
+date: 2025-11-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: connection_pool