anthropic-sdk-beta 0.1.0.pre.beta.6 → 0.1.0.pre.beta.7

data/lib/anthropic/internal/util.rb

@@ -128,6 +128,22 @@ module Anthropic
             input.respond_to?(:to_h) ? input.to_h : input
           end
         end
+
+        # @api private
+        #
+        # @param input [Object]
+        #
+        # @raise [ArgumentError]
+        # @return [Hash{Object=>Object}, nil]
+        def coerce_hash!(input)
+          case coerce_hash(input)
+          in Hash | nil => coerced
+            coerced
+          else
+            message = "Expected a #{Hash} or #{Anthropic::Internal::Type::BaseModel}, got #{data.inspect}"
+            raise ArgumentError.new(message)
+          end
+        end
       end
 
       class << self
@@ -175,18 +191,17 @@ module Anthropic
         # @api private
         #
         # @param data [Hash{Symbol=>Object}, Array<Object>, Object]
-        # @param pick [Symbol, Integer, Array<Symbol, Integer>, nil]
-        # @param sentinel [Object, nil]
+        # @param pick [Symbol, Integer, Array<Symbol, Integer>, Proc, nil]
         # @param blk [Proc, nil]
         #
         # @return [Object, nil]
-        def dig(data, pick, sentinel = nil, &blk)
-          case [data, pick, blk]
-          in [_, nil, nil]
+        def dig(data, pick, &blk)
+          case [data, pick]
+          in [_, nil]
             data
-          in [Hash, Symbol, _] | [Array, Integer, _]
-            blk.nil? ? data.fetch(pick, sentinel) : data.fetch(pick, &blk)
-          in [Hash | Array, Array, _]
+          in [Hash, Symbol] | [Array, Integer]
+            data.fetch(pick) { blk&.call }
+          in [Hash | Array, Array]
             pick.reduce(data) do |acc, key|
               case acc
               in Hash if acc.key?(key)
@@ -194,11 +209,13 @@ module Anthropic
               in Array if key.is_a?(Integer) && key < acc.length
                 acc[key]
               else
-                return blk.nil? ? sentinel : blk.call
+                return blk&.call
               end
             end
-          in _
-            blk.nil? ? sentinel : blk.call
+          in [_, Proc]
+            pick.call(data)
+          else
+            blk&.call
           end
         end
       end
@@ -348,27 +365,6 @@ module Anthropic
         end
       end
 
-      # @api private
-      class SerializationAdapter
-        # @return [Pathname, IO]
-        attr_reader :inner
-
-        # @param a [Object]
-        #
-        # @return [String]
-        def to_json(*a) = (inner.is_a?(IO) ? inner.read : inner.read(binmode: true)).to_json(*a)
-
-        # @param a [Object]
-        #
-        # @return [String]
-        def to_yaml(*a) = (inner.is_a?(IO) ? inner.read : inner.read(binmode: true)).to_yaml(*a)
-
-        # @api private
-        #
-        # @param inner [Pathname, IO]
-        def initialize(inner) = @inner = inner
-      end
-
       # @api private
       #
       # An adapter that satisfies the IO interface required by `::IO.copy_stream`
@@ -448,7 +444,7 @@ module Anthropic
             else
               src
             end
-          @buf = String.new.b
+          @buf = String.new
           @blk = blk
         end
       end
@@ -460,7 +456,7 @@ module Anthropic
         # @return [Enumerable<String>]
         def writable_enum(&blk)
           Enumerator.new do |y|
-            buf = String.new.b
+            buf = String.new
             y.define_singleton_method(:write) do
               self << buf.replace(_1)
               buf.bytesize
@@ -471,53 +467,81 @@ module Anthropic
         end
       end
 
+      # @type [Regexp]
+      JSON_CONTENT = %r{^application/(?:vnd(?:\.[^.]+)*\+)?json(?!l)}
+      # @type [Regexp]
+      JSONL_CONTENT = %r{^application/(:?x-(?:n|l)djson)|(:?(?:x-)?jsonl)}
+
       class << self
         # @api private
         #
         # @param y [Enumerator::Yielder]
-        # @param boundary [String]
-        # @param key [Symbol, String]
         # @param val [Object]
         # @param closing [Array<Proc>]
-        private def write_multipart_chunk(y, boundary:, key:, val:, closing:)
-          val = val.inner if val.is_a?(Anthropic::Internal::Util::SerializationAdapter)
+        # @param content_type [String, nil]
+        private def write_multipart_content(y, val:, closing:, content_type: nil)
+          content_type ||= "application/octet-stream"
 
