turbopuffer 0.1.0 → 1.2.0

data/rbi/turbopuffer/models/namespace_explain_query_response.rbi

@@ -0,0 +1,34 @@
+# typed: strong
+
+module Turbopuffer
+  module Models
+    class NamespaceExplainQueryResponse < Turbopuffer::Internal::Type::BaseModel
+      OrHash =
+        T.type_alias do
+          T.any(
+            Turbopuffer::Models::NamespaceExplainQueryResponse,
+            Turbopuffer::Internal::AnyHash
+          )
+        end
+
+      # The textual representation of the query plan.
+      sig { returns(T.nilable(String)) }
+      attr_reader :plan_text
+
+      sig { params(plan_text: String).void }
+      attr_writer :plan_text
+
+      # The response to a successful query explain.
+      sig { params(plan_text: String).returns(T.attached_class) }
+      def self.new(
+        # The textual representation of the query plan.
+        plan_text: nil
+      )
+      end
+
+      sig { override.returns({ plan_text: String }) }
+      def to_hash
+      end
+    end
+  end
+end

data/rbi/turbopuffer/models/namespace_hint_cache_warm_params.rbi

@@ -0,0 +1,41 @@
+# typed: strong
+
+module Turbopuffer
+  module Models
+    class NamespaceHintCacheWarmParams < Turbopuffer::Internal::Type::BaseModel
+      extend Turbopuffer::Internal::Type::RequestParameters::Converter
+      include Turbopuffer::Internal::Type::RequestParameters
+
+      OrHash =
+        T.type_alias do
+          T.any(
+            Turbopuffer::NamespaceHintCacheWarmParams,
+            Turbopuffer::Internal::AnyHash
+          )
+        end
+
+      sig { returns(T.nilable(String)) }
+      attr_reader :namespace
+
+      sig { params(namespace: String).void }
+      attr_writer :namespace
+
+      sig do
+        params(
+          namespace: String,
+          request_options: Turbopuffer::RequestOptions::OrHash
+        ).returns(T.attached_class)
+      end
+      def self.new(namespace: nil, request_options: {})
+      end
+
+      sig do
+        override.returns(
+          { namespace: String, request_options: Turbopuffer::RequestOptions }
+        )
+      end
+      def to_hash
+      end
+    end
+  end
+end

data/rbi/turbopuffer/models/namespace_hint_cache_warm_response.rbi

@@ -0,0 +1,38 @@
+# typed: strong
+
+module Turbopuffer
+  module Models
+    class NamespaceHintCacheWarmResponse < Turbopuffer::Internal::Type::BaseModel
+      OrHash =
+        T.type_alias do
+          T.any(
+            Turbopuffer::Models::NamespaceHintCacheWarmResponse,
+            Turbopuffer::Internal::AnyHash
+          )
+        end
+
+      # The status of the request.
+      sig { returns(Symbol) }
+      attr_accessor :status
+
+      sig { returns(T.nilable(String)) }
+      attr_reader :message
+
+      sig { params(message: String).void }
+      attr_writer :message
+
+      # The response to a successful cache warm request.
+      sig { params(message: String, status: Symbol).returns(T.attached_class) }
+      def self.new(
+        message: nil,
+        # The status of the request.
+        status: :ACCEPTED
+      )
+      end
+
+      sig { override.returns({ status: Symbol, message: String }) }
+      def to_hash
+      end
+    end
+  end
+end

data/rbi/turbopuffer/models/namespace_metadata.rbi

@@ -0,0 +1,62 @@
+# typed: strong
+
+module Turbopuffer
+  module Models
+    class NamespaceMetadata < Turbopuffer::Internal::Type::BaseModel
+      OrHash =
+        T.type_alias do
+          T.any(Turbopuffer::NamespaceMetadata, Turbopuffer::Internal::AnyHash)
+        end
+
+      # The approximate number of logical bytes in the namespace.
+      sig { returns(Integer) }
+      attr_accessor :approx_logical_bytes
+
+      # The approximate number of rows in the namespace.
+      sig { returns(Integer) }
+      attr_accessor :approx_row_count
+
+      # The timestamp when the namespace was created.
+      sig { returns(Time) }
+      attr_accessor :created_at
+
+      # The schema of the namespace.
+      sig { returns(T::Hash[Symbol, Turbopuffer::AttributeSchemaConfig]) }
+      attr_accessor :schema
+
+      # 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]
+        ).returns(T.attached_class)
+      end
+      def self.new(
+        # The approximate number of logical bytes in the namespace.
+        approx_logical_bytes:,
+        # The approximate number of rows in the namespace.
+        approx_row_count:,
+        # The timestamp when the namespace was created.
+        created_at:,
+        # The schema of the namespace.
+        schema:
+      )
+      end
+
+      sig do
+        override.returns(
+          {
+            approx_logical_bytes: Integer,
+            approx_row_count: Integer,
+            created_at: Time,
+            schema: T::Hash[Symbol, Turbopuffer::AttributeSchemaConfig]
+          }
+        )
+      end
+      def to_hash
+      end
+    end
+  end
+end

