class Riffer::StreamEvents::TokenUsageDone

Represents completion of token usage tracking during streaming.

Emitted when the LLM has finished and token usage data is available.

event.token_usage.input_tokens   # => 100
event.token_usage.output_tokens  # => 50
event.token_usage.total_tokens   # => 150

Attributes

token_usage [R]

The token usage data for this response.

Public Class Methods

new (token_usage:, role: :assistant)

Source

# File lib/riffer/stream_events/token_usage_done.rb, line 18
def initialize(token_usage:, role: :assistant)
  super(role: role)
  @token_usage = token_usage
end

Calls superclass method Riffer::StreamEvents::Base::new

Public Instance Methods

to_h ()

Source

# File lib/riffer/stream_events/token_usage_done.rb, line 25
def to_h
  {role: @role, token_usage: @token_usage.to_h}
end