turbopuffer 2.4.0 → 2.5.0

data/rbi/turbopuffer/models/write_result.rbi

@@ -0,0 +1,159 @@
+# typed: strong
+
+module Turbopuffer
+  module Models
+    class WriteResult < Turbopuffer::Internal::Type::BaseModel
+      OrHash =
+        T.type_alias do
+          T.any(Turbopuffer::WriteResult, Turbopuffer::Internal::AnyHash)
+        end
+
+      # The billing information for a write request.
+      sig { returns(Turbopuffer::WriteBilling) }
+      attr_reader :billing
+
+      sig { params(billing: Turbopuffer::WriteBilling::OrHash).void }
+      attr_writer :billing
+
+      # A message describing the result of the write request.
+      sig { returns(String) }
+      attr_accessor :message
+
+      # The number of rows affected by the write request.
+      sig { returns(Integer) }
+      attr_accessor :rows_affected
+
+      # The status of the request.
+      sig { returns(Symbol) }
+      attr_accessor :status
+
+      # The IDs of documents that were deleted. Only included when `return_affected_ids`
+      # is true and at least one document was deleted.
+      sig { returns(T.nilable(T::Array[Turbopuffer::ID::Variants])) }
+      attr_reader :deleted_ids
+
+      sig { params(deleted_ids: T::Array[Turbopuffer::ID::Variants]).void }
+      attr_writer :deleted_ids
+
+      # The IDs of documents that were patched. Only included when `return_affected_ids`
+      # is true and at least one document was patched.
+      sig { returns(T.nilable(T::Array[Turbopuffer::ID::Variants])) }
+      attr_reader :patched_ids
+
+      sig { params(patched_ids: T::Array[Turbopuffer::ID::Variants]).void }
+      attr_writer :patched_ids
+
+      # The performance information for a write request.
+      sig { returns(T.nilable(Turbopuffer::WritePerformance)) }
+      attr_reader :performance
+
+      sig { params(performance: Turbopuffer::WritePerformance::OrHash).void }
+      attr_writer :performance
+
+      # The number of rows deleted by the write request.
+      sig { returns(T.nilable(Integer)) }
+      attr_reader :rows_deleted
+
+      sig { params(rows_deleted: Integer).void }
+      attr_writer :rows_deleted
+
+      # The number of rows patched by the write request.
+      sig { returns(T.nilable(Integer)) }
+      attr_reader :rows_patched
+
+      sig { params(rows_patched: Integer).void }
+      attr_writer :rows_patched
+
+      # Whether more documents match the filter for partial operations.
+      sig { returns(T.nilable(T::Boolean)) }
+      attr_reader :rows_remaining
+
+      sig { params(rows_remaining: T::Boolean).void }
+      attr_writer :rows_remaining
+
+      # The number of rows upserted by the write request.
+      sig { returns(T.nilable(Integer)) }
+      attr_reader :rows_upserted
+
+      sig { params(rows_upserted: Integer).void }
+      attr_writer :rows_upserted
+
+      # The IDs of documents that were upserted. Only included when
+      # `return_affected_ids` is true and at least one document was upserted.
+      sig { returns(T.nilable(T::Array[Turbopuffer::ID::Variants])) }
+      attr_reader :upserted_ids
+
+      sig { params(upserted_ids: T::Array[Turbopuffer::ID::Variants]).void }
+      attr_writer :upserted_ids
+
+      # The response to a successful write request.
+      sig do
+        params(
+          billing: Turbopuffer::WriteBilling::OrHash,
+          message: String,
+          rows_affected: Integer,
+          deleted_ids: T::Array[Turbopuffer::ID::Variants],
+          patched_ids: T::Array[Turbopuffer::ID::Variants],
+          performance: Turbopuffer::WritePerformance::OrHash,
+          rows_deleted: Integer,
+          rows_patched: Integer,
+          rows_remaining: T::Boolean,
+          rows_upserted: Integer,
+          upserted_ids: T::Array[Turbopuffer::ID::Variants],
+          status: Symbol
+        ).returns(T.attached_class)
+      end
+      def self.new(
+        # The billing information for a write request.
+        billing:,
+        # A message describing the result of the write request.
+        message:,
+        # The number of rows affected by the write request.
+        rows_affected:,
+        # The IDs of documents that were deleted. Only included when `return_affected_ids`
+        # is true and at least one document was deleted.
+        deleted_ids: nil,
+        # The IDs of documents that were patched. Only included when `return_affected_ids`
+        # is true and at least one document was patched.
+        patched_ids: nil,
+        # The performance information for a write request.
+        performance: nil,
+        # The number of rows deleted by the write request.
+        rows_deleted: nil,
+        # The number of rows patched by the write request.
+        rows_patched: nil,
+        # Whether more documents match the filter for partial operations.
+        rows_remaining: nil,
+        # The number of rows upserted by the write request.
+        rows_upserted: nil,
+        # The IDs of documents that were upserted. Only included when
+        # `return_affected_ids` is true and at least one document was upserted.
+        upserted_ids: nil,
+        # The status of the request.
+        status: :OK
+      )
+      end
+
+      sig do
+        override.returns(
+          {
+            billing: Turbopuffer::WriteBilling,
+            message: String,
+            rows_affected: Integer,
+            status: Symbol,
+            deleted_ids: T::Array[Turbopuffer::ID::Variants],
+            patched_ids: T::Array[Turbopuffer::ID::Variants],
+            performance: Turbopuffer::WritePerformance,
+            rows_deleted: Integer,
+            rows_patched: Integer,
+            rows_remaining: T::Boolean,
+            rows_upserted: Integer,
+            upserted_ids: T::Array[Turbopuffer::ID::Variants]
+          }
+        )
+      end
+      def to_hash
+      end
+    end
+  end
+end

