@zhushanwen/pi-system-prompt-trace 0.1.0 → 0.1.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zhushanwen/pi-system-prompt-trace",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "description": "System prompt trace extension: appends xyz:system-prompt entries to the session JSONL whenever the effective system prompt is established or changes",
   "type": "module",
   "main": "index.ts",
@@ -17,18 +17,24 @@
     "src/**/*.ts",
     "README.md"
   ],
+  "xyz-agent": {
+    "role": "taiji"
+  },
   "pi": {
     "extensions": [
       "./index.ts"
     ]
   },
   "peerDependencies": {
-    "@earendil-works/pi-coding-agent": "*"
+    "@earendil-works/pi-coding-agent": "^0.84.1"
   },
   "devDependencies": {
     "@vitest/coverage-v8": "^4.1.9",
     "vitest": "^4.1.8"
   },
+  "dependencies": {
+    "@zhushanwen/pi-extension-logger": "0.3.0"
+  },
   "scripts": {
     "typecheck": "npx tsc --noEmit",
     "test": "vitest run"

package/src/__tests__/index-wiring.test.ts

@@ -25,6 +25,15 @@ import type { SystemPromptTraceEntryData } from "../types.js";
 const P1 = "wiring prompt\nline-1";
 const P2 = "wiring prompt\nline-1\nline-2-added";
 
+const { loggerMock } = vi.hoisted(() => {
+	const mock = { debug: vi.fn(), warn: vi.fn(), error: vi.fn() };
+	return { loggerMock: mock };
+})
+vi.mock("@zhushanwen/pi-extension-logger", () => ({
+	getLogger: () => loggerMock,
+	createLogger: () => loggerMock,
+}))
+
 const agentDirRef = vi.hoisted(() => ({ current: "" }));
 
 vi.mock("@earendil-works/pi-coding-agent", () => ({
@@ -277,17 +286,15 @@ describe("index.ts wiring SDK 契约", () => {
 			throw new Error("prompt boom");
 		}, "sess-w7");
 
-		let logged = 0;
-		const errSpy = vi.spyOn(console, "error").mockImplementation(() => {
-			logged++;
-		});
+		loggerMock.error.mockClear();
 		try {
 			await emit(h, "session_start", { type: "session_start", reason: "startup" }, boomCtx);
 			await emit(h, "turn_start", { type: "turn_start", turnIndex: 0, timestamp: 0 }, boomCtx);
 		} finally {
-			errSpy.mockRestore();
+			// no-op: cleanup handled after assertions
 		}
 		expect(h.entries).toHaveLength(0);
-		expect(logged).toBe(1);
+		expect(loggerMock.error).toHaveBeenCalledTimes(1);
+		loggerMock.error.mockClear();
 	});
 });

package/src/baseline.ts

@@ -6,13 +6,16 @@
  * - readPersistedBaseline / writePersistedBaseline：dataDir 自持久化小文件
  *   （app 重启直 spawn resume 时唯一可用的基线来源），原子写入。
  *
- * 所有函数不抛错：读失败返回 null、写失败 console.error 后静默——基线丢失的代价只是
+ * 所有函数不抛错：读失败返回 null、写失败 logger.error 后静默——基线丢失的代价只是
  * 下次 resume 多写一条留痕（设计 D2 已接受），不允许影响 agent 主流程。
  */
 
 import { existsSync, mkdirSync, readFileSync, renameSync, unlinkSync, writeFileSync } from "node:fs";
 import { dirname } from "node:path";
 
+import { getLogger } from "@zhushanwen/pi-extension-logger";
+
+const logger = getLogger("pi-system-prompt-trace");
 import { isRecord, isSystemPromptTraceEntryData, SYSTEM_PROMPT_CUSTOM_TYPE } from "./types.js";
 import type { PromptBaseline } from "./types.js";
 
@@ -152,8 +155,8 @@ export function writePersistedBaseline(
 		}
 	} catch (e) {
 		// best-effort 降级：基线写失败只影响下次 app 重启 resume 的去重（多写一条留痕，
-		// 设计 D2 已接受），不阻断 agent 主流程；错误进 pi stdout 随日志落盘供排查
-		console.error("[pi-system-prompt-trace] write baseline failed:", e);
+		// 设计 D2 已接受），不阻断 agent 主流程；logger.error → appendEntry 持久化供排查
+		logger.error("write baseline failed", { detail: String(e) });
 	}
 }
 

package/src/trace.ts

@@ -12,6 +12,9 @@
 
 import { createHash } from "node:crypto";
 
+import { getLogger } from "@zhushanwen/pi-extension-logger";
+
+const logger = getLogger("pi-system-prompt-trace");
 import { summarizePromptDiff } from "./diff.js";
 import {
 	mapReasonForFirstWrite,
@@ -145,8 +148,8 @@ export function createSystemPromptTrace(env: TraceEnv, stash: SwitchStash): Syst
 				write(ctx, text, hash, reason, version, parentFullText);
 				current = { version, hash, fullText: text };
 			} catch (e) {
-				// 留痕是诊断性旁路：任何失败都不允许影响 agent 主流程（错误进 pi stdout，随日志落盘）
-				console.error("[pi-system-prompt-trace] turn_start handler failed:", e);
+				// 留痕是诊断性旁路：任何失败都不允许影响 agent 主流程（logger.error → appendEntry 持久化）
+				logger.error("turn_start handler failed", { detail: String(e) });
 			}
 		},
 	};