anthropic 0.4.0 → 1.0.0

data/rbi/anthropic/helpers/bedrock/client.rbi

@@ -0,0 +1,82 @@
+# typed: strong
+
+module Anthropic
+  module Helpers
+    module Bedrock
+      class Client < Anthropic::Client
+        DEFAULT_VERSION = "bedrock-2023-05-31"
+
+        sig { returns(String) }
+        attr_reader :aws_region
+
+        sig { returns(Anthropic::Resources::Completions) }
+        attr_reader :completions
+
+        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::RequestComponents, opts: Anthropic::Internal::AnyHash)
+            .returns(Anthropic::Internal::Transport::BaseClient::RequestInput)
+        end
+        private def build_request(req, opts); end
+
+        sig do
+          params(
+            aws_region: T.nilable(String),
+            aws_access_key: T.nilable(String),
+            aws_secret_key: T.nilable(String),
+            aws_session_token: T.nilable(String),
+            aws_profile: T.nilable(String)
+          ).returns(T::Array[T.anything])
+        end
+        private def resolve_region_and_credentials(
+          aws_region:,
+          aws_access_key:,
+          aws_secret_key:,
+          aws_session_token:,
+          aws_profile:
+        )
+        end
+
+        sig do
+          params(request_components: Anthropic::Internal::Transport::BaseClient::RequestComponents)
+            .returns(Anthropic::Internal::Transport::BaseClient::RequestComponents)
+        end
+        private def fit_req_to_bedrock_specs!(request_components) end
+
+        sig do
+          params(
+            aws_region: T.nilable(String),
+            base_url: T.nilable(String),
+            max_retries: Integer,
+            timeout: Float,
+            initial_retry_delay: Float,
+            max_retry_delay: Float,
+            aws_access_key: T.nilable(String),
+            aws_secret_key: T.nilable(String),
+            aws_session_token: T.nilable(String),
+            aws_profile: T.nilable(String)
+          ).returns(T.attached_class)
+        end
+        def self.new(
+          aws_region: nil,
+          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,
+          aws_access_key: nil,
+          aws_secret_key: nil,
+          aws_session_token: nil,
+          aws_profile: nil
+        ) end
+      end
+    end
+  end
+end

data/rbi/anthropic/helpers/vertex/client.rbi

@@ -0,0 +1,58 @@
+# typed: strong
+
+module Anthropic
+  module Helpers
+    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::RequestComponents, opts: Anthropic::Internal::AnyHash)
+            .returns(Anthropic::Internal::Transport::BaseClient::RequestInput)
+        end
+        private def build_request(req, opts); end
+
+        sig do
+          params(request_components: Anthropic::Internal::Transport::BaseClient::RequestComponents)
+            .returns(Anthropic::Internal::Transport::BaseClient::RequestComponents)
+        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
+end

data/rbi/anthropic/internal/jsonl_stream.rbi

@@ -0,0 +1,17 @@
+# typed: strong
+
+module Anthropic
+  module Internal
+    class JsonLStream
+      Message = type_member(:in) { { fixed: T.anything } }
+      Elem = type_member(:out)
+
+      include Anthropic::Internal::Type::BaseStream
+
+      # @api private
+      sig { override.returns(T::Enumerable[Elem]) }
+      private def iterator
+      end
+    end
+  end
+end

data/rbi/anthropic/internal/page.rbi

@@ -0,0 +1,28 @@
+# typed: strong
+
+module Anthropic
+  module Internal
+    class Page
+      include Anthropic::Internal::Type::BasePage
+
+      Elem = type_member
+
+      sig { returns(T.nilable(T::Array[Elem])) }
+      attr_accessor :data
+
+      sig { returns(T::Boolean) }
+      attr_accessor :has_more
+
+      sig { returns(T.nilable(String)) }
+      attr_accessor :first_id
+
+      sig { returns(T.nilable(String)) }
+      attr_accessor :last_id
+
+      # @api private
+      sig { returns(String) }
+      def inspect
+      end
+    end
+  end
+end

data/rbi/anthropic/internal/stream.rbi

@@ -0,0 +1,20 @@
+# typed: strong
+
+module Anthropic
+  module Internal
+    class Stream
+      Message =
+        type_member(:in) do
+          { fixed: Anthropic::Internal::Util::ServerSentEvent }
+        end
+      Elem = type_member(:out)
+
+      include Anthropic::Internal::Type::BaseStream
+
+      # @api private
+      sig { override.returns(T::Enumerable[Elem]) }
+      private def iterator
+      end
+    end
+  end
+end

data/rbi/anthropic/internal/transport/base_client.rbi

@@ -0,0 +1,310 @@
+# typed: strong
+
+module Anthropic
+  module Internal
+    module Transport
+      # @api private
+      class BaseClient
+        extend Anthropic::Internal::Util::SorbetRuntimeSupport
+
+        abstract!
+
+        RequestComponents =
+          T.type_alias do
+            {
+              method: Symbol,
+              path: T.any(String, T::Array[String]),
+              query:
+                T.nilable(
+                  T::Hash[String, T.nilable(T.any(T::Array[String], String))]
+                ),
+              headers:
+                T.nilable(
+                  T::Hash[
+                    String,
+                    T.nilable(
+                      T.any(
+                        String,
+                        Integer,
+                        T::Array[T.nilable(T.any(String, Integer))]
+                      )
+                    )
+                  ]
+                ),
+              body: T.nilable(T.anything),
+              unwrap:
+                T.nilable(
+                  T.any(
+                    Symbol,
+                    Integer,
+                    T::Array[T.any(Symbol, Integer)],
+                    T.proc.params(arg0: T.anything).returns(T.anything)
+                  )
+                ),
+              page:
+                T.nilable(
+                  T::Class[
+                    Anthropic::Internal::Type::BasePage[
+                      Anthropic::Internal::Type::BaseModel
+                    ]
+                  ]
+                ),
+              stream:
+                T.nilable(
+                  T::Class[
+                    Anthropic::Internal::Type::BaseStream[
+                      T.anything,
+                      Anthropic::Internal::Type::BaseModel
+                    ]
+                  ]
+                ),
+              model: T.nilable(Anthropic::Internal::Type::Converter::Input),
+              options: T.nilable(Anthropic::RequestOptions::OrHash)
+            }
+          end
+
+        RequestInput =
+          T.type_alias do
+            {
+              method: Symbol,
+              url: URI::Generic,
+              headers: T::Hash[String, String],
+              body: T.anything,
+              max_retries: Integer,
+              timeout: Float
+            }
+          end
+
+        # from whatwg fetch spec
+        MAX_REDIRECTS = 20
+
+        PLATFORM_HEADERS = T::Hash[String, String]
+
+        class << self
+          # @api private
+          sig do
+            params(
+              req: Anthropic::Internal::Transport::BaseClient::RequestComponents
+            ).void
+          end
+          def validate!(req)
+          end
+
+          # @api private
+          sig do
+            params(
+              status: Integer,
+              headers: T.any(T::Hash[String, String], Net::HTTPHeader)
+            ).returns(T::Boolean)
+          end
+          def should_retry?(status, headers:)
+          end
+
+          # @api private
+          sig do
+            params(
+              request: Anthropic::Internal::Transport::BaseClient::RequestInput,
+              status: Integer,
+              response_headers: T.any(T::Hash[String, String], Net::HTTPHeader)
+            ).returns(Anthropic::Internal::Transport::BaseClient::RequestInput)
+          end
+          def follow_redirect(request, status:, response_headers:)
+          end
+
+          # @api private
+          sig do
+            params(
+              status: T.any(Integer, Anthropic::Errors::APIConnectionError),
+              stream: T.nilable(T::Enumerable[String])
+            ).void
+          end
+          def reap_connection!(status, stream:)
+          end
+        end
+
+        sig { returns(URI::Generic) }
+        attr_reader :base_url
+
+        sig { returns(Float) }
+        attr_reader :timeout
+
+        sig { returns(Integer) }
+        attr_reader :max_retries
+
+        sig { returns(Float) }
+        attr_reader :initial_retry_delay
+
+        sig { returns(Float) }
+        attr_reader :max_retry_delay
+
+        sig { returns(T::Hash[String, String]) }
+        attr_reader :headers
+
+        sig { returns(T.nilable(String)) }
+        attr_reader :idempotency_header
+
+        # @api private
+        sig { returns(Anthropic::Internal::Transport::PooledNetRequester) }
+        attr_reader :requester
+
+        # @api private
+        sig do
+          params(
+            base_url: String,
+            timeout: Float,
+            max_retries: Integer,
+            initial_retry_delay: Float,
+            max_retry_delay: Float,
+            headers:
+              T::Hash[
+                String,
+                T.nilable(
+                  T.any(
+                    String,
+                    Integer,
+                    T::Array[T.nilable(T.any(String, Integer))]
+                  )
+                )
+              ],
+            idempotency_header: T.nilable(String)
+          ).returns(T.attached_class)
+        end
+        def self.new(
+          base_url:,
+          timeout: 0.0,
+          max_retries: 0,
+          initial_retry_delay: 0.0,
+          max_retry_delay: 0.0,
+          headers: {},
+          idempotency_header: nil
+        )
+        end
+
+        # @api private
+        sig { overridable.returns(T::Hash[String, String]) }
+        private def auth_headers
+        end
+
+        # @api private
+        sig { returns(String) }
+        private def generate_idempotency_key
+        end
+
+        # @api private
+        sig do
+          overridable
+            .params(
+              req:
+                Anthropic::Internal::Transport::BaseClient::RequestComponents,
+              opts: Anthropic::Internal::AnyHash
+            )
+            .returns(Anthropic::Internal::Transport::BaseClient::RequestInput)
+        end
+        private def build_request(req, opts)
+        end
+
+        # @api private
+        sig do
+          params(
+            headers: T::Hash[String, String],
+            retry_count: Integer
+          ).returns(Float)
+        end
+        private def retry_delay(headers, retry_count:)
+        end
+
+        # @api private
+        sig do
+          params(
+            request: Anthropic::Internal::Transport::BaseClient::RequestInput,
+            redirect_count: Integer,
+            retry_count: Integer,
+            send_retry_header: T::Boolean
+          ).returns([Integer, Net::HTTPResponse, T::Enumerable[String]])
+        end
+        private def send_request(
+          request,
+          redirect_count:,
+          retry_count:,
+          send_retry_header:
+        )
+        end
+
+        # Execute the request specified by `req`. This is the method that all resource
+        # methods call into.
+        #
+        # @overload request(method, path, query: {}, headers: {}, body: nil, unwrap: nil, page: nil, stream: nil, model: Anthropic::Internal::Type::Unknown, options: {})
+        sig do
+          params(
+            method: Symbol,
+            path: T.any(String, T::Array[String]),
+            query:
+              T.nilable(
+                T::Hash[String, T.nilable(T.any(T::Array[String], String))]
+              ),
+            headers:
+              T.nilable(
+                T::Hash[
+                  String,
+                  T.nilable(
+                    T.any(
+                      String,
+                      Integer,
+                      T::Array[T.nilable(T.any(String, Integer))]
+                    )
+                  )
+                ]
+              ),
+            body: T.nilable(T.anything),
+            unwrap:
+              T.nilable(
+                T.any(
+                  Symbol,
+                  Integer,
+                  T::Array[T.any(Symbol, Integer)],
+                  T.proc.params(arg0: T.anything).returns(T.anything)
+                )
+              ),
+            page:
+              T.nilable(
+                T::Class[
+                  Anthropic::Internal::Type::BasePage[
+                    Anthropic::Internal::Type::BaseModel
+                  ]
+                ]
+              ),
+            stream:
+              T.nilable(
+                T::Class[
+                  Anthropic::Internal::Type::BaseStream[
+                    T.anything,
+                    Anthropic::Internal::Type::BaseModel
+                  ]
+                ]
+              ),
+            model: T.nilable(Anthropic::Internal::Type::Converter::Input),
+            options: T.nilable(Anthropic::RequestOptions::OrHash)
+          ).returns(T.anything)
+        end
+        def request(
+          method,
+          path,
+          query: {},
+          headers: {},
+          body: nil,
+          unwrap: nil,
+          page: nil,
+          stream: nil,
+          model: Anthropic::Internal::Type::Unknown,
+          options: {}
+        )
+        end
+
+        # @api private
+        sig { returns(String) }
+        def inspect
+        end
+      end
+    end
+  end
+end

