Support attachments when accumulating streams

Author: tpaulshippy

lib/ruby_llm/stream_accumulator.rb

@@ -6,6 +6,7 @@ class StreamAccumulator
     attr_reader :content, :model_id, :tool_calls
 
     def initialize
+      #@content = nil
       @content = +''
       @tool_calls = {}
       @input_tokens = 0
@@ -21,6 +22,7 @@ def add(chunk)
         accumulate_tool_calls chunk.tool_calls
       else
         @content << (chunk.content || '')
+        # accumulate_content(chunk.content)
       end
 
       count_tokens chunk
@@ -30,6 +32,7 @@ def add(chunk)
     def to_message(response)
       Message.new(
         role: :assistant,
+        #content: final_content,
         content: content.empty? ? nil : content,
         model_id: model_id,
         tool_calls: tool_calls_from_stream,
@@ -41,6 +44,53 @@ def to_message(response)
 
     private
 
+    def accumulate_content(new_content)
+      return unless new_content
+
+      if @content.nil?
+        @content = new_content.is_a?(String) ? +new_content : new_content
+      else
+        case [@content.class, new_content.class]
+        when [String, String]
+          @content << new_content
+        when [String, Content]
+          # Convert accumulated string to Content and merge
+          @content = Content.new(@content)
+          merge_content(new_content)
+        when [Content, String]
+          # Append string to existing Content's text
+          @content.instance_variable_set(:@text, (@content.text || '') + new_content)
+        when [Content, Content]
+          merge_content(new_content)
+        end
+      end
+    end
+
+    def merge_content(new_content)
+      # Merge text
+      current_text = @content.text || ''
+      new_text = new_content.text || ''
+      @content.instance_variable_set(:@text, current_text + new_text)
+
+      # Merge attachments
+      new_content.attachments.each do |attachment|
+        @content.attach(attachment)
+      end
+    end
+
+    def final_content
+      case @content
+      when nil
+        nil
+      when String
+        @content.empty? ? nil : @content
+      when Content
+        @content.text.nil? && @content.attachments.empty? ? nil : @content
+      else
+        @content
+      end
+    end
+
     def tool_calls_from_stream
       tool_calls.transform_values do |tc|
         arguments = if tc.arguments.is_a?(String) && !tc.arguments.empty?