omniai 3.1.2 → 3.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d4564620d223dc2b398e45479742eb8e2ab9873716a946af44a069b1df84ec96
-  data.tar.gz: d03aae31d15a19f1f14b0bf9643f79e85ac9f31aa4d8c0fa5b0d2ad0b959e982
+  metadata.gz: 8d7dc6f3dc47d5f0385f1f3cc26c509364c55c42bd6290a40a4616c16e69eb66
+  data.tar.gz: 6b6bb6cddbcdcb699495d23137127e3f90fedd2ef179db0a504b2290ff09f1f8
 SHA512:
-  metadata.gz: e09fcf9083efd4b6de9bd6e8658f04f4b6a6a92f27a935d30e345fed635609d0dd1b21bb0c32e78c25f3fd2e9b7a8f6728b367db6190fa4f69bf7dc47a717f42
-  data.tar.gz: 026e7f2fbddab552fc2635cd11856d652e06f2a3a3a9e30990612da223fe979aee105495be6d39a163dc8040c7bca927e16a6cddec63efe761570ab7eabc30ab
+  metadata.gz: 3c0a4f6a44bb5d205cda589cb1c85be12921ba1ff8fa9ef0c056b9071af05555db2cd872116597197805a43ac731afeee50925d9643546b9f77150a3465f2d06
+  data.tar.gz: ccb72e06a11d9eb6b2f4059b95130afea945aaddacd1f90bc357670a5f052ff93d4b3fbf48612caf5e6c228ad5b9802384913f888558b8a6df3167f0d08db6b1

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_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.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.0"
 end

metadata

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