turbopuffer-ruby 0.2.0.pre.alpha.1

data/lib/turbopuffer/resources/namespaces.rb

@@ -0,0 +1,311 @@
+# frozen_string_literal: true
+
+module Turbopuffer
+  module Resources
+    class Namespaces
+      # Delete namespace.
+      #
+      # @overload delete_all(namespace: nil, request_options: {})
+      #
+      # @param namespace [String] The name of the namespace.
+      #
+      # @param request_options [Turbopuffer::RequestOptions, Hash{Symbol=>Object}, nil]
+      #
+      # @return [Turbopuffer::Models::NamespaceDeleteAllResponse]
+      #
+      # @see Turbopuffer::Models::NamespaceDeleteAllParams
+      def delete_all(params = {})
+        parsed, options = Turbopuffer::NamespaceDeleteAllParams.dump_request(params)
+        namespace =
+          parsed.delete(:namespace) do
+            @client.default_namespace
+          end
+        @client.request(
+          method: :delete,
+          path: ["v2/namespaces/%1$s", namespace],
+          model: Turbopuffer::Models::NamespaceDeleteAllResponse,
+          options: options
+        )
+      end
+
+      # Warm the cache for a namespace.
+      #
+      # @overload hint_cache_warm(namespace: nil, request_options: {})
+      #
+      # @param namespace [String] The name of the namespace.
+      #
+      # @param request_options [Turbopuffer::RequestOptions, Hash{Symbol=>Object}, nil]
+      #
+      # @return [Turbopuffer::Models::NamespaceHintCacheWarmResponse]
+      #
+      # @see Turbopuffer::Models::NamespaceHintCacheWarmParams
+      def hint_cache_warm(params = {})
+        parsed, options = Turbopuffer::NamespaceHintCacheWarmParams.dump_request(params)
+        namespace =
+          parsed.delete(:namespace) do
+            @client.default_namespace
+          end
+        @client.request(
+          method: :get,
+          path: ["v1/namespaces/%1$s/hint_cache_warm", namespace],
+          model: Turbopuffer::Models::NamespaceHintCacheWarmResponse,
+          options: options
+        )
+      end
+
+      # Get metadata about a namespace.
+      #
+      # @overload metadata(namespace: nil, request_options: {})
+      #
+      # @param namespace [String] The name of the namespace.
+      #
+      # @param request_options [Turbopuffer::RequestOptions, Hash{Symbol=>Object}, nil]
+      #
+      # @return [Turbopuffer::Models::NamespaceMetadata]
+      #
+      # @see Turbopuffer::Models::NamespaceMetadataParams
+      def metadata(params = {})
+        parsed, options = Turbopuffer::NamespaceMetadataParams.dump_request(params)
+        namespace =
+          parsed.delete(:namespace) do
+            @client.default_namespace
+          end
+        @client.request(
+          method: :get,
+          path: ["v1/namespaces/%1$s/metadata", namespace],
+          model: Turbopuffer::NamespaceMetadata,
+          options: options
+        )
+      end
+
+      # Issue multiple concurrent queries filter or search documents.
+      #
+      # @overload multi_query(queries:, namespace: nil, consistency: nil, vector_encoding: nil, request_options: {})
+      #
+      # @param queries [Array<Turbopuffer::Models::Query>] Body param:
+      #
+      # @param namespace [String] Path param: The name of the namespace.
+      #
+      # @param consistency [Turbopuffer::Models::NamespaceMultiQueryParams::Consistency] Body param: The consistency level for a query.
+      #
+      # @param vector_encoding [Symbol, Turbopuffer::Models::VectorEncoding] Body param: The encoding to use for vectors in the response.
+      #
+      # @param request_options [Turbopuffer::RequestOptions, Hash{Symbol=>Object}, nil]
+      #
+      # @return [Turbopuffer::Models::NamespaceMultiQueryResponse]
+      #
+      # @see Turbopuffer::Models::NamespaceMultiQueryParams
+      def multi_query(params)
+        parsed, options = Turbopuffer::NamespaceMultiQueryParams.dump_request(params)
+        namespace =
+          parsed.delete(:namespace) do
+            @client.default_namespace
+          end
+        @client.request(
+          method: :post,
+          path: ["v2/namespaces/%1$s/query?stainless_overload=multiQuery", namespace],
+          body: parsed,
+          model: Turbopuffer::Models::NamespaceMultiQueryResponse,
+          options: options
+        )
+      end
+
+      # Some parameter documentations has been truncated, see
+      # {Turbopuffer::Models::NamespaceQueryParams} for more details.
+      #
+      # Query, filter, full-text search and vector search documents.
+      #
+      # @overload query(namespace: nil, aggregate_by: nil, consistency: nil, distance_metric: nil, filters: nil, include_attributes: nil, rank_by: nil, top_k: nil, vector_encoding: nil, request_options: {})
+      #
+      # @param namespace [String] Path param: The name of the namespace.
+      #
+      # @param aggregate_by [Hash{Symbol=>Object}] Body param: Aggregations to compute over all documents in the namespace that mat
+      #
+      # @param consistency [Turbopuffer::Models::NamespaceQueryParams::Consistency] Body param: The consistency level for a query.
+      #
+      # @param distance_metric [Symbol, Turbopuffer::Models::DistanceMetric] Body param: A function used to calculate vector similarity.
+      #
+      # @param filters [Object] Body param: Exact filters for attributes to refine search results for. Think of
+      #
+      # @param include_attributes [Boolean, Array<String>] Body param: Whether to include attributes in the response.
+      #
+      # @param rank_by [Object] Body param: How to rank the documents in the namespace.
+      #
+      # @param top_k [Integer] Body param: The number of results to return.
+      #
+      # @param vector_encoding [Symbol, Turbopuffer::Models::VectorEncoding] Body param: The encoding to use for vectors in the response.
+      #
+      # @param request_options [Turbopuffer::RequestOptions, Hash{Symbol=>Object}, nil]
+      #
+      # @return [Turbopuffer::Models::NamespaceQueryResponse]
+      #
+      # @see Turbopuffer::Models::NamespaceQueryParams
+      def query(params = {})
+        parsed, options = Turbopuffer::NamespaceQueryParams.dump_request(params)
+        namespace =
+          parsed.delete(:namespace) do
+            @client.default_namespace
+          end
+        @client.request(
+          method: :post,
+          path: ["v2/namespaces/%1$s/query", namespace],
+          body: parsed,
+          model: Turbopuffer::Models::NamespaceQueryResponse,
+          options: options
+        )
+      end
+
+      # Some parameter documentations has been truncated, see
+      # {Turbopuffer::Models::NamespaceRecallParams} for more details.
+      #
+      # Evaluate recall.
+      #
+      # @overload recall(namespace: nil, filters: nil, num: nil, queries: nil, top_k: nil, request_options: {})
+      #
+      # @param namespace [String] Path param: The name of the namespace.
+      #
+      # @param filters [Object] Body param: Filter by attributes. Same syntax as the query endpoint.
+      #
+      # @param num [Integer] Body param: The number of searches to run.
+      #
+      # @param queries [Array<Float>] Body param: Use specific query vectors for the measurement. If omitted, sampled
+      #
+      # @param top_k [Integer] Body param: Search for `top_k` nearest neighbors.
+      #
+      # @param request_options [Turbopuffer::RequestOptions, Hash{Symbol=>Object}, nil]
+      #
+      # @return [Turbopuffer::Models::NamespaceRecallResponse]
+      #
+      # @see Turbopuffer::Models::NamespaceRecallParams
+      def recall(params = {})
+        parsed, options = Turbopuffer::NamespaceRecallParams.dump_request(params)
+        namespace =
+          parsed.delete(:namespace) do
+            @client.default_namespace
+          end
+        @client.request(
+          method: :post,
+          path: ["v1/namespaces/%1$s/_debug/recall", namespace],
+          body: parsed,
+          model: Turbopuffer::Models::NamespaceRecallResponse,
+          options: options
+        )
+      end
+
+      # Get namespace schema.
+      #
+      # @overload schema(namespace: nil, request_options: {})
+      #
+      # @param namespace [String] The name of the namespace.
+      #
+      # @param request_options [Turbopuffer::RequestOptions, Hash{Symbol=>Object}, nil]
+      #
+      # @return [Hash{Symbol=>Turbopuffer::Models::AttributeSchemaConfig}]
+      #
+      # @see Turbopuffer::Models::NamespaceSchemaParams
+      def schema(params = {})
+        parsed, options = Turbopuffer::NamespaceSchemaParams.dump_request(params)
+        namespace =
+          parsed.delete(:namespace) do
+            @client.default_namespace
+          end
+        @client.request(
+          method: :get,
+          path: ["v1/namespaces/%1$s/schema", namespace],
+          model: Turbopuffer::Internal::Type::HashOf[Turbopuffer::AttributeSchemaConfig],
+          options: options
+        )
+      end
+
+      # Update namespace schema.
+      #
+      # @overload update_schema(namespace: nil, schema: nil, request_options: {})
+      #
+      # @param namespace [String] Path param: The name of the namespace.
+      #
+      # @param schema [Hash{Symbol=>String, Turbopuffer::Models::AttributeSchemaConfig}] Body param: The desired schema for the namespace.
+      #
+      # @param request_options [Turbopuffer::RequestOptions, Hash{Symbol=>Object}, nil]
+      #
+      # @return [Hash{Symbol=>Turbopuffer::Models::AttributeSchemaConfig}]
+      #
+      # @see Turbopuffer::Models::NamespaceUpdateSchemaParams
+      def update_schema(params = {})
+        parsed, options = Turbopuffer::NamespaceUpdateSchemaParams.dump_request(params)
+        namespace =
+          parsed.delete(:namespace) do
+            @client.default_namespace
+          end
+        @client.request(
+          method: :post,
+          path: ["v1/namespaces/%1$s/schema", namespace],
+          body: parsed[:schema],
+          model: Turbopuffer::Internal::Type::HashOf[Turbopuffer::AttributeSchemaConfig],
+          options: options
+        )
+      end
+
+      # Some parameter documentations has been truncated, see
+      # {Turbopuffer::Models::NamespaceWriteParams} for more details.
+      #
+      # Create, update, or delete documents.
+      #
+      # @overload write(namespace: nil, copy_from_namespace: nil, delete_by_filter: nil, delete_condition: nil, deletes: nil, distance_metric: nil, encryption: nil, patch_columns: nil, patch_condition: nil, patch_rows: nil, schema: nil, upsert_columns: nil, upsert_condition: nil, upsert_rows: nil, request_options: {})
+      #
+      # @param namespace [String] Path param: The name of the namespace.
+      #
+      # @param copy_from_namespace [String] Body param: The namespace to copy documents from.
+      #
+      # @param delete_by_filter [Object] Body param: The filter specifying which documents to delete.
+      #
+      # @param delete_condition [Object] Body param: A condition evaluated against the current value of each document tar
+      #
+      # @param deletes [Array<String, Integer>] Body param:
+      #
+      # @param distance_metric [Symbol, Turbopuffer::Models::DistanceMetric] Body param: A function used to calculate vector similarity.
+      #
+      # @param encryption [Turbopuffer::Models::NamespaceWriteParams::Encryption] Body param: The encryption configuration for a namespace.
+      #
+      # @param patch_columns [Turbopuffer::Models::Columns] Body param: A list of documents in columnar format. Each key is a column name, m
+      #
+      # @param patch_condition [Object] Body param: A condition evaluated against the current value of each document tar
+      #
+      # @param patch_rows [Array<Turbopuffer::Models::Row>] Body param:
+      #
+      # @param schema [Hash{Symbol=>String, Turbopuffer::Models::AttributeSchemaConfig}] Body param: The schema of the attributes attached to the documents.
+      #
+      # @param upsert_columns [Turbopuffer::Models::Columns] Body param: A list of documents in columnar format. Each key is a column name, m
+      #
+      # @param upsert_condition [Object] Body param: A condition evaluated against the current value of each document tar
+      #
+      # @param upsert_rows [Array<Turbopuffer::Models::Row>] Body param:
+      #
+      # @param request_options [Turbopuffer::RequestOptions, Hash{Symbol=>Object}, nil]
+      #
+      # @return [Turbopuffer::Models::NamespaceWriteResponse]
+      #
+      # @see Turbopuffer::Models::NamespaceWriteParams
+      def write(params = {})
+        parsed, options = Turbopuffer::NamespaceWriteParams.dump_request(params)
+        namespace =
+          parsed.delete(:namespace) do
+            @client.default_namespace
+          end
+        @client.request(
+          method: :post,
+          path: ["v2/namespaces/%1$s", namespace],
+          body: parsed,
+          model: Turbopuffer::Models::NamespaceWriteResponse,
+          options: options
+        )
+      end
+
+      # @api private
+      #
+      # @param client [Turbopuffer::Client]
+      def initialize(client:)
+        @client = client
+      end
+    end
+  end
+end

