omniai 2.9.1 → 3.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7f355f49e0fcac63485f6e017ae90eb6f499cd1223cb4ec125422cdef08e14cb
-  data.tar.gz: f07aca376c9982b6f9b071ab426e775c73bc8a20a04d6d4aa2aa5a50b3edd0be
+  metadata.gz: 498fab4d1dc0776ca9b28f970145e9184dd503b7d3032fc45617c0459a442dee
+  data.tar.gz: b0ba0873060444dc59ceb5fe946eaa7b087eb0709a0bd47b80f22bfcc1d5a30c
 SHA512:
-  metadata.gz: 7a6d2bdd83346d8c1b5a66d9351d2e11ede416901b276c85b3ff2f3997c390cff566ced9a1f841df5639347fe6f0f9f6b4586b97d8e9cfcf88d91c5f02d8c086
-  data.tar.gz: 770012e9af20274e3f64f89819695cc918f1d3f827137244e76604f37bc5510f58a7ac2ca8a66a16336a3d6448230e0deb61f2c90e36f2bff1d867043f3bc702
+  metadata.gz: 820e63cbc2a5e4f2cc5d6729ae2d3a0fff1950a16bd22d80eab5fec420975ab79992a6cba1fb332dc0643f4ace515ed389d45be225774b7ae59f5ca911b77b12
+  data.tar.gz: 965a88b9114ee0b215d25a98bcdf4354ac60931154fe2d87859cf423b8d953fc679bdcdb3cb51d82a0118fab926553dffb19eb140cb43ff973e946b76f294549

data/lib/omniai/chat/content.rb

@@ -13,9 +13,10 @@ module OmniAI
       end
 
       # @param context [Context] optional
+      # @param direction [String] optional either "input" or "output"
       #
       # @return [String]
-      def serialize(context: nil)
+      def serialize(context: nil, direction: nil)
         raise NotImplementedError, "#{self.class}#serialize undefined"
       end
 

data/lib/omniai/chat/file.rb

@@ -29,14 +29,16 @@ module OmniAI
       end
 
       # @param context [Context]
+      # @param direction [String] optional - either "input" or "output"
+      #
       # @return [Hash]
-      def serialize(context: nil)
+      def serialize(context: nil, direction: nil)
         if text?
           content = fetch!
           Text.new("<file>#{filename}: #{content}</file>").serialize(context:)
         else
           serializer = context&.serializer(:file)
-          return serializer.call(self, context:) if serializer
+          return serializer.call(self, context:, direction:) if serializer
 
           { type: "#{kind}_url", "#{kind}_url": { url: data_uri } }
         end

data/lib/omniai/chat/message.rb

@@ -157,6 +157,14 @@ module OmniAI
 
         [object]
       end
+
+      # @return [String] either "input" or "output"
+      def direction
+        case role
+        when Role::ASSISTANT then "output"
+        else "input"
+        end
+      end
     end
   end
 end

data/lib/omniai/chat/response.rb

@@ -102,18 +102,17 @@ module OmniAI
         !message.nil? && message.text?
       end
 
-      # @param index [Integer]
-      #
-      # @return [ToolCallList]
-      def tool_call_list(index: 0)
-        message(index:)&.tool_call_list
+      # @return [ToolCallList, nil]
+      def tool_call_list
+        tool_call_lists = messages.map(&:tool_call_list).compact
+        return if tool_call_lists.empty?
+
+        tool_call_lists.reduce(&:+)
       end
 
       # @return [Boolean]
-      def tool_call_list?(index: 0)
-        tool_call_list = tool_call_list(index:)
-
-        !tool_call_list.nil? && tool_call_list.any?
+      def tool_call_list?
+        !tool_call_list.nil?
       end
     end
   end

data/lib/omniai/chat/text.rb

@@ -23,7 +23,8 @@ module OmniAI
         @text
       end
 
-      # @param data [Hash]
+      # @param context [Context] optional
+      # @param data [Hash] required
       def self.deserialize(data, context: nil)
         deserialize = context&.deserializer(:text)
         return deserialize.call(data, context:) if deserialize
@@ -32,11 +33,12 @@ module OmniAI
       end
 
       # @param context [Context] optional
+      # @param direction [String] optional - either "input" or "output"
       #
       # @return [Hash]
-      def serialize(context: nil)
+      def serialize(context: nil, direction: nil)
         serializer = context&.serializer(:text)
-        return serializer.call(self, context:) if serializer
+        return serializer.call(self, context:, direction:) if serializer
 
         { type: "text", text: @text }
       end

data/lib/omniai/chat/tool_call_list.rb

@@ -47,6 +47,16 @@ module OmniAI
       def each(&)
         @entries.each(&)
       end
+
+      # @return [ToolCall]
+      def [](index)
+        @entries[index]
+      end
+
+      # @param [other] [ToolCallList]
+      def +(other)
+        self.class.new(entries: entries + other.entries)
+      end
     end
   end
 end

