Class: LLM::Shell::Renderer

Inherits:
Object
  • Object
show all
Includes:
Command::Utils
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

Returns an instance of the renderer

Parameters:

  • message (LLM::Message) —

    The message to render



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

def initialize(message)
  @message = message
end

Instance Method Details

#render ⇒ Object 



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

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