token-stats-timer 1.0.8 → 1.0.9

package/README.md

@@ -57,10 +57,13 @@ Footer 下行：cwd + git 分支 + 其他扩展状态。
 ## 任务计时（step-timer）
 
 - 任务执行中：工作指示器（spinner 文案）显示 `Working... 01:02`（整体已耗时，每秒刷新）
-- 逐条内联标注：每条 assistant 消息末尾自动追加一行 `> ⏱ 思考 00:05`——当前正在生成的消息该行实时更新，提交后永久定格；历史消息各自保留提交时的耗时、互不影响。因 pi 的 hiddenThinkingLabel 文案是全局的、无法逐条保留，故采用该内联方式（resume 回放不显示）
-- 完成后：会话末尾插入一条汇总 `总耗时 01:23`（失败时 ❌，中止时 ⏹），经 `appendEntry` 持久化、不进入 LLM 上下文，`/resume` 后仍在
+- 完成后：会话末尾插入一条汇总，仅显示总耗时：
 
-无独立开关，随包启用；计时口径与 run-timer 一致（一次 run = 首个 agent_start → agent_settled，含重试/压缩/排队提示）。早期版本的每 turn 摘要（timing-turn）已移除，历史条目自动隐藏。
+  `总耗时 01:23`
+
+  经 `appendEntry` 持久化、不进入 LLM 上下文，`/resume` 后仍在；无主题色样式。
+
+无独立开关，随包启用；计时口径与 run-timer 一致（一次 run = 首个 agent_start → agent_settled，含重试/压缩/排队提示）。早期版本的每 turn 摘要（timing-turn）、“⏱ 思考”每条消息内联标注与 Thinking 实时标签均已移除，历史条目自动隐藏。
 
 ## 命令
 

package/notify.ts

@@ -9,7 +9,7 @@
 //   enabled       总开关（默认 true）
 //   minDurationSec 时长低于该秒数的 run 不通知（默认 0 = 每次都通知）
 //   sound         通知声音（默认 "Glass"，"" 表示静音）
-//   onSuccess / onFailure / onAbort / onSessionEnd  各类通知开关
+//   onSuccess / onAbort / onSessionEnd  各类通知开关
 //
 // 命令：/notify on | off | status
 
@@ -27,7 +27,6 @@ export interface NotifyConfig {
   minDurationSec: number;
   sound: string;
   onSuccess: boolean;
-  onFailure: boolean;
   onAbort: boolean;
   onSessionEnd: boolean;
 }
@@ -37,7 +36,6 @@ const DEFAULT_CONFIG: NotifyConfig = {
   minDurationSec: 0,
   sound: "Glass",
   onSuccess: true,
-  onFailure: true,
   onAbort: false,
   onSessionEnd: false,
 };
@@ -80,9 +78,7 @@ export function createNotifier(pi: ExtensionAPI): void {
 
   // 当前 run 的状态（agent_start 重置，agent_end 收集，agent_settled 消费）
   let runStartMs: number | undefined;
-  let runFailed = false;
   let runAborted = false;
-  let runErrorMsg: string | undefined;
 
   // 进程内去抖计时
   let lastNotifyAt = 0;
@@ -102,9 +98,7 @@ export function createNotifier(pi: ExtensionAPI): void {
   }
 
   pi.on("agent_start", () => {
-    runFailed = false;
     runAborted = false;
-    runErrorMsg = undefined;
     if (runStartMs === undefined) runStartMs = Date.now();
   });
 
@@ -115,17 +109,7 @@ export function createNotifier(pi: ExtensionAPI): void {
       if (m.role === "assistant") {
         if (m.stopReason === "aborted") {
           runAborted = true;
-          runErrorMsg = runErrorMsg || "被用户中止";
-        } else if (m.stopReason === "error" || m.errorMessage) {
-          runFailed = true;
-          runErrorMsg = runErrorMsg || m.errorMessage || "模型调用出错";
         }
-      } else if (m.role === "toolResult" && m.isError) {
-        runFailed = true;
-        const toolMsg = `工具 ${m.toolName} 执行失败`;
-        runErrorMsg = runErrorMsg && !runErrorMsg.startsWith("工具 ")
-          ? runErrorMsg
-          : toolMsg;
       }
     }
   });
