@rohirik/pi-ltm 2.12.0 → 2.12.3

package/dist/index.js

@@ -1,9 +1,11 @@
-import{spawn as w,execSync as b}from"node:child_process";import{createRequire as y}from"node:module";import{resolve as x,dirname as T,join as p}from"node:path";import{homedir as v}from"node:os";import{existsSync as u,readdirSync as k}from"node:fs";var m="LTM_BRIDGE_DEPTH";function P(){let c=[process.env.BUN_INSTALL?p(process.env.BUN_INSTALL,"bin","bun"):null,p(v(),".bun","bin","bun"),"/opt/homebrew/bin/bun","/usr/local/bin/bun","/usr/bin/bun"].filter(t=>t!==null&&u(t));if(c.length>0)return c[0];for(let t of["which bun","command -v bun"])try{let e=b(t,{encoding:"utf-8",stdio:"pipe"}).trim();if(e&&u(e)&&!/\bpi\b/.test(e))return e}catch{}return null}function E(){let c=p(v(),".claude","plugins","cache","ltm","ltm");if(u(c))try{let t=k(c).filter(e=>/^\d/.test(e)).sort().reverse();for(let e of t){let n=p(c,e,"src","mcp-server.ts");if(u(n))return n}}catch{}try{let e=y(import.meta.url).resolve("@rohirik/openltm-core/package.json"),n=x(T(e),"src","mcp-server.ts");if(u(n))return n}catch{}return null}var f=class{constructor(t,e){this.runtime=t;this.script=e}child=null;pending=new Map;buffer="";nextId=1;exited=!1;start(){if(this.child)return;let t=parseInt(process.env[m]??"0",10),e={...process.env,[m]:String(t+1)};this.child=w(this.runtime,[this.script],{stdio:["pipe","pipe","pipe"],env:e}),this.child.stdout?.on("data",n=>{this.buffer+=n.toString("utf-8");let r=this.buffer.split(`
-`);this.buffer=r.pop()??"";for(let o of r){let l=o.trim();if(l)try{let i=JSON.parse(l);if(i.id!==void 0){let s=this.pending.get(i.id);s&&(this.pending.delete(i.id),i.error?s.reject(new Error(i.error.message)):s.resolve(i.result))}}catch{}}}),this.child.on("exit",()=>{this.exited=!0;for(let n of this.pending.values())n.reject(new Error("LTM MCP server exited"));this.pending.clear()}),this.child.on("error",()=>{this.exited=!0;for(let n of this.pending.values())n.reject(new Error("LTM MCP server error"));this.pending.clear()})}send(t,e,n){let r=JSON.stringify({jsonrpc:"2.0",id:n,method:t,params:e})+`
-`;this.child?.stdin?.write(r)}request(t,e,n=3e4){let r=this.nextId++;return new Promise((o,l)=>{let i=setTimeout(()=>{this.pending.delete(r),l(new Error(`LTM MCP timeout: ${t}`))},n);this.pending.set(r,{resolve:s=>{clearTimeout(i),o(s)},reject:s=>{clearTimeout(i),l(s)}}),this.send(t,e,r)})}async initialize(){await this.request("initialize",{protocolVersion:"2024-11-05",capabilities:{},clientInfo:{name:"pi-ltm-bridge",version:"1.0.0"}}),this.send("notifications/initialized")}async listTools(){return(await this.request("tools/list"))?.tools??[]}async callTool(t,e){let n=await this.request("tools/call",{name:t,arguments:e},6e4),r=(n?.content??[]).filter(o=>o?.type==="text"&&o.text).map(o=>o.text).join(`
-`);if(n?.isError)throw new Error(r||`${t} returned an error`);return r}};function S(c){let t=c;if(parseInt(process.env[m]??"0",10)>0)return;let e=P(),n=E();if(!e||!n)return;let r=new f(e,n);r.start();let o=(async()=>{await r.initialize();let l=await r.listTools();for(let i of l){let s=i.name;t.registerTool({name:s,label:s,description:i.description??"",parameters:i.inputSchema??{type:"object",properties:{}},async execute(g,a){return{content:[{type:"text",text:await r.callTool(s,a??{})}],details:{}}}})}})().catch(()=>{});t.on("before_agent_start",async l=>{await o;let i=l;try{let g=String(i?.cwd??process.cwd()).replace(/\/$/,"").split("/").pop()??"",a=await r.callTool("ltm_recall",{project:g,limit:8,sort_by:"relevance"});if(!a||a==="[]")return;let d=`## Prior Knowledge (LTM)
+import{spawn as y,execSync as x}from"node:child_process";import{createRequire as T}from"node:module";import{resolve as P,dirname as k,join as g}from"node:path";import{homedir as b}from"node:os";import{existsSync as p,readdirSync as S}from"node:fs";var h="LTM_BRIDGE_DEPTH";function E(){let a=[process.env.BUN_INSTALL?g(process.env.BUN_INSTALL,"bin","bun"):null,g(b(),".bun","bin","bun"),"/opt/homebrew/bin/bun","/usr/local/bin/bun","/usr/bin/bun"].filter(e=>e!==null&&p(e));if(a.length>0)return a[0];for(let e of["which bun","command -v bun"])try{let t=x(e,{encoding:"utf-8",stdio:"pipe"}).trim();if(t&&p(t)&&!/\bpi\b/.test(t))return t}catch{}return null}function M(){let a=g(b(),".claude","plugins","cache","ltm","ltm");if(p(a))try{let e=S(a).filter(t=>/^\d/.test(t)).sort().reverse();for(let t of e){let n=g(a,t,"src","mcp-server.ts");if(p(n))return{script:n,args:[]}}}catch{}try{let t=T(import.meta.url).resolve("@rohirik/openltm-core/package.json"),n=P(k(t),"src","cli","bin.ts");if(p(n))return{script:n,args:["mcp-serve"]}}catch{}return null}var v=class{constructor(e,t,n=[]){this.runtime=e;this.script=t;this.args=n}child=null;pending=new Map;buffer="";nextId=1;exited=!1;start(){if(this.child)return;let e=parseInt(process.env[h]??"0",10),t={...process.env,[h]:String(e+1)};this.child=y(this.runtime,[this.script,...this.args],{stdio:["pipe","pipe","pipe"],env:t}),this.child.stdout?.on("data",n=>{this.buffer+=n.toString("utf-8");let r=this.buffer.split(`
+`);this.buffer=r.pop()??"";for(let s of r){let o=s.trim();if(o)try{let c=JSON.parse(o);if(c.id!==void 0){let i=this.pending.get(c.id);i&&(this.pending.delete(c.id),c.error?i.reject(new Error(c.error.message)):i.resolve(c.result))}}catch{}}}),this.child.on("exit",()=>{this.exited=!0;for(let n of this.pending.values())n.reject(new Error("LTM MCP server exited"));this.pending.clear()}),this.child.on("error",()=>{this.exited=!0;for(let n of this.pending.values())n.reject(new Error("LTM MCP server error"));this.pending.clear()})}send(e,t,n){let r=JSON.stringify({jsonrpc:"2.0",id:n,method:e,params:t})+`
+`;this.child?.stdin?.write(r)}request(e,t,n=3e4){let r=this.nextId++;return new Promise((s,o)=>{let c=setTimeout(()=>{this.pending.delete(r),o(new Error(`LTM MCP timeout: ${e}`))},n);this.pending.set(r,{resolve:i=>{clearTimeout(c),s(i)},reject:i=>{clearTimeout(c),o(i)}}),this.send(e,t,r)})}async initialize(){await this.request("initialize",{protocolVersion:"2024-11-05",capabilities:{},clientInfo:{name:"pi-ltm-bridge",version:"1.0.0"}}),this.send("notifications/initialized")}async listTools(){return(await this.request("tools/list"))?.tools??[]}async callTool(e,t){let n=await this.request("tools/call",{name:e,arguments:t},6e4),r=(n?.content??[]).filter(s=>s?.type==="text"&&s.text).map(s=>s.text).join(`
+`);if(n?.isError)throw new Error(r||`${e} returned an error`);return r}};function j(a,e){try{let t=JSON.parse(e),n=Array.isArray(t.globals)?t.globals:[],r=Array.isArray(t.scoped)?t.scoped:[];if(n.length===0&&r.length===0)return"";let s=["## Prior Knowledge (LTM)",""];if(n.length>0){s.push("Global:");for(let o of n.slice(0,3))s.push(`- [${o.id}] ${String(o.content).replace(/\s+/g," ").trim()}`);s.push("")}if(r.length>0){s.push(`Project (${a}):`);for(let o of r.slice(0,7))s.push(`- [${o.id}] ${String(o.content).replace(/\s+/g," ").trim()}`);s.push("")}return s.join(`
+`).trimEnd()+`
+`}catch{return""}}function _(a){let e=a;if(parseInt(process.env[h]??"0",10)>0)return;let t=E(),n=M();if(!t||!n)return;let r=new v(t,n.script,n.args);r.start();let s=new Set,o=(async()=>{await r.initialize();let c=await r.listTools();for(let i of c){let l=i.name;s.add(l),e.registerTool({name:l,label:l,description:i.description??"",parameters:i.inputSchema??{type:"object",properties:{}},async execute(d,u){return{content:[{type:"text",text:await r.callTool(l,u??{})}],details:{}}}})}})().catch(()=>{});e.on("before_agent_start",async c=>{await o;let i=c;try{let d=String(i?.cwd??process.cwd()).replace(/\/$/,"").split("/").pop()??"",u=s.has("context")?"context":s.has("recall")?"recall":"";if(!u)return;let m=u==="context"?await r.callTool("context",{project:d}):await r.callTool("recall",{project:d,limit:8,sort_by:"relevance"}),f=u==="context"?j(d,m):`## Prior Knowledge (LTM)
 
-${a}
-`,h=String(i?.systemPrompt??"");return{systemPrompt:h?`${h}
+${m}
+`;if(!f.trim())return;let w=String(i?.systemPrompt??"");return{systemPrompt:w?`${w}
 
-${d}`:d}}catch{}})}export{S as default};
+${f}`:f}}catch{}})}export{_ as default};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rohirik/pi-ltm",
-  "version": "2.12.0",
+  "version": "2.12.3",
   "description": "LTM (Long-Term Memory) native extension for Pi — share memories with Claude Code and OpenCode",
   "type": "module",
   "main": "./dist/index.js",
