Class: Riffer::Tracing::StreamRecorder

Inherits:
Object
  • Object
show all
Defined in:
lib/riffer/tracing/stream_recorder.rb,
sig/generated/riffer/tracing/stream_recorder.rbs

Overview

Wraps a stream yielder to observe terminal events for span stamping while forwarding every event downstream untouched.

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(yielder) ⇒ StreamRecorder

-- : (Enumerator::Yielder) -> void

Parameters:

  • (Enumerator::Yielder) 


21
22
23
24
 
# File 'lib/riffer/tracing/stream_recorder.rb', line 21

def initialize(yielder)
  @yielder = yielder
  @tool_calls = [] #: Array[Riffer::Messages::Assistant::ToolCall]
end

Instance Attribute Details

#content ⇒ String? (readonly)

: String?

Returns:

  • (String, nil) 


15
16
17
 
# File 'lib/riffer/tracing/stream_recorder.rb', line 15

def content
  @content
end

#finish_reason ⇒ Symbol? (readonly)

: Symbol?

Returns:

  • (Symbol, nil) 


11
12
13
 
# File 'lib/riffer/tracing/stream_recorder.rb', line 11

def finish_reason
  @finish_reason
end

#raw_finish_reason ⇒ String? (readonly)

: String?

Returns:

  • (String, nil) 


13
14
15
 
# File 'lib/riffer/tracing/stream_recorder.rb', line 13

def raw_finish_reason
  @raw_finish_reason
end

#token_usage ⇒ Riffer::Providers::TokenUsage? (readonly)

Returns the value of attribute token_usage.

Returns:



9
10
11
 
# File 'lib/riffer/tracing/stream_recorder.rb', line 9

def token_usage
  @token_usage
end

#tool_calls ⇒ Array[Riffer::Messages::Assistant::ToolCall] (readonly)

: Array

Returns:



17
18
19
 
# File 'lib/riffer/tracing/stream_recorder.rb', line 17

def tool_calls
  @tool_calls
end

Instance Method Details

#<<(event) ⇒ self

-- : (Riffer::StreamEvents::Base) -> self

Parameters:

Returns:

  • (self) 


28
29
30
31
32
 
# File 'lib/riffer/tracing/stream_recorder.rb', line 28

def <<(event)
  record(event)
  @yielder << event
  self
end

#record(event) ⇒ void

This method returns an undefined value.

-- : (Riffer::StreamEvents::Base) -> void

Parameters:



38
39
40
41
42
43
44
45
46
47
48
49
50
 
# File 'lib/riffer/tracing/stream_recorder.rb', line 38

def record(event)
  case event
  when Riffer::StreamEvents::TextDone
    @content = event.content
  when Riffer::StreamEvents::ToolCallDone
    @tool_calls << Riffer::Messages::Assistant::ToolCall.new(call_id: event.call_id, name: event.name, arguments: event.arguments)
  when Riffer::StreamEvents::TokenUsageDone
    @token_usage = event.token_usage
  when Riffer::StreamEvents::FinishReasonDone
    @finish_reason = event.finish_reason
    @raw_finish_reason = event.raw_finish_reason
  end
end