@@ -142,19 +126,11 @@ export function createNotifier(pi: ExtensionAPI): void {
 
     const dur = startMs !== undefined ? `${formatDuration(durationMs)}` : "-";
 
-    if (runFailed) {
-      if (config.onFailure) {
-        notify(
-          "❌ pi 执行失败",
-          `耗时：${dur}\n${truncate(runErrorMsg || "未知错误")}`,
-          config.sound,
-        );
-      }
-    } else if (runAborted) {
+    if (runAborted) {
       if (config.onAbort) {
         notify(
           "⏹ pi 已中止",
-          `耗时：${dur}\n${truncate(runErrorMsg || "执行被中止")}`,
+          `耗时：${dur}\n${truncate("执行被中止")}`,
           config.sound,
         );
       }
@@ -167,9 +143,7 @@ export function createNotifier(pi: ExtensionAPI): void {
     }
 
     // 消费本次结果，避免串扰下一次 run
-    runFailed = false;
     runAborted = false;
-    runErrorMsg = undefined;
   });
 
   // 会话关闭时提示（可选）
@@ -196,7 +170,7 @@ export function createNotifier(pi: ExtensionAPI): void {
         const sound = config.sound ? `, 声音=${config.sound}` : ", 静音";
         ctx.ui.notify(
           `完成通知: ${config.enabled ? "开" : "关"}, 最短时长=${config.minDurationSec}s, ` +
-            `成功=${config.onSuccess ? "开" : "关"}, 失败=${config.onFailure ? "开" : "关"}, ` +
+            `成功=${config.onSuccess ? "开" : "关"}, ` +
             `中止=${config.onAbort ? "开" : "关"}, 会话结束=${config.onSessionEnd ? "开" : "关"}${sound}`,
           "info",
         );

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "token-stats-timer",
-  "version": "1.0.8",
+  "version": "1.0.9",
   "description": "pi extension to display token usage and time spend.",
   "keywords": [
     "pi-package",

package/step-timer.ts

@@ -2,13 +2,8 @@
 // =============================================================================
 // 实时显示(仅交互 TUI):
 //   - 任务执行中:工作指示器文案显示 "Working... 01:02"(整体已耗时,每秒刷新)
-// 逐条内联标注(每条 assistant 消息末尾自动追加一行):
-//   - 例:`ai 回复内容...` 后另起一行 `> ⏱ 思考 00:05`
-//   - 当前正在生成的消息:该行随流式内容实时更新(思考计时);消息提交后永久定格
-//   - 历史消息各自保留提交时的耗时,互不影响、不再变化(标注已存在则跳过,resume 回放不显示)
-//     (pi 的 hiddenThinkingLabel 文案是全局的、无法逐条保留,故改用内联标注)
 // 完成后汇总(appendEntry 持久化,不进入 LLM 上下文,resume 后可回看):
-//   - agent_settled  appendEntry("timing-final", …):本次任务总耗时
+//   - agent_settled  appendEntry("timing-final", …):仅本次任务总耗时
 //
 // 一次 run = 空闲后的首个 agent_start → agent_settled(与 run-timer 语义一致,
 // 包含重试、压缩恢复和排队的 steering/follow-up 提示)。
@@ -19,8 +14,6 @@ import { Box, Text } from "@earendil-works/pi-tui";
 /** 一次 run 的总耗时汇总(appendEntry "timing-final") */
 export interface FinalTimingData {
   totalMs: number;
-  failed: boolean;
-  aborted: boolean;
 }
 
 const FINAL_TYPE = "timing-final";
@@ -44,29 +37,6 @@ export function createStepTimer(pi: ExtensionAPI): void {
 
   let runActive = false;
   let runStartMs = 0;
-  let runFailed = false;
-  let runAborted = false;
-
-  // 逐条内联标注:当前消息的思考计时(每条 assistant 消息开始时重置)
-  // curThinkingMs:已结算的思考毫秒;thinkingStart:进行中思考块的起表时间
-  let curThinkingMs = 0;
-  let thinkingStart: number | undefined;
-
-  /** 结算当前进行中的思考块(若有) */
-  function settleThinking(now: number): void {
-    if (thinkingStart !== undefined) {
-      curThinkingMs += now - thinkingStart;
-      thinkingStart = undefined;
-    }
-  }
-
-  /** 当前消息的思考总耗时(含进行中块):有思考数据才显示 */
-  function currentThinkingMs(now: number): number | undefined {
-    if (curThinkingMs === 0 && thinkingStart === undefined) return undefined;
-    return curThinkingMs + (thinkingStart !== undefined ? now - thinkingStart : 0);
-  }
-
-  const MARK_TEXT = "⏱ 思考";
 
   function startTicker(): void {
     stopTicker();
@@ -85,8 +55,6 @@ export function createStepTimer(pi: ExtensionAPI): void {
   function appendFinal(): void {
     pi.appendEntry<FinalTimingData>(FINAL_TYPE, {
       totalMs: Date.now() - runStartMs,
-      failed: runFailed,
-      aborted: runAborted,
     });
   }
 
@@ -101,47 +69,10 @@ export function createStepTimer(pi: ExtensionAPI): void {
     if (runActive) return;
     runActive = true;
     runStartMs = Date.now();
-    runFailed = false;
-    runAborted = false;
-    curThinkingMs = 0;
-    thinkingStart = undefined;
     startTicker();
     if (lastCtx?.hasUI) lastCtx.ui.setWorkingMessage("Working... 00:00");
   });
 
-  // 每条 assistant 消息开始时重置思考计时
-  pi.on("message_start", (event, ctx) => {
-    if (!runActive || event.message.role !== "assistant") return;
-    lastCtx = ctx;
-    curThinkingMs = 0;
-    thinkingStart = undefined;
-  });
-
-  pi.on("message_update", (event, ctx) => {
-    if (!runActive) return;
-    lastCtx = ctx;
-    const ev = event.assistantMessageEvent;
-    if (!ev) return;
-    const now = Date.now();
-    if (ev.type === "thinking_start") {
-      if (thinkingStart === undefined) thinkingStart = now;
-    } else if (ev.type === "thinking_end" || ev.type === "text_start") {
-      settleThinking(now);
-    }
-  });
-
-  pi.on("tool_execution_start", (_event, ctx) => {
-    if (!runActive) return;
-    lastCtx = ctx;
-    // 工具执行通常在同一消息的 toolCall 内容之后:结算思考计时
-    settleThinking(Date.now());
-  });
-
-  pi.on("agent_end", (event, ctx) => {
-    if (!runActive) return;
-    lastCtx = ctx;
-  });
-
   pi.on("agent_settled", (_event, ctx) => {
     lastCtx = ctx;
     if (!runActive) return;
@@ -161,23 +92,11 @@ export function createStepTimer(pi: ExtensionAPI): void {
     lastCtx = undefined;
   });
 
-  // ---------- 逐条内联标注 ----------
-  // 只在 assistant 文本上追加一行耗时;已带标注(历史消息重渲染)则跳过保持原值
-  pi.registerMarkdownTransformer((markdown, context) => {
-    if (context.messageType !== "assistant") return markdown;
-    if (markdown.includes(MARK_TEXT)) return markdown;
-    const now = Date.now();
-    const ms = currentThinkingMs(now);
-    if (ms === undefined) return markdown;
-    return `${markdown}\n\n> ${MARK_TEXT} ${formatDuration(ms)}`;
-  });
-
-  pi.registerEntryRenderer<FinalTimingData>(FINAL_TYPE, (entry, _opts, theme) => {
+  pi.registerEntryRenderer<FinalTimingData>(FINAL_TYPE, (entry, _opts, _theme) => {
     const d = entry.data;
     if (!d) return undefined;
-    const title = theme.fg("accent", "总耗时");
-    const box = new Box(1, 1, (text) => theme.bg("customMessageBg", text));
-    box.addChild(new Text(`${title} ${formatDuration(d.totalMs)}`, 0, 0));
+    const box = new Box(1, 1);
+    box.addChild(new Text(`总耗时 ${formatDuration(d.totalMs)}`, 0, 0));
     return box;
   });
 }