data/rbi/turbopuffer/models.rbi

@@ -31,6 +31,11 @@ module Turbopuffer
   ContainsAnyTokenFilterParams =
     Turbopuffer::Models::ContainsAnyTokenFilterParams
 
+  CopyFromNamespaceOperation = Turbopuffer::Models::CopyFromNamespaceOperation
+
+  CopyFromNamespaceOperationResult =
+    Turbopuffer::Models::CopyFromNamespaceOperationResult
+
   CopyFromNamespaceParams = Turbopuffer::Models::CopyFromNamespaceParams
 
   DecayParams = Turbopuffer::Models::DecayParams
@@ -82,12 +87,17 @@ module Turbopuffer
 
   NamespaceMultiQueryParams = Turbopuffer::Models::NamespaceMultiQueryParams
 
+  NamespacePollCopyFromParams = Turbopuffer::Models::NamespacePollCopyFromParams
+
   NamespaceQueryParams = Turbopuffer::Models::NamespaceQueryParams
 
   NamespaceRecallParams = Turbopuffer::Models::NamespaceRecallParams
 
   NamespaceSchemaParams = Turbopuffer::Models::NamespaceSchemaParams
 
+  NamespaceStartCopyFromParams =
+    Turbopuffer::Models::NamespaceStartCopyFromParams
+
   NamespaceSummary = Turbopuffer::Models::NamespaceSummary
 
   NamespaceUpdateMetadataParams =
@@ -97,12 +107,16 @@ module Turbopuffer
 
   NamespaceWriteParams = Turbopuffer::Models::NamespaceWriteParams
 
+  OperationError = Turbopuffer::Models::OperationError
+
   PinningConfig = Turbopuffer::Models::PinningConfig
 
   QueryBilling = Turbopuffer::Models::QueryBilling
 
   QueryPerformance = Turbopuffer::Models::QueryPerformance
 
+  RerankLimit = Turbopuffer::Models::RerankLimit
+
   Row = Turbopuffer::Models::Row
 
   RrfParams = Turbopuffer::Models::RrfParams
@@ -122,4 +136,6 @@ module Turbopuffer
   WriteBilling = Turbopuffer::Models::WriteBilling
 
   WritePerformance = Turbopuffer::Models::WritePerformance
+
+  WriteResult = Turbopuffer::Models::WriteResult
 end

data/rbi/turbopuffer/resources/namespaces.rbi

@@ -80,6 +80,7 @@ module Turbopuffer
           group_by: T::Array[T.anything],
           include_attributes: Turbopuffer::IncludeAttributes::Variants,
           limit: T.any(Integer, Turbopuffer::Limit::OrHash),
+          offset: Integer,
           rank_by: T.anything,
           top_k: Integer,
           vector_encoding: Turbopuffer::VectorEncoding::OrSymbol,
@@ -113,6 +114,9 @@ module Turbopuffer
         include_attributes: nil,
         # Body param: Limits the documents returned by a query.
         limit: nil,
