omniai 2.9.2 → 3.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 92012d7a509078046f920380845855721e2a27124604d65a27b5e6eaeb8b66b0
-  data.tar.gz: 2a2bcd66dbf8a9f86932ecec620bd30933e19d31c4c87ff6767db6ae4facaecf
+  metadata.gz: d531aa6fb65bf379e80e1ee381a9e0bcdc593ff50368023284516b1d63c12d2e
+  data.tar.gz: cfb766bb5a41be87fa2a24766f1a14ce03fb48bda3aeb5bf747f6f6d23cecc55
 SHA512:
-  metadata.gz: 21e52fdd7d754e1c2d1e358d7180a99a3ac12caed8cc26d35b241f73fec6974df07385e60a24b20b836707fb78af916fa3bfe1a9b5588ca2497e6c1df783ab1b
-  data.tar.gz: f8d2d079bc66b84c242d222825675d9dc5bd74aee4dc7ac1371dd54b5e9a3c971801bb996cb49bb40332407b961695806d791161b0f04da353f0d7cd00d87aba
+  metadata.gz: 4128c986161424706af7cdd57ef1d2a23541761cebd94e798e15fcd5959d8e55b79c7c7bac3aa7a69efe6c65ebb24b47154f37077a28cc233f22e0ec7c7af1f9
+  data.tar.gz: e4d043eb1ca554d3da937b0a72b45ef659125f7d7e52adc855ffa4ee446f2735a3981f00e361fc75c55573bfc72e46c98403d459acc7b732579fe06f2bdaa658

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,22 +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 reasoning [Hash, nil] optional reasoning configuration (provider-specific)
-    # @param verbosity [Hash, nil] optional verbosity configuration (provider-specific)
-    # @param kwargs [Hash] additional provider-specific options
+    # @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,
-      reasoning: nil, verbosity: nil, **kwargs, &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
@@ -87,9 +94,7 @@ module OmniAI
       @stream = stream
       @tools = tools
       @format = format
-      @reasoning = reasoning
-      @verbosity = verbosity
-      @kwargs = kwargs
+      @options = options || {}
     end
 
     # @raise [HTTPError]
@@ -151,9 +156,7 @@ module OmniAI
         stream: @stream,
         tools: @tools,
         format: @format,
-        reasoning: @reasoning,
-        verbosity: @verbosity,
-        **@kwargs
+        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/array.rb

@@ -71,12 +71,14 @@ module OmniAI
       # @example
       #   array.serialize # => { type: "array", items: { type: "string" } }
       #
+      # @param options [Hash] optional
+      #
       # @return [Hash]
-      def serialize
+      def serialize(additional_properties: false)
         {
           type: TYPE,
           description: @description,
-          items: @items.serialize,
+          items: @items.serialize(additional_properties:),
           maxItems: @max_items,
           minItems: @min_items,
         }.compact

data/lib/omniai/schema/format.rb

@@ -51,11 +51,13 @@ module OmniAI
       # @example
       #   format.serialize # => { name: "...", schema: { ... } }
       #
+      # @param options [Hash] optional (e.g. `{ additional_properties: nil }``)
+      #
       # @return [Hash]
-      def serialize
+      def serialize(additional_properties: false)
         {
           name:,
-          schema: schema.serialize,
+          schema: schema.serialize(additional_properties:),
         }
       end
 

data/lib/omniai/schema/object.rb

@@ -74,14 +74,17 @@ module OmniAI
         )
       end
 
+      # @param options [Hash] optional
+      #
       # @return [Hash]
-      def serialize
+      def serialize(additional_properties: false)
         {
           type: TYPE,
           title: @title,
           description: @description,
-          properties: @properties.transform_values(&:serialize),
+          properties: @properties.transform_values { |value| value.serialize(additional_properties:) },
           required: @required,
+          additionalProperties: additional_properties,
         }.compact
       end
 

data/lib/omniai/schema/scalar.rb

@@ -91,7 +91,7 @@ module OmniAI
       #   property.serialize #=> { type: "string" }
       #
       # @return [Hash]
-      def serialize
+      def serialize(*)
         {
           type: @type,
           description: @description,

data/lib/omniai/version.rb

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

metadata

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