omniai 3.1.2 → 3.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d4564620d223dc2b398e45479742eb8e2ab9873716a946af44a069b1df84ec96
-  data.tar.gz: d03aae31d15a19f1f14b0bf9643f79e85ac9f31aa4d8c0fa5b0d2ad0b959e982
+  metadata.gz: 329e024c0319179ac8ff6757755d8c3d4275f68a759ec864200b88d1930e8e52
+  data.tar.gz: beac9879c26d58b6ff6cc5dd71e32cd29a2e32f511971d8982e78c9ece33d654
 SHA512:
-  metadata.gz: e09fcf9083efd4b6de9bd6e8658f04f4b6a6a92f27a935d30e345fed635609d0dd1b21bb0c32e78c25f3fd2e9b7a8f6728b367db6190fa4f69bf7dc47a717f42
-  data.tar.gz: 026e7f2fbddab552fc2635cd11856d652e06f2a3a3a9e30990612da223fe979aee105495be6d39a163dc8040c7bca927e16a6cddec63efe761570ab7eabc30ab
+  metadata.gz: f57c9c5daa85d4cb77c8569abc79cb01203c531fbe9e0849a38884229c45b555df3c286cf7646c071ec000356ff748f7b53a0c0a0c26da5feba402fdf4a0cfe0
+  data.tar.gz: c9abb1aa60e64e663428f42c66bbc78b639dbbaabdfb0510f3ced5ba25ecddbe9497a3d0389a632816748c54b209142c1d4ccc15271a0f313fb32e7a3d8c87b3

data/lib/omniai/chat/message.rb

@@ -165,6 +165,11 @@ module OmniAI
         else "input"
         end
       end
+
+      # @return [Boolean]
+      def tool_call_list?
+        @tool_call_list&.any?
+      end
     end
   end
 end

data/lib/omniai/chat/response.rb

@@ -81,7 +81,7 @@ module OmniAI
 
       # @return [ToolCallList, nil]
       def tool_call_list
-        tool_call_lists = messages.map(&:tool_call_list).compact
+        tool_call_lists = messages.filter(&:tool_call_list?).map(&:tool_call_list)
         return if tool_call_lists.empty?
 
         tool_call_lists.reduce(&:+)

data/lib/omniai/chat/tool_call.rb

@@ -12,11 +12,17 @@ module OmniAI
       #   @return [Function]
       attr_accessor :function
 
+      # @!attribute [r] options
+      #   @return [Hash] provider-specific options
+      attr_reader :options
+
       # @param id [String]
       # @param function [Function]
-      def initialize(id:, function:)
+      # @param options [Hash] optional - used for provider-specific options
+      def initialize(id:, function:, **options)
         @id = id
         @function = function
+        @options = options
       end
 
       # @return [String]

data/lib/omniai/chat/tool_call_list.rb

@@ -42,6 +42,11 @@ module OmniAI
         "#<#{self.class.name} entries=#{@entries.inspect}>"
       end
 
+      # @return [Boolean]
+      def any?
+        @entries.any?
+      end
+
       # @yield toolcall
       # @yieldparam toolcall [ToolCall]
       def each(&)

data/lib/omniai/chat/tool_call_result.rb

@@ -12,11 +12,17 @@ module OmniAI
       #   @return [ToolCall]
       attr_accessor :tool_call_id
 
+      # @!attribute [r] options
+      #   @return [Hash] provider-specific options
+      attr_reader :options
+
       # @param content [Object] e.g. a string, hash, array, boolean, numeric, etc.
       # @param tool_call_id [String]
-      def initialize(content:, tool_call_id:)
+      # @param options [Hash] optional - used for provider-specific options
+      def initialize(content:, tool_call_id:, **options)
         @content = content
         @tool_call_id = tool_call_id
+        @options = options
       end
 
       # @return [String]

data/lib/omniai/chat.rb

@@ -112,7 +112,7 @@ module OmniAI
         raise SSLError, e.message, cause: e
       end
 
-      if @tools && completion.tool_call_list?
+      if tools? && completion.tool_call_list?
         spawn!(
           @prompt.dup.tap do |prompt|
             prompt.messages += completion.messages
@@ -126,6 +126,11 @@ module OmniAI
 
   protected
 
+    # @return [Boolean]
+    def tools?
+      @tools&.any?
+    end
+
     # @return [Boolean]
     def stream?
       !@stream.nil?

data/lib/omniai/version.rb

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

metadata

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