anthropic 0.0.0 → 0.1.0.pre.beta.9

data/lib/anthropic/errors.rb

@@ -0,0 +1,192 @@
+# frozen_string_literal: true
+
+module Anthropic
+  module Errors
+    class Error < StandardError
+      # @!attribute cause
+      #
+      #   @return [StandardError, nil]
+    end
+
+    class ConversionError < Anthropic::Errors::Error
+    end
+
+    class APIError < Anthropic::Errors::Error
+      # @return [URI::Generic]
+      attr_accessor :url
+
+      # @return [Integer, nil]
+      attr_accessor :status
+
+      # @return [Object, nil]
+      attr_accessor :body
+
+      # @api private
+      #
+      # @param url [URI::Generic]
+      # @param status [Integer, nil]
+      # @param body [Object, nil]
+      # @param request [nil]
+      # @param response [nil]
+      # @param message [String, nil]
+      def initialize(url:, status: nil, body: nil, request: nil, response: nil, message: nil)
+        @url = url
+        @status = status
+        @body = body
+        @request = request
+        @response = response
+        super(message)
+      end
+    end
+
+    class APIConnectionError < Anthropic::Errors::APIError
+      # @!attribute status
+      #
+      #   @return [nil]
+
+      # @!attribute body
+      #
+      #   @return [nil]
+
+      # @api private
+      #
+      # @param url [URI::Generic]
+      # @param status [nil]
+      # @param body [nil]
+      # @param request [nil]
+      # @param response [nil]
+      # @param message [String, nil]
+      def initialize(
+        url:,
+        status: nil,
+        body: nil,
+        request: nil,
+        response: nil,
+        message: "Connection error."
+      )
+        super
+      end
+    end
+
+    class APITimeoutError < Anthropic::Errors::APIConnectionError
+      # @api private
+      #
+      # @param url [URI::Generic]
+      # @param status [nil]
+      # @param body [nil]
+      # @param request [nil]
+      # @param response [nil]
+      # @param message [String, nil]
+      def initialize(
+        url:,
+        status: nil,
+        body: nil,
+        request: nil,
+        response: nil,
+        message: "Request timed out."
+      )
+        super
+      end
+    end
+
+    class APIStatusError < Anthropic::Errors::APIError
+      # @api private
+      #
+      # @param url [URI::Generic]
+      # @param status [Integer]
+      # @param body [Object, nil]
+      # @param request [nil]
+      # @param response [nil]
+      # @param message [String, nil]
+      #
+      # @return [self]
+      def self.for(url:, status:, body:, request:, response:, message: nil)
+        kwargs = {
+          url: url,
+          status: status,
+          body: body,
+          request: request,
+          response: response,
+          message: message
+        }
+
+        case status
+        in 400
+          Anthropic::Errors::BadRequestError.new(**kwargs)
+        in 401
+          Anthropic::Errors::AuthenticationError.new(**kwargs)
+        in 403
+          Anthropic::Errors::PermissionDeniedError.new(**kwargs)
+        in 404
+          Anthropic::Errors::NotFoundError.new(**kwargs)
+        in 409
+          Anthropic::Errors::ConflictError.new(**kwargs)
+        in 422
+          Anthropic::Errors::UnprocessableEntityError.new(**kwargs)
+        in 429
+          Anthropic::Errors::RateLimitError.new(**kwargs)
+        in (500..)
+          Anthropic::Errors::InternalServerError.new(**kwargs)
+        else
+          Anthropic::Errors::APIStatusError.new(**kwargs)
+        end
+      end
+
+      # @!parse
+      #   # @return [Integer]
+      #   attr_accessor :status
+
+      # @api private
+      #
+      # @param url [URI::Generic]
+      # @param status [Integer]
+      # @param body [Object, nil]
+      # @param request [nil]
+      # @param response [nil]
+      # @param message [String, nil]
+      def initialize(url:, status:, body:, request:, response:, message: nil)
+        message ||= {url: url.to_s, status: status, body: body}
+        super(
+          url: url,
+          status: status,
+          body: body,
+          request: request,
+          response: response,
+          message: message&.to_s
+        )
+      end
+    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 = (500..)
+    end
+  end
+end

data/lib/anthropic/file_part.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+module Anthropic
+  class FilePart
+    # @return [Pathname, StringIO, IO, String]
+    attr_reader :content
+
+    # @return [String, nil]
+    attr_reader :content_type
+
+    # @return [String, nil]
+    attr_reader :filename
+
+    # @api private
+    #
+    # @return [String]
+    private def read
+      case content
+      in Pathname
+        content.read(binmode: true)
+      in StringIO
+        content.string
+      in IO
+        content.read
+      in String
+        content
+      end
+    end
+
+    # @param a [Object]
+    #
+    # @return [String]
+    def to_json(*a) = read.to_json(*a)
+
+    # @param a [Object]
+    #
+    # @return [String]
+    def to_yaml(*a) = read.to_yaml(*a)
+
+    # @param content [Pathname, StringIO, IO, String]
+    # @param filename [String, nil]
+    # @param content_type [String, nil]
+    def initialize(content, filename: nil, content_type: nil)
+      @content = content
+      @filename =
+        case content
+        in Pathname
+          filename.nil? ? content.basename.to_path : ::File.basename(filename)
+        else
+          filename.nil? ? nil : ::File.basename(filename)
+        end
+      @content_type = content_type
+    end
+  end
+end

