Module: RubyLLM::ActiveRecord::ChatMethods

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

Overview

Methods mixed into chat models to handle message persistence and provide a conversation interface.

Instance Method Summary collapse

Instance Method Details

#ask(message, &block) ⇒ Object Also known as: say 



59
60
61
62
63
 
# File 'lib/ruby_llm/active_record/acts_as.rb', line 59

def ask(message, &block)
  message = { role: :user, content: message }
  messages.create!(**message)
  chat.complete(&block)
end

#to_llm ⇒ Object 



46
47
48
49
50
51
52
53
54
55
56
57
 
# File 'lib/ruby_llm/active_record/acts_as.rb', line 46

def to_llm
  chat = RubyLLM.chat(model: model_id)

  # Load existing messages into chat
  messages.each do |msg|
    chat.add_message(msg.to_llm)
  end

  # Set up message persistence
  chat.on_new_message { persist_new_message }
      .on_end_message { |msg| persist_message_completion(msg) }
end