data/lib/turbopuffer/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module Turbopuffer
+  VERSION = "0.2.0.pre.alpha.1"
+end

data/lib/turbopuffer.rb

@@ -0,0 +1,94 @@
+# frozen_string_literal: true
+
+# Standard libraries.
+# rubocop:disable Lint/RedundantRequireStatement
+require "English"
+require "cgi"
+require "date"
+require "erb"
+require "etc"
+require "json"
+require "net/http"
+require "pathname"
+require "rbconfig"
+require "securerandom"
+require "set"
+require "stringio"
+require "time"
+require "uri"
+# rubocop:enable Lint/RedundantRequireStatement
+
+# We already ship the preferred sorbet manifests in the package itself.
+# `tapioca` currently does not offer us a way to opt out of unnecessary compilation.
+if Object.const_defined?(:Tapioca) &&
+   caller.chain([$PROGRAM_NAME]).chain(ARGV).any?(/tapioca/) &&
+   ARGV.none?(/dsl/)
+  return
+end
+
+# Gems.
+require "connection_pool"
+
+# Package files.
+require_relative "turbopuffer/version"
+require_relative "turbopuffer/internal/util"
+require_relative "turbopuffer/internal/type/converter"
+require_relative "turbopuffer/internal/type/unknown"
+require_relative "turbopuffer/internal/type/boolean"
+require_relative "turbopuffer/internal/type/file_input"
+require_relative "turbopuffer/internal/type/enum"
+require_relative "turbopuffer/internal/type/union"
+require_relative "turbopuffer/internal/type/array_of"
+require_relative "turbopuffer/internal/type/hash_of"
+require_relative "turbopuffer/internal/type/base_model"
+require_relative "turbopuffer/internal/type/base_page"
+require_relative "turbopuffer/internal/type/request_parameters"
+require_relative "turbopuffer/internal"
+require_relative "turbopuffer/request_options"
+require_relative "turbopuffer/file_part"
+require_relative "turbopuffer/errors"
+require_relative "turbopuffer/internal/transport/base_client"
+require_relative "turbopuffer/internal/transport/pooled_net_requester"
+require_relative "turbopuffer/client"
+require_relative "turbopuffer/internal/namespace_page"
+require_relative "turbopuffer/models/attribute_schema"
+require_relative "turbopuffer/models/attribute_schema_config"
+require_relative "turbopuffer/models/attribute_type"
+require_relative "turbopuffer/models/client_namespaces_params"
+require_relative "turbopuffer/models/columns"
+require_relative "turbopuffer/models/distance_metric"
+require_relative "turbopuffer/models/full_text_search"
+require_relative "turbopuffer/models/full_text_search_config"
+require_relative "turbopuffer/models/id"
+require_relative "turbopuffer/models/include_attributes"
+require_relative "turbopuffer/models/language"
+require_relative "turbopuffer/models/namespace_delete_all_params"
+require_relative "turbopuffer/models/namespace_delete_all_response"
+require_relative "turbopuffer/models/namespace_hint_cache_warm_params"
+require_relative "turbopuffer/models/namespace_hint_cache_warm_response"
+require_relative "turbopuffer/models/namespace_metadata"
+require_relative "turbopuffer/models/namespace_metadata_params"
+require_relative "turbopuffer/models/namespace_multi_query_params"
+require_relative "turbopuffer/models/namespace_multi_query_response"
+require_relative "turbopuffer/models/namespace_query_params"
+require_relative "turbopuffer/models/namespace_query_response"
+require_relative "turbopuffer/models/namespace_recall_params"
+require_relative "turbopuffer/models/namespace_recall_response"
+require_relative "turbopuffer/models/namespace_schema_params"
+require_relative "turbopuffer/models/namespace_schema_response"
+require_relative "turbopuffer/models/namespace_summary"
+require_relative "turbopuffer/models/namespace_update_schema_params"
+require_relative "turbopuffer/models/namespace_update_schema_response"
+require_relative "turbopuffer/models/namespace_write_params"
+require_relative "turbopuffer/models/namespace_write_response"
+require_relative "turbopuffer/models/query"
+require_relative "turbopuffer/models/query_billing"
+require_relative "turbopuffer/models/query_performance"
+require_relative "turbopuffer/models/row"
+require_relative "turbopuffer/models/tokenizer"
+require_relative "turbopuffer/models/vector"
+require_relative "turbopuffer/models/vector_encoding"
+require_relative "turbopuffer/models/write_billing"
+require_relative "turbopuffer/models"
+require_relative "turbopuffer/resources/namespaces"
+require_relative "turbopuffer/namespace"

