@josephyoung/pi-file-reference 0.1.0 → 0.1.1

package/extensions/index.ts

@@ -1,10 +1,9 @@
-import type { ExtensionAPI, CustomEntry } from "@earendil-works/pi-coding-agent";
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 import * as fs from "node:fs";
 import * as path from "node:path";
 import * as os from "node:os";
 
 const AGENTS_FILE = "AGENTS.md";
-const STORE_KEY = "agents-at-context";
 
 /**
  * Parse @filepath references from a line.
@@ -130,45 +129,31 @@ function loadRefs(cwd: string): RefContent[] {
     if (fs.existsSync(resolvedPath)) {
       results.push({ ref, content: fs.readFileSync(resolvedPath, "utf-8") });
     } else {
-      console.warn(`[agents-at-reader] @${ref} -> ${resolvedPath} not found, skipping`);
+      console.warn(`[pi-file-reference] @${ref} -> ${resolvedPath} not found, skipping`);
     }
   }
 
   return results;
 }
 
+let cachedRefs: RefContent[] = [];
+
 export default function (pi: ExtensionAPI) {
   pi.on("session_start", (_event, ctx) => {
-    // Check if custom entry already exists (e.g., after reload)
-    const entries = ctx.sessionManager.getEntries();
-    const existing = entries.find(
-      (e): e is CustomEntry<{ refs: RefContent[] }> =>
-        e.type === "custom" && e.customType === STORE_KEY,
-    );
-    if (existing) return;
-
-    const refs = loadRefs(ctx.cwd);
-    if (refs.length === 0) return;
-
-    pi.appendEntry(STORE_KEY, { refs });
+    cachedRefs = loadRefs(ctx.cwd);
   });
 
-  pi.on("before_agent_start", (event, ctx) => {
-    const entries = ctx.sessionManager.getEntries();
-    const entry = entries.find(
-      (e): e is CustomEntry<{ refs: RefContent[] }> =>
-        e.type === "custom" && e.customType === STORE_KEY,
-    );
-    if (!entry?.data?.refs?.length) return;
+  pi.on("before_agent_start", (event, _ctx) => {
+    if (!cachedRefs.length) return;
 
-    const injected = entry.data.refs
-      .map((r) => `// @${r.ref}\n${r.content}`)
+    const injected = cachedRefs
+      .map((r) => `## @${r.ref}\n\n${r.content}`)
       .join("\n\n");
 
     return {
       systemPrompt:
         event.systemPrompt +
-        `\n\n---\n// @AGENTS.md 引用文件 — 注入以下文件内容供参考\n${injected}\n---`,
+        `\n\n# Context References\n\n${injected}`,
     };
   });
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@josephyoung/pi-file-reference",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "private": false,
   "description": "Pi extension that resolves @filepath references in AGENTS.md and injects file content into system prompt",
   "type": "module",

package/tsconfig.json

@@ -7,5 +7,5 @@
     "noEmit": true,
     "skipLibCheck": true
   },
-  "include": ["agents-at-reader.ts"]
+  "include": ["extensions/**/*.ts"]
 }