+        # Body param: Number of documents to skip before returning results. Supported only
+        # in v2 queries with an explicit `rank_by` and `top_k` or `limit`.
+        offset: nil,
         # Body param: How to rank the documents in the namespace.
         rank_by: nil,
         # Body param: The number of results to return.
@@ -159,6 +163,8 @@ module Turbopuffer
           namespace: String,
           consistency:
             Turbopuffer::NamespaceMultiQueryParams::Consistency::OrHash,
+          limit: T.any(Integer, Turbopuffer::RerankLimit::OrHash),
+          offset: Integer,
           rerank_by: T.anything,
           vector_encoding: Turbopuffer::VectorEncoding::OrSymbol,
           request_options: Turbopuffer::RequestOptions::OrHash
@@ -171,6 +177,11 @@ module Turbopuffer
         namespace: nil,
         # Body param: The consistency level for a query.
         consistency: nil,
+        # Body param: Limits the total number of reranked documents returned.
+        limit: nil,
+        # Body param: Number of reranked documents to skip before returning results.
+        # Requires `rerank_by` and `limit`.
+        offset: nil,
         # Body param: How to combine the rows returned by each sub-query into a single
         # ranked list.
         rerank_by: nil,
@@ -180,6 +191,23 @@ module Turbopuffer
       )
       end
 
+      # Retrieve the current status of a copy operation.
+      sig do
+        params(
+          token: String,
+          namespace: String,
+          request_options: Turbopuffer::RequestOptions::OrHash
+        ).returns(Turbopuffer::CopyFromNamespaceOperation::Variants)
+      end
+      def poll_copy_from(
+        # The operation token obtained when starting the copy.
+        token,
+        # The name of the namespace.
+        namespace: nil,
+        request_options: {}
+      )
+      end
+
       # Query, filter, full-text search and vector search documents.
       sig do
         params(
@@ -193,6 +221,7 @@ module Turbopuffer
           group_by: T::Array[T.anything],
           include_attributes: Turbopuffer::IncludeAttributes::Variants,
           limit: T.any(Integer, Turbopuffer::Limit::OrHash),
+          offset: Integer,
           rank_by: T.anything,
           top_k: Integer,
           vector_encoding: Turbopuffer::VectorEncoding::OrSymbol,
@@ -226,6 +255,9 @@ module Turbopuffer
         include_attributes: nil,
         # Body param: Limits the documents returned by a query.
         limit: nil,
+        # Body param: Number of documents to skip before returning results. Supported only
+        # in v2 queries with an explicit `rank_by` and `top_k` or `limit`.
+        offset: nil,
         # Body param: How to rank the documents in the namespace.
         rank_by: nil,
         # Body param: The number of results to return.
@@ -281,12 +313,47 @@ module Turbopuffer
       )
       end
 
+      # Start copying all documents from another namespace into this one. Returns an
+      # operation token without waiting for the copy to finish. Use the token to poll
+      # for progress and the result.
+      sig do
+        params(
+          source_namespace: String,
+          namespace: String,
+          dest_encryption:
+            T.any(
+              Turbopuffer::Encryption::CustomerManaged::OrHash,
+              Turbopuffer::Encryption::Default::OrHash
+            ),
+          source_api_key: String,
+          source_region: String,
+          request_options: Turbopuffer::RequestOptions::OrHash
+        ).returns(Turbopuffer::Models::NamespaceStartCopyFromResponse)
+      end
+      def start_copy_from(
+        # Body param: The namespace to copy documents from.
+        source_namespace:,
+        # Path param: The name of the namespace.
+        namespace: nil,
+        # Body param: (Optional) The encryption configuration for the destination
+        # namespace.
+        dest_encryption: nil,
+        # Body param: (Optional) An API key for the organization containing the source
+        # namespace
+        source_api_key: nil,
+        # Body param: (Optional) The region of the source namespace.
+        source_region: nil,
+        request_options: {}
+      )
+      end
+
       # Update metadata configuration for a namespace.
       sig do
         params(
           namespace: String,
           pinning:
             T.nilable(T.any(T::Boolean, Turbopuffer::PinningConfig::OrHash)),
+          read_only: T::Boolean,
           request_options: Turbopuffer::RequestOptions::OrHash
         ).returns(Turbopuffer::NamespaceMetadata)
       end
