omniai 3.2.1 → 3.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 329e024c0319179ac8ff6757755d8c3d4275f68a759ec864200b88d1930e8e52
-  data.tar.gz: beac9879c26d58b6ff6cc5dd71e32cd29a2e32f511971d8982e78c9ece33d654
+  metadata.gz: b3e745edbb32e4102309da753fb4ccdb4cec5b6ab69a4bebf7260a39d99b31b4
+  data.tar.gz: d8546d88f6c5cdc1c665d08bd7624f284c0e5baec04162f3a7ee29474f25beba
 SHA512:
-  metadata.gz: f57c9c5daa85d4cb77c8569abc79cb01203c531fbe9e0849a38884229c45b555df3c286cf7646c071ec000356ff748f7b53a0c0a0c26da5feba402fdf4a0cfe0
-  data.tar.gz: c9abb1aa60e64e663428f42c66bbc78b639dbbaabdfb0510f3ced5ba25ecddbe9497a3d0389a632816748c54b209142c1d4ccc15271a0f313fb32e7a3d8c87b3
+  metadata.gz: 59798981414c5fb32fd3f59fcc2d159c21b37cc1b6ce979424b91e7154147d05ba36e6e048ef3ff002f70022be5b02af0c7e66eb1c2e6dab6a434cb1be9ee794
+  data.tar.gz: 45f18be41f25a010810901fc9fe001f31698549fdfff6c0a1b59d5eae612f53ad6cd4ff9a857cb11b2a77fe50a1e8f0d000a388f836604e7e26d1d8212635258

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,50 @@ module OmniAI
       def tool_call_list?
         !tool_call_list.nil?
       end
+
+      # Links this response chain to a parent response.
+      # Walks up to find the oldest response (no parent) and sets its parent.
+      #
+      # @param parent [Response]
+      def link_to(parent)
+        current = self
+        current = current.parent while current.parent
+        current.parent = parent
+      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.link_to(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.3"
 end

metadata

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