anthropic 0.4.0 → 1.0.0

data/lib/anthropic/internal/type/base_model.rb

@@ -0,0 +1,481 @@
+# frozen_string_literal: true
+
+module Anthropic
+  module Internal
+    module Type
+      # @abstract
+      class BaseModel
+        extend Anthropic::Internal::Type::Converter
+        extend Anthropic::Internal::Util::SorbetRuntimeSupport
+
+        class << self
+          # @api private
+          #
+          # Assumes superclass fields are totally defined before fields are accessed /
+          # defined on subclasses.
+          #
+          # @param child [Class<Anthropic::Internal::Type::BaseModel>]
+          def inherited(child)
+            super
+            child.known_fields.replace(known_fields.dup)
+          end
+
+          # @api private
+          #
+          # @return [Hash{Symbol=>Hash{Symbol=>Object}}]
+          def known_fields = @known_fields ||= {}
+
+          # @api private
+          #
+          # @return [Hash{Symbol=>Hash{Symbol=>Object}}]
+          def fields
+            known_fields.transform_values do |field|
+              {**field.except(:type_fn), type: field.fetch(:type_fn).call}
+            end
+          end
+
+          # @api private
+          #
+          # @param name_sym [Symbol]
+          #
+          # @param required [Boolean]
+          #
+          # @param type_info [Hash{Symbol=>Object}, Proc, Anthropic::Internal::Type::Converter, Class]
+          #
+          # @param spec [Hash{Symbol=>Object}] .
+          #
+          #   @option spec [NilClass, TrueClass, FalseClass, Integer, Float, Symbol] :const
+          #
+          #   @option spec [Proc] :enum
+          #
+          #   @option spec [Proc] :union
+          #
+          #   @option spec [Boolean] :"nil?"
+          private def add_field(name_sym, required:, type_info:, spec:)
+            type_fn, info =
+              case type_info
+              in Proc | Anthropic::Internal::Type::Converter | Class
+                [Anthropic::Internal::Type::Converter.type_info({**spec, union: type_info}), spec]
+              in Hash
+                [Anthropic::Internal::Type::Converter.type_info(type_info), type_info]
+              end
+
+            setter = "#{name_sym}="
+            api_name = info.fetch(:api_name, name_sym)
+            nilable = info.fetch(:nil?, false)
+            const = if required && !nilable
+              info.fetch(
+                :const,
+                Anthropic::Internal::OMIT
+              )
+            else
+              Anthropic::Internal::OMIT
+            end
+
+            [name_sym, setter].each { undef_method(_1) } if known_fields.key?(name_sym)
+
+            known_fields[name_sym] =
+              {
+                mode: @mode,
+                api_name: api_name,
+                required: required,
+                nilable: nilable,
+                const: const,
+                type_fn: type_fn
+              }
+
+            define_method(setter) { @data.store(name_sym, _1) }
+
+            define_method(name_sym) do
+              target = type_fn.call
+              value = @data.fetch(name_sym) { const == Anthropic::Internal::OMIT ? nil : const }
+              state = {strictness: :strong, exactness: {yes: 0, no: 0, maybe: 0}, branched: 0}
+              if (nilable || !required) && value.nil?
+                nil
+              else
+                Anthropic::Internal::Type::Converter.coerce(
+                  target, value, state: state
+                )
+              end
+            rescue StandardError => e
+              cls = self.class.name.split("::").last
+              message = [
+                "Failed to parse #{cls}.#{__method__} from #{value.class} to #{target.inspect}.",
+                "To get the unparsed API response, use #{cls}[#{__method__.inspect}].",
+                "Cause: #{e.message}"
+              ].join(" ")
+              raise Anthropic::Errors::ConversionError.new(message)
+            end
+          end
+
+          # @api private
+          #
+          # @param name_sym [Symbol]
+          #
+          # @param type_info [Hash{Symbol=>Object}, Proc, Anthropic::Internal::Type::Converter, Class]
+          #
+          # @param spec [Hash{Symbol=>Object}] .
+          #
+          #   @option spec [NilClass, TrueClass, FalseClass, Integer, Float, Symbol] :const
+          #
+          #   @option spec [Proc] :enum
+          #
+          #   @option spec [Proc] :union
+          #
+          #   @option spec [Boolean] :"nil?"
+          def required(name_sym, type_info, spec = {})
+            add_field(name_sym, required: true, type_info: type_info, spec: spec)
+          end
+
+          # @api private
+          #
+          # @param name_sym [Symbol]
+          #
+          # @param type_info [Hash{Symbol=>Object}, Proc, Anthropic::Internal::Type::Converter, Class]
+          #
+          # @param spec [Hash{Symbol=>Object}] .
+          #
+          #   @option spec [NilClass, TrueClass, FalseClass, Integer, Float, Symbol] :const
+          #
+          #   @option spec [Proc] :enum
+          #
+          #   @option spec [Proc] :union
+          #
+          #   @option spec [Boolean] :"nil?"
+          def optional(name_sym, type_info, spec = {})
+            add_field(name_sym, required: false, type_info: type_info, spec: spec)
+          end
+
+          # @api private
+          #
+          # `request_only` attributes not excluded from `.#coerce` when receiving responses
+          # even if well behaved servers should not send them
+          #
+          # @param blk [Proc]
+          private def request_only(&blk)
+            @mode = :dump
+            blk.call
+          ensure
+            @mode = nil
+          end
+
+          # @api private
+          #
+          # `response_only` attributes are omitted from `.#dump` when making requests
+          #
+          # @param blk [Proc]
+          private def response_only(&blk)
+            @mode = :coerce
+            blk.call
+          ensure
+            @mode = nil
+          end
+
+          # @api public
+          #
+          # @param other [Object]
+          #
+          # @return [Boolean]
+          def ==(other)
+            other.is_a?(Class) && other <= Anthropic::Internal::Type::BaseModel && other.fields == fields
+          end
+
+          # @api public
+          #
+          # @return [Integer]
+          def hash = fields.hash
+        end
+
+        # @api public
+        #
+        # @param other [Object]
+        #
+        # @return [Boolean]
+        def ==(other) = self.class == other.class && @data == other.to_h
+
+        # @api public
+        #
+        # @return [Integer]
+        def hash = [self.class, @data].hash
+
+        class << self
+          # @api private
+          #
+          # @param value [Anthropic::Internal::Type::BaseModel, Hash{Object=>Object}, Object]
+          #
+          # @param state [Hash{Symbol=>Object}] .
+          #
+          #   @option state [Boolean, :strong] :strictness
+          #
+          #   @option state [Hash{Symbol=>Object}] :exactness
+          #
+          #   @option state [Integer] :branched
+          #
+          # @return [self, Object]
+          def coerce(value, state:)
+            exactness = state.fetch(:exactness)
+
+            if value.is_a?(self.class)
+              exactness[:yes] += 1
+              return value
+            end
+
+            unless (val = Anthropic::Internal::Util.coerce_hash(value)).is_a?(Hash)
+              exactness[:no] += 1
+              return value
+            end
+            exactness[:yes] += 1
+
+            keys = val.keys.to_set
+            instance = new
+            data = instance.to_h
+
+            # rubocop:disable Metrics/BlockLength
+            fields.each do |name, field|
+              mode, required, target = field.fetch_values(:mode, :required, :type)
+              api_name, nilable, const = field.fetch_values(:api_name, :nilable, :const)
+
+              unless val.key?(api_name)
+                if required && mode != :dump && const == Anthropic::Internal::OMIT
+                  exactness[nilable ? :maybe : :no] += 1
+                else
+                  exactness[:yes] += 1
+                end
+                next
+              end
+
+              item = val.fetch(api_name)
+              keys.delete(api_name)
+
+              converted =
+                if item.nil? && (nilable || !required)
+                  exactness[nilable ? :yes : :maybe] += 1
+                  nil
+                else
+                  coerced = Anthropic::Internal::Type::Converter.coerce(target, item, state: state)
+                  case target
+                  in Anthropic::Internal::Type::Converter | Symbol
+                    coerced
+                  else
+                    item
+                  end
+                end
+              data.store(name, converted)
+            end
+            # rubocop:enable Metrics/BlockLength
+
+            keys.each { data.store(_1, val.fetch(_1)) }
+            instance
+          end
+
+          # @api private
+          #
+          # @param value [self, Object]
+          #
+          # @param state [Hash{Symbol=>Object}] .
+          #
+          #   @option state [Boolean] :can_retry
+          #
+          # @return [Hash{Object=>Object}, Object]
+          def dump(value, state:)
+            unless (coerced = Anthropic::Internal::Util.coerce_hash(value)).is_a?(Hash)
+              return super
+            end
+
+            acc = {}
+
+            coerced.each do |key, val|
+              name = key.is_a?(String) ? key.to_sym : key
+              case (field = known_fields[name])
+              in nil
+                acc.store(name, super(val, state: state))
+              else
+                api_name, mode, type_fn = field.fetch_values(:api_name, :mode, :type_fn)
+                case mode
+                in :coerce
+                  next
+                else
+                  target = type_fn.call
+                  acc.store(api_name, Anthropic::Internal::Type::Converter.dump(target, val, state: state))
+                end
+              end
+            end
+
+            known_fields.each_value do |field|
+              api_name, mode, const = field.fetch_values(:api_name, :mode, :const)
+              next if mode == :coerce || acc.key?(api_name) || const == Anthropic::Internal::OMIT
+              acc.store(api_name, const)
+            end
+
+            acc
+          end
+
+          # @api private
+          #
+          # @return [Object]
+          def to_sorbet_type
+            self
+          end
+        end
+
+        class << self
+          # @api private
+          #
+          # @param model [Anthropic::Internal::Type::BaseModel]
+          # @param convert [Boolean]
+          #
+          # @return [Hash{Symbol=>Object}]
+          def recursively_to_h(model, convert:)
+            rec = ->(x) do
+              case x
+              in Anthropic::Internal::Type::BaseModel
+                if convert
+                  fields = x.class.known_fields
+                  x.to_h.to_h do |key, val|
+                    [key, rec.call(fields.key?(key) ? x.public_send(key) : val)]
+                  rescue Anthropic::Errors::ConversionError
+                    [key, rec.call(val)]
+                  end
+                else
+                  rec.call(x.to_h)
+                end
+              in Hash
+                x.transform_values(&rec)
+              in Array
+                x.map(&rec)
+              else
+                x
+              end
+            end
+            rec.call(model)
+          end
+        end
+
+        # @api public
+        #
+        # Returns the raw value associated with the given key, if found. Otherwise, nil is
+        # returned.
+        #
+        # It is valid to lookup keys that are not in the API spec, for example to access
+        # undocumented features. This method does not parse response data into
+        # higher-level types. Lookup by anything other than a Symbol is an ArgumentError.
+        #
+        # @param key [Symbol]
+        #
+        # @return [Object, nil]
+        def [](key)
+          unless key.instance_of?(Symbol)
+            raise ArgumentError.new("Expected symbol key for lookup, got #{key.inspect}")
+          end
+
+          @data[key]
+        end
+
+        # @api public
+        #
+        # Returns a Hash of the data underlying this object. O(1)
+        #
+        # Keys are Symbols and values are the raw values from the response. The return
+        # value indicates which values were ever set on the object. i.e. there will be a
+        # key in this hash if they ever were, even if the set value was nil.
+        #
+        # This method is not recursive. The returned value is shared by the object, so it
+        # should not be mutated.
+        #
+        # @return [Hash{Symbol=>Object}]
+        def to_h = @data
+
+        alias_method :to_hash, :to_h
+
+        # @api public
+        #
+        # In addition to the behaviour of `#to_h`, this method will recursively call
+        # `#to_h` on nested models.
+        #
+        # @return [Hash{Symbol=>Object}]
+        def deep_to_h = self.class.recursively_to_h(@data, convert: false)
+
+        # @param keys [Array<Symbol>, nil]
+        #
+        # @return [Hash{Symbol=>Object}]
+        #
+        # @example
+        #   # `api_error_object` is a `Anthropic::APIErrorObject`
+        #   api_error_object => {
+        #     message: message,
+        #     type: type
+        #   }
+        def deconstruct_keys(keys)
+          (keys || self.class.known_fields.keys)
+            .filter_map do |k|
+              unless self.class.known_fields.key?(k)
+                next
+              end
+
+              [k, public_send(k)]
+            end
+            .to_h
+        end
+
+        # @api public
+        #
+        # @param a [Object]
+        #
+        # @return [String]
+        def to_json(*a) = Anthropic::Internal::Type::Converter.dump(self.class, self).to_json(*a)
+
+        # @api public
+        #
+        # @param a [Object]
+        #
+        # @return [String]
+        def to_yaml(*a) = Anthropic::Internal::Type::Converter.dump(self.class, self).to_yaml(*a)
+
+        # Create a new instance of a model.
+        #
+        # @param data [Hash{Symbol=>Object}, self]
+        def initialize(data = {}) = (@data = Anthropic::Internal::Util.coerce_hash!(data).to_h)
+
+        class << self
+          # @api private
+          #
+          # @param depth [Integer]
+          #
+          # @return [String]
+          def inspect(depth: 0)
+            return super() if depth.positive?
+
+            depth = depth.succ
+            deferred = fields.transform_values do |field|
+              type, required, nilable = field.fetch_values(:type, :required, :nilable)
+              inspected = [
+                Anthropic::Internal::Type::Converter.inspect(type, depth: depth),
+                !required || nilable ? "nil" : nil
+              ].compact.join(" | ")
+              -> { inspected }.tap { _1.define_singleton_method(:inspect) { call } }
+            end
+
+            "#{name}[#{deferred.inspect}]"
+          end
+        end
+
+        # @api public
+        #
+        # @return [String]
+        def to_s = deep_to_h.to_s
+
+        # @api private
+        #
+        # @return [String]
+        def inspect
+          converted = self.class.recursively_to_h(self, convert: true)
+          "#<#{self.class}:0x#{object_id.to_s(16)} #{converted}>"
+        end
+
+        define_sorbet_constant!(:KnownField) do
+          T.type_alias { {mode: T.nilable(Symbol), required: T::Boolean, nilable: T::Boolean} }
+        end
+      end
+    end
+  end
+end

