turbopuffer 2.3.0 → 2.5.0

data/sig/turbopuffer/models/namespace_explain_query_params.rbs

@@ -4,6 +4,7 @@ module Turbopuffer
       {
         namespace: String,
         aggregate_by: ::Hash[Symbol, top],
+        compute_attributes: ::Hash[Symbol, top],
         consistency: Turbopuffer::NamespaceExplainQueryParams::Consistency,
         distance_metric: Turbopuffer::Models::distance_metric,
         exclude_attributes: ::Array[String],
@@ -11,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
@@ -29,6 +31,10 @@ module Turbopuffer
 
       def aggregate_by=: (::Hash[Symbol, top]) -> ::Hash[Symbol, top]
 
+      attr_reader compute_attributes: ::Hash[Symbol, top]?
+
+      def compute_attributes=: (::Hash[Symbol, top]) -> ::Hash[Symbol, top]
+
       attr_reader consistency: Turbopuffer::NamespaceExplainQueryParams::Consistency?
 
       def consistency=: (
@@ -65,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
@@ -82,6 +92,7 @@ module Turbopuffer
       def initialize: (
         ?namespace: String,
         ?aggregate_by: ::Hash[Symbol, top],
+        ?compute_attributes: ::Hash[Symbol, top],
         ?consistency: Turbopuffer::NamespaceExplainQueryParams::Consistency,
         ?distance_metric: Turbopuffer::Models::distance_metric,
         ?exclude_attributes: ::Array[String],
@@ -89,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,
@@ -98,6 +110,7 @@ module Turbopuffer
       def to_hash: -> {
         namespace: String,
         aggregate_by: ::Hash[Symbol, top],
+        compute_attributes: ::Hash[Symbol, top],
         consistency: Turbopuffer::NamespaceExplainQueryParams::Consistency,
         distance_metric: Turbopuffer::Models::distance_metric,
         exclude_attributes: ::Array[String],
@@ -105,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

@@ -9,7 +9,9 @@ module Turbopuffer
         index: Turbopuffer::Models::NamespaceMetadata::index,
         schema: ::Hash[Symbol, Turbopuffer::AttributeSchemaConfig],
         updated_at: Time,
-        pinning: Turbopuffer::NamespaceMetadata::Pinning
+        pinning: Turbopuffer::NamespaceMetadata::Pinning,
+        read_only: bool,
+        sharding: Turbopuffer::ShardingConfig
       }
 
     class NamespaceMetadata < Turbopuffer::Internal::Type::BaseModel
@@ -33,6 +35,16 @@ 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=: (
+        Turbopuffer::ShardingConfig
+      ) -> Turbopuffer::ShardingConfig
+
       def initialize: (
         approx_logical_bytes: Integer,
         approx_row_count: Integer,
@@ -41,7 +53,9 @@ module Turbopuffer
         index: Turbopuffer::Models::NamespaceMetadata::index,
         schema: ::Hash[Symbol, Turbopuffer::AttributeSchemaConfig],
         updated_at: Time,
-        ?pinning: Turbopuffer::NamespaceMetadata::Pinning
+        ?pinning: Turbopuffer::NamespaceMetadata::Pinning,
+        ?read_only: bool,
+        ?sharding: Turbopuffer::ShardingConfig
       ) -> void
 
       def to_hash: -> {
@@ -52,7 +66,9 @@ module Turbopuffer
         index: Turbopuffer::Models::NamespaceMetadata::index,
         schema: ::Hash[Symbol, Turbopuffer::AttributeSchemaConfig],
         updated_at: Time,
-        pinning: Turbopuffer::NamespaceMetadata::Pinning
+        pinning: Turbopuffer::NamespaceMetadata::Pinning,
+        read_only: bool,
+        sharding: Turbopuffer::ShardingConfig
       }
 
       type index =
@@ -105,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
@@ -57,12 +73,14 @@ module Turbopuffer
       type query =
         {
           aggregate_by: ::Hash[Symbol, top],
+          compute_attributes: ::Hash[Symbol, top],
           distance_metric: Turbopuffer::Models::distance_metric,
           exclude_attributes: ::Array[String],
           filters: top,
           group_by: ::Array[top],
           include_attributes: Turbopuffer::Models::include_attributes,
           limit: Turbopuffer::Models::NamespaceMultiQueryParams::Query::limit,
+          offset: Integer,
           rank_by: top,
           top_k: Integer
         }
@@ -72,6 +90,10 @@ module Turbopuffer
 
         def aggregate_by=: (::Hash[Symbol, top]) -> ::Hash[Symbol, top]
 
+        attr_reader compute_attributes: ::Hash[Symbol, top]?
+
+        def compute_attributes=: (::Hash[Symbol, top]) -> ::Hash[Symbol, top]
+
         attr_reader distance_metric: Turbopuffer::Models::distance_metric?
 
         def distance_metric=: (
@@ -102,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
@@ -112,24 +138,28 @@ module Turbopuffer
 
         def initialize: (
           ?aggregate_by: ::Hash[Symbol, top],
+          ?compute_attributes: ::Hash[Symbol, top],
           ?distance_metric: Turbopuffer::Models::distance_metric,
           ?exclude_attributes: ::Array[String],
           ?filters: top,
           ?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
 
         def to_hash: -> {
           aggregate_by: ::Hash[Symbol, top],
+          compute_attributes: ::Hash[Symbol, top],
           distance_metric: Turbopuffer::Models::distance_metric,
           exclude_attributes: ::Array[String],
           filters: top,
           group_by: ::Array[top],
           include_attributes: Turbopuffer::Models::include_attributes,
           limit: Turbopuffer::Models::NamespaceMultiQueryParams::Query::limit,
+          offset: Integer,
           rank_by: top,
           top_k: Integer
         }
@@ -177,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

data/sig/turbopuffer/models/namespace_poll_copy_from_params.rbs

@@ -0,0 +1,30 @@
+module Turbopuffer
+  module Models
+    type namespace_poll_copy_from_params =
+      { namespace: String, token: String }
+      & Turbopuffer::Internal::Type::request_parameters
+
+    class NamespacePollCopyFromParams < Turbopuffer::Internal::Type::BaseModel
+      extend Turbopuffer::Internal::Type::RequestParameters::Converter
+      include Turbopuffer::Internal::Type::RequestParameters
+
+      attr_reader namespace: String?
+
+      def namespace=: (String) -> String
+
+      attr_accessor token: String
+
+      def initialize: (
+        token: String,
+        ?namespace: String,
+        ?request_options: Turbopuffer::request_opts
+      ) -> void
+
+      def to_hash: -> {
+        namespace: String,
+        token: String,
+        request_options: Turbopuffer::RequestOptions
+      }
+    end
+  end
+end

data/sig/turbopuffer/models/namespace_query_params.rbs

@@ -4,6 +4,7 @@ module Turbopuffer
       {
         namespace: String,
         aggregate_by: ::Hash[Symbol, top],
+        compute_attributes: ::Hash[Symbol, top],
         consistency: Turbopuffer::NamespaceQueryParams::Consistency,
         distance_metric: Turbopuffer::Models::distance_metric,
         exclude_attributes: ::Array[String],
@@ -11,6 +12,7 @@ module Turbopuffer
         group_by: ::Array[top],
         include_attributes: Turbopuffer::Models::include_attributes,
         limit: Turbopuffer::Models::NamespaceQueryParams::limit,
+        offset: Integer,
         rank_by: top,
         top_k: Integer,
         vector_encoding: Turbopuffer::Models::vector_encoding
@@ -29,6 +31,10 @@ module Turbopuffer
 
       def aggregate_by=: (::Hash[Symbol, top]) -> ::Hash[Symbol, top]
 
+      attr_reader compute_attributes: ::Hash[Symbol, top]?
+
+      def compute_attributes=: (::Hash[Symbol, top]) -> ::Hash[Symbol, top]
+
       attr_reader consistency: Turbopuffer::NamespaceQueryParams::Consistency?
 
       def consistency=: (
@@ -65,6 +71,10 @@ module Turbopuffer
         Turbopuffer::Models::NamespaceQueryParams::limit
       ) -> Turbopuffer::Models::NamespaceQueryParams::limit
 
+      attr_reader offset: Integer?
+
+      def offset=: (Integer) -> Integer
+
       attr_reader rank_by: top?
 
       def rank_by=: (top) -> top
@@ -82,6 +92,7 @@ module Turbopuffer
       def initialize: (
         ?namespace: String,
         ?aggregate_by: ::Hash[Symbol, top],
+        ?compute_attributes: ::Hash[Symbol, top],
         ?consistency: Turbopuffer::NamespaceQueryParams::Consistency,
         ?distance_metric: Turbopuffer::Models::distance_metric,
         ?exclude_attributes: ::Array[String],
@@ -89,6 +100,7 @@ module Turbopuffer
         ?group_by: ::Array[top],
         ?include_attributes: Turbopuffer::Models::include_attributes,
         ?limit: Turbopuffer::Models::NamespaceQueryParams::limit,
+        ?offset: Integer,
         ?rank_by: top,
         ?top_k: Integer,
         ?vector_encoding: Turbopuffer::Models::vector_encoding,
@@ -98,6 +110,7 @@ module Turbopuffer
       def to_hash: -> {
         namespace: String,
         aggregate_by: ::Hash[Symbol, top],
+        compute_attributes: ::Hash[Symbol, top],
         consistency: Turbopuffer::NamespaceQueryParams::Consistency,
         distance_metric: Turbopuffer::Models::distance_metric,
         exclude_attributes: ::Array[String],
@@ -105,6 +118,7 @@ module Turbopuffer
         group_by: ::Array[top],
         include_attributes: Turbopuffer::Models::include_attributes,
         limit: Turbopuffer::Models::NamespaceQueryParams::limit,
+        offset: Integer,
         rank_by: top,
         top_k: Integer,
         vector_encoding: Turbopuffer::Models::vector_encoding,

data/sig/turbopuffer/models/namespace_start_copy_from_params.rbs

@@ -0,0 +1,56 @@
+module Turbopuffer
+  module Models
+    type namespace_start_copy_from_params =
+      {
+        namespace: String,
+        source_namespace: String,
+        dest_encryption: Turbopuffer::Models::encryption,
+        source_api_key: String,
+        source_region: String
+      }
+      & Turbopuffer::Internal::Type::request_parameters
+
+    class NamespaceStartCopyFromParams < Turbopuffer::Internal::Type::BaseModel
+      extend Turbopuffer::Internal::Type::RequestParameters::Converter
+      include Turbopuffer::Internal::Type::RequestParameters
+
+      attr_reader namespace: String?
+
+      def namespace=: (String) -> String
+
+      attr_accessor source_namespace: String
+
+      attr_reader dest_encryption: Turbopuffer::Models::encryption?
+
+      def dest_encryption=: (
+        Turbopuffer::Models::encryption
+      ) -> Turbopuffer::Models::encryption
+
+      attr_reader source_api_key: String?
+
+      def source_api_key=: (String) -> String
+
+      attr_reader source_region: String?
+
+      def source_region=: (String) -> String
+
+      def initialize: (
+        source_namespace: String,
+        ?namespace: String,
+        ?dest_encryption: Turbopuffer::Models::encryption,
+        ?source_api_key: String,
+        ?source_region: String,
+        ?request_options: Turbopuffer::request_opts
+      ) -> void
+
+      def to_hash: -> {
+        namespace: String,
+        source_namespace: String,
+        dest_encryption: Turbopuffer::Models::encryption,
+        source_api_key: String,
+        source_region: String,
+        request_options: Turbopuffer::RequestOptions
+      }
+    end
+  end
+end

data/sig/turbopuffer/models/namespace_start_copy_from_response.rbs

@@ -0,0 +1,13 @@
+module Turbopuffer
+  module Models
+    type namespace_start_copy_from_response = { token: String }
+
+    class NamespaceStartCopyFromResponse < Turbopuffer::Internal::Type::BaseModel
+      attr_accessor token: String
+
+      def initialize: (token: String) -> void
+
+      def to_hash: -> { token: String }
+    end
+  end
+end

data/sig/turbopuffer/models/namespace_write_params.rbs

@@ -19,6 +19,7 @@ module Turbopuffer
         patch_rows: ::Array[Turbopuffer::Row],
         return_affected_ids: bool,
         schema: ::Hash[Symbol, Turbopuffer::Models::attribute_schema],
+        sharding: Turbopuffer::ShardingConfig,
         upsert_columns: Turbopuffer::Columns,
         upsert_condition: top,
         upsert_rows: ::Array[Turbopuffer::Row]
@@ -111,6 +112,12 @@ module Turbopuffer
         ::Hash[Symbol, Turbopuffer::Models::attribute_schema]
       ) -> ::Hash[Symbol, Turbopuffer::Models::attribute_schema]
 
+      attr_reader sharding: Turbopuffer::ShardingConfig?
+
+      def sharding=: (
+        Turbopuffer::ShardingConfig
+      ) -> Turbopuffer::ShardingConfig
+
       attr_reader upsert_columns: Turbopuffer::Columns?
 
       def upsert_columns=: (Turbopuffer::Columns) -> Turbopuffer::Columns
@@ -141,6 +148,7 @@ module Turbopuffer
         ?patch_rows: ::Array[Turbopuffer::Row],
         ?return_affected_ids: bool,
         ?schema: ::Hash[Symbol, Turbopuffer::Models::attribute_schema],
+        ?sharding: Turbopuffer::ShardingConfig,
         ?upsert_columns: Turbopuffer::Columns,
         ?upsert_condition: top,
         ?upsert_rows: ::Array[Turbopuffer::Row],
@@ -165,6 +173,7 @@ module Turbopuffer
         patch_rows: ::Array[Turbopuffer::Row],
         return_affected_ids: bool,
         schema: ::Hash[Symbol, Turbopuffer::Models::attribute_schema],
+        sharding: Turbopuffer::ShardingConfig,
         upsert_columns: Turbopuffer::Columns,
         upsert_condition: top,
         upsert_rows: ::Array[Turbopuffer::Row],

data/sig/turbopuffer/models/operation_error.rbs

@@ -0,0 +1,34 @@
+module Turbopuffer
+  module Models
+    type operation_error =
+      { detail: Turbopuffer::OperationError::Detail, status_code: Integer }
+
+    class OperationError < Turbopuffer::Internal::Type::BaseModel
+      attr_accessor detail: Turbopuffer::OperationError::Detail
+
+      attr_accessor status_code: Integer
+
+      def initialize: (
+        detail: Turbopuffer::OperationError::Detail,
+        status_code: Integer
+      ) -> void
+
+      def to_hash: -> {
+        detail: Turbopuffer::OperationError::Detail,
+        status_code: Integer
+      }
+
+      type detail = { error: String, status: :error }
+
+      class Detail < Turbopuffer::Internal::Type::BaseModel
+        attr_accessor error: String
+
+        attr_accessor status: :error
+
+        def initialize: (error: String, ?status: :error) -> void
+
+        def to_hash: -> { error: String, status: :error }
+      end
+    end
+  end
+end

data/sig/turbopuffer/models/rerank_limit.rbs

@@ -0,0 +1,13 @@
+module Turbopuffer
+  module Models
+    type rerank_limit = { total: Integer }
+
+    class RerankLimit < Turbopuffer::Internal::Type::BaseModel
+      attr_accessor total: Integer
+
+      def initialize: (total: Integer) -> void
+
+      def to_hash: -> { total: Integer }
+    end
+  end
+end

data/sig/turbopuffer/models/rrf_params.rbs

@@ -1,15 +1,22 @@
 module Turbopuffer
   module Models
-    type rrf_params = { rank_constant: Integer }
+    type rrf_params = { rank_constant: Integer, weights: ::Array[Float] }
 
     class RrfParams < Turbopuffer::Internal::Type::BaseModel
       attr_reader rank_constant: Integer?
 
       def rank_constant=: (Integer) -> Integer
 
-      def initialize: (?rank_constant: Integer) -> void
+      attr_reader weights: ::Array[Float]?
 
-      def to_hash: -> { rank_constant: Integer }
+      def weights=: (::Array[Float]) -> ::Array[Float]
+
+      def initialize: (
+        ?rank_constant: Integer,
+        ?weights: ::Array[Float]
+      ) -> void
+
+      def to_hash: -> { rank_constant: Integer, weights: ::Array[Float] }
     end
   end
 end

data/sig/turbopuffer/models/sharding_config.rbs

@@ -0,0 +1,13 @@
+module Turbopuffer
+  module Models
+    type sharding_config = { num_shards: Integer }
+
+    class ShardingConfig < Turbopuffer::Internal::Type::BaseModel
+      attr_accessor num_shards: Integer
+
+      def initialize: (num_shards: Integer) -> void
+
+      def to_hash: -> { num_shards: Integer }
+    end
+  end
+end