Class: LLM::Google::StreamParser

Inherits:
Object
  • Object
show all
Defined in:
lib/llm/providers/google/stream_parser.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(stream) ⇒ LLM::Google::StreamParser

Parameters:

  • stream (#<<, LLM::Stream)

    A stream sink that implements #<< or the Stream interface



16
17
18
19
20
21
22
23
# File 'lib/llm/providers/google/stream_parser.rb', line 16

def initialize(stream)
  @body = {"candidates" => []}
  @stream = stream
  @emits = {tools: []}
  @can_emit_content = stream.respond_to?(:on_content)
  @can_emit_tool_call = stream.respond_to?(:on_tool_call)
  @can_push_content = stream.respond_to?(:<<)
end

Instance Attribute Details

#bodyHash (readonly)

Returns the fully constructed response body

Returns:

  • (Hash)


10
11
12
# File 'lib/llm/providers/google/stream_parser.rb', line 10

def body
  @body
end

Instance Method Details

#freevoid

This method returns an undefined value.

Frees internal parser state used during streaming.



35
36
37
# File 'lib/llm/providers/google/stream_parser.rb', line 35

def free
  @emits.clear
end

#parse!(chunk) ⇒ LLM::Google::StreamParser

Parameters:

  • chunk (Hash)

Returns:



28
29
30
# File 'lib/llm/providers/google/stream_parser.rb', line 28

def parse!(chunk)
  tap { merge_chunk!(chunk) }
end