Class: ConsoleAgent::Tools::MemoryTools

Inherits:
Object
  • Object
show all
Defined in:
lib/console_agent/tools/memory_tools.rb

Constant Summary collapse

MEMORIES_DIR = 
'memories'

Instance Method Summary collapse

Constructor Details

#initialize(storage = nil) ⇒ MemoryTools

Returns a new instance of MemoryTools.



8
9
10
 
# File 'lib/console_agent/tools/memory_tools.rb', line 8

def initialize(storage = nil)
  @storage = storage || ConsoleAgent.storage
end

Instance Method Details

#delete_memory(name:) ⇒ Object 



37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
 
# File 'lib/console_agent/tools/memory_tools.rb', line 37

def delete_memory(name:)
  key = memory_key(name)
  unless @storage.exists?(key)
    # Try to find by name match across all memory files
    found_key = find_memory_key_by_name(name)
    return "No memory found: \"#{name}\"" unless found_key
    key = found_key
  end

  memory = load_memory(key)
  @storage.delete(key)
  "Memory deleted: \"#{memory ? memory['name'] : name}\""
rescue Storage::StorageError => e
  "FAILED to delete memory (#{e.message})."
end

#memory_summaries ⇒ Object 



81
82
83
84
85
86
87
88
89
90
 
# File 'lib/console_agent/tools/memory_tools.rb', line 81

def memory_summaries
  memories = load_all_memories
  return nil if memories.empty?

  memories.map { |m|
    tags = Array(m['tags'])
    tag_str = tags.empty? ? '' : " [#{tags.join(', ')}]"
    "- #{m['name']}#{tag_str}"
  }
end

#recall_memories(query: nil, tag: nil) ⇒ Object 



53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
 
# File 'lib/console_agent/tools/memory_tools.rb', line 53

def recall_memories(query: nil, tag: nil)
  memories = load_all_memories
  return "No memories stored yet." if memories.empty?

  results = memories
  if tag && !tag.empty?
    results = results.select { |m|
      Array(m['tags']).any? { |t| t.downcase.include?(tag.downcase) }
    }
  end
  if query && !query.empty?
    q = query.downcase
    results = results.select { |m|
      m['name'].to_s.downcase.include?(q) ||
      m['description'].to_s.downcase.include?(q) ||
      Array(m['tags']).any? { |t| t.downcase.include?(q) }
    }
  end

  return "No memories matching your search." if results.empty?

  results.map { |m|
    line = "**#{m['name']}**\n#{m['description']}"
    line += "\nTags: #{m['tags'].join(', ')}" if m['tags'] && !m['tags'].empty?
    line
  }.join("\n\n")
end

#save_memory(name:, description:, tags: []) ⇒ Object 



12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
 
# File 'lib/console_agent/tools/memory_tools.rb', line 12

def save_memory(name:, description:, tags: [])
  key = memory_key(name)
  existing = load_memory(key)

  frontmatter = {
    'name' => name,
    'tags' => Array(tags).empty? && existing ? (existing['tags'] || []) : Array(tags),
    'created_at' => existing ? existing['created_at'] : Time.now.utc.iso8601
  }
  frontmatter['updated_at'] = Time.now.utc.iso8601 if existing

  content = "---\n#{YAML.dump(frontmatter).sub("---\n", '').strip}\n---\n\n#{description}\n"
  @storage.write(key, content)

  path = @storage.respond_to?(:root_path) ? File.join(@storage.root_path, key) : key
  if existing
    "Memory updated: \"#{name}\" (#{path})"
  else
    "Memory saved: \"#{name}\" (#{path})"
  end
rescue Storage::StorageError => e
  "FAILED to save (#{e.message}). Add this manually to .console_agent/#{key}:\n" \
  "---\nname: #{name}\ntags: #{Array(tags).inspect}\n---\n\n#{description}"
end