data/manifest.yaml

@@ -0,0 +1,15 @@
+dependencies:
+  - English
+  - cgi
+  - date
+  - erb
+  - etc
+  - json
+  - net/http
+  - pathname
+  - rbconfig
+  - securerandom
+  - set
+  - stringio
+  - time
+  - uri

data/rbi/turbopuffer/client.rbi

@@ -0,0 +1,91 @@
+# typed: strong
+
+module Turbopuffer
+  class Client < Turbopuffer::Internal::Transport::BaseClient
+    DEFAULT_MAX_RETRIES = 2
+
+    DEFAULT_TIMEOUT_IN_SECONDS = T.let(60.0, Float)
+
+    DEFAULT_INITIAL_RETRY_DELAY = T.let(0.5, Float)
+
+    DEFAULT_MAX_RETRY_DELAY = T.let(8.0, Float)
+
+    # API key used for authentication
+    sig { returns(String) }
+    attr_reader :api_key
+
+    # The turbopuffer region to use.
+    sig { returns(T.nilable(String)) }
+    attr_reader :region
+
+    sig { returns(T.nilable(String)) }
+    attr_reader :default_namespace
+
+    sig do
+      params(
+        namespace: String
+      ).returns(Turbopuffer::Namespace)
+    end
+    def namespace(
+      namespace
+    )
+    end
+
+    # List namespaces.
+    sig do
+      params(
+        cursor: String,
+        page_size: Integer,
+        prefix: String,
+        request_options: Turbopuffer::RequestOptions::OrHash
+      ).returns(
+        Turbopuffer::Internal::NamespacePage[Turbopuffer::NamespaceSummary]
+      )
+    end
+    def namespaces(
+      # Retrieve the next page of results.
+      cursor: nil,
+      # Limit the number of results per page.
+      page_size: nil,
+      # Retrieve only the namespaces that match the prefix.
+      prefix: nil,
+      request_options: {}
+    )
+    end
+
+    # @api private
+    sig { override.returns(T::Hash[String, String]) }
+    private def auth_headers
+    end
+
+    # Creates and returns a new client for interacting with the API.
+    sig do
+      params(
+        api_key: T.nilable(String),
+        region: T.nilable(String),
+        default_namespace: T.nilable(String),
+        base_url: T.nilable(String),
+        max_retries: Integer,
+        timeout: Float,
+        initial_retry_delay: Float,
+        max_retry_delay: Float
+      ).returns(T.attached_class)
+    end
+    def self.new(
+      # API key used for authentication Defaults to `ENV["TURBOPUFFER_API_KEY"]`
+      api_key: ENV["TURBOPUFFER_API_KEY"],
+      # The turbopuffer region to use. Defaults to `ENV["TURBOPUFFER_REGION"]`
+      region: ENV["TURBOPUFFER_REGION"],
+      default_namespace: nil,
+      # Override the default base URL for the API, e.g.,
+      # `"https://api.example.com/v2/"`. Defaults to `ENV["TURBOPUFFER_BASE_URL"]`
+      base_url: ENV["TURBOPUFFER_BASE_URL"],
+      # Max number of retries to attempt after a failed retryable request.
+      max_retries: Turbopuffer::Client::DEFAULT_MAX_RETRIES,
+      timeout: Turbopuffer::Client::DEFAULT_TIMEOUT_IN_SECONDS,
+      initial_retry_delay: Turbopuffer::Client::DEFAULT_INITIAL_RETRY_DELAY,
+      max_retry_delay: Turbopuffer::Client::DEFAULT_MAX_RETRY_DELAY
+    )
+    end
+  end
+end

