anthropic-rb 0.4.0 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b5e066105f3157621c27e2f974ed43bde9f14e6e55187d1d7a4f095b79a5a1ba
-  data.tar.gz: f32d479f7ec06c29a0347f69084a4378a64d69233d78ed280f5a1c924e15d708
+  metadata.gz: 47751095db1f8f6ad2e30c5161f0137dd8eba0838a2c0a2ccce8cb51aefa278e
+  data.tar.gz: 5de9082d8b07c9f19ec20eeb4dee0525f7797585dbcb208563b85ef040bd2f0f
 SHA512:
-  metadata.gz: 90040c86504a3dfd28e0093d1b8525252387f4e978271908aa158d1cffe10a3dd5766aadc243bc31f45ca6dd38ecf1259d009b2da9f2f78670fcc114c76574a3
-  data.tar.gz: 8ca46c62fe204b3b2403cce1267b6661f43c52159c8d31877b9d2522be604427996eedcbac67ba22a1b6df703259ededd98ea2b86d49755d7aa057835458321c
+  metadata.gz: 4d5be01a84c735f0bf815cba09cd591da94085cf65cf5387c839eba1f3b0c52ed1af63f26b7044d0d8a1c968bfb10bb7efe030f560e378c4ca7b1e70a5e37d44
+  data.tar.gz: c8dd64d95f97c348839e12cc52bbdbdbbaccab310eeff232db99120bcc74584c31781296686d3a9fac6077f94492311d6c49c22b9f330f8754038b8ff958f180

data/.reek.yml

@@ -4,10 +4,22 @@ detectors:
       - Anthropic#self.api_key=
   TooManyStatements:
     exclude:
+      - 'Anthropic::Api::Concerns::Validatable#schema'
+      - 'Anthropic::BaseApi#beta_config'
+      - 'Anthropic::BaseApi#version_config'
+      - 'Anthropic::Bootstrapper#self.load_betas'
+      - 'Anthropic::Bootstrapper#self.load_versions'
       - 'Anthropic::Client#self.post'
   BooleanParameter:
     exclude:
       - 'Anthropic::Messages#initialize'
   DuplicateMethodCall:
     exclude:
+      - 'Anthropic::BaseApi#version_config'
       - 'Anthropic::Messages#create'
+  InstanceVariableAssumption:
+    exclude:
+      - 'Anthropic::BaseApi'
+  NestedIterators:
+    exclude:
+      - 'Anthropic::Bootstrapper#self.load_versions'

data/CHANGELOG.md

@@ -2,6 +2,15 @@
 
 ## [Unreleased]
 
+## [0.5.0] - 2024-04-22
+
+### Updated
+
+- Refactored project for maintainability.
+
+### Breaking Changes
+- You must now pass the beta ID when enabling a beta feature. The only current beta is for Tools (id: tools-2024-04-04). Previously, you would pass `true` to enable the beta.
+
 ## [0.4.0] - 2024-04-20
 
 ### Added
@@ -57,7 +66,8 @@
 
 - Initial release
 
-[Unreleased]: https://github.com/dickdavis/anthropic-rb/compare/v0.4.0...HEAD
+[Unreleased]: https://github.com/dickdavis/anthropic-rb/compare/v0.5.0...HEAD
+[0.5.0]: https://github.com/dickdavis/anthropic-rb/compare/v0.4.0...v0.5.0
 [0.4.0]: https://github.com/dickdavis/anthropic-rb/compare/v0.3.0...v0.4.0
 [0.3.0]: https://github.com/dickdavis/anthropic-rb/compare/v0.2.5...v0.3.0
 [0.2.5]: https://github.com/dickdavis/anthropic-rb/compare/v0.2.3...v0.2.5

data/README.md

