Class: Llm::Agent::Rails::Store::Memory

Inherits:
Object
  • Object
show all
Defined in:
lib/llm/agent/rails/store/memory.rb

Instance Method Summary collapse

Constructor Details

#initializeMemory



7
8
9
# File 'lib/llm/agent/rails/store/memory.rb', line 7

def initialize
  @tool_msgs_by_thread = Hash.new { |h,k| h[k] = [] }
end

Instance Method Details

#fetch_tool_messages(thread_id) ⇒ Object



10
# File 'lib/llm/agent/rails/store/memory.rb', line 10

def fetch_tool_messages(thread_id) = @tool_msgs_by_thread[thread_id]

#push_tool_message(thread_id, msg) ⇒ Object



11
# File 'lib/llm/agent/rails/store/memory.rb', line 11

def push_tool_message(thread_id, msg) = @tool_msgs_by_thread[thread_id] << msg