omniai-anthropic 1.8.0 → 1.8.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 40dd84f42676e7000caed3c06c942ab1bca0d3d10763c279e084e5d378930946
-  data.tar.gz: 2233860b1769b297af92fe5c1156fb08ffb61a88452ba39d8ad79e131ae861be
+  metadata.gz: 139e028a0693e4dfdb45dd1ef6ff5fe4706cf34801eb7154671968f3460cb786
+  data.tar.gz: e50f5502ff0ff81c37bd42ce23dafce2e70843cae7a414061cbbd38edd8abeea
 SHA512:
-  metadata.gz: d4c99af8092f169086859edf43d6a9164d410caa0166a43fd2ff0b76156fb28cedca8fbbd84251e3405406336a371a068fd00968d93dc723050950dae4da7c80
-  data.tar.gz: f022585d04e3983ed15ec31cfc3b125bc5e6396be2d417a9cc10af0e827e32a0a19c324062987a83be5cfd306c4659d5e702ae7c931511189518c546b4e31489
+  metadata.gz: 751b0f552007b98d17d03820150afacab7dc98044f1d558140b5f460d58ffe091b757f3cf6e3e559e894ee96e0c7a10928a3f86cc653e4bfd9b886b44d3311d4
+  data.tar.gz: cccd3f87fffdab99a9e5d0369a293439c953a8d3db5bfa1347b2274b5efe94c9faf17a34fcf1bfc0e841d3e9db5decca842fd5e8a0e77c940344fa3cd266ac4b

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

@@ -0,0 +1,49 @@
+# frozen_string_literal: true
+
+module OmniAI
+  module Anthropic
+    class Chat
+      # Overrides message serialize / deserialize.
+      module MessageSerializer
+        # @param message [OmniAI::Chat::Message]
+        # @param context [OmniAI::Context]
+        # @return [Hash]
+        def self.serialize(message, context:)
+          role = message.role
+          parts = arrayify(message.content) + arrayify(message.tool_call_list)
+          content = parts.map do |part|
+            case part
+            when String then { type: 'text', text: part }
+            else part.serialize(context:)
+            end
+          end
+
+          { role:, content: }
+        end
+
+        # @param data [Hash]
+        # @param context [OmniAI::Context]
+        # @return [OmniAI::Chat::Message]
+        def self.deserialize(data, context:)
+          role = data['role']
+          parts = arrayify(data['content']).map do |content|
+            ContentSerializer.deserialize(content, context:)
+          end
+
+          tool_call_list = parts.select { |part| part.is_a?(OmniAI::Chat::ToolCall) }
+          content = parts.reject { |part| part.is_a?(OmniAI::Chat::ToolCall) }
+
+          OmniAI::Chat::Message.new(content:, role:, tool_call_list:)
+        end
+
+        # @param content [Object]
+        # @return [Array<Object>]
+        def self.arrayify(content)
+          return [] if content.nil?
+
+          content.is_a?(Array) ? content : [content]
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module OmniAI
+  module Anthropic
+    class Chat
+      # Overrides tool-call response serialize / deserialize.
+      module ToolCallResultSerializer
+        # @param tool_call_result [OmniAI::Chat::ToolCallResult]
+        # @return [Hash]
+        def self.serialize(tool_call_result, *)
+          {
+            type: 'tool_result',
+            tool_use_id: tool_call_result.tool_call_id,
+            content: tool_call_result.content,
+          }
+        end
+
+        # @param data [Hash]
+        # @return [OmniAI::Chat::ToolCallResult]
+        def self.deserialize(data, *)
+          tool_call_id = data['tool_use_id']
+          content = data['content']
+
+          OmniAI::Chat::ToolCallResult.new(content:, tool_call_id:)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+module OmniAI
+  module Anthropic
+    class Chat
+      # Overrides tool serialize / deserialize.
+      module ToolSerializer
+        # @param tool [OmniAI::Tool]
+        # @return [Hash]
+        def self.serialize(tool, *)
+          {
+            name: tool.name,
+            description: tool.description,
+            input_schema: tool.parameters.is_a?(Tool::Parameters) ? tool.parameters.serialize : tool.parameters,
+          }.compact
+        end
+      end
+    end
+  end
+end

data/lib/omniai/anthropic/chat.rb

@@ -16,19 +16,29 @@ module OmniAI
         CLAUDE_INSTANT_1_0 = 'claude-instant-1.2'
         CLAUDE_2_0 = 'claude-2.0'
         CLAUDE_2_1 = 'claude-2.1'
-        CLAUDE_3_OPUS_20240229 = 'claude-3-opus-20240229'
+
         CLAUDE_3_HAIKU_20240307 = 'claude-3-haiku-20240307'