data/rbi/anthropic/internal/transport/pooled_net_requester.rbi

@@ -0,0 +1,79 @@
+# typed: strong
+
+module Anthropic
+  module Internal
+    module Transport
+      # @api private
+      class PooledNetRequester
+        extend Anthropic::Internal::Util::SorbetRuntimeSupport
+
+        Request =
+          T.type_alias do
+            {
+              method: Symbol,
+              url: URI::Generic,
+              headers: T::Hash[String, String],
+              body: T.anything,
+              deadline: Float
+            }
+          end
+
+        # from the golang stdlib
+        # https://github.com/golang/go/blob/c8eced8580028328fde7c03cbfcb720ce15b2358/src/net/http/transport.go#L49
+        KEEP_ALIVE_TIMEOUT = 30
+
+        DEFAULT_MAX_CONNECTIONS = T.let(T.unsafe(nil), Integer)
+
+        class << self
+          # @api private
+          sig { params(url: URI::Generic).returns(Net::HTTP) }
+          def connect(url)
+          end
+
+          # @api private
+          sig { params(conn: Net::HTTP, deadline: Float).void }
+          def calibrate_socket_timeout(conn, deadline)
+          end
+
+          # @api private
+          sig do
+            params(
+              request:
+                Anthropic::Internal::Transport::PooledNetRequester::Request,
+              blk: T.proc.params(arg0: String).void
+            ).returns([Net::HTTPGenericRequest, T.proc.void])
+          end
+          def build_request(request, &blk)
+          end
+        end
+
+        # @api private
+        sig do
+          params(
+            url: URI::Generic,
+            deadline: Float,
+            blk: T.proc.params(arg0: Net::HTTP).void
+          ).void
+        end
+        private def with_pool(url, deadline:, &blk)
+        end
+
+        # @api private
+        sig do
+          params(
+            request: Anthropic::Internal::Transport::PooledNetRequester::Request
+          ).returns([Integer, Net::HTTPResponse, T::Enumerable[String]])
+        end
+        def execute(request)
+        end
+
+        # @api private
+        sig { params(size: Integer).returns(T.attached_class) }
+        def self.new(
+          size: Anthropic::Internal::Transport::PooledNetRequester::DEFAULT_MAX_CONNECTIONS
+        )
+        end
+      end
+    end
+  end
+end