data/rbi/turbopuffer/errors.rbi

@@ -0,0 +1,178 @@
+# typed: strong
+
+module Turbopuffer
+  module Errors
+    class Error < StandardError
+      sig { returns(T.nilable(StandardError)) }
+      attr_accessor :cause
+    end
+
+    class ConversionError < Turbopuffer::Errors::Error
+      sig { returns(T.nilable(StandardError)) }
+      def cause
+      end
+
+      # @api private
+      sig do
+        params(
+          on: T::Class[StandardError],
+          method: Symbol,
+          target: T.anything,
+          value: T.anything,
+          cause: T.nilable(StandardError)
+        ).returns(T.attached_class)
+      end
+      def self.new(on:, method:, target:, value:, cause: nil)
+      end
+    end
+
+    class APIError < Turbopuffer::Errors::Error
+      sig { returns(URI::Generic) }
+      attr_accessor :url
+
+      sig { returns(T.nilable(Integer)) }
+      attr_accessor :status
+
+      sig { returns(T.nilable(T.anything)) }
+      attr_accessor :body
+
+      # @api private
+      sig do
+        params(
+          url: URI::Generic,
+          status: T.nilable(Integer),
+          body: T.nilable(Object),
+          request: NilClass,
+          response: NilClass,
+          message: T.nilable(String)
+        ).returns(T.attached_class)
+      end
+      def self.new(
+        url:,
+        status: nil,
+        body: nil,
+        request: nil,
+        response: nil,
+        message: nil
+      )
+      end
+    end
+
+    class APIConnectionError < Turbopuffer::Errors::APIError
+      sig { void }
+      attr_accessor :status
+
+      sig { void }
+      attr_accessor :body
+
+      # @api private
+      sig do
+        params(
+          url: URI::Generic,
+          status: NilClass,
+          body: NilClass,
+          request: NilClass,
+          response: NilClass,
+          message: T.nilable(String)
+        ).returns(T.attached_class)
+      end
+      def self.new(
+        url:,
+        status: nil,
+        body: nil,
+        request: nil,
+        response: nil,
+        message: "Connection error."
+      )
+      end
+    end
+
+    class APITimeoutError < Turbopuffer::Errors::APIConnectionError
+      # @api private
+      sig do
+        params(
+          url: URI::Generic,
+          status: NilClass,
+          body: NilClass,
+          request: NilClass,
+          response: NilClass,
+          message: T.nilable(String)
+        ).returns(T.attached_class)
+      end
+      def self.new(
+        url:,
+        status: nil,
+        body: nil,
+        request: nil,
+        response: nil,
+        message: "Request timed out."
+      )
+      end
+    end
+
+    class APIStatusError < Turbopuffer::Errors::APIError
+      # @api private
+      sig do
+        params(
+          url: URI::Generic,
+          status: Integer,
+          body: T.nilable(Object),
+          request: NilClass,
+          response: NilClass,
+          message: T.nilable(String)
+        ).returns(T.attached_class)
+      end
+      def self.for(url:, status:, body:, request:, response:, message: nil)
+      end
+
+      sig { returns(Integer) }
+      attr_accessor :status
+
+      # @api private
+      sig do
+        params(
+          url: URI::Generic,
+          status: Integer,
+          body: T.nilable(Object),
+          request: NilClass,
+          response: NilClass,
+          message: T.nilable(String)
+        ).returns(T.attached_class)
+      end
+      def self.new(url:, status:, body:, request:, response:, message: nil)
+      end
+    end
+
+    class BadRequestError < Turbopuffer::Errors::APIStatusError
+      HTTP_STATUS = 400
+    end
+
+    class AuthenticationError < Turbopuffer::Errors::APIStatusError
+      HTTP_STATUS = 401
+    end
+
+    class PermissionDeniedError < Turbopuffer::Errors::APIStatusError
+      HTTP_STATUS = 403
+    end
+
+    class NotFoundError < Turbopuffer::Errors::APIStatusError
+      HTTP_STATUS = 404
+    end
+
+    class ConflictError < Turbopuffer::Errors::APIStatusError
+      HTTP_STATUS = 409
+    end
+
+    class UnprocessableEntityError < Turbopuffer::Errors::APIStatusError
+      HTTP_STATUS = 422
+    end
+
+    class RateLimitError < Turbopuffer::Errors::APIStatusError
+      HTTP_STATUS = 429
+    end
+
+    class InternalServerError < Turbopuffer::Errors::APIStatusError
+      HTTP_STATUS = T.let((500..), T::Range[Integer])
+    end
+  end
+end