llmemory 0.1.1 → 0.1.7

data/lib/llmemory/cli/commands/users.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+require_relative "base"
+
+module Llmemory
+  module Cli
+    module Commands
+      class Users < Base
+        def option_parser(parser)
+          parser.on("--store TYPE", "Short-term store: memory, redis, postgres, active_record") do |v|
+            @store_type = v
+          end
+        end
+
+        def execute(_argv, _opts)
+          store = short_term_store(@store_type)
+          users = store.list_users
+          if users.empty?
+            puts "No users with short-term memory found."
+          else
+            users.each { |u| puts u }
+          end
+        end
+      end
+    end
+  end
+end

data/lib/llmemory/cli.rb

@@ -0,0 +1,76 @@
+# frozen_string_literal: true
+
+require "optparse"
+require_relative "cli/commands/base"
+require_relative "cli/commands/users"
+require_relative "cli/commands/short_term"
+require_relative "cli/commands/long_term"
+require_relative "cli/commands/stats"
+require_relative "cli/commands/search"
+
+module Llmemory
+  class CLI
+    def self.run(argv)
+      new.run(argv)
+    end
+
+    def run(argv)
+      argv = argv.dup
+      return print_help if argv.empty?
+      first = argv.first
+      if first == "--help" || first == "-h"
+        print_help
+        return
+      end
+      subcommand = argv.shift
+      command_class = find_command(subcommand)
+      if command_class
+        command_class.new.run(argv)
+      else
+        $stderr.puts "llmemory: unknown command '#{subcommand}'"
+        $stderr.puts "Run 'llmemory --help' for usage."
+        exit 1
+      end
+    end
+
+    private
+
+    def find_command(name)
+      normalized = name.tr("-", "_").downcase
+      {
+        "users" => Cli::Commands::Users,
+        "short_term" => Cli::Commands::ShortTerm,
+        "facts" => Cli::Commands::LongTerm::Facts,
+        "categories" => Cli::Commands::LongTerm::Categories,
+        "resources" => Cli::Commands::LongTerm::Resources,
+        "nodes" => Cli::Commands::LongTerm::Nodes,
+        "edges" => Cli::Commands::LongTerm::Edges,
+        "graph" => Cli::Commands::LongTerm::Graph,
+        "search" => Cli::Commands::Search,
+        "stats" => Cli::Commands::Stats
+      }[normalized]
+    end
+
+    def print_help
+      puts <<~HELP
+        llmemory - Inspect llmemory storage
+
+        Usage: llmemory <command> [options] [arguments]
+
+        Commands:
+          users                    List users with memory
+          short-term USER_ID       Inspect short-term memory (use --list-sessions to list sessions)
+          facts USER_ID            List long-term facts (file-based)
+          categories USER_ID      List long-term categories (file-based)
+          resources USER_ID        List long-term resources (file-based)
+          nodes USER_ID            List graph nodes (graph-based)
+          edges USER_ID            List graph edges (graph-based)
+          graph USER_ID            Export graph (--format dot|json)
+          search USER_ID "query"   Search in memory
+          stats [USER_ID]          Show statistics
+
+        Run 'llmemory <command> --help' for command-specific options.
+      HELP
+    end
+  end
+end

data/lib/llmemory/dashboard/app/controllers/llmemory/dashboard/application_controller.rb

