Fix bug in tool call list

ksylvest · Jul 12, 2024 · f5270f9 · f5270f9
1 parent 2609e34
commit f5270f9
Show file tree

Hide file tree

Showing 5 changed files with 3 additions and 13 deletions.
diff --git a/Gemfile.lock b/Gemfile.lock
@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    omniai (1.5.1)
+    omniai (1.5.2)
       event_stream_parser
       http
       zeitwerk

diff --git a/lib/omniai/chat.rb b/lib/omniai/chat.rb
@@ -103,7 +103,7 @@ def parse!(response:)
     def complete!(response:)
       completion = self.class::Response::Completion.new(data: response.parse)
 
-      if @tools && completion.tool_call_required?
+      if @tools && completion.tool_call_list.any?
         @messages = [
           *@messages,
           *completion.choices.map(&:message).map(&:data),

diff --git a/lib/omniai/chat/response/choice.rb b/lib/omniai/chat/response/choice.rb
@@ -20,11 +20,6 @@ def tool_call_list
           part.tool_call_list
         end
 
-        # @return [Boolean]
-        def tool_call_list?
-          tool_call_list.any?
-        end
-
         # @return [String, nil]
         def content
           part.content

diff --git a/lib/omniai/chat/response/payload.rb b/lib/omniai/chat/response/payload.rb
@@ -66,11 +66,6 @@ def content?
         def tool_call_list
           choice.tool_call_list
         end
-
-        # @return [Boolean]
-        def tool_call_list?
-          choice.tool_call_list?
-        end
       end
     end
   end

diff --git a/lib/omniai/version.rb b/lib/omniai/version.rb
@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module OmniAI
-  VERSION = '1.5.1'
+  VERSION = '1.5.2'
 end