turbopuffer 1.19.0 → 2.0.0.pre.alpha.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5b7d18934b6b6f5b4bca87b243216d1ba458598472cc5df4751bf1e23b8eb38c
-  data.tar.gz: 9ae7f3ad01c68fa4c76ad46419949bca2c957362e3e6bc96a2f9becf762632ea
+  metadata.gz: ec352d339fe92ac66fb6632df2bed51b81559d640f3b4a2140e1ded77218f282
+  data.tar.gz: ce362c309ffab91afb01bf3d057f66200b166cc8087bc7055d95f26492cfb941
 SHA512:
-  metadata.gz: c22b27d365408dec9029a0dd8b876fc3701c05e6ac69b9caf7e4a31140ff7af3635d31499421667b1ba026f15ef8f934bbe4a47378a4225720e0624626ec1f8a
-  data.tar.gz: 6b5ddcafe494903ed7bda7b113906feba7a150d7a51663b297c7134160fc2146fc35f7bb3e62b724a950d13e6a8e5cc42b6f420fce244a9a08d80203858507cc
+  metadata.gz: '0854bb51b122e2db1f0df2c44e2625e6486c964489615e730b97dd6a541cdb1dff537c41b3e93b967fc5b04a501068cab9342e787a8d186051d0b5a91096588a'
+  data.tar.gz: 00514217b637eb9f7d8a6dab848489334267aa8ff5b1ae1c3956fb3e1779cab7c6da22db534e57eb349dee4d9f176e872eae52780a90a234b7342bb01b13e52b

data/CHANGELOG.md

@@ -1,5 +1,37 @@
 # Changelog
 