@@ -0,0 +1,83 @@
+# frozen_string_literal: true
+
+module Llmemory
+  module Dashboard
+    class ApplicationController < ActionController::Base
+      helper_method :short_term_store, :file_based_storage, :graph_based_storage
+
+      protected
+
+      def short_term_store
+        @short_term_store ||= build_short_term_store
+      end
+
+      def file_based_storage
+        @file_based_storage ||= build_file_based_storage
+      end
+
+      def graph_based_storage
+        @graph_based_storage ||= build_graph_based_storage
+      end
+
+      def long_term_type
+        Llmemory.configuration.long_term_type.to_s
+      end
+
+      def graph_based?
+        long_term_type == "graph_based"
+      end
+
+      def file_based?
+        long_term_type == "file_based"
+      end
+
+      private
+
+      def build_short_term_store
+        type = (Llmemory.configuration.short_term_store).to_s.to_sym
+        case type
+        when :memory then Llmemory::ShortTerm::Stores::MemoryStore.new
+        when :redis then Llmemory::ShortTerm::Stores::RedisStore.new
+        when :postgres then Llmemory::ShortTerm::Stores::PostgresStore.new
+        when :active_record, :activerecord
+          require "llmemory/short_term/stores/active_record_store"
+          Llmemory::ShortTerm::Stores::ActiveRecordStore.new
+        else
+          Llmemory::ShortTerm::Stores::MemoryStore.new
+        end
+      end
+
+      def build_file_based_storage
+        type = (Llmemory.configuration.long_term_store).to_s.to_sym
+        case type
+        when :memory then Llmemory::LongTerm::FileBased::Storages::MemoryStorage.new
+        when :file
+          Llmemory::LongTerm::FileBased::Storages::FileStorage.new(
+            base_path: Llmemory.configuration.long_term_storage_path
+          )
+        when :postgres, :database
+          Llmemory::LongTerm::FileBased::Storages::DatabaseStorage.new(
+            database_url: Llmemory.configuration.database_url
+          )
+        when :active_record, :activerecord
+          require "llmemory/long_term/file_based/storages/active_record_storage"
+          Llmemory::LongTerm::FileBased::Storages::ActiveRecordStorage.new
+        else
+          Llmemory::LongTerm::FileBased::Storages::MemoryStorage.new
+        end
+      end
+
+      def build_graph_based_storage
+        type = (Llmemory.configuration.long_term_store).to_s.to_sym
+        case type
+        when :memory then Llmemory::LongTerm::GraphBased::Storages::MemoryStorage.new
+        when :active_record, :activerecord
+          require "llmemory/long_term/graph_based/storages/active_record_storage"
+          Llmemory::LongTerm::GraphBased::Storages::ActiveRecordStorage.new
+        else
+          Llmemory::LongTerm::GraphBased::Storages::MemoryStorage.new
+        end
+      end
+    end
+  end
+end

data/lib/llmemory/dashboard/app/controllers/llmemory/dashboard/graph_controller.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module Llmemory
+  module Dashboard
+    class GraphController < ApplicationController
+      def index
+        @user_id = params[:user_id]
+        @nodes = graph_based_storage.list_nodes(@user_id, limit: 200)
+        @edges = graph_based_storage.list_edges(@user_id, limit: 300)
+      end
+
+      def show
+        index
+        render :index
+      end
+    end
+  end
+end

data/lib/llmemory/dashboard/app/controllers/llmemory/dashboard/long_term_controller.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+module Llmemory
+  module Dashboard
+    class LongTermController < ApplicationController
+      before_action :ensure_file_based
+
+      def index
+        @user_id = params[:user_id]
+        @items = file_based_storage.list_items(user_id: @user_id, limit: 100)
+        @resources = file_based_storage.list_resources(user_id: @user_id, limit: 50)
+      end
+
+      def categories
+        @user_id = params[:user_id]
+        @category_names = file_based_storage.list_categories(@user_id)
+        @categories_content = {}
+        @category_names.each do |name|
+          @categories_content[name] = file_based_storage.load_category(@user_id, name)
+        end
+      end
+
+      private
+
+      def ensure_file_based
+        return if file_based?
+        redirect_to root_path, alert: "Long-term file-based view only available when long_term_type is file_based."
+      end
+    end
+  end
+end

data/lib/llmemory/dashboard/app/controllers/llmemory/dashboard/search_controller.rb

