@ia-ccun/code-agent-cli 0.0.52-beta.2 → 0.0.52-beta.3

package/config/agent/extensions/aicode-advisor.ts

@@ -146,6 +146,19 @@ function setDisabledForModels(models: Array<string | { model: string; minEffort?
 }
 
 // ============== 工具清单序列化 ==============
+/**
+ * stableStringify - 稳定序列化对象（键排序）
+ * 
+ * 用于序列化工具参数，确保参数对象的键顺序一致，便于对比和展示。
+ * 
+ * 功能说明：
+ * - 处理 null 和非对象类型：直接 JSON.stringify
+ * - 处理数组：递归序列化每个元素，undefined 替换为 "null"
+ * - 处理对象：键排序后序列化，跳过 undefined 值
+ * 
+ * @param value - 要序列化的值
+ * @returns 稳定的 JSON 字符串（键排序）
+ */
 function stableStringify(value: unknown): string {
   if (value === null || typeof value !== "object") {
     return JSON.stringify(value);

package/config/agent/extensions/aicode-btw.ts

@@ -40,6 +40,18 @@ function matchesBtwFocusShortcut(data: string): boolean {
   return BTW_FOCUS_SHORTCUTS.some((shortcut) => matchesKey(data, shortcut));
 }
 
+/**
+ * 类型守卫：检查 entry 是否是有效的 Message 对象
+ * ✅ 用于安全转换，避免使用危险的 as unknown as
+ */
+function isMessageLike(entry: unknown): entry is Partial<Message> & { role?: string; customType?: string; content?: unknown } {
+  if (!entry || typeof entry !== "object") {
+    return false;
+  }
+  const candidate = entry as { role?: unknown; customType?: string; content?: unknown };
+  return typeof candidate.role === "string" && Array.isArray(candidate.content);
+}
+
 const BTW_SYSTEM_PROMPT = [
   "You are having an aside conversation with the user, separate from their main working session.",
   "If main session messages are provided, they are for context only — that work is being handled by another agent.",
@@ -277,17 +289,17 @@ function buildBtwSeedState(
           (message) => !isVisibleBtwMessage(message),
         ),
       );
-    } catch {
+    } catch (error) {
+      // ✅ 改进错误处理：记录错误而不忽略
+      console.warn('[btw] buildSessionContext failed, using fallback:', error);
       messages.push(
         ...ctx.sessionManager.getEntries().flatMap((entry) => {
-          if (!entry || typeof entry !== "object") {
+          // ✅ 使用类型守卫进行安全转换，避免 as unknown as
+          if (!isMessageLike(entry)) {
             return [];
           }
 
-          const message = entry as unknown as Partial<Message> & { role?: string; customType?: string; content?: unknown };
-          if (typeof message.role !== "string" || !Array.isArray(message.content)) {
-            return [];
-          }
+          const message = entry; // ✅ 类型安全
 
           return isVisibleBtwMessage({ role: message.role, customType: message.customType }) ? [] : [message as Message];
         }),
@@ -380,7 +392,9 @@ function formatToolPreview(value: unknown): string {
       return "";
     }
     return preview.length > 120 ? `${preview.slice(0, 117)}...` : preview;
-  } catch {
+  } catch (error) {
+    // ✅ 改进错误处理：记录错误而不忽略
+    console.warn('[btw] JSON.stringify failed in formatToolPreview:', error);
     return "";
   }
 }
@@ -396,6 +410,26 @@ function createEmptyTranscriptState(): BtwTranscriptState {
   };
 }
 
+/**
+ * clearTranscriptState - 清理 transcriptState 的所有资源
+ * ✅ 清理 entries 数组
+ * ✅ 清理 toolCalls Map（防止内存泄漏）
+ * ✅ 重置所有状态
+ */
+function clearTranscriptState(state: BtwTranscriptState): void {
+  // ✅ 清理 toolCalls Map（防止内存泄漏）
+  state.toolCalls.clear();
+  
+  // 清理 entries 数组
+  state.entries = [];
+  
+  // 重置状态
+  state.nextEntryId = 1;
+  state.nextTurnId = 1;
+  state.currentTurnId = null;
+  state.lastTurnId = null;
+}
+
 function appendTranscriptEntry<T extends BtwTranscriptEntry>(
   state: BtwTranscriptState,
   entry: Omit<T, "id">,
@@ -444,6 +478,13 @@ function finishTranscriptTurn(state: BtwTranscriptState, turnId?: number | null)
   if (state.currentTurnId === resolvedTurnId) {
     state.currentTurnId = null;
   }
+
+  // ✅ 清理当前 turn 的 toolCalls（防止内存泄漏）
+  for (const [toolCallId, toolCall] of state.toolCalls.entries()) {
+    if (toolCall.turnId === resolvedTurnId) {
+      state.toolCalls.delete(toolCallId);
+    }
+  }
 }
 
 function removeTranscriptTurn(state: BtwTranscriptState, turnId: number | null): void {
@@ -574,7 +615,9 @@ function summarizeToolResult(value: unknown, maxLength = 400): { content: string
     } else if (value !== undefined) {
       try {
         content = JSON.stringify(value, null, 2);
-      } catch {
+      } catch (error) {
+        // ✅ 改进错误处理：记录错误而不忽略
+        console.warn('[btw] JSON.stringify failed in summarizeToolResult:', error);
         content = String(value);
       }
     }
@@ -1340,8 +1383,9 @@ export default function (pi: ExtensionAPI) {
 
     try {
       unsubscribe();
-    } catch {
-      // Ignore unsubscribe errors during BTW session replacement/shutdown.
+    } catch (error) {
+      // ✅ 改进错误处理：记录错误而不忽略
+      console.warn('[btw] Unsubscribe failed during session removal:', error);
     }
   }
 
@@ -1402,6 +1446,12 @@ export default function (pi: ExtensionAPI) {
   async function disposeBtwSession(): Promise<void> {
     const current = activeBtwSession;
     activeBtwSession = null;
+
+    // ✅ 清理 transcriptState（防止内存泄漏）
+    if (transcriptState) {
+      clearTranscriptState(transcriptState);
+    }
+
     if (!current) {
       return;
     }
@@ -1410,8 +1460,9 @@ export default function (pi: ExtensionAPI) {
 
     try {
       await current.session.abort();
-    } catch {
-      // Ignore abort errors during BTW session replacement/shutdown.
+    } catch (error) {
+      // ✅ 改进错误处理：记录错误而不忽略
+      console.warn('[btw] Session abort failed during disposal:', error);
     }
 
     current.session.dispose();
@@ -2144,8 +2195,9 @@ export default function (pi: ExtensionAPI) {
     } finally {
       try {
         await session.abort();
-      } catch {
-        // Ignore abort errors during summarize session shutdown.
+      } catch (error) {
+        // ✅ 改进错误处理：记录错误而不忽略
+        console.warn('[btw] Session abort failed during summarize shutdown:', error);
       }
       session.dispose();
     }

package/config/agent/extensions/aicode-pets.ts

@@ -0,0 +1,362 @@
+/**
+ * Random Pet Widget Extension
+ *
+ * Displays a random animal sprite below the input editor.
+ * Each session randomly selects from 18 different animals.
+ * Inspired by claude-petpet's sprite system.
+ */
+
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+
+// 帽子定义（参考 claude-petpet）
+const HATS: Record<string, string> = {
+	none: "",
+	crown: "   \\^^^/    ",
+	tophat: "   [___]    ",
+	propeller: "    -+-     ",
+	halo: "   (   )    ",
+	wizard: "    /^\\     ",
+	beanie: "   (___)    ",
+	tinyduck: "    ,>      ",
+};
+
+// 18 种动物精灵定义（参考 claude-petpet）
+interface PetDefinition {
+	name: string;
+	emoji: string;
+	frames: string[][];
+	speed: number;
+}
+
+const PETS: Record<string, PetDefinition> = {
+	cat: {
+		name: "小猫",
+		emoji: "🐱",
+		frames: [
+			["   /\\_/\\    ", "  ( o   o)  ", "  (  ω  )   ", '  (")_(")   '],
+			["   /\\_/\\    ", "  ( ^   ^)  ", "  (  ω  )   ", '  (")_(")   '],
+			["   /\\_/\\    ", "  ( -   -)  ", "  (  ω  )   ", '  (")_(")   '],
+		],
+		speed: 500,
+	},
+	rabbit: {
+		name: "兔子",
+		emoji: "🐰",
+		frames: [
+			["   (\\__/)   ", "  ( o   o )  ", " =(  ..  )= ", '  (")_(")  '],
+			["   (\\__/)   ", "  ( ^   ^ )  ", " =(  ..  )= ", '  (")_(")  '],
+			["   (\\__/)   ", "  ( -   - )  ", " =(  ..  )= ", '  (")_(")  '],
+		],
+		speed: 400,
+	},
+	dragon: {
+		name: "龙",
+		emoji: "🐉",
+		frames: [
+			["  /^\\  /^\\  ", " <  o   o  > ", " (   ~~   ) ", "  `-vvvv-´  "],
+			["  /^\\  /^\\  ", " <  ^   ^  > ", " (   ~~   ) ", "  `-vvvv-´  "],
+			["  /^\\  /^\\  ", " <  -   -  > ", " (   ~~   ) ", "  `-vvvv-´  "],
+		],
+		speed: 600,
+	},
+	owl: {
+		name: "猫头鹰",
+		emoji: "🦉",
+		frames: [
+			["   /\\  /\\   ", "  ((o)(o))  ", "  (  ><  )  ", "   `----´   "],
+			["   /\\  /\\   ", "  ((^)(^))  ", "  (  ><  )  ", "   `----´   "],
+			["   /\\  /\\   ", "  ((-)(-))  ", "  (  ><  )  ", "   `----´   "],
+		],
+		speed: 700,
+	},
+	penguin: {
+		name: "企鹅",
+		emoji: "🐧",
+		frames: [
+			["  .---.     ", "  (o> o)     ", " /(   )\\    ", "  `---´     "],
+			["  .---.     ", "  (^> ^)     ", " /(   )\\    ", "  `---´     "],
+			["  .---.     ", "  (-> -)     ", " /(   )\\    ", "  `---´     "],
+		],
+		speed: 500,
+	},
+	duck: {
+		name: "鸭子",
+		emoji: "🦆",
+		frames: [
+			["    __      ", "  <(o )___  ", "   (  ._>   ", "    `--´    "],
+			["    __      ", "  <(^ )___  ", "   (  ._>   ", "    `--´    "],
+			["    __      ", "  <(- )___  ", "   (  ._>   ", "    `--´    "],
+		],
+		speed: 450,
+	},
+	goose: {
+		name: "鹅",
+		emoji: "🦢",
+		frames: [
+			["     (o>    ", "     ||     ", "   _(__)_   ", "    ^^^^    "],
+			["     (^>    ", "     ||     ", "   _(__)_   ", "    ^^^^    "],
+			["     (- >    ", "     ||     ", "   _(__)_   ", "    ^^^^    "],
+		],
+		speed: 500,
+	},
+	turtle: {
+		name: "乌龟",
+		emoji: "🐢",
+		frames: [
+			["   _,--._   ", "  ( o   o )  ", " /[______]\\ ", "  ``    ``  "],
+			["   _,--._   ", "  ( ^   ^ )  ", " /[______]\\ ", "  ``    ``  "],
+			["   _,--._   ", "  ( -   - )  ", " /[______]\\ ", "  ``    ``  "],
+		],
+		speed: 800,
+	},
+	snail: {
+		name: "蜗牛",
+		emoji: "🐌",
+		frames: [
+			[" ·    .--.  ", " \\  ( @ )  ", "  \\_`--´   ", " ~~~~~~~   "],
+			[" ✦    .--.  ", " \\  ( @ )  ", "  \\_`--´   ", "  ~~~~~~~  "],
+			[" ◉    .--.  ", " \\  ( @ )  ", "  \\_`--´   ", "   ~~~~~~~ "],
+		],
+		speed: 900,
+	},
+	capybara: {
+		name: "水豚",
+		emoji: "🦫",
+		frames: [
+			["  n______n  ", " ( o     o ) ", " (   oo   ) ", "  `------´  "],
+			["  n______n  ", " ( ^     ^ ) ", " (   oo   ) ", "  `------´  "],
+			["  n______n  ", " ( -     - ) ", " (   oo   ) ", "  `------´  "],
+		],
+		speed: 600,
+	},
+	ghost: {
+		name: "幽灵",
+		emoji: "👻",
+		frames: [
+			["   .----.   ", "  / o   o \\  ", "  |      |  ", "  ~`~``~`~  "],
+			["   .----.   ", "  / ^   ^ \\  ", "  |      |  ", "  ~`~``~`~  "],
+			["   .----.   ", "  / -   - \\  ", "  |      |  ", "  ~`~``~`~  "],
+		],
+		speed: 500,
+	},
+	robot: {
+		name: "机器人",
+		emoji: "🤖",
+		frames: [
+			["   .[||].   ", "  [ o   o ]  ", "  [ ==== ]  ", "  `------´  "],
+			["   .[||].   ", "  [ ^   ^ ]  ", "  [ ==== ]  ", "  `------´  "],
+			["   .[||].   ", "  [ -   - ]  ", "  [ ==== ]  ", "  `------´  "],
+		],
+		speed: 400,
+	},
+	axolotl: {
+		name: "蝾螈",
+		emoji: "🦎",
+		frames: [
+			["}~(______)~{", "}~(o .. o)~{", "  ( .--. )  ", "  (_/  \\_)  "],
+			["}~(______)~{", "}~(^ .. ^)~{", "  ( .--. )  ", "  (_/  \\_)  "],
+			["}~(______)~{", "}~(- .. -)~{", "  ( .--. )  ", "  (_/  \\_)  "],
+		],
+		speed: 600,
+	},
+	mushroom: {
+		name: "蘑菇",
+		emoji: "🍄",
+		frames: [
+			[" .-o-OO-o-. ", "(__________)", "   |o   o|   ", "   |____|   "],
+			[" .-o-OO-o-. ", "(__________)", "   |^   ^|   ", "   |____|   "],
+			[" .-o-OO-o-. ", "(__________)", "   |-   -|   ", "   |____|   "],
+		],
+		speed: 700,
+	},
+	blob: {
+		name: "团子",
+		emoji: "🫠",
+		frames: [
+			["   .----.   ", "  ( o   o )  ", "  (      )  ", "   `----´   "],
+			["   .----.   ", "  ( ^   ^ )  ", "  (      )  ", "   `----´   "],
+			["   .----.   ", "  ( -   - )  ", "  (      )  ", "   `----´   "],
+		],
+		speed: 500,
+	},
+	octopus: {
+		name: "章鱼",
+		emoji: "🐙",
+		frames: [
+			["   .----.   ", "  ( o   o )  ", "  (______)  ", "  /\\/\\/\\/\\  "],
+			["   .----.   ", "  ( ^   ^ )  ", "  (______)  ", "  \\/\\/\\/\\/  "],
+			["   .----.   ", "  ( -   - )  ", "  (______)  ", "  /\\/\\/\\/\\  "],
+		],
+		speed: 400,
+	},
+	chonk: {
+		name: "胖熊",
+		emoji: "🐻",
+		frames: [
+			["  /\\    /\\  ", " ( o     o ) ", " (   ..   ) ", "  `------´  "],
+			["  /\\    /\\  ", " ( ^     ^ ) ", " (   ..   ) ", "  `------´  "],
+			["  /\\    /\\  ", " ( -     - ) ", " (   ..   ) ", "  `------´  "],
+		],
+		speed: 600,
+	},
+	cactus: {
+		name: "仙人掌",
+		emoji: "🌵",
+		frames: [
+			[" n  ____  n ", " | |o  o| | ", " |_|    |_| ", "   |    |   "],
+			[" n  ____  n ", " | |^  ^| | ", " |_|    |_| ", "   |    |   "],
+			[" n  ____  n ", " | |-  -| | ", " |_|    |_| ", "   |    |   "],
+		],
+		speed: 700,
+	},
+};
+
+// 随机选择动物
+const petKeys = Object.keys(PETS);
+let currentPet: string;
+let currentFrame = 0;
+let intervalId: ReturnType<typeof setInterval> | null = null;
+let isShiny = false;
+let currentHat = "none";
+
+function getRandomPet(): string {
+	return petKeys[Math.floor(Math.random() * petKeys.length)]!;
+}
+
+function getRandomHat(): string {
+	const hatChance = Math.random();
+	if (hatChance < 0.05) return "crown";
+	if (hatChance < 0.1) return "wizard";
+	if (hatChance < 0.15) return "tophat";
+	if (hatChance < 0.2) return "halo";
+	if (hatChance < 0.25) return "beanie";
+	if (hatChance < 0.3) return "tinyduck";
+	if (hatChance < 0.35) return "propeller";
+	return "none";
+}
+
+function getHatNames(): Record<string, string> {
+	return {
+		crown: "皇冠",
+		wizard: "魔法帽",
+		tophat: "礼帽",
+		halo: "光环",
+		beanie: "无檐帽",
+		tinyduck: "小鸭帽",
+		propeller: "螺旋桨帽",
+		none: "",
+	};
+}
+
+function getNextFrame(): string[] {
+	const pet = PETS[currentPet]!;
+	const frames = pet.frames;
+	currentFrame = (currentFrame + 1) % frames.length;
+	const frame = [...frames[currentFrame]!];
+
+	// 添加帽子（如果有）
+	if (currentHat !== "none" && frame.length > 0 && frame[0] && !frame[0].trim()) {
+		frame[0] = HATS[currentHat] ?? frame[0]!;
+	}
+
+	// 闪亮特效
+	if (isShiny) {
+		return frame.map((line) => `\x1b[38;2;255;215;0m${line}\x1b[0m`);
+	}
+
+	return frame;
+}
+
+function showPetNotification(ctx: ExtensionAPI["on"] extends (event: string, cb: (event: any, ctx: infer Ctx) => void) => void ? Ctx : never) {
+	const pet = PETS[currentPet]!;
+	const hatNames = getHatNames();
+	let msg = `${pet.emoji} 本次宠物：${pet.name}`;
+	if (isShiny) msg += " ✨（闪亮）";
+	if (currentHat !== "none") msg += ` + ${hatNames[currentHat]}`;
+	ctx.ui.notify(msg, "info");
+}
+
+function startPetAnimation(ctx: any) {
+	const pet = PETS[currentPet]!;
+
+	// 显示初始精灵
+	const initialSprite = getNextFrame();
+	ctx.ui.setWidget("random-pet", initialSprite, { placement: "belowEditor" });
+
+	// 启动动画
+	let animationCtx = ctx;
+	if (intervalId) {
+		clearInterval(intervalId);
+	}
+	intervalId = setInterval(() => {
+		try {
+			const nextSprite = getNextFrame();
+			animationCtx.ui.setWidget("random-pet", nextSprite, { placement: "belowEditor" });
+		} catch (e) {
+			if (intervalId) {
+				clearInterval(intervalId);
+				intervalId = null;
+			}
+		}
+	}, pet.speed);
+}
+
+export default function (pi: ExtensionAPI) {
+	// 注册 /pet 命令
+	pi.registerCommand("pet", {
+		description: "切换随机宠物",
+		usage: "/pet",
+		handler: async (_args, ctx) => {
+			// 清理旧动画
+			if (intervalId) {
+				clearInterval(intervalId);
+				intervalId = null;
+			}
+
+			// 随机选择新宠物
+			const oldPet = PETS[currentPet]!;
+			currentPet = getRandomPet();
+			isShiny = Math.random() < 0.1;
+			currentHat = getRandomHat();
+			currentFrame = 0;
+
+			const pet = PETS[currentPet]!;
+
+			// 启动新动画
+			startPetAnimation(ctx);
+
+			// 显示切换提示
+			const hatNames = getHatNames();
+			let msg = `${oldPet.emoji}${oldPet.name} → ${pet.emoji}${pet.name}`;
+			if (isShiny) msg += " ✨（闪亮）";
+			if (currentHat !== "none") msg += ` + ${hatNames[currentHat]}`;
+			ctx.ui.notify(msg, "info");
+		},
+	});
+
+	pi.on("session_start", (_event, ctx) => {
+		// 随机选择动物
+		currentPet = getRandomPet();
+		isShiny = Math.random() < 0.1;
+		currentHat = getRandomHat();
+		currentFrame = 0;
+
+		// 启动动画
+		startPetAnimation(ctx);
+
+		// 显示提示
+		const pet = PETS[currentPet]!;
+		const hatNames = getHatNames();
+		let msg = `${pet.emoji} 本次宠物：${pet.name}`;
+		if (isShiny) msg += " ✨（闪亮）";
+		if (currentHat !== "none") msg += ` + ${hatNames[currentHat]}`;
+		ctx.ui.notify(msg, "info");
+	});
+
+	pi.on("session_shutdown", () => {
+		if (intervalId) {
+			clearInterval(intervalId);
+			intervalId = null;
+		}
+	});
+}

package/config/agent/extensions/aicode-tips.ts

@@ -164,7 +164,7 @@ export default function (pi: ExtensionAPI) {
   let currentTipIndex = -1;
   let autoHideTimeoutId: NodeJS.Timeout | undefined;
   let autoShowTimeoutId: NodeJS.Timeout | undefined;
-  let currentCtx: ExtensionContext | undefined;
+  let currentUI: any = undefined; // 保存当前有效的 UI context
   let isAutoPlaying = false;
 
   // 基于 #49bccf (73, 188, 207) 的随机颜色生成
@@ -178,8 +178,8 @@ export default function (pi: ExtensionAPI) {
   };
 
   const hideTip = () => {
-    if (!currentCtx?.hasUI) return;
-    currentCtx.ui.setWidget("aicode-tips", [], { placement: "aboveEditor" });
+    if (!currentUI) return;
+    currentUI.setWidget("aicode-tips", [], { placement: "aboveEditor" });
   };
 
   const clearAllTimers = () => {
@@ -202,7 +202,7 @@ export default function (pi: ExtensionAPI) {
   };
 
   const showRandomTip = (isInternalCall = false) => {
-    if (!currentCtx?.hasUI) return;
+    if (!currentUI) return;
 
     // 清除之前的定时器（但内部轮播调用时不清除，避免中断轮播）
     if (!isInternalCall) {
@@ -218,11 +218,20 @@ export default function (pi: ExtensionAPI) {
     // 使用随机颜色
     const color = getRandomTipColor();
     const coloredTip = `${color}tips: ${tip}\x1b[0m`;
-    currentCtx.ui.setWidget("aicode-tips", [coloredTip], { placement: "aboveEditor" });
+    currentUI.setWidget("aicode-tips", [coloredTip], { placement: "aboveEditor" });
 
     // 自动轮播：10 秒后直接显示下一条
     if (isAutoPlaying) {
-      autoHideTimeoutId = setTimeout(() => showRandomTip(true), 10000);
+      autoHideTimeoutId = setTimeout(() => {
+        try {
+          showRandomTip(true);
+        } catch (error) {
+          // 如果出现错误（比如 stale context），清理定时器
+          console.error('aicode-tips extension error:', error);
+          stopAutoPlay();
+          currentUI = undefined;
+        }
+      }, 10000);
     }
   };
 
@@ -238,22 +247,22 @@ export default function (pi: ExtensionAPI) {
 
   // 会话开始时启动自动轮播
   pi.on("session_start", (_event, ctx) => {
-    currentCtx = ctx;
+    currentUI = ctx.ui;
     startAutoPlay();
   });
 
   // 会话结束时清理
   pi.on("session_shutdown", () => {
     stopAutoPlay();
-    currentCtx?.ui.setWidget("aicode-tips", undefined);
-    currentCtx = undefined;
+    currentUI?.setWidget("aicode-tips", undefined);
+    currentUI = undefined;
   });
 
   // 手动切换技巧
   pi.registerCommand("tip", {
     description: "显示一个新的 aicode 使用技巧",
     handler: async (_args, ctx) => {
-      currentCtx = ctx;
+      currentUI = ctx.ui;
       showRandomTip();
     },
   });

package/config/agent/extensions/custom.ts

@@ -15,38 +15,6 @@ export default function (pi: ExtensionAPI) {
   // === 记录开始时间 ===
   let agentStartTime = 0;
 
-  pi.on("agent_start", async () => {
-    agentStartTime = Date.now();
-  });
-
-  // === 消息完成通知（只在最后一条显示）===
-
-  pi.on("agent_end", async (event, ctx) => {
-    try {
-      // 计算耗时
-      const elapsed = Date.now() - agentStartTime;
-      let timeStr: string;
-      if (elapsed < 1000) {
-        timeStr = `${elapsed}ms`;
-      } else if (elapsed < 60000) {
-        timeStr = `${(elapsed / 1000).toFixed(1)}s`;
-      } else {
-        const mins = Math.floor(elapsed / 60000);
-        const secs = Math.floor((elapsed % 60000) / 1000);
-        timeStr = `${mins}m ${secs}s`;
-      }
-
-      // 格式化完整时间
-      const now = new Date();
-      const timeStr2 = `${now.getFullYear()}-${String(now.getMonth() + 1).padStart(2, '0')}-${String(now.getDate()).padStart(2, '0')} ${String(now.getHours()).padStart(2, '0')}:${String(now.getMinutes()).padStart(2, '0')}:${String(now.getSeconds()).padStart(2, '0')}`;
-
-      // agent_end 表示一轮对话结束，显示完成提示
-      ctx.ui.notify(`${timeStr2} | ✨ 消息处理完成 (${timeStr})`, "info");
-    } catch (e) {
-      // 忽略错误（ctx 可能已过期）
-    }
-  });
-
   // === 拦截工具调用 ===
 
   pi.on("tool_call", async (event, ctx) => {

package/config/agent/extensions/tps.ts

@@ -0,0 +1,87 @@
+/**
+ * @author: xujianjiang
+ * Token 性能统计扩展
+ *
+ * 功能说明：
+ * - 在每轮对话结束时显示 Token 使用统计
+ * - 显示处理速度（token/秒）
+ * - 显示输入输出 Token 数量
+ * - 显示耗时（智能格式化：ms/秒/分+秒）
+ *
+ * 输出格式示例：
+ * 2026-06-08T16:19:54 | ✨ 处理完成 | 15.2 token/秒 | ↓out 1,234 | ↑in 567 | 11.2秒
+ *
+ * 放置位置：config/agent/extensions/
+ */
+import type { AssistantMessage } from "@earendil-works/pi-ai";
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+
+function isAssistantMessage(message: unknown): message is AssistantMessage {
+	if (!message || typeof message !== "object") return false;
+	const role = (message as { role?: unknown }).role;
+	return role === "assistant";
+}
+
+function formatDateTime(date: Date): string {
+	const year = date.getFullYear();
+	const month = String(date.getMonth() + 1).padStart(2, "0");
+	const day = String(date.getDate()).padStart(2, "0");
+	const hour = String(date.getHours()).padStart(2, "0");
+	const minute = String(date.getMinutes()).padStart(2, "0");
+	const second = String(date.getSeconds()).padStart(2, "0");
+	return `${year}-${month}-${day}T${hour}:${minute}:${second}`;
+}
+
+function formatDuration(ms: number): string {
+	if (ms < 1000) {
+		return `${ms}ms`;
+	} else if (ms < 60000) {
+		return `${(ms / 1000).toFixed(1)}秒`;
+	} else {
+		const mins = Math.floor(ms / 60000);
+		const secs = Math.floor((ms % 60000) / 1000);
+		return `${mins}分${secs}秒`;
+	}
+}
+
+export default function (pi: ExtensionAPI) {
+	let agentStartMs: number | null = null;
+
+	pi.on("agent_start", () => {
+		agentStartMs = Date.now();
+	});
+
+	pi.on("agent_end", (event, ctx) => {
+		if (!ctx.hasUI) return;
+		if (agentStartMs === null) return;
+
+		const elapsedMs = Date.now() - agentStartMs;
+		agentStartMs = null;
+		if (elapsedMs <= 0) return;
+
+		let input = 0;
+		let output = 0;
+		let cacheRead = 0;
+		let cacheWrite = 0;
+		let totalTokens = 0;
+
+		for (const message of event.messages) {
+			if (!isAssistantMessage(message)) continue;
+			input += message.usage.input || 0;
+			output += message.usage.output || 0;
+			cacheRead += message.usage.cacheRead || 0;
+			cacheWrite += message.usage.cacheWrite || 0;
+			totalTokens += message.usage.totalTokens || 0;
+		}
+
+		if (output <= 0) return;
+
+		const elapsedSeconds = elapsedMs / 1000;
+		const tokensPerSecond = output / elapsedSeconds;
+		const now = new Date();
+		const timeStr = formatDateTime(now);
+		const durationStr = formatDuration(elapsedMs);
+		const message = `${timeStr} | ✨ 处理完成 | ${tokensPerSecond.toFixed(2)} token/秒 | ↑in ${input.toLocaleString()} |↓out ${output.toLocaleString()} | ${durationStr}`;
+		ctx.ui.notify(message, "info");
+	});
+}

package/config/agent/extensions/working-msg.ts

@@ -2,11 +2,15 @@
  * @author: xujianjiang
  * working-message
  *
- * 自定义工作消息，格式类似 Claude Code: ✶ xxx… (时间 · tokens)
+ * 自定义工作消息，格式类似 Claude Code: ✶ xxx… (时间)
+ * 
+ * 改进：使用 dispose 模式防止内存泄漏
+ * - 使用明确类型，避免 any
+ * - 实现可靠的定时器清理机制
+ * - 双重保障确保资源释放
  */
 
-import type { ExtensionAPI, TurnEndEvent } from "@earendil-works/pi-coding-agent";
-import type { AssistantMessage } from "@earendil-works/pi-ai";
+import type { ExtensionAPI, ExtensionContext, TurnEndEvent } from "@earendil-works/pi-coding-agent";
 
 // 主题色 #49bccf
 const accentColor = "\x1b[38;2;73;188;207m";
@@ -44,11 +48,100 @@ const workingMessages = [
 	"Algorithm", "Architecture", "Database", "Cache", "Queue", "Microservice",
 ];
 
-let intervalId: ReturnType<typeof setInterval> | null = null;
-let startTime = 0;
-let totalTokens = 0;
-let currentUI: any = null; // 保存当前有效的 UI context
+/**
+ * 工作消息状态管理器 - 使用 dispose 模式
+ * 确保定时器和 UI 引用的生命周期管理，防止内存泄漏
+ */
+class WorkingMessageStateManager {
+	// ✅ 使用明确类型，避免 any
+	private _ui: ExtensionContext['ui'] | null = null;
+	private _intervalId: ReturnType<typeof setInterval> | null = null;
+	private _startTime: number = 0;
+	private _disposed: boolean = false;
+
+	// Getter 方法
+	get ui(): ExtensionContext['ui'] | null {
+		return this._ui;
+	}
+
+	get intervalId(): ReturnType<typeof setInterval> | null {
+		return this._intervalId;
+	}
+
+	get startTime(): number {
+		return this._startTime;
+	}
+
+	isDisposed(): boolean {
+		return this._disposed;
+	}
+
+	// Setter 方法（检查 disposed 状态）
+	setUI(ui: ExtensionContext['ui'] | null): void {
+		if (this._disposed) {
+			console.warn('[working-msg] State already disposed, cannot set UI');
+			return;
+		}
+		this._ui = ui;
+	}
+
+	setInterval(intervalId: ReturnType<typeof setInterval> | null): void {
+		if (this._disposed) {
+			console.warn('[working-msg] State already disposed, cannot set interval');
+			return;
+		}
+		this._intervalId = intervalId;
+	}
+
+	setStartTime(startTime: number): void {
+		if (this._disposed) {
+			console.warn('[working-msg] State already disposed, cannot set start time');
+			return;
+		}
+		this._startTime = startTime;
+	}
+
+	/**
+	 * dispose() - 清理所有资源
+	 * ✅ 双重保障：清理定时器 + 清理 UI 引用
+	 * ✅ disposed 标记防止重复清理
+	 */
+	dispose(): void {
+		if (this._disposed) {
+			return; // 防止重复 dispose
+		}
 
+		// ✅ 清理定时器（双重保障）
+		if (this._intervalId) {
+			clearInterval(this._intervalId);
+			this._intervalId = null;
+		}
+
+		// ✅ 清理 UI 引用
+		if (this._ui) {
+			try {
+				this._ui.setWorkingMessage(undefined);
+			} catch (error) {
+				// ✅ 忽略错误，确保清理继续
+				console.warn('[working-msg] Error clearing working message:', error);
+			}
+			this._ui = null;
+		}
+
+		this._disposed = true;
+	}
+
+	/**
+	 * reset() - 重置状态（用于新的 turn）
+	 */
+	reset(): void {
+		this.dispose(); // 先清理旧资源
+		this._disposed = false; // 重置 disposed 标记
+		this._startTime = 0;
+	}
+}
+
+// 工具函数
 function getNextMessage(): string {
 	const randomIndex = Math.floor(Math.random() * workingMessages.length);
 	return workingMessages[randomIndex];
@@ -61,76 +154,77 @@ function formatTime(seconds: number): string {
 	return `${mins} m ${secs} s`;
 }
 
-function formatTokens(tokens: number): string {
-	if (tokens < 1000) return `${tokens}`;
-	return `${(tokens / 1000).toFixed(1)}k`;
+/**
+ * ✅ 改进的定时器逻辑
+ * - 不依赖 try-catch
+ * - 检查 disposed 状态
+ * - 使用 state.ui（明确类型）
+ */
+function startInterval(state: WorkingMessageStateManager): void {
+	if (state.isDisposed() || !state.ui) {
+		return;
+	}
+
+	const intervalId = setInterval(() => {
+		// ✅ 每次执行前检查 disposed 和 UI 状态
+		if (state.isDisposed() || !state.ui) {
+			// ✅ 自动清理
+			state.dispose();
+			return;
+		}
+
+		// ✅ 安全执行（明确类型）
+		const elapsed = Math.floor((Date.now() - state.startTime) / 1000);
+		const timeStr = formatTime(elapsed);
+		const msg = `${accentColor} ✶ ${getNextMessage()}…${reset} (${timeStr})`;
+		
+		try {
+			state.ui.setWorkingMessage(msg);
+		} catch (error) {
+			// ✅ 如果 UI 调用失败，自动清理
+			console.warn('[working-msg] UI call failed, disposing:', error);
+			state.dispose();
+		}
+	}, 3000);
+
+	state.setInterval(intervalId);
 }
 
+/**
+ * 扩展主函数 - 使用 dispose 模式管理状态
+ */
 export default function workingMessageExtension(pi: ExtensionAPI) {
-	// 监听 session_shutdown 事件，清理 interval
+	// ✅ 单一状态管理器
+	const state = new WorkingMessageStateManager();
+
+	// ✅ session_shutdown - 清理所有资源
 	pi.on("session_shutdown", async (_event, ctx) => {
-		if (intervalId) {
-			clearInterval(intervalId);
-			intervalId = null;
-		}
-		currentUI = null;
+		state.dispose();
 	});
 
-	// turn_start 时开始循环切换消息
+	// ✅ turn_start - 初始化状态并启动定时器
 	pi.on("turn_start", async (_event, ctx) => {
-		if (!ctx.ui) return;
-
-		// 保存当前有效的 UI context
-		currentUI = ctx.ui;
+		if (!ctx.ui) {
+			return;
+		}
 
-		// 记录开始时间
-		startTime = Date.now();
-		totalTokens = 0;
+		// ✅ 重置状态（清理旧资源）
+		state.reset();
+		state.setUI(ctx.ui); // ✅ 明确类型：ExtensionContext['ui']
+		state.setStartTime(Date.now());
 
-		// 立即设置第一条消息
-		const elapsed = Math.floor((Date.now() - startTime) / 1000);
+		// 设置第一条消息
+		const elapsed = Math.floor((Date.now() - state.startTime) / 1000);
 		const timeStr = formatTime(elapsed);
-		const tokensStr = formatTokens(totalTokens);
-		const msg = `${accentColor} ✶ ${getNextMessage()}…${reset} (${timeStr} · ↓ ${tokensStr} tokens)`;
-		currentUI.setWorkingMessage(msg);
-
-		// 每3秒更新一次 - 使用 try-catch 处理 stale context
-		if (!intervalId) {
-			intervalId = setInterval(() => {
-				try {
-					// 使用保存的 currentUI 而不是捕获的 ctx.ui
-					if (!currentUI) return;
-
-					// 获取当前 tokens 使用量
-					// 注意：这里不再访问 ctx.sessionManager，避免 stale context 错误
-					// tokens 数量通过其他方式获取，或者保持为 0
-
-					const elapsed = Math.floor((Date.now() - startTime) / 1000);
-					const timeStr = formatTime(elapsed);
-					const tokensStr = formatTokens(totalTokens);
-					const msg = `${accentColor} ✶ ${getNextMessage()}…${reset} (${timeStr} · ↓ ${tokensStr} tokens)`;
-					currentUI.setWorkingMessage(msg);
-				} catch (error) {
-					// 如果出现错误（比如 stale context），清理 interval
-					console.error('working-msg extension error:', error);
-					if (intervalId) {
-						clearInterval(intervalId);
-						intervalId = null;
-					}
-					currentUI = null;
-				}
-			}, 3000);
-		}
+		const msg = `${accentColor} ✶ ${getNextMessage()}…${reset} (${timeStr})`;
+		state.ui?.setWorkingMessage(msg);
+
+		// ✅ 启动定时器
+		startInterval(state);
 	});
 
-	// turn结束时停止
+	// ✅ turn_end - 清理所有资源
 	pi.on("turn_end", async (_event: TurnEndEvent, ctx) => {
-		if (intervalId) {
-			clearInterval(intervalId);
-			intervalId = null;
-		}
-		// 恢复默认消息
-		currentUI?.setWorkingMessage(undefined);
-		currentUI = null;
+		state.dispose();
 	});
-}
+}

package/config/agent/settings.json

@@ -10,7 +10,8 @@
     "~/.aicode-cli/agent/extensions/custom-footer.ts",
     "~/.aicode-cli/agent/extensions/custom.ts",
     "~/.aicode-cli/agent/extensions/review.ts",
-    "~/.aicode-cli/agent/extensions/working-msg.ts"
+    "~/.aicode-cli/agent/extensions/working-msg.ts",
+    "-extensions/aicode-pets.ts"
   ],
   "packages":[  
     "npm:pi-subagents@0.17.1",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ia-ccun/code-agent-cli",
-  "version": "0.0.52-beta.2",
+  "version": "0.0.52-beta.3",
   "description": "AI Coding Agent CLI - 基于OpenClaw🦞底层Agent原理封装的编码智能体（供学习使用）。",
   "type": "module",
   "main": "./dist/index.js",