data/rbi/turbopuffer/models/namespace_metadata_params.rbi

@@ -0,0 +1,41 @@
+# typed: strong
+
+module Turbopuffer
+  module Models
+    class NamespaceMetadataParams < Turbopuffer::Internal::Type::BaseModel
+      extend Turbopuffer::Internal::Type::RequestParameters::Converter
+      include Turbopuffer::Internal::Type::RequestParameters
+
+      OrHash =
+        T.type_alias do
+          T.any(
+            Turbopuffer::NamespaceMetadataParams,
+            Turbopuffer::Internal::AnyHash
+          )
+        end
+
+      sig { returns(T.nilable(String)) }
+      attr_reader :namespace
+
+      sig { params(namespace: String).void }
+      attr_writer :namespace
+
+      sig do
+        params(
+          namespace: String,
+          request_options: Turbopuffer::RequestOptions::OrHash
+        ).returns(T.attached_class)
+      end
+      def self.new(namespace: nil, request_options: {})
+      end
+
+      sig do
+        override.returns(
+          { namespace: String, request_options: Turbopuffer::RequestOptions }
+        )
+      end
+      def to_hash
+      end
+    end
+  end
+end

data/rbi/turbopuffer/models/namespace_multi_query_params.rbi

@@ -0,0 +1,175 @@
+# typed: strong
+
+module Turbopuffer
+  module Models
+    class NamespaceMultiQueryParams < Turbopuffer::Internal::Type::BaseModel
+      extend Turbopuffer::Internal::Type::RequestParameters::Converter
+      include Turbopuffer::Internal::Type::RequestParameters
+
+      OrHash =
+        T.type_alias do
+          T.any(
+            Turbopuffer::NamespaceMultiQueryParams,
+            Turbopuffer::Internal::AnyHash
+          )
+        end
+
+      sig { returns(T.nilable(String)) }
+      attr_reader :namespace
+
+      sig { params(namespace: String).void }
+      attr_writer :namespace
+
+      sig { returns(T::Array[Turbopuffer::Query]) }
+      attr_accessor :queries
+
+      # The consistency level for a query.
+      sig do
+        returns(T.nilable(Turbopuffer::NamespaceMultiQueryParams::Consistency))
+      end
+      attr_reader :consistency
+
+      sig do
+        params(
+          consistency:
+            Turbopuffer::NamespaceMultiQueryParams::Consistency::OrHash
+        ).void
+      end
+      attr_writer :consistency
+
+      # The encoding to use for vectors in the response.
+      sig { returns(T.nilable(Turbopuffer::VectorEncoding::OrSymbol)) }
+      attr_reader :vector_encoding
+
+      sig do
+        params(vector_encoding: Turbopuffer::VectorEncoding::OrSymbol).void
+      end
+      attr_writer :vector_encoding
+
+      sig do
+        params(
+          queries: T::Array[Turbopuffer::Query::OrHash],
+          namespace: String,
+          consistency:
+            Turbopuffer::NamespaceMultiQueryParams::Consistency::OrHash,
+          vector_encoding: Turbopuffer::VectorEncoding::OrSymbol,
+          request_options: Turbopuffer::RequestOptions::OrHash
+        ).returns(T.attached_class)
+      end
+      def self.new(
+        queries:,
+        namespace: nil,
+        # The consistency level for a query.
+        consistency: nil,
+        # The encoding to use for vectors in the response.
+        vector_encoding: nil,
+        request_options: {}
+      )
+      end
+
+      sig do
+        override.returns(
+          {
+            namespace: String,
+            queries: T::Array[Turbopuffer::Query],
+            consistency: Turbopuffer::NamespaceMultiQueryParams::Consistency,
+            vector_encoding: Turbopuffer::VectorEncoding::OrSymbol,
+            request_options: Turbopuffer::RequestOptions
+          }
+        )
+      end
+      def to_hash
+      end
+
+      class Consistency < Turbopuffer::Internal::Type::BaseModel
+        OrHash =
+          T.type_alias do
+            T.any(
+              Turbopuffer::NamespaceMultiQueryParams::Consistency,
+              Turbopuffer::Internal::AnyHash
+            )
+          end
+
+        # The query's consistency level.
+        sig do
+          returns(
+            T.nilable(
+              Turbopuffer::NamespaceMultiQueryParams::Consistency::Level::OrSymbol
+            )
+          )
+        end
+        attr_reader :level
+
+        sig do
+          params(
+            level:
+              Turbopuffer::NamespaceMultiQueryParams::Consistency::Level::OrSymbol
+          ).void
+        end
+        attr_writer :level
+
+        # The consistency level for a query.
+        sig do
+          params(
+            level:
+              Turbopuffer::NamespaceMultiQueryParams::Consistency::Level::OrSymbol
+          ).returns(T.attached_class)
+        end
+        def self.new(
+          # The query's consistency level.
+          level: nil
+        )
+        end
+
+        sig do
+          override.returns(
+            {
+              level:
+                Turbopuffer::NamespaceMultiQueryParams::Consistency::Level::OrSymbol
+            }
+          )
+        end
+        def to_hash
+        end
+
+        # The query's consistency level.
+        module Level
+          extend Turbopuffer::Internal::Type::Enum
+
+          TaggedSymbol =
+            T.type_alias do
+              T.all(
+                Symbol,
+                Turbopuffer::NamespaceMultiQueryParams::Consistency::Level
+              )
+            end
+          OrSymbol = T.type_alias { T.any(Symbol, String) }
+
+          # Strong consistency. Requires a round-trip to object storage to fetch the latest writes.
+          STRONG =
+            T.let(
+              :strong,
+              Turbopuffer::NamespaceMultiQueryParams::Consistency::Level::TaggedSymbol
+            )
+
+          # Eventual consistency. Does not require a round-trip to object storage, but may not see the latest writes.
+          EVENTUAL =
+            T.let(
+              :eventual,
+              Turbopuffer::NamespaceMultiQueryParams::Consistency::Level::TaggedSymbol
+            )
+
+          sig do
+            override.returns(
+              T::Array[
+                Turbopuffer::NamespaceMultiQueryParams::Consistency::Level::TaggedSymbol
+              ]
+            )
+          end
+          def self.values
+          end
+        end
+      end
+    end
+  end
+end

