pi-metrics 0.4.2 → 0.5.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-metrics",
-  "version": "0.4.2",
+  "version": "0.5.0",
   "description": "Pi metrics extension: elapsed-time HUD and resilient TPS telemetry",
   "type": "module",
   "main": "./index.ts",
@@ -62,7 +62,7 @@
     "@earendil-works/pi-tui": ">=0.80.0"
   },
   "dependencies": {
-    "pi-extensions-i18n": "^0.4.1"
+    "pi-extensions-i18n": "^0.5.0"
   },
   "devDependencies": {
     "@earendil-works/pi-ai": "0.85.1",

package/src/index.ts

@@ -3,7 +3,8 @@ import type {
   ExtensionCommandContext,
   ExtensionContext,
 } from "@earendil-works/pi-coding-agent";
-import { createTranslator, loadCatalog } from "pi-extensions-i18n";
+import { createTranslator, loadCatalog, notifyWithSource } from "pi-extensions-i18n";
+import { NOTICE_SOURCE } from "./notice.ts";
 import turnElapsed from "./turn-elapsed.ts";
 import tps from "./tps.ts";
 import { configPath, loadConfig, parseConfig, saveConfig, type MetricsConfig } from "./config.ts";
@@ -29,32 +30,42 @@ function registerConfigCommand(pi: ExtensionAPI): void {
     handler: async (args: string, ctx: ExtensionCommandContext): Promise<void> => {
       const value = args.trim();
       if (value && value !== CONFIG_RESET_COMMAND && value !== ENABLE_COMMAND && value !== DISABLE_COMMAND) {
-        ctx.ui.notify(i18n.t("configCommandUsage"), NOTICE_WARNING);
+        notifyWithSource({ ctx, source: NOTICE_SOURCE, level: NOTICE_WARNING, message: i18n.t("configCommandUsage") });
         return;
       }
       if (value === CONFIG_RESET_COMMAND || value === ENABLE_COMMAND || value === DISABLE_COMMAND) {
         try {
           const config = parseConfig(value === CONFIG_RESET_COMMAND ? {} : { enabled: value === ENABLE_COMMAND });
           const path = saveConfig(config);
-          ctx.ui.notify(i18n.t("configCommandSaved", { path }), NOTICE_INFO);
+          notifyWithSource({ ctx, source: NOTICE_SOURCE, level: NOTICE_INFO, message: i18n.t("configCommandSaved", { path }) });
         } catch (error) {
-          ctx.ui.notify(i18n.t("configCommandInvalid", {
-            error: error instanceof Error ? error.message : String(error),
-          }), NOTICE_ERROR);
+          notifyWithSource({
+            ctx,
+            source: NOTICE_SOURCE,
+            level: NOTICE_ERROR,
+            message: i18n.t("configCommandInvalid", {
+              error: error instanceof Error ? error.message : String(error),
+            }),
+          });
         }
         return;
       }
       if (!ctx.hasUI) {
-        ctx.ui.notify(i18n.t("configCommandInteractiveOnly"), NOTICE_WARNING);
+        notifyWithSource({ ctx, source: NOTICE_SOURCE, level: NOTICE_WARNING, message: i18n.t("configCommandInteractiveOnly") });
         return;
       }
       let current: MetricsConfig;
       try {
         current = loadConfig();
       } catch (error) {
-        ctx.ui.notify(i18n.t("configCommandInvalid", {
-          error: error instanceof Error ? error.message : String(error),
-        }), NOTICE_ERROR);
+        notifyWithSource({
+          ctx,
+          source: NOTICE_SOURCE,
+          level: NOTICE_ERROR,
+          message: i18n.t("configCommandInvalid", {
+            error: error instanceof Error ? error.message : String(error),
+          }),
+        });
         return;
       }
       const enabledChoice = i18n.t("configEnabled", { value: i18n.t(current.enabled ? "configOn" : "configOff") });
@@ -69,11 +80,16 @@ function registerConfigCommand(pi: ExtensionAPI): void {
       const choice = selected === enabledChoice ? CONFIG_CHOICE.enabled : CONFIG_CHOICE.disabled;
       try {
         const path = saveConfig({ enabled: choice === CONFIG_CHOICE.enabled });
-        ctx.ui.notify(i18n.t("configCommandSaved", { path }), NOTICE_INFO);
+        notifyWithSource({ ctx, source: NOTICE_SOURCE, level: NOTICE_INFO, message: i18n.t("configCommandSaved", { path }) });
       } catch (error) {
-        ctx.ui.notify(i18n.t("configCommandInvalid", {
-          error: error instanceof Error ? error.message : String(error),
-        }), NOTICE_ERROR);
+        notifyWithSource({
+          ctx,
+          source: NOTICE_SOURCE,
+          level: NOTICE_ERROR,
+          message: i18n.t("configCommandInvalid", {
+            error: error instanceof Error ? error.message : String(error),
+          }),
+        });
       }
     },
   };
@@ -93,10 +109,15 @@ export default function piHud(pi: ExtensionAPI): void {
   }
   if (configError !== undefined) {
     pi.on("session_start", (_event, ctx: ExtensionContext) => {
-      ctx.ui.notify(i18n.t("configLoadFailed", {
-        path: configPath(),
-        error: configError instanceof Error ? configError.message : String(configError),
-      }), NOTICE_WARNING);
+      notifyWithSource({
+        ctx,
+        source: NOTICE_SOURCE,
+        level: NOTICE_WARNING,
+        message: i18n.t("configLoadFailed", {
+          path: configPath(),
+          error: configError instanceof Error ? configError.message : String(configError),
+        }),
+      });
     });
   }
   if (!config.enabled) return;

package/src/notice.ts

@@ -0,0 +1,13 @@
+/**
+ * 本扩展统一的提示来源定义。
+ * 所有 notifyWithSource 调用共用同一份标签与颜色，避免两个入口不一致。
+ */
+
+import type { NoticeColor, NoticeSource } from "pi-extensions-i18n";
+
+/** 本扩展的提示标签；短且唯一，便于在会话里定位来源。 */
+export const NOTICE_TAG = "metrics";
+/** 提示标签颜色；与其它扩展错开，避免看起来像同一条消息。 */
+export const NOTICE_COLOR: NoticeColor = "dim";
+/** 本扩展的提示来源。 */
+export const NOTICE_SOURCE: NoticeSource = { tag: NOTICE_TAG, color: NOTICE_COLOR };

package/src/tps.ts

@@ -11,7 +11,9 @@ import type {
   ExtensionAPI,
   ExtensionContext,
 } from "@earendil-works/pi-coding-agent";
+import { notifyWithSource } from "pi-extensions-i18n";
 import { i18n } from "./i18n.ts";
+import { NOTICE_SOURCE } from "./notice.ts";
 
 interface TurnStartEvent {
   type: "turn_start";
@@ -330,11 +332,11 @@ function restoreTPSNotification(
     const data = entry.data as Record<string, unknown> | null | undefined;
     if (!data) continue;
     if (typeof data.model === "object" && data.model !== null) {
-      schedule(() => ctx.ui.notify(composeDisplayString(data as unknown as TurnTelemetry), "info"));
+      schedule(() => notifyWithSource({ ctx, source: NOTICE_SOURCE, level: "info", message: composeDisplayString(data as unknown as TurnTelemetry) }));
       return;
     }
     if (typeof data.message === "string") {
-      schedule(() => ctx.ui.notify(data.message as string, "info"));
+      schedule(() => notifyWithSource({ ctx, source: NOTICE_SOURCE, level: "info", message: data.message as string }));
       return;
     }
   }
@@ -392,7 +394,7 @@ export default function tpsExtension(pi: ExtensionAPI): void {
     committed.telemetry = corrected;
     pi.appendEntry("tps", corrected);
     pi.events?.emit("tps:telemetry", corrected);
-    if (committed.ctx.hasUI) committed.ctx.ui.notify(composeDisplayString(corrected), "info");
+    if (committed.ctx.hasUI) notifyWithSource({ ctx: committed.ctx, source: NOTICE_SOURCE, level: "info", message: composeDisplayString(corrected) });
   });
 
   pi.on("session_shutdown", () => {
@@ -513,7 +515,7 @@ export default function tpsExtension(pi: ExtensionAPI): void {
     };
     pi.appendEntry("tps", telemetry);
     pi.events?.emit("tps:telemetry", telemetry);
-    if (ctx.hasUI) ctx.ui.notify(composeDisplayString(telemetry), "info");
+    if (ctx.hasUI) notifyWithSource({ ctx, source: NOTICE_SOURCE, level: "info", message: composeDisplayString(telemetry) });
   });
 
 }

package/src/turn-elapsed.ts

@@ -18,8 +18,10 @@
  */
 
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import { notifyWithSource } from "pi-extensions-i18n";
 import { formatDone, formatTick } from "./format-utils.ts";
 import { i18n } from "./i18n.ts";
+import { NOTICE_SOURCE } from "./notice.ts";
 
 const TICK_MS = 1000;
 
@@ -74,7 +76,7 @@ export default function (pi: ExtensionAPI) {
     // 恢复 pi 默认 working 文字（下次 streaming 由 pi 内部重置）
     ctx.ui.setWorkingMessage(undefined);
     // 在 chat 流末尾插入一条 dim 灰文本
-    ctx.ui.notify(i18n.t("elapsedDone", { value: formatDone(elapsed) }), "info");
+    notifyWithSource({ ctx, source: NOTICE_SOURCE, level: "info", message: i18n.t("elapsedDone", { value: formatDone(elapsed) }) });
   });
 
   pi.on("agent_end", async (_event, ctx) => {
@@ -93,7 +95,7 @@ export default function (pi: ExtensionAPI) {
 
     ctx.ui.setWorkingMessage(undefined);
     if (runElapsed > 0) {
-      ctx.ui.notify(i18n.t("elapsedTotal", { value: formatDone(runElapsed) }), "info");
+      notifyWithSource({ ctx, source: NOTICE_SOURCE, level: "info", message: i18n.t("elapsedTotal", { value: formatDone(runElapsed) }) });
     }
   });
 }