-          y << "--#{boundary}\r\n"
-          y << "Content-Disposition: form-data"
-          unless key.nil?
-            name = ERB::Util.url_encode(key.to_s)
-            y << "; name=\"#{name}\""
-          end
-          case val
-          in Pathname | IO
-            filename = ERB::Util.url_encode(File.basename(val.to_path))
-            y << "; filename=\"#{filename}\""
-          else
-          end
-          y << "\r\n"
           case val
+          in Anthropic::FilePart
+            return write_multipart_content(
+              y,
+              val: val.content,
+              closing: closing,
+              content_type: val.content_type
+            )
           in Pathname
-            y << "Content-Type: application/octet-stream\r\n\r\n"
+            y << "Content-Type: #{content_type}\r\n\r\n"
             io = val.open(binmode: true)
             closing << io.method(:close)
             IO.copy_stream(io, y)
           in IO
-            y << "Content-Type: application/octet-stream\r\n\r\n"
+            y << "Content-Type: #{content_type}\r\n\r\n"
             IO.copy_stream(val, y)
           in StringIO
-            y << "Content-Type: application/octet-stream\r\n\r\n"
+            y << "Content-Type: #{content_type}\r\n\r\n"
             y << val.string
           in String
-            y << "Content-Type: application/octet-stream\r\n\r\n"
+            y << "Content-Type: #{content_type}\r\n\r\n"
             y << val.to_s
-          in _ if primitive?(val)
+          in -> { primitive?(_1) }
             y << "Content-Type: text/plain\r\n\r\n"
             y << val.to_s
           else
             y << "Content-Type: application/json\r\n\r\n"
-            y << JSON.fast_generate(val)
+            y << JSON.generate(val)
+          end
+          y << "\r\n"
+        end
+
+        # @api private
+        #
+        # @param y [Enumerator::Yielder]
+        # @param boundary [String]
+        # @param key [Symbol, String]
+        # @param val [Object]
+        # @param closing [Array<Proc>]
+        private def write_multipart_chunk(y, boundary:, key:, val:, closing:)
+          y << "--#{boundary}\r\n"
+          y << "Content-Disposition: form-data"
+
+          unless key.nil?
+            name = ERB::Util.url_encode(key.to_s)
+            y << "; name=\"#{name}\""
+          end
+
+          case val
+          in Anthropic::FilePart unless val.filename.nil?
+            filename = ERB::Util.url_encode(val.filename)
+            y << "; filename=\"#{filename}\""
+          in Pathname | IO
+            filename = ERB::Util.url_encode(::File.basename(val.to_path))
+            y << "; filename=\"#{filename}\""
+          else
           end
           y << "\r\n"
+
+          write_multipart_content(y, val: val, closing: closing)
         end
 
         # @api private
@@ -560,14 +584,12 @@ module Anthropic
         # @return [Object]
         def encode_content(headers, body)
           content_type = headers["content-type"]
-          body = body.inner if body.is_a?(Anthropic::Internal::Util::SerializationAdapter)
-
           case [content_type, body]
-          in [%r{^application/(?:vnd\.api\+)?json}, Hash | Array | -> { primitive?(_1) }]
-            [headers, JSON.fast_generate(body)]
-          in [%r{^application/(?:x-)?jsonl}, Enumerable] unless body.is_a?(StringIO) || body.is_a?(IO)
-            [headers, body.lazy.map { JSON.fast_generate(_1) }]
-          in [%r{^multipart/form-data}, Hash | Pathname | StringIO | IO]
+          in [Anthropic::Internal::Util::JSON_CONTENT, Hash | Array | -> { primitive?(_1) }]
+            [headers, JSON.generate(body)]
+          in [Anthropic::Internal::Util::JSONL_CONTENT, Enumerable] unless body.is_a?(Anthropic::Internal::Type::FileInput)
+            [headers, body.lazy.map { JSON.generate(_1) }]
+          in [%r{^multipart/form-data}, Hash | Anthropic::Internal::Type::FileInput]
             boundary, strio = encode_multipart_streaming(body)
             headers = {**headers, "content-type" => "#{content_type}; boundary=#{boundary}"}
             [headers, strio]
@@ -575,6 +597,8 @@ module Anthropic
             [headers, body.to_s]
           in [_, StringIO]
             [headers, body.string]
+          in [_, Anthropic::FilePart]
+            [headers, body.content]
           else
             [headers, body]
           end
@@ -582,6 +606,27 @@ module Anthropic
 
         # @api private
         #
