omniai-anthropic 1.2.0 → 1.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 94edb41afdba159c925d228d74af4f953ec1fe0685ee91c96604d9f2e811d285
-  data.tar.gz: d60b97840f9fc260fc9a5d2d7301f0f736985b6e56b5b03d470447142ef3fe9d
+  metadata.gz: ae62c7e30da3f0f0ad3fc0e0c3d6c77fb2401281d5630d0d5447a49321416fcf
+  data.tar.gz: 76ed4ac36be43c81788ec82542c10905a04882f76d48b24903899c571989f7e3
 SHA512:
-  metadata.gz: a367fc5489ec208adbaaec7629a877ddf3f2f1f444cd8a0eac40af0a2f2072ce5d7ea82ab7a6d18de56e0a06ec23e9177b84a5856954c5961ad0967c4c527531
-  data.tar.gz: 3b31c67c47b8c367d0207e8f4bf73f7a498fe78e4848505ec287fb42a9307963fa13e33176f68aa160d7296a3bcfa0e5b2bab6a90a05a264b5ba9092a174336f
+  metadata.gz: 0cf54724f2a28628c73fcefcc71e660bc18af0f615d1dbd1c4455a9f001a71fd9bd35801776318c344833d1b9ce47e599441aa6181de20cc07872b51e056595e
+  data.tar.gz: 887d226ca9d48d16595a1ba7b49fca0d3f07062a65372114558d201f4ccd2cbaebe45b814215f88a7ea06620fbbc7d1e9fff8ec6c5b21208825a5756835c635c

data/README.md

@@ -67,7 +67,7 @@ completion.choice.message.content # 'The capital of Canada is Ottawa.'
 `model` takes an optional string (default is `claude-3-haiku-20240307`):
 
 ```ruby
-completion = client.chat('Provide code for fibonacci', model: OmniAI::Anthropic::Chat::Model::CLAUDE_OPUS)
+completion = client.chat('Provide code for fibonacci', model: OmniAI::Anthropic::Chat::Model::CLAUDE_SONET)
 completion.choice.message.content # 'def fibonacci(n)...end'
 ```
 

data/lib/omniai/anthropic/chat/completion.rb

@@ -5,13 +5,13 @@ module OmniAI
     class Chat
       # A completion returned by the API.
       class Completion < OmniAI::Chat::Completion
-        # @return [Array<OmniAI::Chat::Choice>]
+        # @return [Array<OmniAI::Chat::MessageChoice>]
         def choices
           @choices ||= begin
             role = @data['role']
 
             @data['content'].map do |data, index|
-              OmniAI::Chat::Choice.for(data: {
+              OmniAI::Chat::MessageChoice.for(data: {
                 'index' => index,
                 'message' => { 'role' => role, 'content' => data['text'] },
               })

data/lib/omniai/anthropic/chat.rb

@@ -19,9 +19,10 @@ module OmniAI
         CLAUDE_3_OPUS_20240229 = 'claude-3-opus-20240229'
         CLAUDE_3_HAIKU_20240307 = 'claude-3-haiku-20240307'
         CLAUDE_3_SONET_20240307 = 'claude-3-haiku-20240307'
+        CLAUDE_3_5_SONET_20240620 = 'claude-3-5-sonnet-20240620'
         CLAUDE_OPUS = CLAUDE_3_OPUS_20240229
         CLAUDE_HAIKU = CLAUDE_3_HAIKU_20240307
-        CLAUDE_SONET = CLAUDE_3_SONET_20240307
+        CLAUDE_SONET = CLAUDE_3_5_SONET_20240620
       end
 
       protected

data/lib/omniai/anthropic/client.rb

@@ -24,15 +24,19 @@ module OmniAI
 
       # @param api_key [String] optional - defaults to `OmniAI::Anthropic.config.api_key`
       # @param host [String] optional - defaults to `OmniAI::Anthropic.config.host`
+      # @param version [String] optional - defaults to `OmniAI::Anthropic.config.version`
+      # @param logger [Logger] optional - defaults to `OmniAI::Anthropic.config.logger`
+      # @param timeout [Integer] optional - defaults to `OmniAI::Anthropic.config.timeout`
       def initialize(
         api_key: OmniAI::Anthropic.config.api_key,
+        host: OmniAI::Anthropic.config.host,
         version: OmniAI::Anthropic.config.version,
         logger: OmniAI::Anthropic.config.logger,
-        host: OmniAI::Anthropic.config.host
+        timeout: OmniAI::Anthropic.config.timeout
       )
         raise(ArgumentError, %(ENV['ANTHROPIC_API_KEY'] must be defined or `api_key` must be passed)) if api_key.nil?
 
-        super(api_key:, logger:)
+        super(api_key:, host:, logger:, timeout:)
 
         @host = host
         @version = version
@@ -40,10 +44,10 @@ module OmniAI
 
       # @return [HTTP::Client]
       def connection
-        HTTP
-          .headers('x-api-key': @api_key)
-          .headers('anthropic-version': @version)
-          .persistent('https://api.anthropic.com')
+        @connection ||= super.headers({
+          'x-api-key': @api_key,
+          'anthropic-version': @version,
+        }.compact)
       end
 
       # @raise [OmniAI::Error]

data/lib/omniai/anthropic/config.rb

@@ -2,16 +2,30 @@
 
 module OmniAI
   module Anthropic
-    # Config for the Anthropic `api_key` / `host` / `logger` / `version` / `chat_options`.
+    # Configuration for Anthropic.
     class Config < OmniAI::Config
-      attr_accessor :version, :chat_options
+      DEFAULT_HOST = 'https://api.anthropic.com'
+      DEFAULT_VERSION = '2023-06-01'
 
-      def initialize
-        super
-        @api_key = ENV.fetch('ANTHROPIC_API_KEY', nil)
-        @host = ENV.fetch('ANTHROPIC_HOST', 'https://api.anthropic.com')
-        @version = ENV.fetch('ANTHROPIC_VERSION', '2023-06-01')
-        @chat_options = { max_tokens: 4096 }
+      # @!attribute [rw] version
+      #   @return [String, nil] passed as `anthropic-version` if specified
+      attr_accessor :version
+
+      # @param api_key [String, nil] optional - defaults to `ENV['ANTHROPIC_API_KEY']`
+      # @param host [String, nil] optional - defaults to `ENV['ANTHROPIC_HOST'] w/ fallback to `DEFAULT_HOST`
+      # @param version [String, nil] optional - defaults to `ENV['ANTHROPIC_VERSION'] w/ fallback to `DEFAULT_VERSION`
+      # @param logger [Logger, nil] optional - defaults to
+      # @param timeout [Integer, Hash, nil] optional
+      def initialize(
+        api_key: ENV.fetch('ANTHROPIC_API_KEY', nil),
+        host: ENV.fetch('ANTHROPIC_HOST', DEFAULT_HOST),
+        version: ENV.fetch('ANTHROPIC_VERSION', DEFAULT_VERSION),
+        logger: nil,
+        timeout: nil
+      )
+        super(api_key:, host:, logger:, timeout:)
+        @version = version
+        @chat_options[:max_tokens] = 4096
       end
     end
   end

data/lib/omniai/anthropic/version.rb

@@ -2,6 +2,6 @@
 
 module OmniAI
   module Anthropic
-    VERSION = '1.2.0'
+    VERSION = '1.3.0'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: omniai-anthropic
 version: !ruby/object:Gem::Version
-  version: 1.2.0
+  version: 1.3.0
 platform: ruby
 authors:
 - Kevin Sylvestre
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-06-20 00:00:00.000000000 Z
+date: 2024-06-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser