Class: OmniAI::Google::Chat::Stream

Inherits:
Chat::Stream
  • Object
show all
Defined in:
lib/omniai/google/chat/stream.rb

Overview

Combine chunks into a hash. For each chunk yield the text (delta) if a block is given and the chunk is text.

Instance Method Summary collapse

Instance Method Details

#stream! {|delta| ... } ⇒ Hash

Yields:

  • (delta)

Yield Parameters:

  • delta (OmniAI::Chat::Delta)

Returns:

  • (Hash) 


12
13
14
15
16
17
18
19
20
21
22
 
# File 'lib/omniai/google/chat/stream.rb', line 12

def stream!(&block)
  @data = { "candidates" => [] }

  @chunks.each do |chunk|
    parser.feed(chunk) do |type, data, id|
      process!(type, data, id, &block)
    end
  end

  @data
end