data/lib/anthropic/helpers/bedrock/client.rb

@@ -0,0 +1,259 @@
+# frozen_string_literal: true
+
+module Anthropic
+  module Helpers
+    module Bedrock
+      class Client < Anthropic::Client
+        DEFAULT_VERSION = "bedrock-2023-05-31"
+
+        # @return [Anthropic::Resources::Messages]
+        attr_reader :messages
+
+        # @return [Anthropic::Resources::Completions]
+        attr_reader :completions
+
+        # @return [Anthropic::Resources::Beta]
+        attr_reader :beta
+
+        # @return [String]
+        attr_reader :aws_region
+
+        # @return [Aws::Credentials]
+        attr_reader :aws_credentials
+
+        # Creates and returns a new client for interacting with the AWS Bedrock API for Anthropic models.
+        #
+        # AWS credentials are resolved according to the AWS SDK's default resolution order, described at
+        #   https://docs.aws.amazon.com/sdk-for-ruby/v3/developer-guide/setup-config.html#credchain or https://github.com/aws/aws-sdk-ruby?tab=readme-ov-file#configuration
+        #
+        # @param aws_region [String, nil] Enforce the AWS Region to use. If unset, the region is set according to the
+        #   AWS SDK's default resolution order, described at https://github.com/aws/aws-sdk-ruby?tab=readme-ov-file#configuration
+        #
+        # @param aws_access_key [String, nil]  Optional AWS access key to use for authentication. Overrides profile and credential provider chain
+        #
+        # @param aws_secret_key [String, nil] Optional AWS secret access key to use for authentication. Overrides profile and credential provider chain
+        #
+        # @param aws_session_token [String, nil] Optional AWS session token to use for authentication. Overrides profile and credential provider chain
+        #
+        # @param aws_profile [String, nil] Optional AWS profile to use for authentication. Overrides the credential provider chain
+        #
+        # @param base_url [String, nil] Override the default base URL for the API, e.g., `"https://api.example.com/v2/"`
+        #
+        # @param max_retries [Integer] The maximum number of times to retry a request if it fails
+        #
+        # @param timeout [Float] The number of seconds to wait for a response before timing out
+        #
+        # @param initial_retry_delay [Float] The number of seconds to wait before retrying a request
+        #
+        # @param max_retry_delay [Float] The maximum number of seconds to wait before retrying a request
+        #
+        def initialize(
+          aws_region: nil,
+          base_url: nil,
+          max_retries: DEFAULT_MAX_RETRIES,
+          timeout: DEFAULT_TIMEOUT_IN_SECONDS,
+          initial_retry_delay: DEFAULT_INITIAL_RETRY_DELAY,
+          max_retry_delay: DEFAULT_MAX_RETRY_DELAY,
+          aws_access_key: nil,
+          aws_secret_key: nil,
+          aws_session_token: nil,
+          aws_profile: nil
+        )
+          begin
+            require("aws-sdk-bedrockruntime")
+          rescue LoadError
+            raise <<~MSG
+
+              In order to access Anthropic models on Bedrock you must require the `aws-sdk-bedrockruntime` gem.
+              You can install it by adding the following to your Gemfile:
+
+                  gem "aws-sdk-bedrockruntime"
+
+              and then running `bundle install`.
+
+              Alternatively, if you are not using Bundler, simply run:
+
+                  gem install aws-sdk-bedrockruntime
+            MSG
+          end
+
+          @aws_region, @aws_credentials = resolve_region_and_credentials(
+            aws_region: aws_region,
+            aws_secret_key: aws_secret_key,
+            aws_access_key: aws_access_key,
+            aws_session_token: aws_session_token,
+            aws_profile: aws_profile
+          )
+
+          @signer = Aws::Sigv4::Signer.new(
+            service: "bedrock",
+            region: @aws_region,
+            credentials: @aws_credentials
+          )
+
+          base_url ||= ENV.fetch(
+            "ANTHROPIC_BEDROCK_BASE_URL",
+            "https://bedrock-runtime.#{@aws_region}.amazonaws.com"
+          )
+
+          super(
+            base_url: base_url,
+            timeout: timeout,
+            max_retries: max_retries,
+            initial_retry_delay: initial_retry_delay,
+            max_retry_delay: max_retry_delay,
+          )
+
+          @messages = Anthropic::Resources::Messages.new(client: self)
+          @completions = Anthropic::Resources::Completions.new(client: self)
+          @beta = Anthropic::Resources::Beta.new(client: self)
+        end
+
+        # @api private
+        #
+        # @param req [Hash{Symbol=>Object}] .
+        #
+        #   @option req [Symbol] :method
+        #
+        #   @option req [String, Array<String>] :path
+        #
+        #   @option req [Hash{String=>Array<String>, String, nil}, nil] :query
+        #
+        #   @option req [Hash{String=>String, Integer, Array<String, Integer, nil>, nil}, nil] :headers
+        #
+        #   @option req [Object, nil] :body
+        #
+        #   @option req [Symbol, Integer, Array<Symbol, Integer>, Proc, nil] :unwrap
+        #
+        #   @option req [Class<Anthropic::Internal::Type::BasePage>, nil] :page
+        #
+        #   @option req [Class<Anthropic::Internal::Type::BaseStream>, nil] :stream
+        #
+        #   @option req [Anthropic::Internal::Type::Converter, Class, nil] :model
+        #
+        # @param opts [Hash{Symbol=>Object}] .
+        #
+        #   @option opts [String, nil] :idempotency_key
+        #
+        #   @option opts [Hash{String=>Array<String>, String, nil}, nil] :extra_query
+        #
+        #   @option opts [Hash{String=>String, nil}, nil] :extra_headers
+        #
+        #   @option opts [Object, nil] :extra_body
+        #
+        #   @option opts [Integer, nil] :max_retries
+        #
+        #   @option opts [Float, nil] :timeout
+        #
+        # @return [Hash{Symbol=>Object}]
+        private def build_request(req, opts)
+          fit_req_to_bedrock_specs!(req)
+
+          request_input = super
+
+          signed_request = @signer.sign_request(
+            http_method: request_input[:method],
+            url: request_input[:url],
+            headers: request_input[:headers],
+            body: request_input[:body]
+          )
+
+          request_input[:headers].merge!(signed_request.headers)
+
+          request_input
+        end
+
+        # @param aws_region [String, nil]
+        #
+        # @param aws_secret_key [String, nil]
+        #
+        # @param aws_access_key [String, nil]
+        #
+        # @param aws_session_token [String, nil]
+        #
+        # @param aws_profile [String, nil]
+        #
+        # @return [Array<String, Aws::Credentials>]
+        #
+        private def resolve_region_and_credentials(
+          aws_region:,
+          aws_secret_key:,
+          aws_access_key:,
+          aws_session_token:,
+          aws_profile:
+        )
+          client_options = {
+            access_key_id: aws_access_key,
+            secret_access_key: aws_secret_key,
+            session_token: aws_session_token,
+            profile: aws_profile
+          }
+          (client_options[:region] = aws_region) unless aws_region.nil?
+
+          bedrock_client = Aws::BedrockRuntime::Client.new(client_options)
+          [bedrock_client.config.region, bedrock_client.config.credentials.credentials]
+        end
+
+        # @private
+        #
+        # Overrides request components for Bedrock-specific request-shape requirements.
+        #
+        # @param request_components [Hash{Symbol=>Object}] .
+        #
+        #   @option request_components [Symbol] :method
+        #
+        #   @option request_components [String, Array<String>] :path
+        #
+        #   @option request_components [Hash{String=>Array<String>, String, nil}, nil] :query
+        #
+        #   @option request_components [Hash{String=>String, nil}, nil] :headers
+        #
+        #   @option request_components [Object, nil] :body
+        #
+        #   @option request_components [Symbol, nil] :unwrap
+        #
+        #   @option request_components [Class, nil] :page
+        #
+        #   @option request_components [Anthropic::Converter, Class, nil] :model
+        #
+        # @return [Hash{Symbol=>Object}]
+        #
+        private def fit_req_to_bedrock_specs!(request_components)
+          if (body = request_components[:body]).is_a?(Hash)
+            body[:anthropic_version] ||= DEFAULT_VERSION
+            body.transform_keys!("anthropic-beta": :anthropic_beta)
+          end
+
+          case request_components[:path]
+          in %r{^v1/messages/batches}
+            message = "The Batch API is not supported in Bedrock yet"
+            raise NotImplementedError.new(message)
+          in %r{v1/messages/count_tokens}
+            message = "Token counting is not supported in Bedrock yet"
+            raise NotImplementedError.new(message)
+          in %r{v1/models\?beta=true}
+            # rubocop:disable Layout/LineLength
+            message = "Please instead use https://docs.anthropic.com/en/api/claude-on-amazon-bedrock#list-available-models to list available models on Bedrock."
+            # rubocop:enable Layout/LineLength
+            raise NotImplementedError.new(message)
+          else
+          end
+
+          if %w[
+            v1/complete
+            v1/messages
+            v1/messages?beta=true
+          ].include?(request_components[:path]) && request_components[:method] == :post && body.is_a?(Hash)
+            model = body.delete(:model)
+            model = URI.encode_www_form_component(model.to_s)
+            stream = body.delete(:stream) || false
+            request_components[:path] =
+              stream ? "model/#{model}/invoke-with-response-stream" : "model/#{model}/invoke"
+          end
+
+          request_components
+        end
+      end
+    end
+  end
+end