omniai 3.2.1 → 3.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 329e024c0319179ac8ff6757755d8c3d4275f68a759ec864200b88d1930e8e52
-  data.tar.gz: beac9879c26d58b6ff6cc5dd71e32cd29a2e32f511971d8982e78c9ece33d654
+  metadata.gz: 20fb99d5f16d225a2f1107557c4a488341abf5c14e612a42e5091605a93592f2
+  data.tar.gz: 41b178dee887827bb1197c7a6f21c1699a40f592120964d3b6c92983563e0119
 SHA512:
-  metadata.gz: f57c9c5daa85d4cb77c8569abc79cb01203c531fbe9e0849a38884229c45b555df3c286cf7646c071ec000356ff748f7b53a0c0a0c26da5feba402fdf4a0cfe0
-  data.tar.gz: c9abb1aa60e64e663428f42c66bbc78b639dbbaabdfb0510f3ced5ba25ecddbe9497a3d0389a632816748c54b209142c1d4ccc15271a0f313fb32e7a3d8c87b3
+  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.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.1"
+  VERSION = "3.2.2"
 end

metadata

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