@@ -300,6 +367,10 @@ module Turbopuffer
         # - `true`: enable pinning with default configuration
         # - Object: set pinning configuration
         pinning: nil,
+        # Body param: Set to `true` to reject document and schema writes, or `false` to
+        # allow them. Writes already in progress may still commit. Metadata updates remain
+        # available.
+        read_only: nil,
         request_options: {}
       )
       end

data/sig/turbopuffer/models/copy_from_namespace_operation.rbs

@@ -0,0 +1,65 @@
+module Turbopuffer
+  module Models
+    type copy_from_namespace_operation =
+      Turbopuffer::CopyFromNamespaceOperation::Running
+      | Turbopuffer::CopyFromNamespaceOperation::Finished
+
+    module CopyFromNamespaceOperation
+      extend Turbopuffer::Internal::Type::Union
+
+      type running = { start_time: Time, status: :running, progress: String }
+
+      class Running < Turbopuffer::Internal::Type::BaseModel
+        attr_accessor start_time: Time
+
+        attr_accessor status: :running
+
+        attr_reader progress: String?
+
+        def progress=: (String) -> String
+
+        def initialize: (
+          start_time: Time,
+          ?progress: String,
+          ?status: :running
+        ) -> void
+
+        def to_hash: -> { start_time: Time, status: :running, progress: String }
+      end
+
+      type finished =
+        {
+          finish_time: Time,
+          result: Turbopuffer::Models::copy_from_namespace_operation_result,
+          start_time: Time,
+          status: :finished
+        }
+
+      class Finished < Turbopuffer::Internal::Type::BaseModel
+        attr_accessor finish_time: Time
+
+        attr_accessor result: Turbopuffer::Models::copy_from_namespace_operation_result
+
+        attr_accessor start_time: Time
+
+        attr_accessor status: :finished
+
+        def initialize: (
+          finish_time: Time,
+          result: Turbopuffer::Models::copy_from_namespace_operation_result,
+          start_time: Time,
+          ?status: :finished
+        ) -> void
+
+        def to_hash: -> {
+          finish_time: Time,
+          result: Turbopuffer::Models::copy_from_namespace_operation_result,
+          start_time: Time,
+          status: :finished
+        }
+      end
+
+      def self?.variants: -> ::Array[Turbopuffer::Models::copy_from_namespace_operation]
+    end
+  end
+end

data/sig/turbopuffer/models/copy_from_namespace_operation_result.rbs

@@ -0,0 +1,33 @@
+module Turbopuffer
+  module Models
+    type copy_from_namespace_operation_result =
+      Turbopuffer::CopyFromNamespaceOperationResult::Success
+      | Turbopuffer::CopyFromNamespaceOperationResult::Error
+
+    module CopyFromNamespaceOperationResult
+      extend Turbopuffer::Internal::Type::Union
+
+      type success = { success: Turbopuffer::WriteResult }
+
+      class Success < Turbopuffer::Internal::Type::BaseModel
+        attr_accessor success: Turbopuffer::WriteResult
+
+        def initialize: (success: Turbopuffer::WriteResult) -> void
+
+        def to_hash: -> { success: Turbopuffer::WriteResult }
+      end
+
+      type error = { error: Turbopuffer::OperationError }
+
+      class Error < Turbopuffer::Internal::Type::BaseModel
+        attr_accessor error: Turbopuffer::OperationError
+
+        def initialize: (error: Turbopuffer::OperationError) -> void
+
+        def to_hash: -> { error: Turbopuffer::OperationError }
+      end
+
+      def self?.variants: -> ::Array[Turbopuffer::Models::copy_from_namespace_operation_result]
+    end
+  end
+end

data/sig/turbopuffer/models/namespace_explain_query_params.rbs

@@ -12,6 +12,7 @@ module Turbopuffer
         group_by: ::Array[top],
         include_attributes: Turbopuffer::Models::include_attributes,
         limit: Turbopuffer::Models::NamespaceExplainQueryParams::limit,
+        offset: Integer,
         rank_by: top,
         top_k: Integer,
         vector_encoding: Turbopuffer::Models::vector_encoding
@@ -70,6 +71,10 @@ module Turbopuffer
         Turbopuffer::Models::NamespaceExplainQueryParams::limit
       ) -> Turbopuffer::Models::NamespaceExplainQueryParams::limit
 