+## 2.0.0-alpha.2 (2026-04-29)
+
+Full Changelog: [v2.0.0-alpha.1...v2.0.0-alpha.2](https://github.com/turbopuffer/turbopuffer-ruby/compare/v2.0.0-alpha.1...v2.0.0-alpha.2)
+
+### Features
+
+* Spec: Add sparse_distance_metric to the list of models. ([81f2998](https://github.com/turbopuffer/turbopuffer-ruby/commit/81f29982fb72e5f90923a35975841633b40aacc5))
+
+## 2.0.0-alpha.1 (2026-04-29)
+
+Full Changelog: [v1.19.0...v2.0.0-alpha.1](https://github.com/turbopuffer/turbopuffer-ruby/compare/v1.19.0...v2.0.0-alpha.1)
+
+### Features
+
+* Add sparse vectors to the spec. ([060434a](https://github.com/turbopuffer/turbopuffer-ruby/commit/060434a1516b678fdf8090b9c7d37c9b9102c3f3))
+* Enable Stainless normalizer v2 while keeping Query inlined ([20d751b](https://github.com/turbopuffer/turbopuffer-ruby/commit/20d751b10ff3a74dfc27f5f5baf4ac41ee50f71d))
+* Require `fuzzy: true` in the schema to use the `Fuzzy` filter ([7408995](https://github.com/turbopuffer/turbopuffer-ruby/commit/7408995fefd4c32638309feefe39e1c4d474213b))
+* set up branchFrom and copyFrom sugar in the OpenAPI spec ([3b0df07](https://github.com/turbopuffer/turbopuffer-ruby/commit/3b0df076ee081ed3ddf129be8762d683a949845c))
+* spec: flatten branchFrom/copyFrom sugar endpoint params ([06214bc](https://github.com/turbopuffer/turbopuffer-ruby/commit/06214bcbded87f2c214b3fc7b2cf2309540b197b))
+* spec: update OpenAPI spec for v2 encryption API ([0c2654f](https://github.com/turbopuffer/turbopuffer-ruby/commit/0c2654fdb779231b867c4c40f9ff9e904601ca3c))
+* support setting headers via env ([a0c4d53](https://github.com/turbopuffer/turbopuffer-ruby/commit/a0c4d53342da77dfbf07f60350246ab418926cff))
+
+
+### Bug Fixes
+
+* multipart encoding for file arrays ([bca18a3](https://github.com/turbopuffer/turbopuffer-ruby/commit/bca18a3d40304187b478ec130b274316fb95125a))
+
+
+### Chores
+
+* **internal:** more robust bootstrap script ([0b727d5](https://github.com/turbopuffer/turbopuffer-ruby/commit/0b727d5f42db0363d4143b8a912621c2c892c93f))
+
 ## 1.19.0 (2026-04-06)
 
 Full Changelog: [v1.18.0...v1.19.0](https://github.com/turbopuffer/turbopuffer-ruby/compare/v1.18.0...v1.19.0)

data/README.md

@@ -26,7 +26,7 @@ To use this gem, install via Bundler by adding the following to your application
 <!-- x-release-please-start-version -->
 
 ```ruby
-gem "turbopuffer", "~> 1.19.0"
+gem "turbopuffer", "~> 2.0.0.pre.alpha.2"
 ```
 
 <!-- x-release-please-end -->

data/lib/turbopuffer/client.rb

@@ -109,6 +109,19 @@ module Turbopuffer
         raise ArgumentError.new("api_key is required, and can be set via environ: \"TURBOPUFFER_API_KEY\"")
       end
 
+      headers = compression ? {} : {"accept-encoding" => "identity"}
+      custom_headers_env = ENV["TURBOPUFFER_CUSTOM_HEADERS"]
+      unless custom_headers_env.nil?
+        parsed = {}
+        custom_headers_env.split("\n").each do |line|
+          colon = line.index(":")
+          unless colon.nil?
+            parsed[line[0...colon].strip] = line[(colon + 1)..].strip
+          end
+        end
+        headers = parsed.merge(headers)
+      end
+
       if base_url.include?("{region}")
         if region.nil?
           raise ArgumentError.new("region is required when base_url contains {region} placeholder: #{base_url}")
@@ -128,7 +141,7 @@ module Turbopuffer
         max_retries: max_retries,
         initial_retry_delay: initial_retry_delay,
         max_retry_delay: max_retry_delay,
-        headers: compression ? {} : {"accept-encoding" => "identity"}
+        headers: headers
       )
     end
   end

data/lib/turbopuffer/internal/util.rb

@@ -610,6 +610,7 @@ module Turbopuffer
         #
         # @return [Array(String, Enumerable<String>)]
         private def encode_multipart_streaming(body)
+          # rubocop:disable Style/CaseEquality
           # RFC 1521 Section 7.2.1 says we should have 70 char maximum for boundary length
           boundary = SecureRandom.urlsafe_base64(46)
 
@@ -619,7 +620,7 @@ module Turbopuffer
             in Hash
               body.each do |key, val|
                 case val
-                in Array if val.all? { primitive?(_1) }
+                in Array if val.all? { primitive?(_1) || Turbopuffer::Internal::Type::FileInput === _1 }
                   val.each do |v|
                     write_multipart_chunk(y, boundary: boundary, key: key, val: v, closing: closing)
                   end
@@ -635,6 +636,7 @@ module Turbopuffer
 
           fused_io = fused_enum(strio) { closing.each(&:call) }
           [boundary, fused_io]
+          # rubocop:enable Style/CaseEquality
         end
 
         # @api private

data/lib/turbopuffer/models/attribute_schema.rb

@@ -6,7 +6,7 @@ module Turbopuffer
     module AttributeSchema
       extend Turbopuffer::Internal::Type::Union
 
-      # 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.
+      # 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, {}f16.
       variant -> { Turbopuffer::AttributeType }
 
       # Detailed configuration for an attribute attached to a document.

data/lib/turbopuffer/models/attribute_schema_config.rb

@@ -6,7 +6,7 @@ module Turbopuffer
       # @!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.
+      #   [DIMS]f16, [DIMS]f32, {}f16.
       #
       #   @return [String]
       required :type, String
@@ -32,6 +32,12 @@ module Turbopuffer
       #   @return [Boolean, Turbopuffer::Models::FullTextSearchConfig, nil]
       optional :full_text_search, union: -> { Turbopuffer::FullTextSearch }
 
+      # @!attribute fuzzy
+      #   Whether to enable Fuzzy filters on this attribute.
+      #
+      #   @return [Boolean, nil]
+      optional :fuzzy, Turbopuffer::Internal::Type::Boolean
+
       # @!attribute glob
       #   Whether to enable Glob filters on this attribute.
       #
@@ -44,7 +50,14 @@ module Turbopuffer
       #   @return [Boolean, nil]
       optional :regex, Turbopuffer::Internal::Type::Boolean
 
-      # @!method initialize(type:, ann: nil, filterable: nil, full_text_search: nil, glob: nil, regex: nil)
+      # @!attribute sparse_knn
+      #   Whether to create a sparse kNN index for the attribute. Requires the `{}f16`
+      #   type.
+      #
+      #   @return [Turbopuffer::Models::AttributeSchemaConfig::SparseKnn, nil]
+      optional :sparse_knn, -> { Turbopuffer::AttributeSchemaConfig::SparseKnn }
+
+      # @!method initialize(type:, ann: nil, filterable: nil, full_text_search: nil, fuzzy: nil, glob: nil, regex: nil, sparse_knn: nil)
       #   Some parameter documentations has been truncated, see
       #   {Turbopuffer::Models::AttributeSchemaConfig} for more details.
       #
@@ -58,9 +71,13 @@ module Turbopuffer
       #
       #   @param full_text_search [Boolean, Turbopuffer::Models::FullTextSearchConfig] Whether this attribute can be used as part of a BM25 full-text search. Requires
       #
+      #   @param fuzzy [Boolean] Whether to enable Fuzzy filters on this attribute.
+      #
       #   @param glob [Boolean] Whether to enable Glob filters on this attribute.
       #
       #   @param regex [Boolean] Whether to enable Regex filters on this attribute.
+      #
+      #   @param sparse_knn [Turbopuffer::Models::AttributeSchemaConfig::SparseKnn] Whether to create a sparse kNN index for the attribute. Requires the `{}f16` typ
 
       # Whether to create an approximate nearest neighbor index for the attribute. Can
       # be a boolean or a detailed configuration object.
@@ -90,6 +107,21 @@ module Turbopuffer
         # @!method self.variants
         #   @return [Array(Boolean, Turbopuffer::Models::AttributeSchemaConfig::Ann::AnnConfig)]
       end
+
+      # @see Turbopuffer::Models::AttributeSchemaConfig#sparse_knn
+      class SparseKnn < Turbopuffer::Internal::Type::BaseModel
+        # @!attribute distance_metric
+        #   A function used to calculate sparse vector similarity.
+        #
+        #   @return [Symbol, Turbopuffer::Models::SparseDistanceMetric]
+        required :distance_metric, enum: -> { Turbopuffer::SparseDistanceMetric }
+
+        # @!method initialize(distance_metric:)
+        #   Whether to create a sparse kNN index for the attribute. Requires the `{}f16`
+        #   type.
+        #
+        #   @param distance_metric [Symbol, Turbopuffer::Models::SparseDistanceMetric] A function used to calculate sparse vector similarity.
+      end
     end
   end
 end

data/lib/turbopuffer/models/columns.rb

@@ -32,20 +32,14 @@ module Turbopuffer
         # The vector embeddings of the documents.
         variant -> { Turbopuffer::Models::Columns::Vector::VectorArray }
 
-        # A dense vector encoded as an array of floats.
-        variant -> { Turbopuffer::Models::Columns::Vector::FloatArray }
-
-        # A dense vector encoded as a base64 string.
-        variant String
+        # A vector embedding associated with a document.
+        variant union: -> { Turbopuffer::Vector }
 
         # @!method self.variants
         #   @return [Array(Array<Array<Float>, String>, Array<Float>, String)]
 
         # @type [Turbopuffer::Internal::Type::Converter]
         VectorArray = Turbopuffer::Internal::Type::ArrayOf[union: -> { Turbopuffer::Vector }]
-
-        # @type [Turbopuffer::Internal::Type::Converter]
-        FloatArray = Turbopuffer::Internal::Type::ArrayOf[Float]
       end
 
       def method_missing(name, *args, &block)

data/lib/turbopuffer/models/encryption.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+module Turbopuffer
+  module Models
+    # The encryption configuration for a namespace.
+    module Encryption
+      extend Turbopuffer::Internal::Type::Union
+
+      # Encrypt the namespace with a customer-managed encryption key (CMEK).
+      variant -> { Turbopuffer::Encryption::CustomerManaged }
+
+      # Use the default server-side encryption (SSE).
+      variant -> { Turbopuffer::Encryption::Default }
+
+      class CustomerManaged < Turbopuffer::Internal::Type::BaseModel
+        # @!attribute key_name
+        #   The identifier of the CMEK key to use for encryption. For GCP, the
+        #   fully-qualified resource name of the key. For AWS, the ARN of the key.
+        #
+        #   @return [String]
+        required :key_name, String
+
+        # @!attribute mode
+        #
+        #   @return [Symbol, :"customer-managed"]
+        required :mode, const: :"customer-managed"
+
+        # @!method initialize(key_name:, mode: :"customer-managed")
+        #   Some parameter documentations has been truncated, see
+        #   {Turbopuffer::Models::Encryption::CustomerManaged} for more details.
+        #
+        #   Encrypt the namespace with a customer-managed encryption key (CMEK).
+        #
+        #   @param key_name [String] The identifier of the CMEK key to use for encryption. For GCP, the fully-qualifi
+        #
+        #   @param mode [Symbol, :"customer-managed"]
+      end
+
+      class Default < Turbopuffer::Internal::Type::BaseModel
+        # @!attribute mode
+        #
+        #   @return [Symbol, :default]
+        required :mode, const: :default
+
+        # @!method initialize(mode: :default)
+        #   Use the default server-side encryption (SSE).
+        #
+        #   @param mode [Symbol, :default]
+      end
+
+      # @!method self.variants
+      #   @return [Array(Turbopuffer::Models::Encryption::CustomerManaged, Turbopuffer::Models::Encryption::Default)]
+    end
+  end
+end

data/lib/turbopuffer/models/id.rb

@@ -6,7 +6,7 @@ module Turbopuffer
     module ID
       extend Turbopuffer::Internal::Type::Union
 
-      # A UUID.
+      # A string ID.
       variant String
 
       # An integer ID.

data/lib/turbopuffer/models/limit.rb

@@ -41,9 +41,6 @@ module Turbopuffer
         required :limit, Integer
 
         # @!method initialize(attributes:, limit:)
-        #   Some parameter documentations has been truncated, see
-        #   {Turbopuffer::Models::Limit::Per} for more details.
-        #
         #   Limits the number of documents with the same value for a set of attributes (the
         #   "limit key") that can appear in the results.
         #

data/lib/turbopuffer/models/namespace_branch_from_params.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module Turbopuffer
+  module Models
+    # @see Turbopuffer::Resources::Namespaces#branch_from
+    class NamespaceBranchFromParams < Turbopuffer::Internal::Type::BaseModel
+      extend Turbopuffer::Internal::Type::RequestParameters::Converter
+      include Turbopuffer::Internal::Type::RequestParameters
+
+      # @!attribute namespace
+      #
+      #   @return [String, nil]
+      optional :namespace, String
+
+      # @!attribute source_namespace
+      #   The namespace to create an instant, copy-on-write clone of.
+      #
+      #   @return [String]
+      required :source_namespace, String
+
+      # @!method initialize(source_namespace:, namespace: nil, request_options: {})
+      #   @param source_namespace [String] The namespace to create an instant, copy-on-write clone of.
+      #
+      #   @param namespace [String]
+      #
+      #   @param request_options [Turbopuffer::RequestOptions, Hash{Symbol=>Object}]
+    end
+  end
+end

data/lib/turbopuffer/models/namespace_branch_from_response.rb

@@ -0,0 +1,113 @@
+# frozen_string_literal: true
+
+module Turbopuffer
+  module Models
+    # @see Turbopuffer::Resources::Namespaces#branch_from
+    class NamespaceBranchFromResponse < Turbopuffer::Internal::Type::BaseModel
+      # @!attribute billing
+      #   The billing information for a write request.
+      #
+      #   @return [Turbopuffer::Models::WriteBilling]
+      required :billing, -> { Turbopuffer::WriteBilling }
+
+      # @!attribute message
+      #   A message describing the result of the write request.
+      #
+      #   @return [String]
+      required :message, String
+
+      # @!attribute rows_affected
+      #   The number of rows affected by the write request.
+      #
+      #   @return [Integer]
+      required :rows_affected, Integer
+
+      # @!attribute status
+      #   The status of the request.
+      #
+      #   @return [Symbol, :OK]
+      required :status, const: :OK
+
+      # @!attribute deleted_ids
+      #   The IDs of documents that were deleted. Only included when `return_affected_ids`
+      #   is true and at least one document was deleted.
+      #
+      #   @return [Array<String, Integer>, nil]
+      optional :deleted_ids, -> { Turbopuffer::Internal::Type::ArrayOf[union: Turbopuffer::ID] }
+
+      # @!attribute patched_ids
+      #   The IDs of documents that were patched. Only included when `return_affected_ids`
+      #   is true and at least one document was patched.
+      #
+      #   @return [Array<String, Integer>, nil]
+      optional :patched_ids, -> { Turbopuffer::Internal::Type::ArrayOf[union: Turbopuffer::ID] }
+
+      # @!attribute performance
+      #   The performance information for a write request.
+      #
+      #   @return [Turbopuffer::Models::WritePerformance, nil]
+      optional :performance, -> { Turbopuffer::WritePerformance }
+
+      # @!attribute rows_deleted
+      #   The number of rows deleted by the write request.
+      #
+      #   @return [Integer, nil]
+      optional :rows_deleted, Integer
+
+      # @!attribute rows_patched
+      #   The number of rows patched by the write request.
+      #
+      #   @return [Integer, nil]
+      optional :rows_patched, Integer
+
+      # @!attribute rows_remaining
+      #   Whether more documents match the filter for partial operations.
+      #
+      #   @return [Boolean, nil]
+      optional :rows_remaining, Turbopuffer::Internal::Type::Boolean
+
+      # @!attribute rows_upserted
+      #   The number of rows upserted by the write request.
+      #
+      #   @return [Integer, nil]
+      optional :rows_upserted, Integer
+
+      # @!attribute upserted_ids
+      #   The IDs of documents that were upserted. Only included when
+      #   `return_affected_ids` is true and at least one document was upserted.
+      #
+      #   @return [Array<String, Integer>, nil]
+      optional :upserted_ids, -> { Turbopuffer::Internal::Type::ArrayOf[union: Turbopuffer::ID] }
+
+      # @!method initialize(billing:, message:, rows_affected:, deleted_ids: nil, patched_ids: nil, performance: nil, rows_deleted: nil, rows_patched: nil, rows_remaining: nil, rows_upserted: nil, upserted_ids: nil, status: :OK)
+      #   Some parameter documentations has been truncated, see
+      #   {Turbopuffer::Models::NamespaceBranchFromResponse} for more details.
+      #
+      #   The response to a successful write request.
+      #
+      #   @param billing [Turbopuffer::Models::WriteBilling] The billing information for a write request.
+      #
+      #   @param message [String] A message describing the result of the write request.
+      #
+      #   @param rows_affected [Integer] The number of rows affected by the write request.
+      #
+      #   @param deleted_ids [Array<String, Integer>] The IDs of documents that were deleted. Only included when `return_affected_ids`
+      #
+      #   @param patched_ids [Array<String, Integer>] The IDs of documents that were patched. Only included when `return_affected_ids`
+      #
+      #   @param performance [Turbopuffer::Models::WritePerformance] The performance information for a write request.
+      #
+      #   @param rows_deleted [Integer] The number of rows deleted by the write request.
+      #
+      #   @param rows_patched [Integer] The number of rows patched by the write request.
+      #
+      #   @param rows_remaining [Boolean] Whether more documents match the filter for partial operations.
+      #
+      #   @param rows_upserted [Integer] The number of rows upserted by the write request.
+      #
+      #   @param upserted_ids [Array<String, Integer>] The IDs of documents that were upserted. Only included when `return_affected_ids
+      #
+      #   @param status [Symbol, :OK] The status of the request.
+    end
+  end
+end

data/lib/turbopuffer/models/namespace_copy_from_params.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+module Turbopuffer
+  module Models
+    # @see Turbopuffer::Resources::Namespaces#copy_from
+    class NamespaceCopyFromParams < Turbopuffer::Internal::Type::BaseModel
+      extend Turbopuffer::Internal::Type::RequestParameters::Converter
+      include Turbopuffer::Internal::Type::RequestParameters
+
+      # @!attribute namespace
+      #
+      #   @return [String, nil]
+      optional :namespace, String
+
+      # @!attribute source_namespace
+      #   The namespace to copy documents from.
+      #
+      #   @return [String]
+      required :source_namespace, String
+
+      # @!attribute source_api_key
+      #   (Optional) An API key for the organization containing the source namespace
+      #
+      #   @return [String, nil]
+      optional :source_api_key, String
+
+      # @!attribute source_region
+      #   (Optional) The region of the source namespace.
+      #
+      #   @return [String, nil]
+      optional :source_region, String
+
+      # @!method initialize(source_namespace:, namespace: nil, source_api_key: nil, source_region: nil, request_options: {})
+      #   @param source_namespace [String] The namespace to copy documents from.
+      #
+      #   @param namespace [String]
+      #
+      #   @param source_api_key [String] (Optional) An API key for the organization containing the source namespace
+      #
+      #   @param source_region [String] (Optional) The region of the source namespace.
+      #
+      #   @param request_options [Turbopuffer::RequestOptions, Hash{Symbol=>Object}]
+    end
+  end
+end

data/lib/turbopuffer/models/namespace_copy_from_response.rb

@@ -0,0 +1,113 @@
+# frozen_string_literal: true
+
+module Turbopuffer
+  module Models
+    # @see Turbopuffer::Resources::Namespaces#copy_from
+    class NamespaceCopyFromResponse < Turbopuffer::Internal::Type::BaseModel
+      # @!attribute billing
+      #   The billing information for a write request.
+      #
+      #   @return [Turbopuffer::Models::WriteBilling]
+      required :billing, -> { Turbopuffer::WriteBilling }
+
+      # @!attribute message
+      #   A message describing the result of the write request.
+      #
+      #   @return [String]
+      required :message, String
+
+      # @!attribute rows_affected
+      #   The number of rows affected by the write request.
+      #
+      #   @return [Integer]
+      required :rows_affected, Integer
+
+      # @!attribute status
+      #   The status of the request.
+      #
+      #   @return [Symbol, :OK]
+      required :status, const: :OK
+
+      # @!attribute deleted_ids
+      #   The IDs of documents that were deleted. Only included when `return_affected_ids`
+      #   is true and at least one document was deleted.
+      #
+      #   @return [Array<String, Integer>, nil]
+      optional :deleted_ids, -> { Turbopuffer::Internal::Type::ArrayOf[union: Turbopuffer::ID] }
+
+      # @!attribute patched_ids
+      #   The IDs of documents that were patched. Only included when `return_affected_ids`
+      #   is true and at least one document was patched.
+      #
+      #   @return [Array<String, Integer>, nil]
+      optional :patched_ids, -> { Turbopuffer::Internal::Type::ArrayOf[union: Turbopuffer::ID] }
+
+      # @!attribute performance
+      #   The performance information for a write request.
+      #
+      #   @return [Turbopuffer::Models::WritePerformance, nil]
+      optional :performance, -> { Turbopuffer::WritePerformance }
+
+      # @!attribute rows_deleted
+      #   The number of rows deleted by the write request.
+      #
+      #   @return [Integer, nil]
+      optional :rows_deleted, Integer
+
+      # @!attribute rows_patched
+      #   The number of rows patched by the write request.
+      #
+      #   @return [Integer, nil]
+      optional :rows_patched, Integer
+
+      # @!attribute rows_remaining
+      #   Whether more documents match the filter for partial operations.
+      #
+      #   @return [Boolean, nil]
+      optional :rows_remaining, Turbopuffer::Internal::Type::Boolean
+
+      # @!attribute rows_upserted
+      #   The number of rows upserted by the write request.
+      #
+      #   @return [Integer, nil]
+      optional :rows_upserted, Integer
+
+      # @!attribute upserted_ids
+      #   The IDs of documents that were upserted. Only included when
+      #   `return_affected_ids` is true and at least one document was upserted.
+      #
+      #   @return [Array<String, Integer>, nil]
+      optional :upserted_ids, -> { Turbopuffer::Internal::Type::ArrayOf[union: Turbopuffer::ID] }
+
+      # @!method initialize(billing:, message:, rows_affected:, deleted_ids: nil, patched_ids: nil, performance: nil, rows_deleted: nil, rows_patched: nil, rows_remaining: nil, rows_upserted: nil, upserted_ids: nil, status: :OK)
+      #   Some parameter documentations has been truncated, see
+      #   {Turbopuffer::Models::NamespaceCopyFromResponse} for more details.
+      #
+      #   The response to a successful write request.
+      #
+      #   @param billing [Turbopuffer::Models::WriteBilling] The billing information for a write request.
+      #
+      #   @param message [String] A message describing the result of the write request.
+      #
+      #   @param rows_affected [Integer] The number of rows affected by the write request.
+      #
+      #   @param deleted_ids [Array<String, Integer>] The IDs of documents that were deleted. Only included when `return_affected_ids`
+      #
+      #   @param patched_ids [Array<String, Integer>] The IDs of documents that were patched. Only included when `return_affected_ids`
+      #
+      #   @param performance [Turbopuffer::Models::WritePerformance] The performance information for a write request.
+      #
+      #   @param rows_deleted [Integer] The number of rows deleted by the write request.
+      #
+      #   @param rows_patched [Integer] The number of rows patched by the write request.
+      #
+      #   @param rows_remaining [Boolean] Whether more documents match the filter for partial operations.
+      #
+      #   @param rows_upserted [Integer] The number of rows upserted by the write request.
+      #
+      #   @param upserted_ids [Array<String, Integer>] The IDs of documents that were upserted. Only included when `return_affected_ids
+      #
+      #   @param status [Symbol, :OK] The status of the request.
+    end
+  end
+end