+        # https://www.iana.org/assignments/character-sets/character-sets.xhtml
+        #
+        # @param content_type [String]
+        # @param text [String]
+        def force_charset!(content_type, text:)
+          charset = /charset=([^;\s]+)/.match(content_type)&.captures&.first
+
+          return unless charset
+
+          begin
+            encoding = Encoding.find(charset)
+            text.force_encoding(encoding)
+          rescue ArgumentError
+            nil
+          end
+        end
+
+        # @api private
+        #
+        # Assumes each chunk in stream has `Encoding::BINARY`.
+        #
         # @param headers [Hash{String=>String}, Net::HTTPHeader]
         # @param stream [Enumerable<String>]
         # @param suppress_error [Boolean]
@@ -589,8 +634,8 @@ module Anthropic
         # @raise [JSON::ParserError]
         # @return [Object]
         def decode_content(headers, stream:, suppress_error: false)
-          case headers["content-type"]
-          in %r{^application/(?:vnd\.api\+)?json}
+          case (content_type = headers["content-type"])
+          in Anthropic::Internal::Util::JSON_CONTENT
             json = stream.to_a.join
             begin
               JSON.parse(json, symbolize_names: true)
@@ -598,7 +643,7 @@ module Anthropic
               raise e unless suppress_error
               json
             end
-          in %r{^application/(?:x-)?jsonl}
+          in Anthropic::Internal::Util::JSONL_CONTENT
             lines = decode_lines(stream)
             chain_fused(lines) do |y|
               lines.each { y << JSON.parse(_1, symbolize_names: true) }
@@ -606,11 +651,10 @@ module Anthropic
           in %r{^text/event-stream}
             lines = decode_lines(stream)
             decode_sse(lines)
-          in %r{^text/}
-            stream.to_a.join
           else
-            # TODO: parsing other response types
-            StringIO.new(stream.to_a.join)
+            text = stream.to_a.join
+            force_charset!(content_type, text: text)
+            StringIO.new(text)
           end
         end
       end
@@ -675,12 +719,17 @@ module Anthropic
       class << self
         # @api private
         #
+        # Assumes Strings have been forced into having `Encoding::BINARY`.
+        #
+        # This decoder is responsible for reassembling lines split across multiple
+        # fragments.
+        #
         # @param enum [Enumerable<String>]
         #
         # @return [Enumerable<String>]
         def decode_lines(enum)
           re = /(\r\n|\r|\n)/
-          buffer = String.new.b
+          buffer = String.new
           cr_seen = nil
 
           chain_fused(enum) do |y|
@@ -711,6 +760,8 @@ module Anthropic
         #
         # https://html.spec.whatwg.org/multipage/server-sent-events.html#parsing-an-event-stream
         #
+        # Assumes that `lines` has been decoded with `#decode_lines`.
+        #
         # @param lines [Enumerable<String>]
         #
         # @return [Enumerable<Hash{Symbol=>Object}>]
@@ -734,7 +785,7 @@ module Anthropic
                 in "event"
                   current.merge!(event: value)
                 in "data"
-                  (current[:data] ||= String.new.b) << (value << "\n")
+                  (current[:data] ||= String.new) << (value << "\n")
                 in "id" unless value.include?("\0")
                   current.merge!(id: value)
                 in "retry" if /^\d+$/ =~ value
@@ -750,6 +801,62 @@ module Anthropic
           end
         end
       end
+
+      # @api private
+      module SorbetRuntimeSupport
+        class MissingSorbetRuntimeError < ::RuntimeError
+        end
+
+        # @api private
+        #
+        # @return [Hash{Symbol=>Object}]
+        private def sorbet_runtime_constants = @sorbet_runtime_constants ||= {}
+
+        # @api private
+        #
+        # @param name [Symbol]
+        def const_missing(name)
+          super unless sorbet_runtime_constants.key?(name)
+
+          unless Object.const_defined?(:T)
+            message = "Trying to access a Sorbet constant #{name.inspect} without `sorbet-runtime`."
+            raise MissingSorbetRuntimeError.new(message)
+          end
+
+          sorbet_runtime_constants.fetch(name).call
+        end
+
+        # @api private
+        #
+        # @param name [Symbol]
+        # @param blk [Proc]
+        def define_sorbet_constant!(name, &blk) = sorbet_runtime_constants.store(name, blk)
+      end
+
+      extend Anthropic::Internal::Util::SorbetRuntimeSupport
+
+      define_sorbet_constant!(:ParsedUri) do
+        T.type_alias do
+          {
+            scheme: T.nilable(String),
+            host: T.nilable(String),
+            port: T.nilable(Integer),
+            path: T.nilable(String),
+            query: T::Hash[String, T::Array[String]]
+          }
+        end
+      end
+
+      define_sorbet_constant!(:ServerSentEvent) do
+        T.type_alias do
+          {
+            event: T.nilable(String),
+            data: T.nilable(String),
+            id: T.nilable(String),
+            retry: T.nilable(Integer)
+          }
+        end
+      end
     end
   end
 end