data/lib/omniai/chat/tool_call_message.rb

@@ -40,8 +40,7 @@ module OmniAI
 
       # Usage:
       #
-      #   message.serialize # => { role: :user, content: 'Hello!' }
-      #   message.serialize # => { role: :user, content: [{ type: 'text', text: 'Hello!' }] }
+      #   message.serialize # => { role: :tool, content: 'Hello!', tool_call_id: '...' }
       #
       # @param context [Context] optional
       #

data/lib/omniai/chat/url.rb

@@ -26,7 +26,8 @@ module OmniAI
         "#<#{self.class} uri=#{@uri.inspect}>"
       end
 
-      # @param data [Hash]
+      # @param context [Context] optional
+      # @param data [Hash] required
       def self.deserialize(data, context: nil)
         deserialize = context&.deserializer(:url)
         return deserialize.call(data, context:) if deserialize
@@ -38,15 +39,16 @@ module OmniAI
       end
 
       # @param context [Context] optional
+      # @param direction [String] optional - either "input" or "output"
       #
       # @return [Hash]
-      def serialize(context: nil)
+      def serialize(context: nil, direction: nil)
         if text?
           content = fetch!
           Text.new("<file>#{filename}: #{content}</file>").serialize(context:)
         else
           serializer = context&.serializer(:url)
-          return serializer.call(self, context:) if serializer
+          return serializer.call(self, context:, direction:) if serializer
 
           {
             type: "#{kind}_url",

data/lib/omniai/chat.rb

@@ -60,18 +60,29 @@ module OmniAI
     end
 
     # @param prompt [OmniAI::Chat::Prompt, String, nil] optional
-    # @param client [OmniAI::Client] the client
+    # @param client [OmniAI::Client] required
     # @param model [String] required
     # @param temperature [Float, nil] optional
     # @param stream [Proc, IO, nil] optional
-    # @param tools [Array<OmniAI::Tool>] optional
+    # @param tools [Array<OmniAI::Tool>, nil] optional
     # @param format [:json, :text, OmniAI::Schema::Object, nil] optional
+    # @param options [Hash] optional (used for vendor specific options)
     #
     # @yield [prompt] optional
     # @yieldparam prompt [OmniAI::Chat::Prompt]
     #
     # @return [OmniAi::Chat]
-    def initialize(prompt = nil, client:, model:, temperature: nil, stream: nil, tools: nil, format: nil, &block)
+    def initialize(
+      prompt = nil,
+      client:,
+      model:,
+      temperature: nil,
+      stream: nil,
+      tools: nil,
+      format: nil,
+      **options,
+      &block
+    )
       raise ArgumentError, "prompt or block is required" if !prompt && !block
 
       @prompt = prompt ? Prompt.parse(prompt) : Prompt.new
@@ -83,6 +94,7 @@ module OmniAI
       @stream = stream
       @tools = tools
       @format = format
+      @options = options || {}
     end
 
     # @raise [HTTPError]
@@ -143,7 +155,8 @@ module OmniAI
         temperature: @temperature,
         stream: @stream,
         tools: @tools,
-        format: @format
+        format: @format,
+        options: @options
       )
     end
 

data/lib/omniai/client.rb

@@ -185,18 +185,19 @@ module OmniAI
 
     # @raise [OmniAI::Error]
     #
-    # @param messages [String] optional
-    # @param model [String] optional
-    # @param format [Symbol] optional :text or :json
+    # @param prompt [OmniAI::Chat::Prompt, String, nil] optional
+    # @param model [String, nil] optional
+    # @param format [Symbol, OmniAI::Schema::Format, nil] optional (e.g. `:text`, `:json`, `OmniAI::Schema.format(...)`)
     # @param temperature [Float, nil] optional
     # @param stream [Proc, nil] optional
     # @param tools [Array<OmniAI::Tool>] optional
+    # @param options [Hash, nil] optional
     #
     # @yield [prompt] optional
     # @yieldparam prompt [OmniAI::Chat::Prompt]
     #
     # @return [OmniAI::Chat::Response]
-    def chat(messages = nil, model:, temperature: nil, format: nil, stream: nil, tools: nil, &)
+    def chat(prompt = nil, model:, temperature: nil, format: nil, stream: nil, tools: nil, options: {}, &)
       raise NotImplementedError, "#{self.class.name}#chat undefined"
     end
 

data/lib/omniai/schema/object.rb

@@ -82,6 +82,7 @@ module OmniAI
           description: @description,
           properties: @properties.transform_values(&:serialize),
           required: @required,
+          additionalProperties: false,
         }.compact
       end
 

data/lib/omniai/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module OmniAI
-  VERSION = "2.9.1"
+  VERSION = "3.0.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: omniai
 version: !ruby/object:Gem::Version
-  version: 2.9.1
+  version: 3.0.0
 platform: ruby
 authors:
 - Kevin Sylvestre