omniai 3.2.0 → 3.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8d7dc6f3dc47d5f0385f1f3cc26c509364c55c42bd6290a40a4616c16e69eb66
-  data.tar.gz: 6b6bb6cddbcdcb699495d23137127e3f90fedd2ef179db0a504b2290ff09f1f8
+  metadata.gz: 20fb99d5f16d225a2f1107557c4a488341abf5c14e612a42e5091605a93592f2
+  data.tar.gz: 41b178dee887827bb1197c7a6f21c1699a40f592120964d3b6c92983563e0119
 SHA512:
-  metadata.gz: 3c0a4f6a44bb5d205cda589cb1c85be12921ba1ff8fa9ef0c056b9071af05555db2cd872116597197805a43ac731afeee50925d9643546b9f77150a3465f2d06
-  data.tar.gz: ccb72e06a11d9eb6b2f4059b95130afea945aaddacd1f90bc357670a5f052ff93d4b3fbf48612caf5e6c228ad5b9802384913f888558b8a6df3167f0d08db6b1
+  metadata.gz: 6572fd2a7453a1cf838781187f607fb92e5675c9a206e85a46379089576837724181bcc52792370f676d876860df791dda840881fab7df6e53d2f509830ff711
+  data.tar.gz: 4de9bda7b59d8a2b529ad2c91fa5b4a930c2e00c8667907ba3e2a9a997d0b749f6221c6e72c47ea422caf2e7f6f15b10cc8bce1c9b388a9dcc9715fd73dc9460

data/lib/omniai/chat/response.rb

@@ -16,6 +16,10 @@ module OmniAI
       #   @return [Array<Choice>]
       attr_accessor :choices
 
+      # @!attribute [parent]
+      #   @return [Response, nil] the parent response in a tool call chain
+      attr_accessor :parent
+
       # @param data [Hash]
       # @param choices [Array<Choice>]
       # @param usage [Usage, nil]
@@ -91,6 +95,40 @@ module OmniAI
       def tool_call_list?
         !tool_call_list.nil?
       end
+
+      # Returns the chain of responses from oldest (first) to newest (self).
+      #
+      # @return [Array<Response>]
+      def response_chain
+        chain = []
+        current = self
+
+        while current
+          chain.unshift(current)
+          current = current.parent
+        end
+
+        chain
+      end
+
+      # Returns aggregated usage across all responses in the chain.
+      # Walks the parent chain and sums all token counts.
+      #
+      # @return [Usage, nil]
+      def total_usage
+        chain = response_chain
+        usages = chain.map(&:usage).compact
+        return nil if usages.empty?
+
+        input_tokens = usages.sum { |u| u.input_tokens || 0 }
+        output_tokens = usages.sum { |u| u.output_tokens || 0 }
+
+        Usage.new(
+          input_tokens:,
+          output_tokens:,
+          total_tokens: input_tokens + output_tokens
+        )
+      end
     end
   end
 end

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_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

@@ -113,12 +113,16 @@ module OmniAI
       end
 
       if tools? && completion.tool_call_list?
-        spawn!(
+        next_completion = spawn!(
           @prompt.dup.tap do |prompt|
             prompt.messages += completion.messages
             prompt.messages += build_tool_call_messages(completion.tool_call_list)
           end
         ).process!
+
+        next_completion.parent = completion
+
+        next_completion
       else
         completion
       end

data/lib/omniai/version.rb

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

metadata

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