token-stats-timer 1.0.7 → 1.0.8

package/README.md

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

package/package.json

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

package/step-timer.ts

@@ -2,15 +2,16 @@
 // =============================================================================
 // 实时显示(仅交互 TUI):
 //   - 任务执行中:工作指示器文案显示 "Working... 01:02"(整体已耗时,每秒刷新)
-//   - 思考阶段:隐藏思考块文案显示 "Thinking... 00:45"(当前思考块已耗时)
+// 逐条内联标注(每条 assistant 消息末尾自动追加一行):
+//   - 例:`ai 回复内容...` 后另起一行 `> ⏱ 思考 00:05`
+//   - 当前正在生成的消息:该行随流式内容实时更新(思考计时);消息提交后永久定格
+//   - 历史消息各自保留提交时的耗时,互不影响、不再变化(标注已存在则跳过,resume 回放不显示)
+//     (pi 的 hiddenThinkingLabel 文案是全局的、无法逐条保留,故改用内联标注)
 // 完成后汇总(appendEntry 持久化,不进入 LLM 上下文,resume 后可回看):
 //   - agent_settled  appendEntry("timing-final", …):本次任务总耗时
 //
 // 一次 run = 空闲后的首个 agent_start → agent_settled(与 run-timer 语义一致,
 // 包含重试、压缩恢复和排队的 steering/follow-up 提示)。
-//
-// 注意:早期版本曾用 "timing-turn" 每 turn 摘要,该渲染器已移除,
-// 历史遗留的 timing-turn 条目因无渲染器而自动不再显示。
 
 import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
 import { Box, Text } from "@earendil-works/pi-tui";
@@ -46,26 +47,32 @@ export function createStepTimer(pi: ExtensionAPI): void {
   let runFailed = false;
   let runAborted = false;
 
-  // 思考阶段状态(仅用于 Thinking... 文案计时)
-  let thinkingSinceMs: number | undefined;
+  // 逐条内联标注:当前消息的思考计时(每条 assistant 消息开始时重置)
+  // curThinkingMs:已结算的思考毫秒;thinkingStart:进行中思考块的起表时间
+  let curThinkingMs = 0;
+  let thinkingStart: number | undefined;
+
+  /** 结算当前进行中的思考块(若有) */
+  function settleThinking(now: number): void {
+    if (thinkingStart !== undefined) {
+      curThinkingMs += now - thinkingStart;
+      thinkingStart = undefined;
+    }
+  }
 
-  /** 结束思考阶段并恢复默认 "Thinking..." 文案(若当前处于思考中) */
-  function endThinking(): void {
-    if (thinkingSinceMs === undefined) return;
-    thinkingSinceMs = undefined;
-    if (lastCtx?.hasUI) lastCtx.ui.setHiddenThinkingLabel();
+  /** 当前消息的思考总耗时(含进行中块):有思考数据才显示 */
+  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();
     tick = setInterval(() => {
       if (!lastCtx?.hasUI || !runActive) return;
-      const now = Date.now();
-      const ui = lastCtx.ui;
-      ui.setWorkingMessage(`Working... ${formatDuration(now - runStartMs)}`);
-      if (thinkingSinceMs !== undefined) {
-        ui.setHiddenThinkingLabel(`Thinking... ${formatDuration(now - thinkingSinceMs)}`);
-      }
+      lastCtx.ui.setWorkingMessage(`Working... ${formatDuration(Date.now() - runStartMs)}`);
     }, TICK_MS);
   }
 
@@ -96,11 +103,20 @@ export function createStepTimer(pi: ExtensionAPI): void {
     runStartMs = Date.now();
     runFailed = false;
     runAborted = false;
-    thinkingSinceMs = undefined;
+    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;
@@ -108,43 +124,27 @@ export function createStepTimer(pi: ExtensionAPI): void {
     if (!ev) return;
     const now = Date.now();
     if (ev.type === "thinking_start") {
-      thinkingSinceMs = now;
-      if (lastCtx?.hasUI) {
-        lastCtx.ui.setHiddenThinkingLabel("Thinking... 00:00");
-      }
+      if (thinkingStart === undefined) thinkingStart = now;
     } else if (ev.type === "thinking_end" || ev.type === "text_start") {
-      endThinking();
+      settleThinking(now);
     }
   });
 
   pi.on("tool_execution_start", (_event, ctx) => {
     if (!runActive) return;
     lastCtx = ctx;
-    // 工具执行时恢复正常文案,避免 Thinking 计时残留
-    endThinking();
+    // 工具执行通常在同一消息的 toolCall 内容之后:结算思考计时
+    settleThinking(Date.now());
   });
 
   pi.on("agent_end", (event, ctx) => {
     if (!runActive) return;
     lastCtx = ctx;
-    // 扫描消息找失败/中止信号,仅用于最终摘要素材
-    for (const msg of event.messages ?? []) {
-      if (msg.type === "assistant") {
-        if (msg.stopReason === "error" || msg.stopReason === "aborted") {
-          if (msg.stopReason === "error") runFailed = true;
-          else runAborted = true;
-        }
-        if (msg.errorMessage) runFailed = true;
-      } else if (msg.type === "toolResult" && msg.isError) {
-        runFailed = true;
-      }
-    }
   });
 
   pi.on("agent_settled", (_event, ctx) => {
     lastCtx = ctx;
     if (!runActive) return;
-    endThinking();
     appendFinal();
     if (lastCtx?.hasUI) lastCtx.ui.setWorkingMessage(); // 恢复默认 "Working..."
     runActive = false;
@@ -154,7 +154,6 @@ export function createStepTimer(pi: ExtensionAPI): void {
   pi.on("session_shutdown", (_event, _ctx) => {
     // 会话提前结束(如 /exit 或崩溃):尽力补一条最终汇总
     if (runActive) {
-      endThinking();
       appendFinal();
       runActive = false;
     }
@@ -162,14 +161,21 @@ 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) => {
     const d = entry.data;
     if (!d) return undefined;
-    const title = d.failed
-      ? theme.fg("error", "❌ 任务失败 总耗时")
-      : d.aborted
-        ? theme.fg("dim", "⏹ 任务中止 总耗时")
-        : theme.fg("accent", "✅ 任务完成 总耗时");
+    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));
     return box;