@yalieny/pi-better-cost-display-footer 2.0.1 → 2.1.1

package/extensions/pi-better-cost-display-footer.ts

@@ -3,8 +3,9 @@
 //
 // 配置驱动：读取 <configDir>/pi-better-cost-display-footer.json（configDir 默认 ~/.pi/agent，
 // 可用 PI_CODING_AGENT_DIR 覆盖）。配置文件不存在时自动生成默认配置（内置 DeepSeek 官方价目）落盘。
-// 在 session_start 和每次发送消息（input 事件）时，按各 provider 的 tierFn 脚本函数计算当前档位并重注册其 cost。
-// 跨时段边界的请求按发送时刻计费。effectiveFrom（含）之前不做任何事。
+// 在 session_start、每次发送消息（input 事件）和每次 LLM 调用前（turn_start，含工具循环内
+// 的每次调用）按各 provider 的 tierFn 脚本函数计算当前档位并重注册其 cost，对齐按调用时刻计费。
+// effectiveFrom（含）之前不做任何事。
 //
 // 档位 = 任意字符串 id（如 "peak" / "flat" / "valley" / "sharp"），不绑定峰值/谷值语义：
 // {
@@ -665,7 +666,14 @@ export default function (pi: ExtensionAPI): void {
     installCustomFooter(ctx);
     apply(pi, ctx);
   });
+  pi.on("session_shutdown", () => {
+    // Pi invalidates session-bound ctx before the next footer render.
+    activeCtx = null;
+  });
   pi.on("input", (_e, ctx) => apply(pi, ctx));
   // 切换模型时刷新档位标签（深色/深绿 ⇄ 其他模型）
   pi.on("model_select", (_e, ctx) => apply(pi, ctx));
+  // 每个 LLM 调用（含工具循环内的每次调用）开始前按调用时刻重算档位，对齐按调用点计费；
+  // lastAppliedKey 守卫保证档位未变时零开销。
+  pi.on("turn_start", (_e, ctx) => apply(pi, ctx));
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yalieny/pi-better-cost-display-footer",
-  "version": "2.0.1",
+  "version": "2.1.1",
   "description": "pi extension: scriptable peak/off-peak tier rules per provider with an enhanced footer (tier label, cache-hit-rate precision, custom currency symbol)",
   "keywords": [
     "pi-package",