data/lib/anthropic/internal/type/base_page.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+module Anthropic
+  module Internal
+    module Type
+      # @api private
+      #
+      # @generic Elem
+      #
+      # This module provides a base implementation for paginated responses in the SDK.
+      module BasePage
+        # rubocop:disable Lint/UnusedMethodArgument
+
+        # @api public
+        #
+        # @return [Boolean]
+        def next_page? = (raise NotImplementedError)
+
+        # @api public
+        #
+        # @raise [Anthropic::Errors::APIError]
+        # @return [self]
+        def next_page = (raise NotImplementedError)
+
+        # @api public
+        #
+        # @param blk [Proc]
+        #
+        # @yieldparam [generic<Elem>]
+        # @return [void]
+        def auto_paging_each(&blk) = (raise NotImplementedError)
+
+        # @return [Enumerable<generic<Elem>>]
+        def to_enum = super(:auto_paging_each)
+
+        alias_method :enum_for, :to_enum
+
+        # @api private
+        #
+        # @param client [Anthropic::Internal::Transport::BaseClient]
+        # @param req [Hash{Symbol=>Object}]
+        # @param headers [Hash{String=>String}, Net::HTTPHeader]
+        # @param page_data [Object]
+        def initialize(client:, req:, headers:, page_data:)
+          @client = client
+          @req = req
+          @model = req.fetch(:model)
+          super()
+        end
+
+        # rubocop:enable Lint/UnusedMethodArgument
+      end
+    end
+  end
+end

