Class: LLM::Message

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

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(role, content, extra = {}) ⇒ LLM::Message

Returns a new message

Parameters:

  • role (Symbol)
  • content (String)
  • extra (Hash) (defaults to: {})


26
27
28
29
30
 
# File 'lib/llm/message.rb', line 26

def initialize(role, content, extra = {})
  @role = role.to_s
  @content = content
  @extra = LLM::Object.from(extra)
end

Instance Attribute Details

#role ⇒ Symbol (readonly)

Returns the role of the message

Returns:

  • (Symbol)


8
9
10
 
# File 'lib/llm/message.rb', line 8

def role
  @role
end

#content ⇒ String (readonly)

Returns the content of the message

Returns:

  • (String)


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

def content
  @content
end

#extra ⇒ Hash (readonly)

Returns extra context associated with the message

Returns:

  • (Hash)


18
19
20
 
# File 'lib/llm/message.rb', line 18

def extra
  @extra
end

Instance Method Details

#inspect ⇒ String

Returns a string representation of the message

Returns:

  • (String)


165
166
167
168
169
 
# File 'lib/llm/message.rb', line 165

def inspect
  "#<#{self.class.name}:0x#{object_id.to_s(16)} " \
  "tool_call=#{tool_calls.any?} role=#{role.inspect} " \
  "content=#{content.inspect} reasoning_content=#{reasoning_content.inspect}>"
end

#to_h ⇒ Hash

Returns a Hash representation of the message.

Returns:

  • (Hash)


35
36
37
38
39
40
 
# File 'lib/llm/message.rb', line 35

def to_h
  {role:, content:, reasoning_content:,
   tools: extra.tool_calls,
   usage:,
   original_tool_calls: extra.original_tool_calls}.compact
end

#to_json ⇒ String

Returns:

  • (String)


44
45
46
 
# File 'lib/llm/message.rb', line 44

def to_json(...)
  LLM.json.dump(to_h, ...)
end

#==(other) ⇒ Boolean Also known as: eql?

Returns true when two objects have the same role and content

Parameters:

  • other (Object) —

    The other object to compare

Returns:

  • (Boolean)


53
54
55
56
57
58
59
 
# File 'lib/llm/message.rb', line 53

def ==(other)
  if other.respond_to?(:to_h)
    to_h == other.to_h
  else
    false
  end
end

#content! ⇒ Hash

Try to parse JSON content

Returns:

  • (Hash) —

    Returns the parsed content as a Hash



66
67
68
 
# File 'lib/llm/message.rb', line 66

def content!
  LLM.json.load(content)
end

#reasoning_content ⇒ String?

Returns reasoning content associated with the message

Returns:

  • (String, nil)


73
74
75
 
# File 'lib/llm/message.rb', line 73

def reasoning_content
  extra.reasoning_content
end

#functions ⇒ Array<LLM::Function>

Returns:



79
80
81
82
83
84
85
86
 
# File 'lib/llm/message.rb', line 79

def functions
  @functions ||= tool_calls.filter_map do |fn|
    function = available_tools.find { _1.name.to_s == fn["name"] } || next
    function = function.dup
    function.tap { _1.id = fn.id }
    function.tap { _1.arguments = fn.arguments }
  end
end

#assistant? ⇒ Boolean

Returns true when the message is an assistant message

Returns:

  • (Boolean)


91
92
93
 
# File 'lib/llm/message.rb', line 91

def assistant?
  role == "assistant" || role == "model"
end

#system? ⇒ Boolean

Returns true when the message is a system message

Returns:

  • (Boolean)


98
99
100
 
# File 'lib/llm/message.rb', line 98

def system?
  role == "system"
end

#user? ⇒ Boolean

Returns true when the message is a user message

Returns:

  • (Boolean)


105
106
107
 
# File 'lib/llm/message.rb', line 105

def user?
  role == "user"
end

#tool_call? ⇒ Boolean

Returns true when the message requests a function call

Returns:

  • (Boolean) —

    Returns true when the message requests a function call



112
113
114
 
# File 'lib/llm/message.rb', line 112

def tool_call?
  tool_calls.any?
end

#tool_return? ⇒ Boolean

Returns true when the message represents a function return

Returns:

  • (Boolean) —

    Returns true when the message represents a function return



119
120
121
122
 
# File 'lib/llm/message.rb', line 119

def tool_return?
  LLM::Function::Return === content ||
    [*content].grep(LLM::Function::Return).any?
end

#response ⇒ LLM::Response?

Note:

This method returns a response for assistant messages, and it returns nil for non-assistant messages

Returns the response associated with the message, or nil

Returns:

  • (LLM::Response, nil) —

    Returns the response associated with the message, or nil



130
131
132
 
# File 'lib/llm/message.rb', line 130

def response
  extra.response
end

#annotations ⇒ Array<LLM::Object>

Note:

This method might return annotations for assistant messages, and it returns an empty array for non-assistant messages

Returns annotations associated with the message

Returns:



140
141
142
 
# File 'lib/llm/message.rb', line 140

def annotations
  @annotations ||= LLM::Object.from(extra.annotations || [])
end

#usage ⇒ LLM::Object? Also known as: token_usage

Note:

This method returns token usage for assistant messages, and it returns nil for non-assistant messages

Returns token usage statistics

Returns:



150
151
152
 
# File 'lib/llm/message.rb', line 150

def usage
  @usage ||= extra.usage || response&.usage
end

#model ⇒ String?

Returns the model associated with a message

Returns:

  • (String, nil) —

    Returns the model associated with a message



158
159
160
 
# File 'lib/llm/message.rb', line 158

def model
  response&.model
end