@@ -62,7 +62,7 @@ end
 # { type: 'message_stop' }
 ```
 
-You can also experiment with the new tools beta by passing the `beta` flag when calling the API. This will ensure each request includes the correct beta header.
+You can also experiment with the new tools beta by passing the `beta` name when calling the API. This will ensure each request includes the correct beta header and validate properly.
 
 ```ruby
 tools = [
@@ -79,7 +79,7 @@ tools = [
   }
 ]
 
-Anthropic.messages(beta: true).create(
+Anthropic.messages(beta: 'tools-2024-04-04').create(
   model: 'claude-3-opus-20240229',
   max_tokens: 200,
   tools:,

data/anthropic-rb.gemspec

@@ -24,7 +24,7 @@ Gem::Specification.new do |spec|
       (File.expand_path(f) == __FILE__) ||
         f.start_with?(*%w[bin/ spec/ .git .github/ Gemfile])
     end
-  end
+  end + Dir.glob('lib/**/*.json')
   spec.bindir = 'exe'
   spec.executables = spec.files.grep(%r{\Aexe/}) { |f| File.basename(f) }
   spec.require_paths = ['lib']

data/lib/anthropic/api/base_api.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+require_relative 'concerns/requestable'
+require_relative 'concerns/validatable'
+
+module Anthropic
+  ##
+  # Provides a base class for APIs
+  class BaseApi
+    include Anthropic::Api::Concerns::Requestable
+    include Anthropic::Api::Concerns::Validatable
+
+    def initialize(beta: nil)
+      @beta = beta
+    end
+
+    private
+
+    attr_reader :beta
+
+    def api
+      self.class.name.split('::').last.downcase
+    end
+
+    def version_config
+      return @version_config if defined?(@version_config)
+
+      @version_config ||= catch(:version_found) do
+        found_config = Anthropic.versions[api.to_sym].find { |config| config['version'] == Anthropic.api_version }
+        unless found_config
+          raise Anthropic::Errors::UnsupportedApiVersionError, "Unsupported API version: #{Anthropic.api_version}"
+        end
+
+        throw :version_found, found_config
+      end
+    end
+
+    def beta_config
+      return @beta_config if defined?(@beta_config)
+
+      @beta_config = catch(:beta_found) do
+        found_config = Anthropic.betas.find { |config| config['id'] == beta }
+        raise Anthropic::Errors::UnsupportedBetaError, "#{beta} not supported" unless found_config
+
+        throw :beta_found, found_config
+      end
+    end
+
+    def beta_loaded?(name)
+      return false unless beta
+
+      beta_config['id'] == name
+    end
+  end
+end

data/lib/anthropic/api/completions.rb

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+module Anthropic
+  module Api
+    ##
+    # Provides bindings for the Anthropic completions API
+    class Completions < BaseApi
+      def create(**params, &)
+        validate!(params)
+        return post(params) unless params[:stream]
+
+        post_as_stream(params, &)
+      end
+    end
+  end
+end

data/lib/anthropic/api/concerns/requestable.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+module Anthropic
+  module Api
+    module Concerns
+      ##
+      # Provides helpers for sending API requests
+      module Requestable
+        def post(params)
+          Anthropic::Client.post(uri, params, additional_headers)
+        end
+
+        def post_as_stream(params, &)
+          Anthropic::Client.post_as_stream(uri, params, additional_headers, &)
+        end
+
+        def uri
+          "#{Anthropic.api_host}#{version_config['endpoint']}"
+        end
+
+        def additional_headers
+          {}.merge(beta_headers)
+        end
+
+        def beta_headers
+          return {} unless beta
+
+          header = beta_config['header']
+          raise Anthropic::Errors::InvalidBetaConfigurationError, "Missing header: #{beta}" unless header
+
+          header
+        end
+      end
+    end
+  end
+end

data/lib/anthropic/api/concerns/validatable.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+module Anthropic
+  module Api
+    module Concerns
+      ##
+      # Provides helpers for validating params against the API schema
+      module Validatable
+        def validate!(params)
+          JSON::Validator.validate!(schema, params)
+        rescue JSON::Schema::ValidationError => error
+          raise Anthropic::Errors::SchemaValidationError, error.message
+        end
+
+        def schema
+          api_schema = version_config['schema']
+
+          unless api_schema
+            raise Anthropic::Errors::MissingSchemaError, "Missing schema for API version: #{Anthropic.api_version}"
+          end
+
+          if beta
+            beta_schema = beta_config['schema']
+            raise Anthropic::Errors::InvalidBetaConfigurationError, "Missing beta schema: #{beta}" unless beta_schema
+
+            api_schema['properties'].merge!(beta_schema)
+          end
+
+          api_schema
+        end
+      end
+    end
+  end
+end

data/lib/anthropic/api/messages.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module Anthropic
+  module Api
+    ##
+    # Provides bindings for the Anthropic messages API
+    class Messages < BaseApi
+      def create(**params, &)
+        streaming = params[:stream]
+        if streaming && beta_loaded?('tools-2024-04-04')
+          raise Anthropic::Errors::UnsupportedBetaUseError, 'Tool use is not yet supported in streaming mode'
+        end
+
+        validate!(params)
+        return post(params) unless streaming
+
+        post_as_stream(params, &)
+      end
+    end
+  end
+end

data/lib/anthropic/betas/tools-2024-04-04.json

@@ -0,0 +1,26 @@
+{
+  "id": "tools-2024-04-04",
+  "name": "Tools",
+  "description": "Equips Claude with custom tools for interacting with clients for a wide variety of tasks.",
+  "documentation": "https://docs.anthropic.com/claude/docs/tool-use",
+  "header" : { "anthropic-beta": "tools-2024-04-04"  },
+  "schema": {
+    "tools": {
+      "type": "array",
+      "items": {
+        "type": "object",
+        "properties": {
+          "name": {
+            "type": "string"
+          },
+          "description": {
+            "type": "string"
+          },
+          "input_schema": {
+            "type": "object"
+          }
+        }
+      }
+    }
+  }
+}

data/lib/anthropic/bootstrapper.rb

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+require 'httpx'
+
+module Anthropic
+  ##
+  # Provides methods for bootstrapping the Anthropic gem
+  module Bootstrapper
+    def self.load_betas
+      current_dir = File.dirname(__FILE__)
+      directory_path = File.join(current_dir, 'betas')
+
+      raise "Directory not found: #{directory_path}" unless Dir.exist?(directory_path)
+
+      file_paths = Dir.glob(File.join(directory_path, '*.json'))
+
+      file_paths.map do |file_path|
+        JSON.parse(File.read(file_path))
+      end
+    end
+
+    # rubocop:disable Metrics/MethodLength, Metrics/AbcSize
+    def self.load_versions
+      current_dir = File.dirname(__FILE__)
+      directory_path = File.join(current_dir, 'versions')
+
+      raise "Directory not found: #{directory_path}" unless Dir.exist?(directory_path)
+
+      versions = {}
+
+      Dir.glob(File.join(directory_path, '*')).each do |subdirectory_path|
+        next unless File.directory?(subdirectory_path)
+
+        subdirectory_name = File.basename(subdirectory_path)
+        file_paths = Dir.glob(File.join(subdirectory_path, '*.json'))
+
+        versions[subdirectory_name.to_sym] = file_paths.map do |file_path|
+          JSON.parse(File.read(file_path))
+        end
+      end
+
+      versions
+    end
+    # rubocop:enable Metrics/MethodLength, Metrics/AbcSize
+  end
+end

data/lib/anthropic/client.rb

@@ -3,31 +3,6 @@
 require 'httpx'
 
 module Anthropic
-  ##
-  # Error when the request is malformed.
-  class BadRequestError < StandardError; end
-  ##
-  # Error when the API key is invalid.
-  class AuthenticationError < StandardError; end
-  ##
-  # Error when the account does not have permission for the operation.
-  class PermissionDeniedError < StandardError; end
-  ##
-  # Error when the resource is not found.
-  class NotFoundError < StandardError; end
-  ##
-  # Error when the resource already exists.
-  class ConflictError < StandardError; end
-  ##
-  # Error when the resource cannot be processed.
-  class UnprocessableEntityError < StandardError; end
-  ##
-  # Error when the request exceeds the rate limit.
-  class RateLimitError < StandardError; end
-  ##
-  # Error when the server experienced an internal error.
-  class InternalServerError < StandardError; end
-
   ##
   # Provides a client for sending HTTP requests.
   class Client
@@ -47,21 +22,21 @@ module Anthropic
       when 200
         response_body
       when 400
-        raise Anthropic::BadRequestError, response_body
+        raise Anthropic::Errors::BadRequestError, response_body
       when 401
-        raise Anthropic::AuthenticationError, response_body
+        raise Anthropic::Errors::AuthenticationError, response_body
       when 403
-        raise Anthropic::PermissionDeniedError, response_body
+        raise Anthropic::Errors::PermissionDeniedError, response_body
       when 404
-        raise Anthropic::NotFoundError, response_body
+        raise Anthropic::Errors::NotFoundError, response_body
       when 409
-        raise Anthropic::ConflictError, response_body
+        raise Anthropic::Errors::ConflictError, response_body
       when 422
-        raise Anthropic::UnprocessableEntityError, response_body
+        raise Anthropic::Errors::UnprocessableEntityError, response_body
       when 429
-        raise Anthropic::RateLimitError, response_body
+        raise Anthropic::Errors::RateLimitError, response_body
       when 500
-        raise Anthropic::InternalServerError, response_body
+        raise Anthropic::Errors::InternalServerError, response_body
       end
     end
 
@@ -86,21 +61,21 @@ module Anthropic
     rescue HTTPX::HTTPError => error
       case error.response.status
       when 400
-        raise Anthropic::BadRequestError
+        raise Anthropic::Errors::BadRequestError
       when 401
-        raise Anthropic::AuthenticationError
+        raise Anthropic::Errors::AuthenticationError
       when 403
-        raise Anthropic::PermissionDeniedError
+        raise Anthropic::Errors::PermissionDeniedError
       when 404
-        raise Anthropic::NotFoundError
+        raise Anthropic::Errors::NotFoundError
       when 409
-        raise Anthropic::ConflictError
+        raise Anthropic::Errors::ConflictError
       when 422
-        raise Anthropic::UnprocessableEntityError
+        raise Anthropic::Errors::UnprocessableEntityError
       when 429
-        raise Anthropic::RateLimitError
+        raise Anthropic::Errors::RateLimitError
       when 500
-        raise Anthropic::InternalServerError
+        raise Anthropic::Errors::InternalServerError
       end
     end
     # rubocop:enable Metrics/AbcSize, Metrics/MethodLength, Metrics/CyclomaticComplexity

data/lib/anthropic/errors/betas.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+module Anthropic
+  module Errors
+    # Error for when the provided beta is not supported
+    class UnsupportedBetaError < StandardError; end
+
+    # Error for when a beta feature is configured incorrectly
+    class InvalidBetaConfigurationError < StandardError; end
+
+    # Error for when a beta feature is not used correctly
+    class UnsupportedBetaUseError < StandardError; end
+  end
+end

data/lib/anthropic/errors/requests.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+module Anthropic
+  module Errors
+    ##
+    # Error when the request is malformed.
+    class BadRequestError < StandardError; end
+    ##
+    # Error when the API key is invalid.
+    class AuthenticationError < StandardError; end
+    ##
+    # Error when the account does not have permission for the operation.
+    class PermissionDeniedError < StandardError; end
+    ##
+    # Error when the resource is not found.
+    class NotFoundError < StandardError; end
+    ##
+    # Error when the resource already exists.
+    class ConflictError < StandardError; end
+    ##
+    # Error when the resource cannot be processed.
+    class UnprocessableEntityError < StandardError; end
+    ##
+    # Error when the request exceeds the rate limit.
+    class RateLimitError < StandardError; end
+    ##
+    # Error when the server experienced an internal error.
+    class InternalServerError < StandardError; end
+  end
+end

data/lib/anthropic/errors/versions.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+module Anthropic
+  module Errors
+    # Error for when the API version is not supported.
+    class UnsupportedApiVersionError < StandardError; end
+
+    # Error for when API version is missing a schema.
+    class MissingSchemaError < StandardError; end
+
+    # Error for when the provided params do not match the API schema
+    class SchemaValidationError < StandardError; end
+  end
+end

data/lib/anthropic/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Anthropic
-  VERSION = '0.4.0'
+  VERSION = '0.5.0'
 end

data/lib/anthropic/versions/completions/2023-06-01.json

@@ -0,0 +1,45 @@
+{
+  "version": "2023-06-01",
+  "endpoint": "/v1/complete",
+  "schema": {
+    "type": "object",
+    "required": [
+      "model",
+      "prompt",
+      "max_tokens_to_sample"
+    ],
+    "properties": {
+      "model": {
+        "type": "string"
+      },
+      "prompt": {
+        "type": "string"
+      },
+      "max_tokens_to_sample": {
+        "type": "integer"
+      },
+      "stop_sequences": {
+        "type": "array",
+        "items": {
+          "type": "string"
+        }
+      },
+      "temperature": {
+        "type": "number"
+      },
+      "top_k": {
+        "type": "integer"
+      },
+      "top_p": {
+        "type": "number"
+      },
+      "metadata": {
+        "type": "object"
+      },
+      "stream": {
+        "type": "boolean"
+      }
+    },
+    "additionalProperties": false
+  }
+}

data/lib/anthropic/versions/messages/2023-06-01.json

@@ -0,0 +1,48 @@
+{
+  "version": "2023-06-01",
+  "endpoint": "/v1/messages",
+  "schema": {
+    "type": "object",
+    "required": [
+      "model",
+      "messages",
+      "max_tokens"
+    ],
+    "properties": {
+      "model": {
+        "type": "string"
+      },
+      "messages": {
+        "type": "array"
+      },
+      "max_tokens": {
+        "type": "integer"
+      },
+      "system": {
+        "type": "string"
+      },
+      "stop_sequences": {
+        "type": "array",
+        "items": {
+          "type": "string"
+        }
+      },
+      "temperature": {
+        "type": "number"
+      },
+      "top_k": {
+        "type": "integer"
+      },
+      "top_p": {
+        "type": "number"
+      },
+      "metadata": {
+        "type": "object"
+      },
+      "stream": {
+        "type": "boolean"
+      }
+    },
+    "additionalProperties": false
+  }
+}

data/lib/anthropic.rb

@@ -4,24 +4,20 @@ require 'httpx'
 require 'json'
 require 'json-schema'
 
-require_relative 'anthropic/client'
-require_relative 'anthropic/completions'
-require_relative 'anthropic/messages'
-require_relative 'anthropic/version'
+Dir[File.join(__dir__, 'anthropic/', '**', '*.rb')].each { |file| require_relative file }
 
 ##
 # Namespace for anthropic-rb gem
 module Anthropic
-  ##
-  # Default error class
-  class Error < StandardError; end
-
   def self.setup
     yield self
+    @betas = Bootstrapper.load_betas
+    @versions = Bootstrapper.load_versions
   end
 
   def self.reset
     @api_key = nil
+    @api_host = nil
     @api_version = nil
   end
 
@@ -33,6 +29,14 @@ module Anthropic
     @api_key = api_key
   end
 
+  def self.api_host
+    @api_host || ENV.fetch('ANTHROPIC_API_HOST', 'https://api.anthropic.com')
+  end
+
+  def self.api_host=(api_host = nil)
+    @api_host = api_host
+  end
+
   def self.api_version
     @api_version || ENV.fetch('ANTHROPIC_API_VERSION', '2023-06-01')
   end
@@ -41,11 +45,19 @@ module Anthropic
     @api_version = api_version
   end
 
+  def self.betas
+    @betas
+  end
+
+  def self.versions
+    @versions
+  end
+
   def self.completions
-    Completions.new
+    Anthropic::Api::Completions.new
   end
 
   def self.messages(...)
-    Messages.new(...)
+    Anthropic::Api::Messages.new(...)
   end
 end

data/sanity_check.rb

@@ -0,0 +1,59 @@
+# frozen_string_literal: true
+
+require 'anthropic'
+
+Anthropic.setup do |config|
+  config.api_key = ENV.fetch('ANTHROPIC_API_KEY')
+end
+
+puts "\nTesting non-streaming messages API"
+puts Anthropic.messages.create(
+  model: 'claude-2.1',
+  max_tokens: 200,
+  messages: [{ role: 'user', content: 'Yo what up?' }]
+)
+
+puts "\nTesting streaming messages API"
+Anthropic.messages.create(
+  model: 'claude-2.1',
+  max_tokens: 200,
+  messages: [{ role: 'user', content: 'Yo what up?' }],
+  stream: true
+) { |event| puts event }
+
+puts "\nTesting tools beta"
+tools = [
+  {
+    name: 'get_weather',
+    description: 'Get the current weather in a given location',
+    input_schema: {
+      type: 'object',
+      properties: {
+        location: { type: 'string' }
+      },
+      required: ['location']
+    }
+  }
+]
+
+puts Anthropic.messages(beta: 'tools-2024-04-04').create(
+  model: 'claude-3-opus-20240229',
+  max_tokens: 200,
+  tools:,
+  messages: [{ role: 'user', content: 'What is the weather like in Nashville?' }]
+)
+
+puts "\nTesting completions API"
+puts Anthropic.completions.create(
+  model: 'claude-2',
+  max_tokens_to_sample: 200,
+  prompt: 'Human: Yo what up?\n\nAssistant:'
+)
+
+puts "\nTesting streaming completions API"
+Anthropic.completions.create(
+  model: 'claude-2',
+  max_tokens_to_sample: 200,
+  prompt: 'Human: Yo what up?\n\nAssistant:',
+  stream: true
+) { |event| puts event }

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: anthropic-rb
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.5.0
 platform: ruby
 authors:
 - Dick Davis
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-04-20 00:00:00.000000000 Z
+date: 2024-04-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: httpx
@@ -56,10 +56,21 @@ files:
 - Rakefile
 - anthropic-rb.gemspec
 - lib/anthropic.rb
+- lib/anthropic/api/base_api.rb
+- lib/anthropic/api/completions.rb
+- lib/anthropic/api/concerns/requestable.rb
+- lib/anthropic/api/concerns/validatable.rb
+- lib/anthropic/api/messages.rb
+- lib/anthropic/betas/tools-2024-04-04.json
+- lib/anthropic/bootstrapper.rb
 - lib/anthropic/client.rb
-- lib/anthropic/completions.rb
-- lib/anthropic/messages.rb
+- lib/anthropic/errors/betas.rb
+- lib/anthropic/errors/requests.rb
+- lib/anthropic/errors/versions.rb
 - lib/anthropic/version.rb
+- lib/anthropic/versions/completions/2023-06-01.json
+- lib/anthropic/versions/messages/2023-06-01.json
+- sanity_check.rb
 - sig/anthropic/rb.rbs
 homepage: https://github.com/dickdavis/anthropic-rb
 licenses:

data/lib/anthropic/completions.rb

@@ -1,54 +0,0 @@
-# frozen_string_literal: true
-
-module Anthropic
-  ##
-  # Provides bindings for the Anthropic completions API
-  class Completions
-    # Error for when the API version is not supported.
-    class UnsupportedApiVersionError < StandardError; end
-
-    V1_SCHEMA = {
-      type: 'object',
-      required: %w[model prompt max_tokens_to_sample],
-      properties: {
-        model: { type: 'string' },
-        prompt: { type: 'string' },
-        max_tokens_to_sample: { type: 'integer' },
-        stop_sequences: { type: 'array', items: { type: 'string' } },
-        temperature: { type: 'number' },
-        top_k: { type: 'integer' },
-        top_p: { type: 'number' },
-        metadata: { type: 'object' },
-        stream: { type: 'boolean' }
-      },
-      additionalProperties: false
-    }.freeze
-
-    def initialize
-      @endpoint = 'https://api.anthropic.com/v1/complete'
-    end
-
-    def create(**params, &)
-      JSON::Validator.validate!(schema_for_api_version, params)
-      return Anthropic::Client.post(endpoint, params) unless params[:stream]
-
-      Anthropic::Client.post_as_stream(endpoint, params, &)
-    rescue JSON::Schema::ValidationError => error
-      raise ArgumentError, error.message
-    end
-
-    private
-
-    attr_reader :endpoint
-
-    def schema_for_api_version
-      api_version = Anthropic.api_version
-      case api_version
-      when '2023-06-01'
-        V1_SCHEMA
-      else
-        raise UnsupportedApiVersionError, "Unsupported API version: #{api_version}"
-      end
-    end
-  end
-end

data/lib/anthropic/messages.rb

@@ -1,76 +0,0 @@
-# frozen_string_literal: true
-
-module Anthropic
-  ##
-  # Provides bindings for the Anthropic messages API
-  class Messages
-    # Error for when the API version is not supported.
-    class UnsupportedApiVersionError < StandardError; end
-
-    # Error for when a beta feature is not used correctly
-    class UnsupportedBetaOptionError < StandardError; end
-
-    ENDPOINT = 'https://api.anthropic.com/v1/messages'
-    V1_SCHEMA = {
-      type: 'object',
-      required: %w[model messages max_tokens],
-      properties: {
-        model: { type: 'string' },
-        messages: { type: 'array' },
-        max_tokens: { type: 'integer' },
-        system: { type: 'string' },
-        stop_sequences: { type: 'array', items: { type: 'string' } },
-        temperature: { type: 'number' },
-        tools: {
-          type: 'array',
-          items: {
-            name: { type: 'string' },
-            description: { type: 'string' },
-            input_schema: { type: 'object' }
-          }
-        },
-        top_k: { type: 'integer' },
-        top_p: { type: 'number' },
-        metadata: { type: 'object' },
-        stream: { type: 'boolean' }
-      },
-      additionalProperties: false
-    }.freeze
-
-    def initialize(beta: false)
-      @beta = beta
-    end
-
-    def create(**params, &)
-      raise UnsupportedBetaOptionError, 'Tool use is not yet supported in streaming mode' if params[:stream] && beta
-
-      JSON::Validator.validate!(schema_for_api_version, params)
-
-      return Anthropic::Client.post(ENDPOINT, params, additional_headers) unless params[:stream]
-
-      Anthropic::Client.post_as_stream(ENDPOINT, params, additional_headers, &)
-    rescue JSON::Schema::ValidationError => error
-      raise ArgumentError, error.message
-    end
-
-    private
-
-    attr_reader :beta
-
-    def schema_for_api_version
-      api_version = Anthropic.api_version
-      case api_version
-      when '2023-06-01'
-        V1_SCHEMA
-      else
-        raise UnsupportedApiVersionError, "Unsupported API version: #{api_version}"
-      end
-    end
-
-    def additional_headers
-      return {} unless beta
-
-      { 'anthropic-beta' => 'tools-2024-04-04' }
-    end
-  end
-end