Module: RubyLLM::ActiveRecord::MessageMethods

Extended by:
ActiveSupport::Concern
Defined in:
lib/ruby_llm/active_record/message_methods.rb

Overview

Methods mixed into message models.

Instance Method Summary collapse

Instance Method Details

#to_llm ⇒ Object 



13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
 
# File 'lib/ruby_llm/active_record/message_methods.rb', line 13

def to_llm
  cached = has_attribute?(:cached_tokens) ? self[:cached_tokens] : nil
  cache_creation = has_attribute?(:cache_creation_tokens) ? self[:cache_creation_tokens] : nil

  RubyLLM::Message.new(
    role: role.to_sym,
    content: extract_content,
    tool_calls: extract_tool_calls,
    tool_call_id: extract_tool_call_id,
    input_tokens: input_tokens,
    output_tokens: output_tokens,
    cached_tokens: cached,
    cache_creation_tokens: cache_creation,
    model_id: model_association&.model_id
  )
end