data/lib/anthropic/internal/type/base_stream.rb

@@ -0,0 +1,90 @@
+# frozen_string_literal: true
+
+module Anthropic
+  module Internal
+    module Type
+      # @api private
+      #
+      # @generic Elem
+      #
+      # This module provides a base implementation for streaming responses in the SDK.
+      #
+      # @see https://rubyapi.org/3.2/o/enumerable
+      module BaseStream
+        include Enumerable
+
+        class << self
+          # Attempt to close the underlying transport when the stream itself is garbage
+          # collected.
+          #
+          # This should not be relied upon for resource clean up, as the garbage collector
+          # is not guaranteed to run.
+          #
+          # @param stream [Enumerable<Object>]
+          #
+          # @return [Proc]
+          #
+          # @see https://rubyapi.org/3.2/o/objectspace#method-c-define_finalizer
+          def defer_closing(stream) = ->(_id) { Anthropic::Internal::Util.close_fused!(stream) }
+        end
+
+        # @api public
+        #
+        # @return [void]
+        def close = Anthropic::Internal::Util.close_fused!(@iterator)
+
+        # @api private
+        #
+        # @return [Enumerable<generic<Elem>>]
+        private def iterator = (raise NotImplementedError)
+
+        # @api public
+        #
+        # @param blk [Proc]
+        #
+        # @yieldparam [generic<Elem>]
+        # @return [void]
+        def each(&blk)
+          unless block_given?
+            raise ArgumentError.new("A block must be given to ##{__method__}")
+          end
+          @iterator.each(&blk)
+        end
+
+        # @api public
+        #
+        # @return [Enumerator<generic<Elem>>]
+        def to_enum = @iterator
+
+        alias_method :enum_for, :to_enum
+
+        # @api private
+        #
+        # @param model [Class, Anthropic::Internal::Type::Converter]
+        # @param url [URI::Generic]
+        # @param status [Integer]
+        # @param response [Net::HTTPResponse]
+        # @param stream [Enumerable<Object>]
+        def initialize(model:, url:, status:, response:, stream:)
+          @model = model
+          @url = url
+          @status = status
+          @response = response
+          @stream = stream
+          @iterator = iterator
+
+          ObjectSpace.define_finalizer(self, Anthropic::Internal::Type::BaseStream.defer_closing(@stream))
+        end
+
+        # @api private
+        #
+        # @return [String]
+        def inspect
+          model = Anthropic::Internal::Type::Converter.inspect(@model, depth: 1)
+
+          "#<#{self.class}[#{model}]:0x#{object_id.to_s(16)}>"
+        end
+      end
+    end
+  end
+end

