claude-ruby 0.3.1 → 0.4.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f0914fd554c4c01686d477c44cf2de423f0faad30d673e39a767d0296da1cac4
-  data.tar.gz: fe9e0aa7f262c1af5fe3b0036136739a87f718c040f107c7ad792cb1492fe752
+  metadata.gz: f9a25095b21f1e2a394c2c783902e72db6f8c442fe9f9e47d7196a02956875db
+  data.tar.gz: 5a7900af12bfaf9cd1d59292641dd9e79af25e8ba5aec21c259f93f85a1b8e54
 SHA512:
-  metadata.gz: 5dba3fa917472c533a8d515624346ec40ccd95ebd79e803ad9d0fd536a4aaab3b4bb0c9281d97a9baf0c12c87c7c2849b23dec5a51128e28a2099f47274fbf05
-  data.tar.gz: a0e033b78ced858eae7d8fb196f40ecb04ef540dadcd76e77ba8e2cff177b12642579cbcb0114d27da1336dd26846fa10ebb3cfa2ae0641870e0db33b2c278e0
+  metadata.gz: faa79c579a7180b187afca76c94cf6a972dfb016f32f0355767d69c4716b81611879fdf6bb4ded4e98e6ce136f8ff56abfab66fc1a623b717e847e229e34a944
+  data.tar.gz: d797536e40b66eb4724071fbf92bce78c282aa7c822da4fc2b4685865f0d87061731c7e48f3dcd7fcf82348766f647615164138e1c8a5e115f9d6c1f3cc5bf09

data/CHANGELOG.md

@@ -1,3 +1,11 @@
+## [0.4.1] - 2024-11-06
+- Added the latest Claude models
+- 
+## [0.4.0] - 2024-07-09
+- Refactored model constants into a separate Model module
+- Support custom endpoints
+- Allow timeout to be specified
+
 ## [0.3.1] - 2024-06-29
 - Updated documentation
  

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    claude-ruby (0.3.1)
+    claude-ruby (0.4.1)
       httparty
 
 GEM
@@ -51,6 +51,7 @@ GEM
 PLATFORMS
   arm64-darwin-22
   ruby
+  x86_64-linux
 
 DEPENDENCIES
   claude-ruby!

data/README.md

@@ -132,7 +132,7 @@ response = claude_client.messages(messages, { system: system })
 
 ## Models
 
-If you don't specify a model, then the gem will use the latest version of Clause Sonnet by default, which is currently ```claude-3-5-sonnet-20240620```
+If you don't specify a model, then the gem will use the latest version of Claude Sonnet by default, which is currently ```claude-3-5-sonnet-20240620```
 
 You can use a different model by specifying it as a parameter in the messages call:
 