data/rbi/turbopuffer/models/namespace_multi_query_response.rbi

@@ -0,0 +1,121 @@
+# typed: strong
+
+module Turbopuffer
+  module Models
+    class NamespaceMultiQueryResponse < Turbopuffer::Internal::Type::BaseModel
+      OrHash =
+        T.type_alias do
+          T.any(
+            Turbopuffer::Models::NamespaceMultiQueryResponse,
+            Turbopuffer::Internal::AnyHash
+          )
+        end
+
+      # The billing information for a query.
+      sig { returns(Turbopuffer::QueryBilling) }
+      attr_reader :billing
+
+      sig { params(billing: Turbopuffer::QueryBilling::OrHash).void }
+      attr_writer :billing
+
+      # The performance information for a query.
+      sig { returns(Turbopuffer::QueryPerformance) }
+      attr_reader :performance
+
+      sig { params(performance: Turbopuffer::QueryPerformance::OrHash).void }
+      attr_writer :performance
+
+      sig do
+        returns(
+          T::Array[Turbopuffer::Models::NamespaceMultiQueryResponse::Result]
+        )
+      end
+      attr_accessor :results
+
+      # The result of a multi-query.
+      sig do
+        params(
+          billing: Turbopuffer::QueryBilling::OrHash,
+          performance: Turbopuffer::QueryPerformance::OrHash,
+          results:
+            T::Array[
+              Turbopuffer::Models::NamespaceMultiQueryResponse::Result::OrHash
+            ]
+        ).returns(T.attached_class)
+      end
+      def self.new(
+        # The billing information for a query.
+        billing:,
+        # The performance information for a query.
+        performance:,
+        results:
+      )
+      end
+
+      sig do
+        override.returns(
+          {
+            billing: Turbopuffer::QueryBilling,
+            performance: Turbopuffer::QueryPerformance,
+            results:
+              T::Array[Turbopuffer::Models::NamespaceMultiQueryResponse::Result]
+          }
+        )
+      end
+      def to_hash
+      end
+
+      class Result < Turbopuffer::Internal::Type::BaseModel
+        OrHash =
+          T.type_alias do
+            T.any(
+              Turbopuffer::Models::NamespaceMultiQueryResponse::Result,
+              Turbopuffer::Internal::AnyHash
+            )
+          end
+
+        sig { returns(T.nilable(T::Array[T::Hash[Symbol, T.anything]])) }
+        attr_reader :aggregation_groups
+
+        sig do
+          params(aggregation_groups: T::Array[T::Hash[Symbol, T.anything]]).void
+        end
+        attr_writer :aggregation_groups
+
+        sig { returns(T.nilable(T::Hash[Symbol, T.anything])) }
+        attr_reader :aggregations
+
+        sig { params(aggregations: T::Hash[Symbol, T.anything]).void }
+        attr_writer :aggregations
+
+        sig { returns(T.nilable(T::Array[Turbopuffer::Row])) }
+        attr_reader :rows
+
+        sig { params(rows: T::Array[Turbopuffer::Row::OrHash]).void }
+        attr_writer :rows
+
+        sig do
+          params(
+            aggregation_groups: T::Array[T::Hash[Symbol, T.anything]],
+            aggregations: T::Hash[Symbol, T.anything],
+            rows: T::Array[Turbopuffer::Row::OrHash]
+          ).returns(T.attached_class)
+        end
+        def self.new(aggregation_groups: nil, aggregations: nil, rows: nil)
+        end
+
+        sig do
+          override.returns(
+            {
+              aggregation_groups: T::Array[T::Hash[Symbol, T.anything]],
+              aggregations: T::Hash[Symbol, T.anything],
+              rows: T::Array[Turbopuffer::Row]
+            }
+          )
+        end
+        def to_hash
+        end
+      end
+    end
+  end
+end