@@ -0,0 +1,58 @@
+# frozen_string_literal: true
+
+module Llmemory
+  module Dashboard
+    class SearchController < ApplicationController
+      def index
+        @query = params[:q].to_s.strip
+        @user_id = params[:user_id].presence
+
+        if @query.empty?
+          @short_term_results = []
+          @file_based_results = []
+          @graph_results = []
+          return
+        end
+
+        if @user_id.present?
+          @short_term_results = search_short_term(@user_id)
+          if file_based?
+            @file_based_results = file_based_storage.search_items(@user_id, @query) +
+              file_based_storage.search_resources(@user_id, @query).map { |r| r.merge(type: "resource") }
+          else
+            @file_based_results = []
+          end
+          if graph_based?
+            nodes = graph_based_storage.list_nodes(@user_id)
+            q = @query.downcase
+            @graph_results = nodes.select { |n| (n.respond_to?(:name) ? n.name : n[:name]).to_s.downcase.include?(q) }
+          else
+            @graph_results = []
+          end
+        else
+          @short_term_results = []
+          @file_based_results = []
+          @graph_results = []
+        end
+      end
+
+      private
+
+      def search_short_term(user_id)
+        sessions = short_term_store.list_sessions(user_id: user_id)
+        results = []
+        sessions.each do |session_id|
+          state = short_term_store.load(user_id, session_id)
+          next unless state
+          messages = state[:messages] || state["messages"] || []
+          messages.each do |m|
+            content = (m[:content] || m["content"]).to_s
+            next unless content.downcase.include?(@query.downcase)
+            results << { session_id: session_id, role: m[:role] || m["role"], content: content }
+          end
+        end
+        results
+      end
+    end
+  end
+end

data/lib/llmemory/dashboard/app/controllers/llmemory/dashboard/short_term_controller.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module Llmemory
+  module Dashboard
+    class ShortTermController < ApplicationController
+      def show
+        @user_id = params[:user_id]
+        @session_id = params[:session_id].presence || "default"
+        state = short_term_store.load(@user_id, @session_id)
+        @messages = state ? (state[:messages] || state["messages"] || []) : []
+        @sessions = short_term_store.list_sessions(user_id: @user_id)
+      end
+    end
+  end
+end

data/lib/llmemory/dashboard/app/controllers/llmemory/dashboard/stats_controller.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module Llmemory
+  module Dashboard
+    class StatsController < ApplicationController
+      def index
+        @user_id = params[:user_id]
+        @sessions = short_term_store.list_sessions(user_id: @user_id)
+
+        if graph_based?
+          @nodes_count = graph_based_storage.count_nodes(@user_id)
+          @edges_count = graph_based_storage.count_edges(@user_id)
+        else
+          @items_count = file_based_storage.count_items(user_id: @user_id)
+          @categories_count = file_based_storage.list_categories(@user_id).size
+          @resources_count = file_based_storage.list_resources(user_id: @user_id).size
+        end
+      end
+    end
+  end
+end

data/lib/llmemory/dashboard/app/controllers/llmemory/dashboard/users_controller.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+module Llmemory
+  module Dashboard
+    class UsersController < ApplicationController
+      def index
+        @users = short_term_store.list_users
+        @users = (@users + long_term_user_ids).uniq
+      end
+
+      def show
+        @user_id = params[:user_id]
+        @sessions = short_term_store.list_sessions(user_id: @user_id)
+      end
+
+      protected
+
+      def long_term_user_ids
+        if graph_based?
+          graph_based_storage.list_users
+        else
+          file_based_storage.list_users
+        end
+      end
+    end
+  end
+end

data/lib/llmemory/dashboard/app/views/layouts/application.html.erb