+      attr_reader offset: Integer?
+
+      def offset=: (Integer) -> Integer
+
       attr_reader rank_by: top?
 
       def rank_by=: (top) -> top
@@ -95,6 +100,7 @@ module Turbopuffer
         ?group_by: ::Array[top],
         ?include_attributes: Turbopuffer::Models::include_attributes,
         ?limit: Turbopuffer::Models::NamespaceExplainQueryParams::limit,
+        ?offset: Integer,
         ?rank_by: top,
         ?top_k: Integer,
         ?vector_encoding: Turbopuffer::Models::vector_encoding,
@@ -112,6 +118,7 @@ module Turbopuffer
         group_by: ::Array[top],
         include_attributes: Turbopuffer::Models::include_attributes,
         limit: Turbopuffer::Models::NamespaceExplainQueryParams::limit,
+        offset: Integer,
         rank_by: top,
         top_k: Integer,
         vector_encoding: Turbopuffer::Models::vector_encoding,

data/sig/turbopuffer/models/namespace_metadata.rbs

@@ -10,6 +10,7 @@ module Turbopuffer
         schema: ::Hash[Symbol, Turbopuffer::AttributeSchemaConfig],
         updated_at: Time,
         pinning: Turbopuffer::NamespaceMetadata::Pinning,
+        read_only: bool,
         sharding: Turbopuffer::ShardingConfig
       }
 
@@ -34,6 +35,10 @@ module Turbopuffer
         Turbopuffer::NamespaceMetadata::Pinning
       ) -> Turbopuffer::NamespaceMetadata::Pinning
 
+      attr_reader read_only: bool?
+
+      def read_only=: (bool) -> bool
+
       attr_reader sharding: Turbopuffer::ShardingConfig?
 
       def sharding=: (
@@ -49,6 +54,7 @@ module Turbopuffer
         schema: ::Hash[Symbol, Turbopuffer::AttributeSchemaConfig],
         updated_at: Time,
         ?pinning: Turbopuffer::NamespaceMetadata::Pinning,
+        ?read_only: bool,
         ?sharding: Turbopuffer::ShardingConfig
       ) -> void
 
@@ -61,6 +67,7 @@ module Turbopuffer
         schema: ::Hash[Symbol, Turbopuffer::AttributeSchemaConfig],
         updated_at: Time,
         pinning: Turbopuffer::NamespaceMetadata::Pinning,
+        read_only: bool,
         sharding: Turbopuffer::ShardingConfig
       }
 
@@ -114,23 +121,32 @@ module Turbopuffer
         }
 
         type status =
-          { ready_replicas: Integer, updated_at: Time, utilization: Float }
+          {
+            ready_replicas: Integer,
+            replicas: Integer,
+            updated_at: Time,
+            utilization: Float
+          }
 
         class Status < Turbopuffer::Internal::Type::BaseModel
           attr_accessor ready_replicas: Integer
 
+          attr_accessor replicas: Integer
+
           attr_accessor updated_at: Time
 
           attr_accessor utilization: Float
 
           def initialize: (
             ready_replicas: Integer,
+            replicas: Integer,
             updated_at: Time,
             utilization: Float
           ) -> void
 
           def to_hash: -> {
             ready_replicas: Integer,
+            replicas: Integer,
             updated_at: Time,
             utilization: Float
           }

data/sig/turbopuffer/models/namespace_metadata_patch.rbs

@@ -1,17 +1,26 @@
 module Turbopuffer
   module Models
     type namespace_metadata_patch =
-      { pinning: Turbopuffer::Models::NamespaceMetadataPatch::pinning? }
+      {
+        pinning: Turbopuffer::Models::NamespaceMetadataPatch::pinning?,
+        read_only: bool
+      }
 
     class NamespaceMetadataPatch < Turbopuffer::Internal::Type::BaseModel
       attr_accessor pinning: Turbopuffer::Models::NamespaceMetadataPatch::pinning?
 
+      attr_reader read_only: bool?
+
+      def read_only=: (bool) -> bool
+
       def initialize: (
-        ?pinning: Turbopuffer::Models::NamespaceMetadataPatch::pinning?
+        ?pinning: Turbopuffer::Models::NamespaceMetadataPatch::pinning?,
+        ?read_only: bool
       ) -> void
 
       def to_hash: -> {
-        pinning: Turbopuffer::Models::NamespaceMetadataPatch::pinning?
+        pinning: Turbopuffer::Models::NamespaceMetadataPatch::pinning?,
+        read_only: bool
       }
 
       type pinning = bool | Turbopuffer::PinningConfig

