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

data/lib/anthropic/client.rb

@@ -0,0 +1,107 @@
+# frozen_string_literal: true
+
+module Anthropic
+  class Client < Anthropic::Internal::Transport::BaseClient
+    # Default max number of retries to attempt after a failed retryable request.
+    DEFAULT_MAX_RETRIES = 2
+
+    # 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
+
+    # @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
+    #
+    # @return [Hash{String=>String}]
+    private def auth_headers
+      {**api_key_auth, **bearer_auth}
+    end
+
+    # @api private
+    #
+    # @return [Hash{String=>String}]
+    private def api_key_auth
+      {"x-api-key" => @api_key}
+    end
+
+    # @api private
+    #
+    # @return [Hash{String=>String}]
+    private def bearer_auth
+      return {} if @auth_token.nil?
+
+      {"authorization" => "Bearer #{@auth_token}"}
+    end
+
+    # Creates and returns a new client for interacting with the API.
+    #
+    # @param api_key [String, nil] Defaults to `ENV["ANTHROPIC_API_KEY"]`
+    #
+    # @param auth_token [String, nil] Defaults to `ENV["ANTHROPIC_AUTH_TOKEN"]`
+    #
+    # @param base_url [String, nil] Override the default base URL for the API, e.g., `"https://api.example.com/v2/"`
+    #
+    # @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: 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
+    )
+      base_url ||= "https://api.anthropic.com"
+
+      headers = {
+        "anthropic-version" => "2023-06-01"
+      }
+
+      @api_key = api_key&.to_s
+      @auth_token = auth_token&.to_s
+
+      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
+      )
+
+      @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
+      # @!parse
+      #   # @return [StandardError, nil]
+      #   attr_accessor :cause
+    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
+      # @!parse
+      #   # @return [nil]
+      #   attr_accessor :status
+
+      # @!parse
+      #   # @return [nil]
+      #   attr_accessor :body
+
+      # @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 [Anthropic::Errors::APIStatusError]
+      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/internal/jsonl_stream.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+module Anthropic
+  module Internal
+    # @generic Elem
+    #
+    # @example
+    #   stream.each do |batch|
+    #     puts(batch)
+    #   end
+    class JsonLStream
+      include Anthropic::Internal::Type::BaseStream
+
+      # @api private
+      #
+      # @return [Enumerable<generic<Elem>>]
+      private def iterator
+        @iterator ||= Anthropic::Internal::Util.chain_fused(@stream) do |y|
+          @stream.each do
+            y << Anthropic::Internal::Type::Converter.coerce(@model, _1)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/anthropic/internal/page.rb

@@ -0,0 +1,109 @@
+# frozen_string_literal: true
+
+module Anthropic
+  module Internal
+    # @generic Elem
+    #
+    # @example
+    #   if page.has_next?
+    #     page = page.next_page
+    #   end
+    #
+    # @example
+    #   page.auto_paging_each do |batch|
+    #     puts(batch)
+    #   end
+    class Page
+      include Anthropic::Internal::Type::BasePage
+
+      # @return [Array<generic<Elem>>, nil]
+      attr_accessor :data
+
+      # @return [Boolean]
+      attr_accessor :has_more
+
+      # @return [String, nil]
+      attr_accessor :first_id
+
+      # @return [String, nil]
+      attr_accessor :last_id
+
+      # @api private
+      #
+      # @param client [Anthropic::Internal::Transport::BaseClient]
+      # @param req [Hash{Symbol=>Object}]
+      # @param headers [Hash{String=>String}, Net::HTTPHeader]
+      # @param page_data [Hash{Symbol=>Object}]
+      def initialize(client:, req:, headers:, page_data:)
+        super
+        model = req.fetch(:model)
+
+        case page_data
+        in {data: Array | nil => data}
+          @data = data&.map { Anthropic::Internal::Type::Converter.coerce(model, _1) }
+        else
+        end
+
+        case page_data
+        in {has_more: true | false | nil => has_more}
+          @has_more = has_more
+        else
+        end
+
+        case page_data
+        in {first_id: String | nil => first_id}
+          @first_id = first_id
+        else
+        end
+
+        case page_data
+        in {last_id: String | nil => last_id}
+          @last_id = last_id
+        else
+        end
+      end
+
+      # @return [Boolean]
+      def next_page?
+        has_more
+      end
+
+      # @raise [Anthropic::HTTP::Error]
+      # @return [Anthropic::Internal::Page]
+      def next_page
+        unless next_page?
+          message = "No more pages available. Please check #next_page? before calling ##{__method__}"
+          raise RuntimeError.new(message)
+        end
+
+        req = Anthropic::Internal::Util.deep_merge(
+          @req,
+          {query: first_id.nil? ? {after_id: last_id} : {before_id: first_id}}
+        )
+        @client.request(req)
+      end
+
+      # @param blk [Proc]
+      #
+      # @yieldparam [generic<Elem>]
+      def auto_paging_each(&blk)
+        unless block_given?
+          raise ArgumentError.new("A block must be given to ##{__method__}")
+        end
+        page = self
+        loop do
+          page.data&.each { blk.call(_1) }
+          break unless page.next_page?
+          page = page.next_page
+        end
+      end
+
+      # @return [String]
+      def inspect
+        # rubocop:disable Layout/LineLength
+        "#<#{self.class}:0x#{object_id.to_s(16)} data=#{data.inspect} has_more=#{has_more.inspect} first_id=#{first_id.inspect} last_id=#{last_id.inspect}>"
+        # rubocop:enable Layout/LineLength
+      end
+    end
+  end
+end

data/lib/anthropic/internal/stream.rb

@@ -0,0 +1,58 @@
+# frozen_string_literal: true
+
+module Anthropic
+  module Internal
+    # @generic Elem
+    #
+    # @example
+    #   stream.each do |event|
+    #     puts(event)
+    #   end
+    class Stream
+      include Anthropic::Internal::Type::BaseStream
+
+      # @api private
+      #
+      # @return [Enumerable<generic<Elem>>]
+      private def iterator
+        # rubocop:disable Metrics/BlockLength
+        # rubocop:disable Layout/LineLength
+        # rubocop:disable Lint/DuplicateBranch
+        @iterator ||= Anthropic::Internal::Util.chain_fused(@stream) do |y|
+          @stream.each do |msg|
+            case msg
+            in {event: "completion", data: String => data}
+              decoded = JSON.parse(data, symbolize_names: true)
+              y << Anthropic::Internal::Type::Converter.coerce(@model, decoded)
+            in {
+              event: "message_start" | "message_delta" | "message_stop" | "content_block_start" | "content_block_delta" | "content_block_stop",
+              data: String => data
+            }
+              decoded = JSON.parse(data, symbolize_names: true)
+              y << Anthropic::Internal::Type::Converter.coerce(@model, decoded)
+            in {event: "ping"}
+              next
+            in {event: "error", data: String => data}
+              decoded = Kernel.then do
+                JSON.parse(data, symbolize_names: true)
+              rescue JSON::ParserError
+                data
+              end
+              Anthropic::Errors::APIStatusError.for(
+                url: @url,
+                status: @status,
+                body: decoded,
+                request: nil,
+                response: @response
+              )
+            else
+            end
+          end
+        end
+        # rubocop:enable Lint/DuplicateBranch
+        # rubocop:enable Layout/LineLength
+        # rubocop:enable Metrics/BlockLength
+      end
+    end
+  end
+end