@hasna/mementos 0.14.13 → 0.14.14

package/dist/cli/index.js

@@ -15088,6 +15088,155 @@ var init_projects = __esm(() => {
   init_database();
 });
 
+// src/db/agents.ts
+var exports_agents = {};
+__export(exports_agents, {
+  updateAgent: () => updateAgent,
+  touchAgent: () => touchAgent,
+  registerAgent: () => registerAgent,
+  listAgentsByProject: () => listAgentsByProject,
+  listAgents: () => listAgents,
+  getAgent: () => getAgent
+});
+function parseAgentRow(row) {
+  return {
+    id: row["id"],
+    name: row["name"],
+    session_id: row["session_id"] || null,
+    description: row["description"] || null,
+    role: row["role"] || null,
+    metadata: JSON.parse(row["metadata"] || "{}"),
+    active_project_id: row["active_project_id"] || null,
+    created_at: row["created_at"],
+    last_seen_at: row["last_seen_at"]
+  };
+}
+function registerAgent(name, sessionId, description, role, projectId, db) {
+  const d = db || getDatabase();
+  const timestamp = now();
+  const normalizedName = name.trim().toLowerCase();
+  if (projectId) {
+    const resolvedProjectId = resolvePartialId(d, "projects", projectId);
+    if (!resolvedProjectId) {
+      throw new Error(`Project not found: ${projectId}`);
+    }
+    projectId = resolvedProjectId;
+  }
+  const existing = d.query("SELECT * FROM agents WHERE LOWER(name) = ?").get(normalizedName);
+  if (existing) {
+    const existingId = existing["id"];
+    const existingSessionId = existing["session_id"] || null;
+    const existingLastSeen = existing["last_seen_at"];
+    if (sessionId && existingSessionId && existingSessionId !== sessionId) {
+      const lastSeenMs = new Date(existingLastSeen).getTime();
+      const nowMs = Date.now();
+      if (nowMs - lastSeenMs < CONFLICT_WINDOW_MS) {
+        throw new AgentConflictError({
+          existing_id: existingId,
+          existing_name: normalizedName,
+          last_seen_at: existingLastSeen,
+          session_hint: existingSessionId.slice(0, 8),
+          working_dir: null
+        });
+      }
+    }
+    d.run("UPDATE agents SET last_seen_at = ?, session_id = ? WHERE id = ?", [
+      timestamp,
+      sessionId ?? existingSessionId,
+      existingId
+    ]);
+    if (description) {
+      d.run("UPDATE agents SET description = ? WHERE id = ?", [description, existingId]);
+    }
+    if (role) {
+      d.run("UPDATE agents SET role = ? WHERE id = ?", [role, existingId]);
+    }
+    if (projectId !== undefined) {
+      d.run("UPDATE agents SET active_project_id = ? WHERE id = ?", [projectId, existingId]);
+    }
+    return getAgent(existingId, d);
+  }
+  const id = shortUuid();
+  d.run("INSERT INTO agents (id, name, session_id, description, role, active_project_id, created_at, last_seen_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?)", [id, normalizedName, sessionId ?? null, description || null, role || "agent", projectId ?? null, timestamp, timestamp]);
+  return getAgent(id, d);
+}
+function getAgent(idOrName, db) {
+  const d = db || getDatabase();
+  let row = d.query("SELECT * FROM agents WHERE id = ?").get(idOrName);
+  if (row)
+    return parseAgentRow(row);
+  row = d.query("SELECT * FROM agents WHERE LOWER(name) = ?").get(idOrName.trim().toLowerCase());
+  if (row)
+    return parseAgentRow(row);
+  const rows = d.query("SELECT * FROM agents WHERE id LIKE ?").all(`${idOrName}%`);
+  if (rows.length === 1)
+    return parseAgentRow(rows[0]);
+  return null;
+}
+function listAgents(db) {
+  const d = db || getDatabase();
+  const rows = d.query("SELECT * FROM agents ORDER BY last_seen_at DESC").all();
+  return rows.map(parseAgentRow);
+}
+function touchAgent(idOrName, db) {
+  const d = db || getDatabase();
+  const agent = getAgent(idOrName, d);
+  if (!agent)
+    return;
+  d.run("UPDATE agents SET last_seen_at = ? WHERE id = ?", [now(), agent.id]);
+}
+function listAgentsByProject(projectId, db) {
+  const d = db || getDatabase();
+  const resolvedId = resolvePartialId(d, "projects", projectId) || projectId;
+  const rows = d.query("SELECT * FROM agents WHERE active_project_id = ? ORDER BY last_seen_at DESC").all(resolvedId);
+  return rows.map(parseAgentRow);
+}
+function updateAgent(id, updates, db) {
+  const d = db || getDatabase();
+  const agent = getAgent(id, d);
+  if (!agent)
+    return null;
+  const timestamp = now();
+  if (updates.name) {
+    const normalizedNewName = updates.name.trim().toLowerCase();
+    if (normalizedNewName !== agent.name) {
+      const existing = d.query("SELECT id FROM agents WHERE LOWER(name) = ? AND id != ?").get(normalizedNewName, agent.id);
+      if (existing) {
+        throw new Error(`Agent name already taken: ${normalizedNewName}`);
+      }
+      d.run("UPDATE agents SET name = ? WHERE id = ?", [normalizedNewName, agent.id]);
+    }
+  }
+  if (updates.description !== undefined) {
+    d.run("UPDATE agents SET description = ? WHERE id = ?", [updates.description, agent.id]);
+  }
+  if (updates.role !== undefined) {
+    d.run("UPDATE agents SET role = ? WHERE id = ?", [updates.role, agent.id]);
+  }
+  if (updates.metadata !== undefined) {
+    d.run("UPDATE agents SET metadata = ? WHERE id = ?", [JSON.stringify(updates.metadata), agent.id]);
+  }
+  if ("active_project_id" in updates) {
+    let resolvedProjectId = updates.active_project_id ?? null;
+    if (resolvedProjectId) {
+      const fullId = resolvePartialId(d, "projects", resolvedProjectId);
+      if (!fullId) {
+        throw new Error(`Project not found: ${resolvedProjectId}`);
+      }
+      resolvedProjectId = fullId;
+    }
+    d.run("UPDATE agents SET active_project_id = ? WHERE id = ?", [resolvedProjectId, agent.id]);
+  }
+  d.run("UPDATE agents SET last_seen_at = ? WHERE id = ?", [timestamp, agent.id]);
+  return getAgent(agent.id, d);
+}
+var CONFLICT_WINDOW_MS;
+var init_agents = __esm(() => {
+  init_types2();
+  init_database();
+  CONFLICT_WINDOW_MS = 30 * 60 * 1000;
+});
+
 // src/db/entities.ts
 function parseEntityRow(row) {
   return {
@@ -16369,155 +16518,6 @@ var init_helpers = __esm(() => {
   VALID_CATEGORIES = ["preference", "fact", "knowledge", "history"];
 });
 
-// src/db/agents.ts
-var exports_agents = {};
-__export(exports_agents, {
-  updateAgent: () => updateAgent,
-  touchAgent: () => touchAgent,
-  registerAgent: () => registerAgent,
-  listAgentsByProject: () => listAgentsByProject,
-  listAgents: () => listAgents,
-  getAgent: () => getAgent
-});
-function parseAgentRow(row) {
-  return {
-    id: row["id"],
-    name: row["name"],
-    session_id: row["session_id"] || null,
-    description: row["description"] || null,
-    role: row["role"] || null,
-    metadata: JSON.parse(row["metadata"] || "{}"),
-    active_project_id: row["active_project_id"] || null,
-    created_at: row["created_at"],
-    last_seen_at: row["last_seen_at"]
-  };
-}
-function registerAgent(name, sessionId, description, role, projectId, db) {
-  const d = db || getDatabase();
-  const timestamp = now();
-  const normalizedName = name.trim().toLowerCase();
-  if (projectId) {
-    const resolvedProjectId = resolvePartialId(d, "projects", projectId);
-    if (!resolvedProjectId) {
-      throw new Error(`Project not found: ${projectId}`);
-    }
-    projectId = resolvedProjectId;
-  }
-  const existing = d.query("SELECT * FROM agents WHERE LOWER(name) = ?").get(normalizedName);
-  if (existing) {
-    const existingId = existing["id"];
-    const existingSessionId = existing["session_id"] || null;
-    const existingLastSeen = existing["last_seen_at"];
-    if (sessionId && existingSessionId && existingSessionId !== sessionId) {
-      const lastSeenMs = new Date(existingLastSeen).getTime();
-      const nowMs = Date.now();
-      if (nowMs - lastSeenMs < CONFLICT_WINDOW_MS) {
-        throw new AgentConflictError({
-          existing_id: existingId,
-          existing_name: normalizedName,
-          last_seen_at: existingLastSeen,
-          session_hint: existingSessionId.slice(0, 8),
-          working_dir: null
-        });
-      }
-    }
-    d.run("UPDATE agents SET last_seen_at = ?, session_id = ? WHERE id = ?", [
-      timestamp,
-      sessionId ?? existingSessionId,
-      existingId
-    ]);
-    if (description) {
-      d.run("UPDATE agents SET description = ? WHERE id = ?", [description, existingId]);
-    }
-    if (role) {
-      d.run("UPDATE agents SET role = ? WHERE id = ?", [role, existingId]);
-    }
-    if (projectId !== undefined) {
-      d.run("UPDATE agents SET active_project_id = ? WHERE id = ?", [projectId, existingId]);
-    }
-    return getAgent(existingId, d);
-  }
-  const id = shortUuid();
-  d.run("INSERT INTO agents (id, name, session_id, description, role, active_project_id, created_at, last_seen_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?)", [id, normalizedName, sessionId ?? null, description || null, role || "agent", projectId ?? null, timestamp, timestamp]);
-  return getAgent(id, d);
-}
-function getAgent(idOrName, db) {
-  const d = db || getDatabase();
-  let row = d.query("SELECT * FROM agents WHERE id = ?").get(idOrName);
-  if (row)
-    return parseAgentRow(row);
-  row = d.query("SELECT * FROM agents WHERE LOWER(name) = ?").get(idOrName.trim().toLowerCase());
-  if (row)
-    return parseAgentRow(row);
-  const rows = d.query("SELECT * FROM agents WHERE id LIKE ?").all(`${idOrName}%`);
-  if (rows.length === 1)
-    return parseAgentRow(rows[0]);
-  return null;
-}
-function listAgents(db) {
-  const d = db || getDatabase();
-  const rows = d.query("SELECT * FROM agents ORDER BY last_seen_at DESC").all();
-  return rows.map(parseAgentRow);
-}
-function touchAgent(idOrName, db) {
-  const d = db || getDatabase();
-  const agent = getAgent(idOrName, d);
-  if (!agent)
-    return;
-  d.run("UPDATE agents SET last_seen_at = ? WHERE id = ?", [now(), agent.id]);
-}
-function listAgentsByProject(projectId, db) {
-  const d = db || getDatabase();
-  const resolvedId = resolvePartialId(d, "projects", projectId) || projectId;
-  const rows = d.query("SELECT * FROM agents WHERE active_project_id = ? ORDER BY last_seen_at DESC").all(resolvedId);
-  return rows.map(parseAgentRow);
-}
-function updateAgent(id, updates, db) {
-  const d = db || getDatabase();
-  const agent = getAgent(id, d);
-  if (!agent)
-    return null;
-  const timestamp = now();
-  if (updates.name) {
-    const normalizedNewName = updates.name.trim().toLowerCase();
-    if (normalizedNewName !== agent.name) {
-      const existing = d.query("SELECT id FROM agents WHERE LOWER(name) = ? AND id != ?").get(normalizedNewName, agent.id);
-      if (existing) {
-        throw new Error(`Agent name already taken: ${normalizedNewName}`);
-      }
-      d.run("UPDATE agents SET name = ? WHERE id = ?", [normalizedNewName, agent.id]);
-    }
-  }
-  if (updates.description !== undefined) {
-    d.run("UPDATE agents SET description = ? WHERE id = ?", [updates.description, agent.id]);
-  }
-  if (updates.role !== undefined) {
-    d.run("UPDATE agents SET role = ? WHERE id = ?", [updates.role, agent.id]);
-  }
-  if (updates.metadata !== undefined) {
-    d.run("UPDATE agents SET metadata = ? WHERE id = ?", [JSON.stringify(updates.metadata), agent.id]);
-  }
-  if ("active_project_id" in updates) {
-    let resolvedProjectId = updates.active_project_id ?? null;
-    if (resolvedProjectId) {
-      const fullId = resolvePartialId(d, "projects", resolvedProjectId);
-      if (!fullId) {
-        throw new Error(`Project not found: ${resolvedProjectId}`);
-      }
-      resolvedProjectId = fullId;
-    }
-    d.run("UPDATE agents SET active_project_id = ? WHERE id = ?", [resolvedProjectId, agent.id]);
-  }
-  d.run("UPDATE agents SET last_seen_at = ? WHERE id = ?", [timestamp, agent.id]);
-  return getAgent(agent.id, d);
-}
-var CONFLICT_WINDOW_MS;
-var init_agents = __esm(() => {
-  init_types2();
-  init_database();
-  CONFLICT_WINDOW_MS = 30 * 60 * 1000;
-});
-
 // src/lib/poll.ts
 var exports_poll = {};
 __export(exports_poll, {
@@ -20957,6 +20957,7 @@ import { fileURLToPath as fileURLToPath3 } from "url";
 init_database();
 init_memories();
 init_projects();
+init_agents();
 init_search();
 import chalk2 from "chalk";
 import { resolve as resolve3 } from "path";
@@ -21048,6 +21049,11 @@ function registerMemoryCommands(program2) {
       }
       const explicitTags = opts.tags ? opts.tags.split(",").map((t) => t.trim()) : undefined;
       const mergedTags = explicitTags ? explicitTags : templateDefaults?.tags && templateDefaults.tags.length > 0 ? templateDefaults.tags : undefined;
+      let resolvedAgentId;
+      if (globalOpts.agent) {
+        const ag = getAgent(globalOpts.agent);
+        resolvedAgentId = ag?.id;
+      }
       const input = {
         key,
         value,
@@ -21058,7 +21064,7 @@ function registerMemoryCommands(program2) {
         summary: opts.summary,
         ttl_ms: opts.ttl ? parseDuration(opts.ttl) : undefined,
         source: opts.source,
-        agent_id: globalOpts.agent,
+        agent_id: resolvedAgentId,
         session_id: globalOpts.session
       };
       if (globalOpts.project) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hasna/mementos",
-  "version": "0.14.13",
+  "version": "0.14.14",
   "description": "Universal memory system for AI agents - CLI + MCP server + library API",
   "type": "module",
   "main": "dist/index.js",