@@ -38,7 +38,7 @@
     "access": "public"
   },
   "dependencies": {
-    "@rohirik/openltm-core": "^2.12.0"
+    "@rohirik/openltm-core": "^2.12.3"
   },
   "peerDependencies": {
     "bun-types": ">=1.0.0"

package/src/hooks.ts

@@ -1,16 +1,9 @@
-import { recall, learn } from "@rohirik/openltm-core";
+import { buildPrefillContext, deriveProjectFromCwd, learn } from "@rohirik/openltm-core";
 
 type PiAny = any;
 
-const MAX_CONTEXT_MEMORIES = 10;
-
-function formatContextBlock(memories: Array<{ id: number; content: string; category: string }>): string {
-  const lines = memories.map((m) => `- [${m.id}] (${m.category}) ${m.content}`);
-  return "## Prior Knowledge (LTM)\n\n" + lines.join("\n") + "\n";
-}
-
 function projectFromCwd(cwd: string): string {
-  return cwd.replace(/\/$/, "").split("/").pop() ?? "";
+  return deriveProjectFromCwd(cwd);
 }
 
 export function registerHooks(pi: PiAny): void {
@@ -19,12 +12,8 @@ export function registerHooks(pi: PiAny): void {
     try {
       const cwd = String(event?.cwd ?? process.cwd());
       const project = projectFromCwd(cwd);
-      const memories = await recall({ project, limit: MAX_CONTEXT_MEMORIES, sort_by: "relevance" });
-      if (memories.length === 0) return;
-
-      const block = formatContextBlock(
-        memories.map((m) => ({ id: m.id, content: m.content, category: m.category })),
-      );
+      const block = buildPrefillContext({ project, maxMemories: 10, maxLines: 18 });
+      if (!block) return;
       const existing = String(event?.systemPrompt ?? "");
       const parts = existing ? [existing, block] : [block];
       return { systemPrompt: parts.join("\n\n") };

package/src/index.ts

@@ -42,7 +42,7 @@ function findBun(): string | null {
   return null;
 }
 
-function findMcpServer(): string | null {
+function findMcpServer(): { script: string; args: string[] } | null {
   // 1. Plugin cache — newest version first (Claude Code users)
   const cacheBase = join(homedir(), ".claude", "plugins", "cache", "ltm", "ltm");
   if (existsSync(cacheBase)) {
@@ -52,19 +52,19 @@ function findMcpServer(): string | null {
         .sort()
         .reverse();
       for (const v of versions) {
-        const s = join(cacheBase, v, "src", "mcp-server.ts");
-        if (existsSync(s)) return s;
+        const script = join(cacheBase, v, "src", "mcp-server.ts");
+        if (existsSync(script)) return { script, args: [] };
       }
     } catch {
       // continue to next strategy
     }
   }
-  // 2. openltm-core package (if mcp-server is added in a future version)
+  // 2. openltm-core package — run the packaged CLI entrypoint with mcp-serve
   try {
     const req = createRequire(import.meta.url);
     const pkgJson = req.resolve("@rohirik/openltm-core/package.json");
-    const script = resolve(dirname(pkgJson), "src", "mcp-server.ts");
-    if (existsSync(script)) return script;
+    const script = resolve(dirname(pkgJson), "src", "cli", "bin.ts");
+    if (existsSync(script)) return { script, args: ["mcp-serve"] };
   } catch {
     // not available
   }
@@ -89,6 +89,7 @@ class LtmMcpClient {
   constructor(
     private readonly runtime: string,
     private readonly script: string,
+    private readonly args: string[] = [],
   ) {}
 
   start(): void {
@@ -96,7 +97,7 @@ class LtmMcpClient {
     const depth = parseInt(process.env[BRIDGE_DEPTH_ENV] ?? "0", 10);
     const env = { ...process.env, [BRIDGE_DEPTH_ENV]: String(depth + 1) };
 
-    this.child = spawn(this.runtime, [this.script], { stdio: ["pipe", "pipe", "pipe"], env });
+    this.child = spawn(this.runtime, [this.script, ...this.args], { stdio: ["pipe", "pipe", "pipe"], env });
 
     this.child.stdout?.on("data", (chunk: Buffer) => {
       this.buffer += chunk.toString("utf-8");
@@ -184,6 +185,33 @@ class LtmMcpClient {
 
 // ── Extension entry point ─────────────────────────────────────────────────────
 
+function formatContextPayload(project: string, raw: string): string {
+  try {
+    const parsed = JSON.parse(raw) as {
+      globals?: Array<{ id: number; content: string }>;
+      scoped?: Array<{ id: number; content: string }>;
+    };
+    const globals = Array.isArray(parsed.globals) ? parsed.globals : [];
+    const scoped = Array.isArray(parsed.scoped) ? parsed.scoped : [];
+    if (globals.length === 0 && scoped.length === 0) return "";
+
+    const lines: string[] = ["## Prior Knowledge (LTM)", ""];
+    if (globals.length > 0) {
+      lines.push("Global:");
+      for (const m of globals.slice(0, 3)) lines.push(`- [${m.id}] ${String(m.content).replace(/\s+/g, " ").trim()}`);
+      lines.push("");
+    }
+    if (scoped.length > 0) {
+      lines.push(`Project (${project}):`);
+      for (const m of scoped.slice(0, 7)) lines.push(`- [${m.id}] ${String(m.content).replace(/\s+/g, " ").trim()}`);
+      lines.push("");
+    }
+    return lines.join("\n").trimEnd() + "\n";
+  } catch {
+    return "";
+  }
+}
+
 export default function ltmExtension(pi: unknown): void {
   const p = pi as {
     registerTool: (def: {
@@ -198,19 +226,21 @@ export default function ltmExtension(pi: unknown): void {
   if (parseInt(process.env[BRIDGE_DEPTH_ENV] ?? "0", 10) > 0) return;
 
   const bun = findBun();
-  const script = findMcpServer();
-  if (!bun || !script) return; // degrade gracefully — no bun or server found
+  const server = findMcpServer();
+  if (!bun || !server) return; // degrade gracefully — no bun or server found
 
-  const client = new LtmMcpClient(bun, script);
+  const client = new LtmMcpClient(bun, server.script, server.args);
   client.start();
 
   // Bootstrap runs async — tools are registered once handshake completes.
   // Pi allows registerTool after extension load; before_agent_start awaits ready.
+  const toolNames = new Set<string>();
   const ready = (async () => {
     await client.initialize();
     const tools = await client.listTools();
     for (const tool of tools) {
       const name = tool.name;
+      toolNames.add(name);
       p.registerTool({
         name,
         label: name,
@@ -231,9 +261,15 @@ export default function ltmExtension(pi: unknown): void {
     try {
       const cwd = String(ev?.cwd ?? process.cwd());
       const project = cwd.replace(/\/$/, "").split("/").pop() ?? "";
-      const text = await client.callTool("ltm_recall", { project, limit: 8, sort_by: "relevance" });
-      if (!text || text === "[]") return;
-      const block = `## Prior Knowledge (LTM)\n\n${text}\n`;
+      const toolName = toolNames.has("context") ? "context" : (toolNames.has("recall") ? "recall" : "");
+      if (!toolName) return;
+      const text = toolName === "context"
+        ? await client.callTool("context", { project })
+        : await client.callTool("recall", { project, limit: 8, sort_by: "relevance" });
+      const block = toolName === "context"
+        ? formatContextPayload(project, text)
+        : `## Prior Knowledge (LTM)\n\n${text}\n`;
+      if (!block.trim()) return;
       const existing = String(ev?.systemPrompt ?? "");
       return { systemPrompt: existing ? `${existing}\n\n${block}` : block };
     } catch {