Class: Llmemory::LongTerm::Episodic::Storages::FileStorage

Inherits:
Base
  • Object
show all
Defined in:
lib/llmemory/long_term/episodic/storages/file_storage.rb

Instance Method Summary collapse

Constructor Details

#initialize(base_path: nil) ⇒ FileStorage

Returns a new instance of FileStorage.



13
14
15
16
 
# File 'lib/llmemory/long_term/episodic/storages/file_storage.rb', line 13

def initialize(base_path: nil)
  @base_path = base_path || Llmemory.configuration.long_term_storage_path || "./llmemory_data"
  @base_path = File.expand_path(@base_path)
end

Instance Method Details

#archive_episodes(user_id, ids) ⇒ Object 



56
57
58
59
60
61
62
63
64
65
66
 
# File 'lib/llmemory/long_term/episodic/storages/file_storage.rb', line 56

def archive_episodes(user_id, ids)
  Array(ids).map(&:to_s).count do |id|
    path = episode_path(user_id, id)
    next false unless File.file?(path)
    data = JSON.parse(File.read(path))
    next false if data["archived_at"]
    data["archived_at"] = Time.now.iso8601
    File.write(path, JSON.generate(data))
    true
  end
end

#count_episodes(user_id) ⇒ Object 



43
44
45
 
# File 'lib/llmemory/long_term/episodic/storages/file_storage.rb', line 43

def count_episodes(user_id)
  active_episodes(user_id).size
end

#delete_episodes(user_id, ids) ⇒ Object 



47
48
49
50
51
52
53
54
 
# File 'lib/llmemory/long_term/episodic/storages/file_storage.rb', line 47

def delete_episodes(user_id, ids)
  Array(ids).map(&:to_s).count do |id|
    path = episode_path(user_id, id)
    next false unless File.file?(path)
    File.delete(path)
    true
  end
end

#expired_episode_ids(user_id, cutoff:) ⇒ Object 



68
69
70
71
72
 
# File 'lib/llmemory/long_term/episodic/storages/file_storage.rb', line 68

def expired_episode_ids(user_id, cutoff:)
  active_episodes(user_id)
    .select { |e| (e[:created_at] || Time.now) < cutoff }
    .map { |e| e[:id].to_s }
end

#get_episode(user_id, id) ⇒ Object 



26
27
28
29
30
 
# File 'lib/llmemory/long_term/episodic/storages/file_storage.rb', line 26

def get_episode(user_id, id)
  path = episode_path(user_id, id)
  return nil unless File.file?(path)
  load_episode(path)
end

#list_episodes(user_id, limit: nil, offset: nil) ⇒ Object 



32
33
34
35
36
 
# File 'lib/llmemory/long_term/episodic/storages/file_storage.rb', line 32

def list_episodes(user_id, limit: nil, offset: nil)
  sorted = active_episodes(user_id).sort_by { |e| e[:created_at] }.reverse
  sorted = sorted.drop(offset.to_i) if offset && offset.to_i.positive?
  limit && limit.to_i.positive? ? sorted.first(limit.to_i) : sorted
end

#list_users ⇒ Object 



74
75
76
77
 
# File 'lib/llmemory/long_term/episodic/storages/file_storage.rb', line 74

def list_users
  return [] unless Dir.exist?(@base_path)
  Dir.children(@base_path).select { |d| Dir.exist?(File.join(@base_path, d, "episodes")) }
end

#save_episode(user_id, episode) ⇒ Object 



18
19
20
21
22
23
24
 
# File 'lib/llmemory/long_term/episodic/storages/file_storage.rb', line 18

def save_episode(user_id, episode)
  id = episode[:id] || episode["id"] || "ep_#{next_seq(user_id)}"
  data = stringify_for_json(episode).merge("id" => id, "user_id" => user_id)
  data["created_at"] ||= Time.now.iso8601
  File.write(episode_path(user_id, id), JSON.generate(data))
  id
end

#search_episodes(user_id, query) ⇒ Object 



38
39
40
41
 
# File 'lib/llmemory/long_term/episodic/storages/file_storage.rb', line 38

def search_episodes(user_id, query)
  return list_episodes(user_id) if query.to_s.strip.empty?
  active_episodes(user_id).select { |e| Llmemory::Tokenizer.matches?(episode_text(e), query) }
end