@@ -143,22 +143,43 @@ response = claude_client.messages(messages, { model: 'claude-3-haiku-20240307' }
 There are some constants defined so you can choose an appropriate model for your use-case and not have to worry about updating it when new Claude models are released:
 
 ```ruby
-Claude::Client::MODEL_CLAUDE_OPUS_LATEST
-Claude::Client::MODEL_CLAUDE_SONNET_LATEST
-Claude::Client::MODEL_CLAUDE_HAIKU_LATEST
-
-Claude::Client::MODEL_CLAUDE_FASTEST
-Claude::Client::MODEL_CLAUDE_CHEAPEST
-Claude::Client::MODEL_CLAUDE_BALANCED
-Claude::Client::MODEL_CLAUDE_SMARTEST
+Claude::Model::CLAUDE_OPUS_LATEST
+Claude::Model::CLAUDE_SONNET_LATEST
+Claude::Model::CLAUDE_HAIKU_LATEST
+
+Claude::Model::CLAUDE_FASTEST
+Claude::Model::CLAUDE_CHEAPEST
+Claude::Model::CLAUDE_BALANCED
+Claude::Model::CLAUDE_SMARTEST
 ````
 
 Example usage:
 
 ```ruby
-response = claude_client.messages(messages, { model: Claude::Client::MODEL_CLAUDE_CHEAPEST })
+response = claude_client.messages(messages, { model: Claude::Model::CLAUDE_CHEAPEST })
 ````
 
+## Timeout
+
+You can optionally set a timeout (integer) which will determine the maximum number of seconds to wait for the API call to complete.
+
+There are two ways to do this:
+
+1. Set a default timeout when instantiating the claude_client \
+This timeout value will be used for all API calls unless overridden.
+
+```ruby
+claude_client = Claude::Client.new(api_key, timeout: 10)
+```
+
+2. Pass in a timeout value as a parameter when calling the messages method. \
+This timeout value will be used only for that specific messages request. 
+
+```ruby
+response = claude_client.messages(messages, { timeout: 10 })
+```
+
+
 ## Parameters
 
 You can pass in any of the following parameters, which will be included in the Anthropic API call: 
@@ -173,17 +194,30 @@ stream
 temperature
 top_p
 top_k
+
+timeout (*)
 ````
+(*) timeout is used for the HTTP request but not passed with the API data
 
 Example:
 
 ```ruby
 response = claude_client.messages(messages, 
-                                  { model: Claude::Client::MODEL_CLAUDE_SMARTEST,
+                                  { model: Claude::Model::CLAUDE_SMARTEST,
                                     max_tokens: 500,
                                     temperature: 0.1 })
 ````
 
+## Custom endpoint
+
+By default claude-ruby will use the latest official Anthropic API endpoint at the time that the gem version is released.
+
+You can optionally optionally override this - e.g. for testing, or for using a beta endpoint.
+
+```ruby
+claude_client = Claude::Client.new(api_key, endpoint: 'your-custom-endpoint')
+```
+
 ## Vision
 
 It's possible to pass an image to the Anthropic API and have Claude describe the image for you.

data/lib/claude/client.rb

@@ -2,36 +2,55 @@ require 'httparty'
 require 'json'
 
 module Claude
-  class Client
-    MODEL_CLAUDE_3_OPUS_20240229 = 'claude-3-opus-20240229'
+  module Model
+    CLAUDE_3_OPUS_20240229 = 'claude-3-opus-20240229'
 
-    MODEL_CLAUDE_3_SONNET_20240229 = 'claude-3-sonnet-20240229'
-    MODEL_CLAUDE_3_5_SONNET_20240620 = 'claude-3-5-sonnet-20240620'
+    CLAUDE_3_5_SONNET_20241022 = 'claude-3-5-sonnet-20241022'
+    CLAUDE_3_5_SONNET_20240620 = 'claude-3-5-sonnet-20240620'
+    CLAUDE_3_SONNET_20240229 = 'claude-3-sonnet-20240229'
 
-    MODEL_CLAUDE_3_HAIKU_20240307 = 'claude-3-haiku-20240307'
+    CLAUDE_3_5_HAIKU_20241022 = 'claude-3-5-haiku-20241022'
+    CLAUDE_3_HAIKU_20240307 = 'claude-3-haiku-20240307'
 
-    MODEL_CLAUDE_OPUS_LATEST = MODEL_CLAUDE_3_OPUS_20240229
-    MODEL_CLAUDE_SONNET_LATEST = MODEL_CLAUDE_3_5_SONNET_20240620
-    MODEL_CLAUDE_HAIKU_LATEST = MODEL_CLAUDE_3_HAIKU_20240307
+    CLAUDE_OPUS_LATEST = CLAUDE_3_OPUS_20240229
+    CLAUDE_SONNET_LATEST = CLAUDE_3_5_SONNET_20241022
+    CLAUDE_HAIKU_LATEST = CLAUDE_3_5_HAIKU_20241022
 
-    MODEL_CLAUDE_FASTEST = MODEL_CLAUDE_HAIKU_LATEST
-    MODEL_CLAUDE_CHEAPEST = MODEL_CLAUDE_HAIKU_LATEST
-    MODEL_CLAUDE_BALANCED = MODEL_CLAUDE_SONNET_LATEST
-    MODEL_CLAUDE_SMARTEST = MODEL_CLAUDE_3_5_SONNET_20240620
+    CLAUDE_FASTEST = CLAUDE_HAIKU_LATEST
+    CLAUDE_CHEAPEST = CLAUDE_3_HAIKU_20240307
+    CLAUDE_BALANCED = CLAUDE_SONNET_LATEST
+    CLAUDE_SMARTEST = CLAUDE_3_5_SONNET_20240620
 
-    MODEL_CLAUDE_DEFAULT = MODEL_CLAUDE_SONNET_LATEST
+    CLAUDE_DEFAULT = CLAUDE_BALANCED
+  end
 
-    def initialize(api_key)
+  class Client
+
+    def initialize(api_key, endpoint: nil, timeout: 60)
       @api_key = api_key
-      @endpoint = 'https://api.anthropic.com/v1'
+      @endpoint = endpoint || anthropic_endpoint
+      @timeout = timeout
+
+      raise(ArgumentError, "api_key is required") if api_key.nil?
+    end
+
+    def version
+      'v1'
+    end
+
+    def anthropic_endpoint
+      "https://api.anthropic.com/#{version}"
+    end
+
+    def messages_endpoint
+      "#{anthropic_endpoint}/messages"
     end
 
     def messages(messages, params = {})
-      model = params[:model] || MODEL_CLAUDE_DEFAULT
+      model = params[:model] || Model::CLAUDE_DEFAULT
       max_tokens = params[:max_tokens] || 4096
       system = params[:system] || "You are a helpful assistant."
-
-      url = "#{@endpoint}/messages"
+      timeout = params[:timeout] || @timeout
 
       data = {
         model: model,
@@ -46,7 +65,7 @@ module Claude
         top_k: params[:top_k],
       }.compact
 
-      post_api(url, data)
+      post_api(messages_endpoint, data, timeout)
     end
 
     def headers
@@ -70,16 +89,30 @@ module Claude
       response['content'][0]['text']
     end
 
+    # for backwards compatibility with version 0.3.1
+    def self.const_missing(const_name)
+      if const_name.to_s.match(/^MODEL_(CLAUDE_.+)$/)
+        new_const_name = $1
+        if Claude::Model.constants.include?(new_const_name.to_sym)
+          warn "[DEPRECATION] `#{const_name}` is deprecated. Please use `Claude::Model::#{new_const_name}` instead."
+          Claude::Model.const_get(new_const_name)
+        else
+          super
+        end
+      else
+        super
+      end
+    end
+
     private
 
-    def post_api(url, data)
-      response = HTTParty.post(url, body: data.to_json, headers: headers)
+    def post_api(url, data, timeout)
+      response = HTTParty.post(url, body: data.to_json, headers: headers, timeout: timeout)
       if response && response['type'] == 'error'
         raise StandardError.new("#{response['error']['type']}: #{response['error']['message']}")
       else
         JSON.parse(response.body)
       end
     end
-
   end
 end

data/lib/claude/ruby/version.rb

@@ -2,6 +2,6 @@
 
 module Claude
   module Ruby
-    VERSION = "0.3.1"
+    VERSION = "0.4.1"
   end
 end

data/sig/claude/client.rbs

@@ -0,0 +1,7 @@
+module Claude
+  class Client
+    @api_key: String
+    @endpoint: String
+    @timeout: Integer
+  end
+end

data/sig/claude/ruby.rbs

@@ -1,6 +1,5 @@
 module Claude
   module Ruby
     VERSION: String
-    # See the writing guide of rbs: https://github.com/ruby/rbs#guides
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: claude-ruby
 version: !ruby/object:Gem::Version
-  version: 0.3.1
+  version: 0.4.1
 platform: ruby
 authors:
 - Web Ventures Ltd
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-06-29 00:00:00.000000000 Z
+date: 2024-11-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: httparty
@@ -24,8 +24,8 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-description: Unofficial ruby SDK for interacting with the Anthropic API, for generating
-  and streaming messages through Claude.
+description: Ruby SDK for interacting with the Anthropic API, for generating and streaming
+  messages through Claude AI.
 email:
 - webven@mailgab.com
 executables: []
@@ -41,6 +41,7 @@ files:
 - lib/claude/client.rb
 - lib/claude/ruby.rb
 - lib/claude/ruby/version.rb
+- sig/claude/client.rbs
 - sig/claude/ruby.rbs
 homepage: https://github.com/webventures/claude-ruby.git
 licenses: