ox-ai-workers 0.8.4 → 0.8.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4c3c03d24ee695dc7f931547e96fefee838a761e58a6f1410ffaffae5674b87a
-  data.tar.gz: 88e9a6a0b4711e49b3d2d168562df273a618fb54129efba683909d2cc8a8c936
+  metadata.gz: ff4f2cb710ff8bf02e912c0354ff62f3f3ce65871edc90f2cfb398805cd8eadb
+  data.tar.gz: fdca1f55dd1cc8721bf8eca0317272c749d7d3f0608b5554f86bd6677d5307e4
 SHA512:
-  metadata.gz: c372c3ae901dcd5d8591633791419cbe2f81ed14f7e2fc1e889d2546db6144954b534504e2ba0e728e844cd758efa67e02a3c4ac8cc91548c45bc3266b64602f
-  data.tar.gz: 01e5efec2e0b6fac0421415a48f4452def47ffcfb135260820c05c39547e71390b032c7af66877620e2e5c38129228af58e7eb80141eb4ecc5f9e64a66a1eb4c
+  metadata.gz: 555e7b22260a4b443c5b05fcacc6e0f3dbe72dca112e887159abcfe12e0efda767d23ca7e995e60fc3236ce545361a37a7ea65a324171f65d28c8cfe0f975d59
+  data.tar.gz: a5d39e26ecc80a315101cabab6e50486f1e05a293360ef79c38e56b022535bb7a769daf3caf851a69fa06e20546caa4ad9c0648e457715f481166eea9e5bf66a

data/CHANGELOG.md

@@ -1,3 +1,7 @@
+## [0.8.5] - 2025-04-18
+
+- Fixed tool calls parsing
+
 ## [0.8.4] - 2025-04-18
 
 - Refine Russian locale for Iterator with enhanced step descriptions and planning guidance

data/config/locales/ru.oxaiworkers.iterator.yml

@@ -26,5 +26,5 @@ ru:
         - Шаг %d. Когда промежуточные шаги завершены и точное содержание предыдущих сообщений больше не важно, сожми историю диалога с помощью ox_ai_workers_iterator__summarize.
         - Шаг %d. После выполнения всех шагов плана и получения финального результата через вызов соответствующей функции (инструмента), оцени полноту и корректность решения.
         - Шаг %d. Если необходимо запросить подтверждение у пользователя используй функцию ox_ai_workers_iterator__action_request для представления финального результата и ожидания ответа.
-        - Шаг %d. После успешного завершения задачи и получения результата (и, возможно, подтверждения пользователя), вызови ox_ai_workers_iterator__finish_it.
+        - Шаг %d. После успешного завершения задачи и получения результата (и подтверждения пользователя), вызови ox_ai_workers_iterator__finish_it.
         - Старайся выполнить работу наилучшим образом, используя доступные инструменты максимально эффективно.

data/lib/oxaiworkers/module_request.rb

@@ -70,25 +70,65 @@ module OxAiWorkers
     end
 
     def parse_choices(response)
+      # Reset instance variables before processing choices
+      @result = nil
+      @tool_calls_raw = []
       @tool_calls = []
-      @result = response.dig('choices', 0, 'message', 'content')
-      @tool_calls_raw = response.dig('choices', 0, 'message', 'tool_calls')
-      @finish_reason = response.dig('choices', 0, 'finish_reason')
-      @is_truncated = @finish_reason == 'length'
-
-      return @tool_calls if @tool_calls_raw.nil?
-
-      @tool_calls_raw.each do |tool|
-        function = tool['function']
-        args = JSON.parse(YAML.load(function['arguments']).to_json, { symbolize_names: true })
-        @tool_calls << {
-          class: function['name'].split('__').first,
-          name: function['name'].split('__').last,
-          args:
-        }
+      @finish_reason = nil
+      @is_truncated = false
+
+      choices = response['choices']
+      return if choices.nil? || choices.empty?
+
+      choices.each do |choice|
+        parse_one_choice(choice)
       end
 
-      @tool_calls
+      # @tool_calls is already populated by parse_one_choice
+      # @result, @finish_reason, @is_truncated are updated with the last relevant values
+    end
+
+    def parse_one_choice(choice)
+      message = choice['message']
+      return unless message # Skip if there's no message in this choice
+
+      # Accumulate raw tool calls
+      if message['tool_calls']
+        @tool_calls_raw.concat(message['tool_calls'])
+        message['tool_calls'].each do |tool_call|
+          next unless tool_call['type'] == 'function' # Ensure it's a function call
+
+          function = tool_call['function']
+          next unless function && function['name'] && function['arguments']
+
+          begin
+            # Attempt to parse arguments, handle potential JSON errors
+            args = JSON.parse(function['arguments'], symbolize_names: true)
+          rescue JSON::ParserError => e
+            OxAiWorkers.logger.error("Failed to parse tool call arguments: #{e.message}", for: self.class)
+            OxAiWorkers.logger.debug("Raw arguments: #{function['arguments']}", for: self.class)
+            args = {} # Assign empty args or handle error as appropriate
+          end
+
+          # Accumulate parsed tool calls
+          @tool_calls << {
+            class: function['name'].split('__').first,
+            name: function['name'].split('__').last,
+            args: args
+          }
+        end
+      end
+
+      # Update result with the content if present
+      current_result = message['content']
+      @result = current_result if current_result.present?
+
+      # Update finish reason and truncation status
+      current_finish_reason = choice['finish_reason']
+      if current_finish_reason.present?
+        @finish_reason = current_finish_reason
+        @is_truncated = @finish_reason == 'length'
+      end
     end
   end
 end

data/lib/oxaiworkers/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module OxAiWorkers
-  VERSION = '0.8.4'
+  VERSION = '0.8.5'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ox-ai-workers
 version: !ruby/object:Gem::Version
-  version: 0.8.4
+  version: 0.8.5
 platform: ruby
 authors:
 - Denis Smolev