anthropic-sdk-beta 0.1.0.pre.beta.3

data/rbi/lib/anthropic/resources/models.rbi

@@ -0,0 +1,53 @@
+# typed: strong
+
+module Anthropic
+  module Resources
+    class Models
+      # Get a specific model.
+      #
+      # The Models API response can be used to determine information about a specific
+      # model or resolve a model alias to a model ID.
+      sig do
+        params(
+          model_id: String,
+          request_options: T.nilable(T.any(Anthropic::RequestOptions, Anthropic::Internal::AnyHash))
+        )
+          .returns(Anthropic::Models::ModelInfo)
+      end
+      def retrieve(
+        # Model identifier or alias.
+        model_id,
+        request_options: {}
+      ); end
+      # List available models.
+      #
+      # The Models API response can be used to determine which models are available for
+      # use in the API. More recently released models are listed first.
+      sig do
+        params(
+          after_id: String,
+          before_id: String,
+          limit: Integer,
+          request_options: T.nilable(T.any(Anthropic::RequestOptions, Anthropic::Internal::AnyHash))
+        )
+          .returns(Anthropic::Internal::Page[Anthropic::Models::ModelInfo])
+      end
+      def list(
+        # ID of the object to use as a cursor for pagination. When provided, returns the
+        # page of results immediately after this object.
+        after_id: nil,
+        # ID of the object to use as a cursor for pagination. When provided, returns the
+        # page of results immediately before this object.
+        before_id: nil,
+        # Number of items to return per page.
+        #
+        # Defaults to `20`. Ranges from `1` to `1000`.
+        limit: nil,
+        request_options: {}
+      ); end
+      # @api private
+      sig { params(client: Anthropic::Client).returns(T.attached_class) }
+      def self.new(client:); end
+    end
+  end
+end

data/rbi/lib/anthropic/version.rbi

@@ -0,0 +1,5 @@
+# typed: strong
+
+module Anthropic
+  VERSION = T.let(T.unsafe(nil), String)
+end

data/rbi/lib/anthropic/vertex/client.rbi