data/sig/turbopuffer/models/namespace_multi_query_params.rbs

@@ -5,6 +5,8 @@ module Turbopuffer
         namespace: String,
         queries: ::Array[Turbopuffer::NamespaceMultiQueryParams::Query],
         consistency: Turbopuffer::NamespaceMultiQueryParams::Consistency,
+        limit: Turbopuffer::Models::NamespaceMultiQueryParams::limit,
+        offset: Integer,
         rerank_by: top,
         vector_encoding: Turbopuffer::Models::vector_encoding
       }
@@ -26,6 +28,16 @@ module Turbopuffer
         Turbopuffer::NamespaceMultiQueryParams::Consistency
       ) -> Turbopuffer::NamespaceMultiQueryParams::Consistency
 
+      attr_reader limit: Turbopuffer::Models::NamespaceMultiQueryParams::limit?
+
+      def limit=: (
+        Turbopuffer::Models::NamespaceMultiQueryParams::limit
+      ) -> Turbopuffer::Models::NamespaceMultiQueryParams::limit
+
+      attr_reader offset: Integer?
+
+      def offset=: (Integer) -> Integer
+
       attr_reader rerank_by: top?
 
       def rerank_by=: (top) -> top
@@ -40,6 +52,8 @@ module Turbopuffer
         queries: ::Array[Turbopuffer::NamespaceMultiQueryParams::Query],
         ?namespace: String,
         ?consistency: Turbopuffer::NamespaceMultiQueryParams::Consistency,
+        ?limit: Turbopuffer::Models::NamespaceMultiQueryParams::limit,
+        ?offset: Integer,
         ?rerank_by: top,
         ?vector_encoding: Turbopuffer::Models::vector_encoding,
         ?request_options: Turbopuffer::request_opts
@@ -49,6 +63,8 @@ module Turbopuffer
         namespace: String,
         queries: ::Array[Turbopuffer::NamespaceMultiQueryParams::Query],
         consistency: Turbopuffer::NamespaceMultiQueryParams::Consistency,
+        limit: Turbopuffer::Models::NamespaceMultiQueryParams::limit,
+        offset: Integer,
         rerank_by: top,
         vector_encoding: Turbopuffer::Models::vector_encoding,
         request_options: Turbopuffer::RequestOptions
@@ -64,6 +80,7 @@ module Turbopuffer
           group_by: ::Array[top],
           include_attributes: Turbopuffer::Models::include_attributes,
           limit: Turbopuffer::Models::NamespaceMultiQueryParams::Query::limit,
+          offset: Integer,
           rank_by: top,
           top_k: Integer
         }
@@ -107,6 +124,10 @@ module Turbopuffer
           Turbopuffer::Models::NamespaceMultiQueryParams::Query::limit
         ) -> Turbopuffer::Models::NamespaceMultiQueryParams::Query::limit
 
+        attr_reader offset: Integer?
+
+        def offset=: (Integer) -> Integer
+
         attr_reader rank_by: top?
 
         def rank_by=: (top) -> top
@@ -124,6 +145,7 @@ module Turbopuffer
           ?group_by: ::Array[top],
           ?include_attributes: Turbopuffer::Models::include_attributes,
           ?limit: Turbopuffer::Models::NamespaceMultiQueryParams::Query::limit,
+          ?offset: Integer,
           ?rank_by: top,
           ?top_k: Integer
         ) -> void
@@ -137,6 +159,7 @@ module Turbopuffer
           group_by: ::Array[top],
           include_attributes: Turbopuffer::Models::include_attributes,
           limit: Turbopuffer::Models::NamespaceMultiQueryParams::Query::limit,
+          offset: Integer,
           rank_by: top,
           top_k: Integer
         }
@@ -184,6 +207,14 @@ module Turbopuffer
           def self?.values: -> ::Array[Turbopuffer::Models::NamespaceMultiQueryParams::Consistency::level]
         end
       end
+
+      type limit = Integer | Turbopuffer::RerankLimit
+
+      module Limit
+        extend Turbopuffer::Internal::Type::Union
+
+        def self?.variants: -> ::Array[Turbopuffer::Models::NamespaceMultiQueryParams::limit]
+      end
     end
   end
 end