data/lib/anthropic/internal/type/boolean.rb

@@ -0,0 +1,71 @@
+# frozen_string_literal: true
+
+module Anthropic
+  module Internal
+    module Type
+      # @api private
+      #
+      # @abstract
+      #
+      # Ruby has no Boolean class; this is something for models to refer to.
+      class Boolean
+        extend Anthropic::Internal::Type::Converter
+        extend Anthropic::Internal::Util::SorbetRuntimeSupport
+
+        private_class_method :new
+
+        # @api public
+        #
+        # @param other [Object]
+        #
+        # @return [Boolean]
+        def self.===(other) = other == true || other == false
+
+        # @api public
+        #
+        # @param other [Object]
+        #
+        # @return [Boolean]
+        def self.==(other) = other.is_a?(Class) && other <= Anthropic::Internal::Type::Boolean
+
+        class << self
+          # @api private
+          #
+          # @param value [Boolean, Object]
+          #
+          # @param state [Hash{Symbol=>Object}] .
+          #
+          #   @option state [Boolean, :strong] :strictness
+          #
+          #   @option state [Hash{Symbol=>Object}] :exactness
+          #
+          #   @option state [Integer] :branched
+          #
+          # @return [Boolean, Object]
+          def coerce(value, state:)
+            state.fetch(:exactness)[value == true || value == false ? :yes : :no] += 1
+            value
+          end
+
+          # @!method dump(value, state:)
+          #   @api private
+          #
+          #   @param value [Boolean, Object]
+          #
+          #   @param state [Hash{Symbol=>Object}] .
+          #
+          #     @option state [Boolean] :can_retry
+          #
+          #   @return [Boolean, Object]
+
+          # @api private
+          #
+          # @return [Object]
+          def to_sorbet_type
+            T::Boolean
+          end
+        end
+      end
+    end
+  end
+end