@@ -0,0 +1,53 @@
+<!DOCTYPE html>
+<html>
+<head>
+  <meta charset="utf-8">
+  <meta name="viewport" content="width=device-width, initial-scale=1">
+  <title>Llmemory Dashboard</title>
+  <style>
+    * { box-sizing: border-box; }
+    body { font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, sans-serif; margin: 0; background: #f5f5f5; color: #333; line-height: 1.5; }
+    .header { background: #2c3e50; color: #ecf0f1; padding: 12px 20px; }
+    .header a { color: #ecf0f1; text-decoration: none; }
+    .header a:hover { text-decoration: underline; }
+    .header h1 { margin: 0; font-size: 1.25rem; font-weight: 600; }
+    .container { max-width: 1200px; margin: 0 auto; padding: 20px; }
+    .card { background: #fff; border-radius: 6px; box-shadow: 0 1px 3px rgba(0,0,0,0.1); padding: 20px; margin-bottom: 20px; }
+    table { width: 100%; border-collapse: collapse; }
+    th, td { padding: 10px 12px; text-align: left; border-bottom: 1px solid #eee; }
+    th { background: #f8f9fa; font-weight: 600; }
+    tr:hover { background: #f8f9fa; }
+    a { color: #3498db; }
+    a:hover { color: #2980b9; }
+    .nav-links { margin-top: 10px; }
+    .nav-links a { margin-right: 16px; }
+    .empty { color: #999; font-style: italic; }
+    .message { padding: 8px 12px; margin: 8px 0; border-radius: 4px; }
+    .message.user { background: #e8f4fd; border-left: 4px solid #3498db; }
+    .message.assistant { background: #f0f9f0; border-left: 4px solid #27ae60; }
+    pre { background: #f4f4f4; padding: 12px; border-radius: 4px; overflow-x: auto; font-size: 13px; }
+    .stats-grid { display: grid; grid-template-columns: repeat(auto-fill, minmax(160px, 1fr)); gap: 16px; }
+    .stat-box { background: #f8f9fa; padding: 16px; border-radius: 6px; text-align: center; }
+    .stat-box .value { font-size: 1.5rem; font-weight: 700; color: #2c3e50; }
+    .stat-box .label { font-size: 0.875rem; color: #666; margin-top: 4px; }
+    form { margin: 16px 0; }
+    input[type=text] { padding: 8px 12px; border: 1px solid #ddd; border-radius: 4px; width: 300px; }
+    input[type=submit] { padding: 8px 16px; background: #3498db; color: #fff; border: none; border-radius: 4px; cursor: pointer; }
+    input[type=submit]:hover { background: #2980b9; }
+  </style>
+</head>
+<body>
+  <div class="header">
+    <h1><a href="<%= main_app.respond_to?(:root_path) ? main_app.root_path : '#' %>">Llmemory</a> Dashboard</h1>
+    <div class="nav-links">
+      <a href="<%= root_path %>">Users</a>
+      <a href="<%= search_path %>">Search</a>
+    </div>
+  </div>
+  <div class="container">
+    <% if flash[:alert] %><p class="message" style="background:#fde8e8;border-left-color:#e74c3c;"><%= flash[:alert] %></p><% end %>
+    <% if flash[:notice] %><p class="message" style="background:#e8fdf0;border-left-color:#27ae60;"><%= flash[:notice] %></p><% end %>
+    <%= yield %>
+  </div>
+</body>
+</html>

data/lib/llmemory/dashboard/app/views/llmemory/dashboard/graph/index.html.erb

@@ -0,0 +1,41 @@
+<div class="card">
+  <h2>Knowledge graph: <%= @user_id %></h2>
+  <h3>Nodes (<%= @nodes.size %>)</h3>
+  <% if @nodes.blank? %>
+    <p class="empty">No nodes.</p>
+  <% else %>
+    <table>
+      <thead>
+        <tr><th>ID</th><th>Type</th><th>Name</th></tr>
+      </thead>
+      <tbody>
+        <% @nodes.each do |n| %>
+          <tr>
+            <td><%= n.respond_to?(:id) ? n.id : n[:id] %></td>
+            <td><%= n.respond_to?(:entity_type) ? n.entity_type : n[:entity_type] %></td>
+            <td><%= n.respond_to?(:name) ? n.name : n[:name] %></td>
+          </tr>
+        <% end %>
+      </tbody>
+    </table>
+  <% end %>
+  <h3>Edges (<%= @edges.size %>)</h3>
+  <% if @edges.blank? %>
+    <p class="empty">No edges.</p>
+  <% else %>
+    <table>
+      <thead>
+        <tr><th>Subject</th><th>Predicate</th><th>Object</th></tr>
+      </thead>
+      <tbody>
+        <% @edges.each do |e| %>
+          <tr>
+            <td><%= e.respond_to?(:subject_id) ? e.subject_id : e[:subject_id] %></td>
+            <td><%= e.respond_to?(:predicate) ? e.predicate : e[:predicate] %></td>
+            <td><%= e.respond_to?(:object_id) ? e.object_id : e[:object_id] %></td>
+          </tr>
+        <% end %>
+      </tbody>
+    </table>
+  <% end %>
+</div>

data/lib/llmemory/dashboard/app/views/llmemory/dashboard/long_term/categories.html.erb

@@ -0,0 +1,12 @@
+<div class="card">
+  <h2>Categories: <%= @user_id %></h2>
+  <p><a href="<%= user_long_term_path(@user_id) %>">Back to long-term</a></p>
+  <% if @category_names.blank? %>
+    <p class="empty">No categories.</p>
+  <% else %>
+    <% @category_names.each do |name| %>
+      <h3><%= name %></h3>
+      <pre><%= ERB::Util.html_escape(@categories_content[name].to_s) %></pre>
+    <% end %>
+  <% end %>
+</div>

data/lib/llmemory/dashboard/app/views/llmemory/dashboard/long_term/index.html.erb

@@ -0,0 +1,42 @@
+<div class="card">
+  <h2>Long-term memory (file-based): <%= @user_id %></h2>
+  <p><a href="<%= user_long_term_categories_path(@user_id) %>">View categories</a></p>
+  <h3>Items (facts)</h3>
+  <% if @items.blank? %>
+    <p class="empty">No items.</p>
+  <% else %>
+    <table>
+      <thead>
+        <tr><th>Category</th><th>Content</th><th>Created</th></tr>
+      </thead>
+      <tbody>
+        <% @items.each do |i| %>
+          <tr>
+            <td><%= i[:category] || i["category"] %></td>
+            <td><%= truncate((i[:content] || i["content"]).to_s, length: 120) %></td>
+            <td><%= i[:created_at] || i["created_at"] %></td>
+          </tr>
+        <% end %>
+      </tbody>
+    </table>
+  <% end %>
+  <h3>Resources</h3>
+  <% if @resources.blank? %>
+    <p class="empty">No resources.</p>
+  <% else %>
+    <table>
+      <thead>
+        <tr><th>ID</th><th>Text (excerpt)</th><th>Created</th></tr>
+      </thead>
+      <tbody>
+        <% @resources.each do |r| %>
+          <tr>
+            <td><%= r[:id] || r["id"] %></td>
+            <td><%= truncate((r[:text] || r["text"]).to_s, length: 100) %></td>
+            <td><%= r[:created_at] || r["created_at"] %></td>
+          </tr>
+        <% end %>
+      </tbody>
+    </table>
+  <% end %>
+</div>

data/lib/llmemory/dashboard/app/views/llmemory/dashboard/search/index.html.erb

@@ -0,0 +1,43 @@
+<div class="card">
+  <h2>Search</h2>
+  <%= form_tag search_path, method: :get do %>
+    <%= text_field_tag :q, @query, placeholder: "Query" %>
+    <%= submit_tag "Search" %>
+  <% end %>
+  <% if @query.present? && @user_id.blank? %>
+    <p class="empty">Specify a user (e.g. add ?user_id=USER_ID to the URL) to search within that user's memory.</p>
+  <% end %>
+  <% if @query.present? && @user_id.present? %>
+    <h3>Short-term</h3>
+    <% if @short_term_results.blank? %>
+      <p class="empty">No matches.</p>
+    <% else %>
+      <% @short_term_results.each do |r| %>
+        <div class="message">
+          <strong>[<%= r[:session_id] %>] <%= r[:role] %>:</strong>
+          <%= truncate(r[:content], length: 200) %>
+        </div>
+      <% end %>
+    <% end %>
+    <% if file_based? %>
+      <h3>Long-term (file)</h3>
+      <% if @file_based_results.blank? %>
+        <p class="empty">No matches.</p>
+      <% else %>
+        <% @file_based_results.each do |r| %>
+          <p><%= r[:type] == "resource" ? "[resource]" : "[#{r[:category]}]" %> <%= truncate(r[:content] || r[:text], length: 150) %></p>
+        <% end %>
+      <% end %>
+    <% end %>
+    <% if graph_based? %>
+      <h3>Graph nodes</h3>
+      <% if @graph_results.blank? %>
+        <p class="empty">No matches.</p>
+      <% else %>
+        <% @graph_results.each do |n| %>
+          <p>[<%= n.respond_to?(:entity_type) ? n.entity_type : n[:entity_type] %>] <%= n.respond_to?(:name) ? n.name : n[:name] %></p>
+        <% end %>
+      <% end %>
+    <% end %>
+  <% end %>
+</div>

data/lib/llmemory/dashboard/app/views/llmemory/dashboard/short_term/show.html.erb

@@ -0,0 +1,25 @@
+<div class="card">
+  <h2>Short-term memory: <%= @user_id %></h2>
+  <p>
+    Session: <%= @session_id %>
+    <% if @sessions.any? %>
+      | Other sessions:
+      <% @sessions.each do |sess| %>
+        <% next if sess == @session_id %>
+        <a href="<%= user_short_term_path(@user_id) %>?session_id=<%= sess %>"><%= sess %></a>
+      <% end %>
+    <% end %>
+  </p>
+  <% if @messages.blank? %>
+    <p class="empty">No messages in this session.</p>
+  <% else %>
+    <% @messages.each do |m| %>
+      <% role = m[:role] || m["role"] %>
+      <% content = (m[:content] || m["content"]).to_s %>
+      <div class="message <%= role %>">
+        <strong><%= role %>:</strong>
+        <%= content.gsub(/\n/, "<br>").html_safe %>
+      </div>
+    <% end %>
+  <% end %>
+</div>

data/lib/llmemory/dashboard/app/views/llmemory/dashboard/stats/index.html.erb

@@ -0,0 +1,32 @@
+<div class="card">
+  <h2>Stats: <%= @user_id %></h2>
+  <div class="stats-grid">
+    <div class="stat-box">
+      <div class="value"><%= @sessions.size %></div>
+      <div class="label">Short-term sessions</div>
+    </div>
+    <% if graph_based? %>
+      <div class="stat-box">
+        <div class="value"><%= @nodes_count %></div>
+        <div class="label">Nodes</div>
+      </div>
+      <div class="stat-box">
+        <div class="value"><%= @edges_count %></div>
+        <div class="label">Edges</div>
+      </div>
+    <% else %>
+      <div class="stat-box">
+        <div class="value"><%= @items_count %></div>
+        <div class="label">Items</div>
+      </div>
+      <div class="stat-box">
+        <div class="value"><%= @categories_count %></div>
+        <div class="label">Categories</div>
+      </div>
+      <div class="stat-box">
+        <div class="value"><%= @resources_count %></div>
+        <div class="label">Resources</div>
+      </div>
+    <% end %>
+  </div>
+</div>

data/lib/llmemory/dashboard/app/views/llmemory/dashboard/users/index.html.erb

@@ -0,0 +1,30 @@
+<div class="card">
+  <h2>Users with memory</h2>
+  <% if @users.blank? %>
+    <p class="empty">No users found.</p>
+  <% else %>
+    <table>
+      <thead>
+        <tr><th>User ID</th><th>Actions</th></tr>
+      </thead>
+      <tbody>
+        <% @users.each do |user_id| %>
+          <tr>
+            <td><%= user_id %></td>
+            <td>
+              <a href="<%= user_path(user_id) %>">View</a>
+              <a href="<%= user_short_term_path(user_id) %>">Short-term</a>
+              <% if file_based? %>
+                <a href="<%= user_long_term_path(user_id) %>">Long-term</a>
+              <% end %>
+              <% if graph_based? %>
+                <a href="<%= user_graph_path(user_id) %>">Graph</a>
+              <% end %>
+              <a href="<%= user_stats_path(user_id) %>">Stats</a>
+            </td>
+          </tr>
+        <% end %>
+      </tbody>
+    </table>
+  <% end %>
+</div>

data/lib/llmemory/dashboard/app/views/llmemory/dashboard/users/show.html.erb

@@ -0,0 +1,14 @@
+<div class="card">
+  <h2>User: <%= @user_id %></h2>
+  <p>
+    <a href="<%= user_short_term_path(@user_id) %>">Short-term memory</a>
+    <% if file_based? %>
+      | <a href="<%= user_long_term_path(@user_id) %>">Long-term (file)</a>
+    <% end %>
+    <% if graph_based? %>
+      | <a href="<%= user_graph_path(@user_id) %>">Graph</a>
+    <% end %>
+    | <a href="<%= user_stats_path(@user_id) %>">Stats</a>
+  </p>
+  <p>Sessions: <%= @sessions.join(", ") %></p>
+</div>