Class: OmniAI::Anthropic::Chat::Stream

Inherits:
Chat::Stream
  • Object
show all
Defined in:
lib/omniai/anthropic/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.

Defined Under Namespace

Modules: ContentBlockDeltaType, ContentBlockType, Type

Instance Method Summary collapse

Instance Method Details

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

Yields:

  • (delta)

Yield Parameters:

  • delta (OmniAI::Chat::Delta)

Returns:

  • (Hash) 


31
32
33
34
35
36
37
38
39
40
41
 
# File 'lib/omniai/anthropic/chat/stream.rb', line 31

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

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

  @data
end