data/lib/anthropic/internal.rb

@@ -1,8 +1,17 @@
 # frozen_string_literal: true
 
 module Anthropic
-  # @api private
   module Internal
-    OMIT = Object.new.freeze
+    extend Anthropic::Internal::Util::SorbetRuntimeSupport
+
+    OMIT =
+      Object.new.tap do
+        _1.define_singleton_method(:inspect) { "#<#{Anthropic::Internal}::OMIT>" }
+      end
+        .freeze
+
+    define_sorbet_constant!(:AnyHash) do
+      T.type_alias { T::Hash[Symbol, T.anything] }
+    end
   end
 end

data/lib/anthropic/models/anthropic_beta.rb

@@ -7,25 +7,28 @@ module Anthropic
 
       variant String
 
-      variant const: -> { Anthropic::Models::AnthropicBeta::MESSAGE_BATCHES_2024_09_24 }
+      variant const: -> { Anthropic::AnthropicBeta::MESSAGE_BATCHES_2024_09_24 }
 
-      variant const: -> { Anthropic::Models::AnthropicBeta::PROMPT_CACHING_2024_07_31 }
+      variant const: -> { Anthropic::AnthropicBeta::PROMPT_CACHING_2024_07_31 }
 
-      variant const: -> { Anthropic::Models::AnthropicBeta::COMPUTER_USE_2024_10_22 }
+      variant const: -> { Anthropic::AnthropicBeta::COMPUTER_USE_2024_10_22 }
 
-      variant const: -> { Anthropic::Models::AnthropicBeta::COMPUTER_USE_2025_01_24 }
+      variant const: -> { Anthropic::AnthropicBeta::COMPUTER_USE_2025_01_24 }
 
-      variant const: -> { Anthropic::Models::AnthropicBeta::PDFS_2024_09_25 }
+      variant const: -> { Anthropic::AnthropicBeta::PDFS_2024_09_25 }
 
-      variant const: -> { Anthropic::Models::AnthropicBeta::TOKEN_COUNTING_2024_11_01 }
+      variant const: -> { Anthropic::AnthropicBeta::TOKEN_COUNTING_2024_11_01 }
 
-      variant const: -> { Anthropic::Models::AnthropicBeta::TOKEN_EFFICIENT_TOOLS_2025_02_19 }
+      variant const: -> { Anthropic::AnthropicBeta::TOKEN_EFFICIENT_TOOLS_2025_02_19 }
 
-      variant const: -> { Anthropic::Models::AnthropicBeta::OUTPUT_128K_2025_02_19 }
+      variant const: -> { Anthropic::AnthropicBeta::OUTPUT_128K_2025_02_19 }
 
-      # @!parse
-      #   # @return [Array(String, Symbol)]
-      #   def self.variants; end
+      # @!method self.variants
+      #   @return [Array(String, Symbol)]
+
+      define_sorbet_constant!(:Variants) do
+        T.type_alias { T.any(String, Anthropic::AnthropicBeta::TaggedSymbol) }
+      end
 
       # @!group
 

data/lib/anthropic/models/api_error_object.rb

@@ -13,13 +13,9 @@ module Anthropic
       #   @return [Symbol, :api_error]
       required :type, const: :api_error
 
-      # @!parse
-      #   # @param message [String]
-      #   # @param type [Symbol, :api_error]
-      #   #
-      #   def initialize(message:, type: :api_error, **) = super
-
-      # def initialize: (Hash | Anthropic::Internal::Type::BaseModel) -> void
+      # @!method initialize(message:, type: :api_error)
+      #   @param message [String]
+      #   @param type [Symbol, :api_error]
     end
   end
 end

data/lib/anthropic/models/authentication_error.rb

@@ -13,13 +13,9 @@ module Anthropic
       #   @return [Symbol, :authentication_error]
       required :type, const: :authentication_error
 
-      # @!parse
-      #   # @param message [String]
-      #   # @param type [Symbol, :authentication_error]
-      #   #
-      #   def initialize(message:, type: :authentication_error, **) = super
-
-      # def initialize: (Hash | Anthropic::Internal::Type::BaseModel) -> void
+      # @!method initialize(message:, type: :authentication_error)
+      #   @param message [String]
+      #   @param type [Symbol, :authentication_error]
     end
   end
 end

