Class: LLM::Shell::Renderer

Inherits:
Object
  • Object
show all
Defined in:
lib/llm/shell/renderer.rb

Constant Summary collapse

RenderError = 
Class.new(RuntimeError)

Instance Method Summary collapse

Constructor Details

#initialize(message) ⇒ LLM::Shell::MessageRenderer 



12
13
14
 
# File 'lib/llm/shell/renderer.rb', line 12

def initialize(message)
  @message = message
end

Instance Method Details

#render ⇒ Object 



16
17
18
19
20
21
22
23
24
25
26
 
# File 'lib/llm/shell/renderer.rb', line 16

def render
  if message.user?
    render_message(message, :yellow)
  elsif message.assistant?
    render_message(message, :green)
  elsif message.system?
    render_message(message, :red)
  else
    raise RenderError.new("no handler for message role '#{message.role}'")
  end
end