data/rbi/anthropic/internal/type/array_of.rbi

@@ -0,0 +1,104 @@
+# typed: strong
+
+module Anthropic
+  module Internal
+    module Type
+      # @api private
+      #
+      # Array of items of a given type.
+      class ArrayOf
+        include Anthropic::Internal::Type::Converter
+        include Anthropic::Internal::Util::SorbetRuntimeSupport
+
+        abstract!
+
+        Elem = type_member(:out)
+
+        sig do
+          params(
+            type_info:
+              T.any(
+                Anthropic::Internal::AnyHash,
+                T.proc.returns(Anthropic::Internal::Type::Converter::Input),
+                Anthropic::Internal::Type::Converter::Input
+              ),
+            spec: Anthropic::Internal::AnyHash
+          ).returns(T.attached_class)
+        end
+        def self.[](type_info, spec = {})
+        end
+
+        sig { params(other: T.anything).returns(T::Boolean) }
+        def ===(other)
+        end
+
+        sig { params(other: T.anything).returns(T::Boolean) }
+        def ==(other)
+        end
+
+        sig { returns(Integer) }
+        def hash
+        end
+
+        # @api private
+        sig do
+          override
+            .params(
+              value: T.any(T::Array[T.anything], T.anything),
+              state: Anthropic::Internal::Type::Converter::CoerceState
+            )
+            .returns(T.any(T::Array[T.anything], T.anything))
+        end
+        def coerce(value, state:)
+        end
+
+        # @api private
+        sig do
+          override
+            .params(
+              value: T.any(T::Array[T.anything], T.anything),
+              state: Anthropic::Internal::Type::Converter::DumpState
+            )
+            .returns(T.any(T::Array[T.anything], T.anything))
+        end
+        def dump(value, state:)
+        end
+
+        # @api private
+        sig { returns(T.anything) }
+        def to_sorbet_type
+        end
+
+        # @api private
+        sig { returns(Elem) }
+        protected def item_type
+        end
+
+        # @api private
+        sig { returns(T::Boolean) }
+        protected def nilable?
+        end
+
+        # @api private
+        sig do
+          params(
+            type_info:
+              T.any(
+                Anthropic::Internal::AnyHash,
+                T.proc.returns(Anthropic::Internal::Type::Converter::Input),
+                Anthropic::Internal::Type::Converter::Input
+              ),
+            spec: Anthropic::Internal::AnyHash
+          ).void
+        end
+        def initialize(type_info, spec = {})
+        end
+
+        # @api private
+        sig { params(depth: Integer).returns(String) }
+        def inspect(depth: 0)
+        end
+      end
+    end
+  end
+end