+        CLAUDE_3_OPUS_20240229 = 'claude-3-opus-20240229'
+        CLAUDE_3_SONNET_20240209 = 'claude-3-sonnet-20240229'
         CLAUDE_3_SONNET_20240307 = 'claude-3-sonnet-20240307'
         CLAUDE_3_5_SONNET_20240620 = 'claude-3-5-sonnet-20240620'
-        CLAUDE_OPUS = CLAUDE_3_OPUS_20240229
-        CLAUDE_HAIKU = CLAUDE_3_HAIKU_20240307
-        CLAUDE_SONNET = CLAUDE_3_5_SONNET_20240620
+        CLAUDE_3_5_SONNET_20241022 = 'claude-3-5-sonnet-20241022'
+
+        CLAUDE_3_HAIKU_LATEST = CLAUDE_3_HAIKU_20240307
+        CLAUDE_3_OPUS_LATEST = 'claude-3-opus-latest'
+        CLAUDE_3_5_SONNET_LATEST = 'claude-3-5-sonnet-latest'
+
+        CLAUDE_HAIKU = CLAUDE_3_HAIKU_LATEST
+        CLAUDE_OPUS = CLAUDE_3_OPUS_LATEST
+        CLAUDE_SONNET = CLAUDE_3_5_SONNET_LATEST
       end
 
       DEFAULT_MODEL = Model::CLAUDE_SONNET
 
       # @return [Context]
       CONTEXT = Context.build do |context|
+        context.serializers[:tool] = ToolSerializer.method(:serialize)
+
         context.serializers[:file] = MediaSerializer.method(:serialize)
         context.serializers[:url] = MediaSerializer.method(:serialize)
 
@@ -38,9 +48,15 @@ module OmniAI
         context.serializers[:tool_call] = ToolCallSerializer.method(:serialize)
         context.deserializers[:tool_call] = ToolCallSerializer.method(:deserialize)
 
+        context.serializers[:tool_call_result] = ToolCallResultSerializer.method(:serialize)
+        context.deserializers[:tool_call_result] = ToolCallResultSerializer.method(:deserialize)
+
         context.serializers[:function] = FunctionSerializer.method(:serialize)
         context.deserializers[:function] = FunctionSerializer.method(:deserialize)
 
+        context.serializers[:message] = MessageSerializer.method(:serialize)
+        context.deserializers[:message] = MessageSerializer.method(:deserialize)
+
         context.deserializers[:content] = ContentSerializer.method(:deserialize)
         context.deserializers[:payload] = PayloadSerializer.method(:deserialize)
       end
@@ -83,17 +99,20 @@ module OmniAI
         CONTEXT
       end
 
+      # @return [Array<Message>]
+      def build_tool_call_messages(tool_call_list)
+        content = tool_call_list.map do |tool_call|
+          ToolCallResult.new(tool_call_id: tool_call.id, content: execute_tool_call(tool_call))
+        end
+
+        [Message.new(role: OmniAI::Chat::Role::USER, content:)]
+      end
+
       private
 
       # @return [Array<Hash>, nil]
       def tools_payload
-        @tools&.map do |tool|
-          {
-            name: tool.name,
-            description: tool.description,
-            input_schema: tool.parameters&.prepare,
-          }.compact
-        end
+        @tools.map { |tool| tool.serialize(context:) } if @tools&.any?
       end
     end
   end

data/lib/omniai/anthropic/version.rb

@@ -2,6 +2,6 @@
 
 module OmniAI
   module Anthropic
-    VERSION = '1.8.0'
+    VERSION = '1.8.2'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: omniai-anthropic
 version: !ruby/object:Gem::Version
-  version: 1.8.0
+  version: 1.8.2
 platform: ruby
 authors:
 - Kevin Sylvestre
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-08-16 00:00:00.000000000 Z
+date: 2024-10-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser
@@ -67,10 +67,13 @@ files:
 - lib/omniai/anthropic/chat/content_serializer.rb
 - lib/omniai/anthropic/chat/function_serializer.rb
 - lib/omniai/anthropic/chat/media_serializer.rb
+- lib/omniai/anthropic/chat/message_serializer.rb
 - lib/omniai/anthropic/chat/payload_serializer.rb
 - lib/omniai/anthropic/chat/stream.rb
 - lib/omniai/anthropic/chat/text_serializer.rb
+- lib/omniai/anthropic/chat/tool_call_result_serializer.rb
 - lib/omniai/anthropic/chat/tool_call_serializer.rb
+- lib/omniai/anthropic/chat/tool_serializer.rb
 - lib/omniai/anthropic/client.rb
 - lib/omniai/anthropic/config.rb
 - lib/omniai/anthropic/version.rb
@@ -96,7 +99,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.14
+rubygems_version: 3.5.18
 signing_key: 
 specification_version: 4
 summary: A generalized framework for interacting with Anthropic