@hasna/mementos 0.10.3 → 0.10.5

package/dist/cli/index.js

@@ -6920,6 +6920,13 @@ function listAgents(db) {
   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 updateAgent(id, updates, db) {
   const d = db || getDatabase();
   const agent = getAgent(id, d);
@@ -6960,6 +6967,40 @@ function updateAgent(id, updates, db) {
   return getAgent(agent.id, d);
 }
 
+// src/lib/focus.ts
+init_hooks();
+var sessionFocus = new Map;
+function setFocus(agentId, projectId) {
+  const previous = getFocusCached(agentId);
+  sessionFocus.set(agentId, projectId);
+  updateAgent(agentId, { active_project_id: projectId });
+  if (projectId && projectId !== previous) {
+    hookRegistry.runHooks("OnSessionStart", {
+      agentId,
+      projectId,
+      timestamp: Date.now()
+    });
+  } else if (!projectId && previous) {
+    hookRegistry.runHooks("OnSessionEnd", {
+      agentId,
+      projectId: previous,
+      timestamp: Date.now()
+    });
+  }
+}
+function getFocusCached(agentId) {
+  return sessionFocus.get(agentId) ?? null;
+}
+function getFocus(agentId) {
+  if (sessionFocus.has(agentId)) {
+    return sessionFocus.get(agentId) ?? null;
+  }
+  const agent = getAgent(agentId);
+  const projectId = agent?.active_project_id ?? null;
+  sessionFocus.set(agentId, projectId);
+  return projectId;
+}
+
 // src/db/projects.ts
 init_database();
 function parseProjectRow(row) {
@@ -10832,4 +10873,77 @@ session_end = "bun ${script}"`));
     console.log("Usage: mementos session setup-hook --claude | --codex");
   }
 });
+program2.command("heartbeat [agent-id]").description("Update last_seen_at to signal this agent is active").action((agentId) => {
+  const globalOpts = program2.opts();
+  const id = agentId || globalOpts.agent;
+  if (!id) {
+    process.stderr.write(`Agent ID required. Use --agent or pass as argument.
+`);
+    process.exit(1);
+  }
+  const agent = getAgent(id);
+  if (!agent) {
+    process.stderr.write(`Agent not found: ${id}
+`);
+    process.exit(1);
+  }
+  touchAgent(agent.id);
+  if (globalOpts.json)
+    console.log(JSON.stringify({ agent_id: agent.id, name: agent.name, last_seen_at: new Date().toISOString() }));
+  else
+    console.log(chalk.green(`\u2665 ${agent.name} (${agent.id}) \u2014 heartbeat sent`));
+});
+program2.command("set-focus [project]").description("Focus this agent on a project (or clear focus if no project given)").option("--agent <id>", "Agent ID").action((project, opts) => {
+  const globalOpts = program2.opts();
+  const agentId = opts?.agent || globalOpts.agent;
+  if (!agentId) {
+    process.stderr.write(`Agent ID required. Use --agent.
+`);
+    process.exit(1);
+  }
+  setFocus(agentId, project ?? null);
+  if (project)
+    console.log(chalk.green(`Focused: ${agentId} \u2192 project ${project}`));
+  else
+    console.log(chalk.dim(`Focus cleared for ${agentId}`));
+});
+program2.command("get-focus").description("Show the current project focus for an agent").option("--agent <id>", "Agent ID").action((opts) => {
+  const globalOpts = program2.opts();
+  const agentId = opts?.agent || globalOpts.agent;
+  if (!agentId) {
+    process.stderr.write(`Agent ID required. Use --agent.
+`);
+    process.exit(1);
+  }
+  const focus = getFocus(agentId);
+  if (focus)
+    console.log(chalk.cyan(`Focus: ${focus}`));
+  else
+    console.log(chalk.dim("No focus set."));
+});
+program2.command("remove <nameOrId>").description("Remove/delete a memory by name or ID (alias for memory delete)").option("--agent <id>", "Agent ID").action((nameOrId, opts) => {
+  const globalOpts = program2.opts();
+  const agentId = opts.agent || globalOpts.agent;
+  const { deleteMemory: deleteMemory2, getMemoryByKey: getMemoryByKey2, resolvePartialMemoryId } = (init_memories(), __toCommonJS(exports_memories));
+  let id = null;
+  try {
+    id = resolvePartialMemoryId?.(nameOrId) || null;
+  } catch {}
+  if (!id) {
+    const mem = getMemoryByKey2?.(nameOrId, agentId);
+    if (mem)
+      id = mem.id;
+  }
+  if (!id) {
+    console.error(chalk.red(`Memory not found: ${nameOrId}`));
+    process.exit(1);
+  }
+  const deleted = deleteMemory2(id);
+  if (deleted)
+    console.log(chalk.green(`\u2713 Memory ${id.slice(0, 8)} removed`));
+  else {
+    console.error(chalk.red(`Memory not found: ${nameOrId}`));
+    process.exit(1);
+  }
+});
 program2.parse(process.argv);

package/dist/mcp/index.js

@@ -11165,6 +11165,17 @@ server.tool("set_focus", "Set focus for an agent on a project. Memory ops will a
     return { content: [{ type: "text", text: formatError(e) }], isError: true };
   }
 });
+server.tool("heartbeat", "Update agent last_seen_at to signal active session. Call periodically during long tasks to prevent being marked stale.", { agent_id: exports_external.string().describe("Agent ID or name") }, async (args) => {
+  try {
+    const agent = getAgent(args.agent_id);
+    if (!agent)
+      return { content: [{ type: "text", text: `Agent not found: ${args.agent_id}` }], isError: true };
+    touchAgent(agent.id);
+    return { content: [{ type: "text", text: `\u2665 ${agent.name} (${agent.id}) \u2014 last_seen_at updated` }] };
+  } catch (e) {
+    return { content: [{ type: "text", text: formatError(e) }], isError: true };
+  }
+});
 server.tool("get_focus", "Get the current focus project for an agent.", { agent_id: exports_external.string() }, async (args) => {
   try {
     const projectId = getFocus(args.agent_id);

package/package.json

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