@@ -0,0 +1,56 @@
+# typed: strong
+
+module Anthropic
+  module Vertex
+    class Client < Anthropic::Client
+      DEFAULT_VERSION = "vertex-2023-10-16"
+
+      sig { returns(String) }
+      attr_reader :region
+
+      sig { returns(String) }
+      attr_reader :project_id
+
+      sig { returns(Anthropic::Resources::Messages) }
+      attr_reader :messages
+
+      sig { returns(Anthropic::Resources::Beta) }
+      attr_reader :beta
+
+      # @api private
+      sig do
+        override
+          .params(req: Anthropic::Internal::Transport::BaseClient::RequestComponentsShape, opts: Anthropic::Internal::AnyHash)
+          .returns(Anthropic::Internal::Transport::BaseClient::RequestInputShape)
+      end
+      private def build_request(req, opts); end
+
+      sig do
+        params(request_components: Anthropic::Internal::Transport::BaseClient::RequestComponentsShape)
+          .returns(Anthropic::Internal::Transport::BaseClient::RequestComponentsShape)
+      end
+      private def fit_req_to_vertex_specs!(request_components) end
+
+      sig do
+        params(
+          region: T.nilable(String),
+          project_id: 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(
+        region: ENV["CLOUD_ML_REGION"],
+        project_id: ENV["ANTHROPIC_VERTEX_PROJECT_ID"],
+        base_url: nil,
+        max_retries: Anthropic::Client::DEFAULT_MAX_RETRIES,
+        timeout: Anthropic::Client::DEFAULT_TIMEOUT_IN_SECONDS,
+        initial_retry_delay: Anthropic::Client::DEFAULT_INITIAL_RETRY_DELAY,
+        max_retry_delay: Anthropic::Client::DEFAULT_MAX_RETRY_DELAY
+      ) end
+    end
+  end
+end

data/sig/anthropic/bedrock/client.rbs

@@ -0,0 +1,41 @@
+module Anthropic
+  module Bedrock
+    class Client < Anthropic::Client
+      DEFAULT_VERSION: String
+
+      attr_reader aws_region: String
+
+      attr_reader completions: Anthropic::Resources::Completions
+
+      attr_reader messages: Anthropic::Resources::Messages
+
+      attr_reader beta: Anthropic::Resources::Beta
+
+      private def build_request: (
+        Anthropic::Internal::Transport::BaseClient::request_components req,
+        Anthropic::request_options opts
+      ) -> Anthropic::Internal::Transport::BaseClient::request_input
+
+      def resolve_region_and_credentials: -> top
+
+      private def signer: -> top
+
+      private def fit_req_to_bedrock_specs!: (
+        Anthropic::Internal::Transport::BaseClient::request_components request_components
+      ) -> Anthropic::Internal::Transport::BaseClient::request_components
+
+      def initialize: (
+        aws_region: String?,
+        base_url: String?,
+        max_retries: Integer,
+        timeout: Float,
+        initial_retry_delay: Float,
+        max_retry_delay: Float,
+        aws_access_key: String?,
+        aws_secret_key: String?,
+        aws_session_token: String?,
+        aws_profile: String?
+      ) -> void
+    end
+  end
+end

data/sig/anthropic/client.rbs

@@ -0,0 +1,39 @@
+module Anthropic
+  class Client < Anthropic::Internal::Transport::BaseClient
+    DEFAULT_MAX_RETRIES: 2
+
+    DEFAULT_TIMEOUT_IN_SECONDS: Float
+
+    DEFAULT_INITIAL_RETRY_DELAY: Float
+
+    DEFAULT_MAX_RETRY_DELAY: Float
+
+    attr_reader api_key: String?
+
+    attr_reader auth_token: String?
+
+    attr_reader completions: Anthropic::Resources::Completions
+
+    attr_reader messages: Anthropic::Resources::Messages
+
+    attr_reader models: Anthropic::Resources::Models
+
+    attr_reader beta: Anthropic::Resources::Beta
+
+    private def auth_headers: -> ::Hash[String, String]
+
+    private def api_key_auth: -> ::Hash[String, String]
+
+    private def bearer_auth: -> ::Hash[String, String]
+
+    def initialize: (
+      ?api_key: String?,
+      ?auth_token: String?,
+      ?base_url: String?,
+      ?max_retries: Integer,
+      ?timeout: Float,
+      ?initial_retry_delay: Float,
+      ?max_retry_delay: Float
+    ) -> void
+  end
+end

data/sig/anthropic/errors.rbs

@@ -0,0 +1,101 @@
+module Anthropic
+  module Errors
+    class Error < StandardError
+      attr_accessor cause: StandardError?
+    end
+
+    class ConversionError < Anthropic::Errors::Error
+    end
+
+    class APIError < Anthropic::Errors::Error
+      attr_accessor url: URI::Generic
+
+      attr_accessor status: Integer?
+
+      attr_accessor body: top?
+
+      def initialize: (
+        url: URI::Generic,
+        ?status: Integer?,
+        ?body: Object?,
+        ?request: nil,
+        ?response: nil,
+        ?message: String?
+      ) -> void
+    end
+
+    class APIConnectionError < Anthropic::Errors::APIError
+      def initialize: (
+        url: URI::Generic,
+        ?status: nil,
+        ?body: nil,
+        ?request: nil,
+        ?response: nil,
+        ?message: String?
+      ) -> void
+    end
+
+    class APITimeoutError < Anthropic::Errors::APIConnectionError
+      def initialize: (
+        url: URI::Generic,
+        ?status: nil,
+        ?body: nil,
+        ?request: nil,
+        ?response: nil,
+        ?message: String?
+      ) -> void
+    end
+
+    class APIStatusError < Anthropic::Errors::APIError
+      def self.for: (
+        url: URI::Generic,
+        status: Integer,
+        body: Object?,
+        request: nil,
+        response: nil,
+        ?message: String?
+      ) -> instance
+
+      def initialize: (
+        url: URI::Generic,
+        status: Integer,
+        body: Object?,
+        request: nil,
+        response: nil,
+        ?message: String?
+      ) -> void
+    end
+
+    class BadRequestError < Anthropic::Errors::APIStatusError
+      HTTP_STATUS: 400
+    end
+
+    class AuthenticationError < Anthropic::Errors::APIStatusError
+      HTTP_STATUS: 401
+    end
+
+    class PermissionDeniedError < Anthropic::Errors::APIStatusError
+      HTTP_STATUS: 403
+    end
+
+    class NotFoundError < Anthropic::Errors::APIStatusError
+      HTTP_STATUS: 404
+    end
+
+    class ConflictError < Anthropic::Errors::APIStatusError
+      HTTP_STATUS: 409
+    end
+
+    class UnprocessableEntityError < Anthropic::Errors::APIStatusError
+      HTTP_STATUS: 422
+    end
+
+    class RateLimitError < Anthropic::Errors::APIStatusError
+      HTTP_STATUS: 429
+    end
+
+    class InternalServerError < Anthropic::Errors::APIStatusError
+      HTTP_STATUS: Range[Integer]
+    end
+  end
+end

data/sig/anthropic/internal/jsonl_stream.rbs

@@ -0,0 +1,9 @@
+module Anthropic
+  module Internal
+    class JsonLStream[Elem]
+      include Anthropic::Internal::Type::BaseStream[top, Elem]
+
+      private def iterator: -> Enumerable[Elem]
+    end
+  end
+end

data/sig/anthropic/internal/page.rbs

@@ -0,0 +1,17 @@
+module Anthropic
+  module Internal
+    class Page[Elem]
+      include Anthropic::Internal::Type::BasePage[Elem]
+
+      attr_accessor data: ::Array[Elem]?
+
+      attr_accessor has_more: bool
+
+      attr_accessor first_id: String?
+
+      attr_accessor last_id: String?
+
+      def inspect: -> String
+    end
+  end
+end

data/sig/anthropic/internal/stream.rbs

@@ -0,0 +1,9 @@
+module Anthropic
+  module Internal
+    class Stream[Elem]
+      include Anthropic::Internal::Type::BaseStream[Anthropic::Internal::Util::server_sent_event, Elem]
+
+      private def iterator: -> Enumerable[Elem]
+    end
+  end
+end

data/sig/anthropic/internal/transport/base_client.rbs

@@ -0,0 +1,110 @@
+module Anthropic
+  module Internal
+    module Transport
+      class BaseClient
+        type request_components =
+          {
+            method: Symbol,
+            path: String | ::Array[String],
+            query: ::Hash[String, (::Array[String] | String)?]?,
+            headers: ::Hash[String, (String
+            | Integer
+            | ::Array[(String | Integer)?])?]?,
+            body: top?,
+            unwrap: Symbol?,
+            page: Class?,
+            stream: Class?,
+            model: Anthropic::Internal::Type::Converter::input?,
+            options: Anthropic::request_opts?
+          }
+
+        type request_input =
+          {
+            method: Symbol,
+            url: URI::Generic,
+            headers: ::Hash[String, String],
+            body: top,
+            max_retries: Integer,
+            timeout: Float
+          }
+
+        MAX_REDIRECTS: 20
+
+        PLATFORM_HEADERS: ::Hash[String, String]
+
+        def self.validate!: (
+          Anthropic::Internal::Transport::BaseClient::request_components req
+        ) -> void
+
+        def self.should_retry?: (
+          Integer status,
+          headers: ::Hash[String, String]
+        ) -> bool
+
+        def self.follow_redirect: (
+          Anthropic::Internal::Transport::BaseClient::request_input request,
+          status: Integer,
+          response_headers: ::Hash[String, String]
+        ) -> Anthropic::Internal::Transport::BaseClient::request_input
+
+        def self.reap_connection!: (
+          Integer | Anthropic::Errors::APIConnectionError status,
+          stream: Enumerable[String]?
+        ) -> void
+
+        # @api private
+        attr_accessor requester: Anthropic::Internal::Transport::PooledNetRequester
+
+        def initialize: (
+          base_url: String,
+          ?timeout: Float,
+          ?max_retries: Integer,
+          ?initial_retry_delay: Float,
+          ?max_retry_delay: Float,
+          ?headers: ::Hash[String, (String
+          | Integer
+          | ::Array[(String | Integer)?])?],
+          ?idempotency_header: String?
+        ) -> void
+
+        private def auth_headers: -> ::Hash[String, String]
+
+        private def generate_idempotency_key: -> String
+
+        private def build_request: (
+          Anthropic::Internal::Transport::BaseClient::request_components req,
+          Anthropic::request_options opts
+        ) -> Anthropic::Internal::Transport::BaseClient::request_input
+
+        private def retry_delay: (
+          ::Hash[String, String] headers,
+          retry_count: Integer
+        ) -> Float
+
+        private def send_request: (
+          Anthropic::Internal::Transport::BaseClient::request_input request,
+          redirect_count: Integer,
+          retry_count: Integer,
+          send_retry_header: bool
+        ) -> [Integer, top, Enumerable[String]]
+
+        def request: (
+          Symbol method,
+          String | ::Array[String] path,
+          ?query: ::Hash[String, (::Array[String] | String)?]?,
+          ?headers: ::Hash[String, (String
+          | Integer
+          | ::Array[(String | Integer)?])?]?,
+          ?body: top?,
+          ?unwrap: Symbol?,
+          ?page: Class?,
+          ?stream: Class?,
+          ?model: Anthropic::Internal::Type::Converter::input?,
+          ?options: Anthropic::request_opts?
+        ) -> top
+
+        def inspect: -> String
+      end
+    end
+  end
+end

data/sig/anthropic/internal/transport/pooled_net_requester.rbs

@@ -0,0 +1,41 @@
+module Anthropic
+  module Internal
+    module Transport
+      class PooledNetRequester
+        type request =
+          {
+            method: Symbol,
+            url: URI::Generic,
+            headers: ::Hash[String, String],
+            body: top,
+            deadline: Float
+          }
+
+        KEEP_ALIVE_TIMEOUT: 30
+
+        def self.connect: (URI::Generic url) -> top
+
+        def self.calibrate_socket_timeout: (top conn, Float deadline) -> void
+
+        def self.build_request: (
+          Anthropic::Internal::Transport::PooledNetRequester::request request
+        ) {
+          (String arg0) -> void
+        } -> [top, (^-> void)]
+
+        private def with_pool: (
+          URI::Generic url,
+          deadline: Float
+        ) {
+          (top arg0) -> void
+        } -> void
+
+        def execute: (
+          Anthropic::Internal::Transport::PooledNetRequester::request request
+        ) -> [Integer, top, Enumerable[String]]
+
+        def initialize: (?size: Integer) -> void
+      end
+    end
+  end
+end

data/sig/anthropic/internal/type/array_of.rbs

@@ -0,0 +1,41 @@
+module Anthropic
+  module Internal
+    module Type
+      class ArrayOf[Elem]
+        include Anthropic::Internal::Type::Converter
+
+        def self.[]: (
+          ::Hash[Symbol, top]
+          | ^-> Anthropic::Internal::Type::Converter::input
+          | Anthropic::Internal::Type::Converter::input type_info,
+          ?::Hash[Symbol, top] spec
+        ) -> instance
+
+        def ===: (top other) -> bool
+
+        def ==: (top other) -> bool
+
+        def coerce: (
+          ::Array[top] | top value,
+          state: Anthropic::Internal::Type::Converter::coerce_state
+        ) -> (::Array[top] | top)
+
+        def dump: (
+          ::Array[top] | top value,
+          state: Anthropic::Internal::Type::Converter::dump_state
+        ) -> (::Array[top] | top)
+
+        def item_type: -> Elem
+
+        def nilable?: -> bool
+
+        def initialize: (
+          ::Hash[Symbol, top]
+          | ^-> Anthropic::Internal::Type::Converter::input
+          | Anthropic::Internal::Type::Converter::input type_info,
+          ?::Hash[Symbol, top] spec
+        ) -> void
+      end
+    end
+  end
+end

data/sig/anthropic/internal/type/base_model.rbs

@@ -0,0 +1,82 @@
+module Anthropic
+  module Internal
+    module Type
+      class BaseModel
+        extend Anthropic::Internal::Type::Converter
+
+        type known_field =
+          { mode: (:coerce | :dump)?, required: bool, nilable: bool }
+
+        def self.known_fields: -> ::Hash[Symbol, (Anthropic::Internal::Type::BaseModel::known_field
+        & { type_fn: (^-> Anthropic::Internal::Type::Converter::input) })]
+
+        def self.fields: -> ::Hash[Symbol, (Anthropic::Internal::Type::BaseModel::known_field
+        & { type: Anthropic::Internal::Type::Converter::input })]
+
+        private def self.add_field: (
+          Symbol name_sym,
+          required: bool,
+          type_info: {
+            const: (nil | bool | Integer | Float | Symbol)?,
+            enum: ^-> Anthropic::Internal::Type::Converter::input?,
+            union: ^-> Anthropic::Internal::Type::Converter::input?,
+            api_name: Symbol
+          }
+          | ^-> Anthropic::Internal::Type::Converter::input
+          | Anthropic::Internal::Type::Converter::input,
+          spec: ::Hash[Symbol, top]
+        ) -> void
+
+        def self.required: (
+          Symbol name_sym,
+          ::Hash[Symbol, top]
+          | ^-> Anthropic::Internal::Type::Converter::input
+          | Anthropic::Internal::Type::Converter::input type_info,
+          ?::Hash[Symbol, top] spec
+        ) -> void
+
+        def self.optional: (
+          Symbol name_sym,
+          ::Hash[Symbol, top]
+          | ^-> Anthropic::Internal::Type::Converter::input
+          | Anthropic::Internal::Type::Converter::input type_info,
+          ?::Hash[Symbol, top] spec
+        ) -> void
+
+        private def self.request_only: { -> void } -> void
+
+        private def self.response_only: { -> void } -> void
+
+        def self.==: (top other) -> bool
+
+        def ==: (top other) -> bool
+
+        def self.coerce: (
+          Anthropic::Internal::Type::BaseModel | ::Hash[top, top] | top value,
+          state: Anthropic::Internal::Type::Converter::coerce_state
+        ) -> (instance | top)
+
+        def self.dump: (
+          instance | top value,
+          state: Anthropic::Internal::Type::Converter::dump_state
+        ) -> (::Hash[top, top] | top)
+
+        def []: (Symbol key) -> top?
+
+        def to_h: -> ::Hash[Symbol, top]
+
+        alias to_hash to_h
+
+        def deconstruct_keys: (::Array[Symbol]? keys) -> ::Hash[Symbol, top]
+
+        def to_json: (*top a) -> String
+
+        def to_yaml: (*top a) -> String
+
+        def initialize: (?::Hash[Symbol, top] | self data) -> void
+
+        def inspect: -> String
+      end
+    end
+  end
+end

data/sig/anthropic/internal/type/base_page.rbs

@@ -0,0 +1,24 @@
+module Anthropic
+  module Internal
+    module Type
+      module BasePage[Elem]
+        def next_page?: -> bool
+
+        def next_page: -> self
+
+        def auto_paging_each: { (Elem arg0) -> void } -> void
+
+        def to_enum: -> Enumerable[Elem]
+
+        alias enum_for to_enum
+
+        def initialize: (
+          client: Anthropic::Internal::Transport::BaseClient,
+          req: Anthropic::Internal::Transport::BaseClient::request_components,
+          headers: ::Hash[String, String],
+          page_data: top
+        ) -> void
+      end
+    end
+  end
+end

data/sig/anthropic/internal/type/base_stream.rbs

@@ -0,0 +1,31 @@
+module Anthropic
+  module Internal
+    module Type
+      module BaseStream[Message, Elem]
+        include Enumerable[Elem]
+
+        def self.defer_closing: (
+          Enumerable[top] stream
+        ) -> (^(Integer arg0) -> void)
+
+        def close: -> void
+
+        private def iterator: -> Enumerable[Elem]
+
+        def each: { (Elem arg0) -> void } -> void
+
+        def to_enum: -> Enumerator[Elem]
+
+        alias enum_for to_enum
+
+        def initialize: (
+          model: Class | Anthropic::Internal::Type::Converter,
+          url: URI::Generic,
+          status: Integer,
+          response: top,
+          stream: Enumerable[Message]
+        ) -> void
+      end
+    end
+  end
+end

data/sig/anthropic/internal/type/boolean.rbs

@@ -0,0 +1,23 @@
+module Anthropic
+  module Internal
+    module Type
+      class Boolean
+        extend Anthropic::Internal::Type::Converter
+
+        def self.===: (top other) -> bool
+
+        def self.==: (top other) -> bool
+
+        def self.coerce: (
+          bool | top value,
+          state: Anthropic::Internal::Type::Converter::coerce_state
+        ) -> (bool | top)
+
+        def self.dump: (
+          bool | top value,
+          state: Anthropic::Internal::Type::Converter::dump_state
+        ) -> (bool | top)
+      end
+    end
+  end
+end