anthropic 0.4.1 → 1.0.0

data/SECURITY.md

@@ -0,0 +1,27 @@
+# Security Policy
+
+## Reporting Security Issues
+
+This SDK is generated by [Stainless Software Inc](http://stainless.com). Stainless takes security seriously, and encourages you to report any security vulnerability promptly so that appropriate action can be taken.
+
+To report a security issue, please contact the Stainless team at security@stainless.com.
+
+## Responsible Disclosure
+
+We appreciate the efforts of security researchers and individuals who help us maintain the security of
+SDKs we generate. If you believe you have found a security vulnerability, please adhere to responsible
+disclosure practices by allowing us a reasonable amount of time to investigate and address the issue
+before making any information public.
+
+## Reporting Non-SDK Related Security Issues
+
+If you encounter security issues that are not directly related to SDKs but pertain to the services
+or products provided by Anthropic, please follow the respective company's security reporting guidelines.
+
+### Anthropic Terms and Policies
+
+Please contact support@anthropic.com for any questions or concerns regarding the security of our services.
+
+---
+
+Thank you for helping us keep the SDKs and systems they interact with secure.

data/lib/anthropic/bedrock.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module Anthropic
+  BedrockClient = Anthropic::Helpers::Bedrock::Client
+end

data/lib/anthropic/client.rb

@@ -1,115 +1,108 @@
-require_relative "messages/batches"
-require_relative "messages/client"
+# frozen_string_literal: true
 
 module Anthropic
-  class Client
-    include Anthropic::HTTP
-
-    CONFIG_KEYS = %i[
-      access_token
-      anthropic_version
-      api_version
-      log_errors
-      uri_base
-      request_timeout
-      extra_headers
-    ].freeze
-    attr_reader(*CONFIG_KEYS)
-
-    def initialize(config = {}, &faraday_middleware)
-      CONFIG_KEYS.each do |key|
-        # Set instance variables like api_type & access_token. Fall back to global config
-        # if not present.
-        instance_variable_set(
-          "@#{key}",
-          config[key].nil? ? Anthropic.configuration.send(key) : config[key]
-        )
-      end
-      @faraday_middleware = faraday_middleware
-      log "[WARNING] Gem `anthropic` was renamed to `ruby-anthropic`.
-        Please update your Gemfile to use `ruby-anthropic` version 0.4.2 or later."
-    end
+  class Client < Anthropic::Internal::Transport::BaseClient
+    # Default max number of retries to attempt after a failed retryable request.
+    DEFAULT_MAX_RETRIES = 2
 
-    # @deprecated (but still works while Anthropic API responds to it)
-    def complete(parameters: {})
-      parameters[:prompt] = wrap_prompt(prompt: parameters[:prompt])
-      json_post(path: "/complete", parameters: parameters)
-    end
+    # Default per-request timeout.
+    DEFAULT_TIMEOUT_IN_SECONDS = 600.0
+
+    # Default initial retry delay in seconds.
+    # Overall delay is calculated using exponential backoff + jitter.
+    DEFAULT_INITIAL_RETRY_DELAY = 0.5
+
+    # Default max retry delay in seconds.
+    DEFAULT_MAX_RETRY_DELAY = 8.0
+
+    # @return [String, nil]
+    attr_reader :api_key
+
+    # @return [String, nil]
+    attr_reader :auth_token
 
-    # Anthropic API Parameters as of 2024-05-07:
-    # @see https://docs.anthropic.com/claude/reference/messages_post
+    # @return [Anthropic::Resources::Completions]
+    attr_reader :completions
+
+    # @return [Anthropic::Resources::Messages]
+    attr_reader :messages
+
+    # @return [Anthropic::Resources::Models]
+    attr_reader :models
+
+    # @return [Anthropic::Resources::Beta]
+    attr_reader :beta
+
+    # @api private
     #
-    # When called without parameters, returns a Messages::Batches instance for batch operations.
-    # When called with parameters, creates a single message.
+    # @return [Hash{String=>String}]
+    private def auth_headers
+      {**api_key_auth, **bearer_auth}
+    end
+
+    # @api private
     #
-    # @param [Hash] parameters
-    # @option parameters [Array] :messages - Required. An array of messages to send to the API. Each
-    #   message should have a role and content. Single message example:
-    #   +[{ role: "user", content: "Hello, Claude!" }]+
-    # @option parameters [String] :model - see https://docs.anthropic.com/claude/docs/models-overview
-    # @option parameters [Integer] :max_tokens - Required, must be less than 4096 - @see https://docs.anthropic.com/claude/docs/models-overview
-    # @option parameters [String] :system - Optional but recommended. @see https://docs.anthropic.com/claude/docs/system-prompts
-    # @option parameters [Float] :temperature - Optional, defaults to 1.0
-    # @option parameters [Proc] :stream - Optional, if present, must be a Proc that will receive the
-    #   content fragments as they come in
-    # @option parameters [String] :preprocess_stream - If true, the streaming Proc will be pre-
-    #   processed. Specifically, instead of being passed a raw Hash like:
-    #   {"type"=>"content_block_delta", "index"=>0, "delta"=>{"type"=>"text_delta", "text"=>" of"}}
-    #   the Proc will instead be passed something nicer. If +preprocess_stream+ is set to +"json"+
-    #   or +:json+, then the Proc will only receive full json objects, one at a time.
-    #   If +preprocess_stream+ is set to +"text"+ or +:text+ then the Proc will receive two
-    #   arguments: the first will be the text accrued so far, and the second will be the delta
-    #   just received in the current chunk.
+    # @return [Hash{String=>String}]
+    private def api_key_auth
+      {"x-api-key" => @api_key}
+    end
+
+    # @api private
     #
-    # @returns [Hash, Messages::Client] Returns a Hash response from the API when creating a message
-    # with parameters, or a Messages::Client instance when called without parameters
-    # @example Creating a message:
-    #   {
-    #     "id" => "msg_013xVudG9xjSvLGwPKMeVXzG",
-    #     "type" => "message",
-    #     "role" => "assistant",
-    #     "content" => [{"type" => "text", "text" => "The sky has no distinct"}],
-    #     "model" => "claude-2.1",
-    #     "stop_reason" => "max_tokens",
-    #     "stop_sequence" => nil,
-    #     "usage" => {"input_tokens" => 15, "output_tokens" => 5}
-    #   }
-    # @example Accessing batches:
-    #   client.messages.batches.create(requests: [...])
-    #   client.messages.batches.get(id: "batch_123")
-    def messages(**args)
-      return @messages ||= Messages::Client.new(self) unless args && args[:parameters]
-
-      json_post(path: "/messages", parameters: args[:parameters])
+    # @return [Hash{String=>String}]
+    private def bearer_auth
+      return {} if @auth_token.nil?
+
+      {"authorization" => "Bearer #{@auth_token}"}
     end
 
-    # Adds Anthropic beta features to API requests. Can be used in two ways:
+    # Creates and returns a new client for interacting with the API.
     #
-    # 1. Multiple betas in one call with comma-separated string:
-    #    client.beta("feature1,feature2").messages
+    # @param api_key [String, nil] Defaults to `ENV["ANTHROPIC_API_KEY"]`
     #
-    # 2. Chaining multiple beta calls:
-    #    client.beta("feature1").beta("feature2").messages
+    # @param auth_token [String, nil] Defaults to `ENV["ANTHROPIC_AUTH_TOKEN"]`
     #
-    # @param version [String] The beta version(s) to enable
-    # @return [Client] A new client instance with the beta header(s)
-    def beta(version)
-      dup.tap do |client|
-        existing_beta = client.extra_headers["anthropic-beta"]
-        combined_beta = [existing_beta, version].compact.join(",")
-        client.add_headers("anthropic-beta" => combined_beta)
-      end
-    end
+    # @param base_url [String, nil] Override the default base URL for the API, e.g.,
+    # `"https://api.example.com/v2/"`. Defaults to `ENV["ANTHROPIC_BASE_URL"]`
+    #
+    # @param max_retries [Integer] Max number of retries to attempt after a failed retryable request.
+    #
+    # @param timeout [Float]
+    #
+    # @param initial_retry_delay [Float]
+    #
+    # @param max_retry_delay [Float]
+    def initialize(
+      api_key: ENV["ANTHROPIC_API_KEY"],
+      auth_token: ENV["ANTHROPIC_AUTH_TOKEN"],
+      base_url: ENV["ANTHROPIC_BASE_URL"],
+      max_retries: self.class::DEFAULT_MAX_RETRIES,
+      timeout: self.class::DEFAULT_TIMEOUT_IN_SECONDS,
+      initial_retry_delay: self.class::DEFAULT_INITIAL_RETRY_DELAY,
+      max_retry_delay: self.class::DEFAULT_MAX_RETRY_DELAY
+    )
+      base_url ||= "https://api.anthropic.com"
+
+      headers = {
+        "anthropic-version" => "2023-06-01"
+      }
 
-    private
+      @api_key = api_key&.to_s
+      @auth_token = auth_token&.to_s
 
-    # Used only by @deprecated +complete+ method
-    def wrap_prompt(prompt:, prefix: "\n\nHuman: ", suffix: "\n\nAssistant:")
-      return if prompt.nil?
+      super(
+        base_url: base_url,
+        timeout: timeout,
+        max_retries: max_retries,
+        initial_retry_delay: initial_retry_delay,
+        max_retry_delay: max_retry_delay,
+        headers: headers
+      )
 
-      prompt.prepend(prefix) unless prompt.start_with?(prefix)
-      prompt.concat(suffix) unless prompt.end_with?(suffix)
-      prompt
+      @completions = Anthropic::Resources::Completions.new(client: self)
+      @messages = Anthropic::Resources::Messages.new(client: self)
+      @models = Anthropic::Resources::Models.new(client: self)
+      @beta = Anthropic::Resources::Beta.new(client: self)
     end
   end
 end

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