Class: Llmemory::Cli::Commands::LongTerm::Edges

Inherits:
Base
  • Object
show all
Defined in:
lib/llmemory/cli/commands/long_term/edges.rb

Instance Method Summary collapse

Methods inherited from Base

#parse_options, #run

Instance Method Details

#execute(argv, _opts) ⇒ Object 



16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
 
# File 'lib/llmemory/cli/commands/long_term/edges.rb', line 16

def execute(argv, _opts)
  user_id = argv.first
  unless user_id
    $stderr.puts "Usage: llmemory edges USER_ID [--subject NODE_ID] [--limit N]"
    exit 1
  end

  storage = graph_based_storage(@store_type)
  edges = storage.list_edges(user_id, subject_id: @subject_id, limit: @limit)

  if edges.empty?
    puts "No edges found for user #{user_id}."
    return
  end

  edges.each do |e|
    id = e.respond_to?(:id) ? e.id : e[:id]
    subj = e.respond_to?(:subject_id) ? e.subject_id : e[:subject_id]
    pred = e.respond_to?(:predicate) ? e.predicate : e[:predicate]
    obj = e.respond_to?(:target_id) ? e.target_id : e[:object_id]
    puts "#{id}: #{subj} --#{pred}--> #{obj}"
  end
end

#option_parser(parser) ⇒ Object 



10
11
12
13
14
 
# File 'lib/llmemory/cli/commands/long_term/edges.rb', line 10

def option_parser(parser)
  parser.on("--subject NODE_ID", "Filter by subject node") { |v| @subject_id = v }
  parser.on("--limit N", Integer, "Max number of edges") { |v| @limit = v }
  parser.on("--store TYPE", "Storage type") { |v| @store_type = v }
end