data/lib/anthropic/models/base64_image_source.rb

@@ -10,24 +10,20 @@ module Anthropic
 
       # @!attribute media_type
       #
-      #   @return [Symbol, Anthropic::Models::Base64ImageSource::MediaType]
-      required :media_type, enum: -> { Anthropic::Models::Base64ImageSource::MediaType }
+      #   @return [Symbol, Anthropic::Base64ImageSource::MediaType]
+      required :media_type, enum: -> { Anthropic::Base64ImageSource::MediaType }
 
       # @!attribute type
       #
       #   @return [Symbol, :base64]
       required :type, const: :base64
 
-      # @!parse
-      #   # @param data [String]
-      #   # @param media_type [Symbol, Anthropic::Models::Base64ImageSource::MediaType]
-      #   # @param type [Symbol, :base64]
-      #   #
-      #   def initialize(data:, media_type:, type: :base64, **) = super
+      # @!method initialize(data:, media_type:, type: :base64)
+      #   @param data [String]
+      #   @param media_type [Symbol, Anthropic::Base64ImageSource::MediaType]
+      #   @param type [Symbol, :base64]
 
-      # def initialize: (Hash | Anthropic::Internal::Type::BaseModel) -> void
-
-      # @see Anthropic::Models::Base64ImageSource#media_type
+      # @see Anthropic::Base64ImageSource#media_type
       module MediaType
         extend Anthropic::Internal::Type::Enum
 
@@ -36,11 +32,8 @@ module Anthropic
         IMAGE_GIF = :"image/gif"
         IMAGE_WEBP = :"image/webp"
 
-        finalize!
-
-        # @!parse
-        #   # @return [Array<Symbol>]
-        #   def self.values; end
+        # @!method self.values
+        #   @return [Array<Symbol>]
       end
     end
   end

data/lib/anthropic/models/base64_pdf_source.rb

@@ -18,14 +18,10 @@ module Anthropic
       #   @return [Symbol, :base64]
       required :type, const: :base64
 
-      # @!parse
-      #   # @param data [String]
-      #   # @param media_type [Symbol, :"application/pdf"]
-      #   # @param type [Symbol, :base64]
-      #   #
-      #   def initialize(data:, media_type: :"application/pdf", type: :base64, **) = super
-
-      # def initialize: (Hash | Anthropic::Internal::Type::BaseModel) -> void
+      # @!method initialize(data:, media_type: :"application/pdf", type: :base64)
+      #   @param data [String]
+      #   @param media_type [Symbol, :"application/pdf"]
+      #   @param type [Symbol, :base64]
     end
   end
 end

data/lib/anthropic/models/beta/beta_base64_image_source.rb

@@ -11,24 +11,20 @@ module Anthropic
 
         # @!attribute media_type
         #
-        #   @return [Symbol, Anthropic::Models::Beta::BetaBase64ImageSource::MediaType]
-        required :media_type, enum: -> { Anthropic::Models::Beta::BetaBase64ImageSource::MediaType }
+        #   @return [Symbol, Anthropic::Beta::BetaBase64ImageSource::MediaType]
+        required :media_type, enum: -> { Anthropic::Beta::BetaBase64ImageSource::MediaType }
 
         # @!attribute type
         #
         #   @return [Symbol, :base64]
         required :type, const: :base64
 
-        # @!parse
-        #   # @param data [String]
-        #   # @param media_type [Symbol, Anthropic::Models::Beta::BetaBase64ImageSource::MediaType]
-        #   # @param type [Symbol, :base64]
-        #   #
-        #   def initialize(data:, media_type:, type: :base64, **) = super
+        # @!method initialize(data:, media_type:, type: :base64)
+        #   @param data [String]
+        #   @param media_type [Symbol, Anthropic::Beta::BetaBase64ImageSource::MediaType]
+        #   @param type [Symbol, :base64]
 
-        # def initialize: (Hash | Anthropic::Internal::Type::BaseModel) -> void
-
-        # @see Anthropic::Models::Beta::BetaBase64ImageSource#media_type
+        # @see Anthropic::Beta::BetaBase64ImageSource#media_type
         module MediaType
           extend Anthropic::Internal::Type::Enum
 
@@ -37,11 +33,8 @@ module Anthropic
           IMAGE_GIF = :"image/gif"
           IMAGE_WEBP = :"image/webp"
 
-          finalize!
-
-          # @!parse
-          #   # @return [Array<Symbol>]
-          #   def self.values; end
+          # @!method self.values
+          #   @return [Array<Symbol>]
         end
       end
     end