@yalieny/pi-better-cost-display-footer 2.1.0 → 2.1.2

package/README.md

@@ -2,7 +2,8 @@
 
 pi 扩展:脚本化多段计价动态列表 + footer 增强。
 
-- **多段动态计价**:每个 provider 一个 `tierFn` 脚本函数(JS 表达式字符串)判定当前档位,**档位是任意字符串 id**(`peak` / `flat` / `valley` / `sharp`…),不绑定峰谷语义;在 `session_start` 和每次发送消息时重注册 provider 的 `cost`(仅影响 pi 本地用量统计)。
+- **多段动态计价**:每个 provider 一个 `tierFn` 脚本函数(JS 表达式字符串)判定当前档位,**档位是任意字符串 id**(`peak` / `flat` / `valley` / `sharp`…),不绑定峰谷语义;在 `session_start`、每次发送消息、每次 LLM 调用前重注册 provider 的 `cost`(仅影响 pi 本地用量统计)。
+- **计费自愈**:每轮比对期望 cost(配置 × 当前档位)与实际参与计费的 `ctx.model.cost`,不一致则重注册纠正;连续 3 次未收敛则停止重试并在 footer 显式告警(金额前 `≈`、标签加 `?`),不再静默错价.
 - **档位标签**:价格后紧跟当前档位标签(如 `¥0.047(梁文峰)`),按档位 id 配置文案与颜色;未配置标签的档位不显示。
 - **CH 精度**:footer 缓存命中率小数位可配(默认 1 位)。
 - **货币符号**:按 provider 配置计费金额符号(默认 `$`)。
@@ -85,10 +86,19 @@ pi install npm:@yalieny/pi-better-cost-display-footer
 
 该命令把抓取官方价目页、解析价格、更新配置文件的完整任务交给 agent 执行(自然语言指令,无需手动维护价格表)。命令会告知 agent 配置文件位置与 JSON 格式;agent 抓取 [DeepSeek 官方价目页](https://api-docs.deepseek.com/zh-cn/quick_start/pricing) 后更新 `providers.deepseek.models` 下各模型的分档价格,其余字段不动。
 
+## 计价诊断
+
+```bash
+/cost-tier
+```
+
+只读打印当前模型的档位状态:档位、期望费率 vs 实际计费费率、是否漂移、tierFn 编译状态、注册守卫与重试预算、最近一次注册结果。扩展的 `console.error` 在 TUI 下不落盘,排查"标签与金额不符"时用这条命令而不是猜。
+
 ## 测试
 
 ```bash
-npm install && npm test
+npm install && npm test                                  # 纯函数单测(档位判定 / 漂移判定)
+node --experimental-strip-types test/drift-probe.ts      # 端到端:守卫重试与自愈(用系统临时目录隔离配置)
 ```
 
 ## 说明
@@ -97,3 +107,7 @@ npm install && npm test
 - provider 必须在 `models.json` 中有定义;未列出的模型保持原价。
 - 模型缺少当前档位价格时保持原价并打日志。
 - cost 仅影响 pi 本地用量统计,不影响 API 实际账单。
+
+## 变更记录
+
+- **2.1.2** 修复"档位标签正确但计费金额按错档结算":注册守卫改为按实际计费 cost 判定漂移(不再只看档位字符串),注册失败不再写入守卫(下一轮自动重试);漂移 3 次未收敛转为 footer 可见告警(`≈` + 标签 `?`);新增 `/cost-tier` 诊断命令与注册失败一次性提示。方案见 `docs/fix-plan-stale-cost-tier.md`。

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

@@ -3,8 +3,14 @@
 //
 // 配置驱动：读取 <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（含）之前不做任何事。
+//
+// 守卫按“状态”而非“档位字符串”判定：每轮比对期望 cost（配置 × 当前档位）与实际 cost
+// （ctx.model.cost，真正参与计费的对象）。不一致 → 重注册当前 provider 自愈（最多 3 次），
+// 重试用尽仍不符 → footer 显示 ≈ + 标签加 ? 显式告警。注册失败不写守卫，下一轮重试。
+// /cost-tier 命令只读打印上述状态。
 //
 // 档位 = 任意字符串 id（如 "peak" / "flat" / "valley" / "sharp"），不绑定峰值/谷值语义：
 // {
@@ -168,7 +174,16 @@ const DEFAULT_CONFIG: DynamicConfig = {
   },
 };
 
-let lastAppliedKey: string | null = null;
+/** 注册守卫：只在全部注册成功时写入，失败保持旧值 → 下一轮 key 不等自动重试 */
+let appliedKey: string | null = null;
+/** 同一漂移状态的连续重注册次数，超 DRIFT_RETRY_LIMIT 后只告警不再重试 */
+let driftRetries = 0;
+/** 最近一次提醒的失败文案，变化时才弹，避免每轮抖动 */
+let lastErrorText: string | null = null;
+/** 最近一次注册结果，供 /cost-tier 诊断 */
+let lastRegisterResult = "尚未注册";
+/** 当前模型计费 cost 与档位不符，footer 降级显示（≈ + 标签加 ?） */
+let footerDrift = false;
 /** 最近一次事件的 ctx，footer 闭包读取；随 input/model_select 事件刷新 */
 let activeCtx: ExtensionContext | null = null;
 /** 当前生效配置，apply() 时刷新，footer 每帧直接读，避免频繁读盘 */
@@ -176,6 +191,17 @@ let currentCfg: DynamicConfig | null = null;
 /** 当前档位标签（null = 不显示），apply() 时刷新 */
 let footerLabel: FooterLabel | null = null;
 
+/** ponytail: 漂移重试预算固定 3 次，不做指数退避；真出现抖动再把预算做成配置项 */
+const DRIFT_RETRY_LIMIT = 3;
+
+/** 一次性提示：TUI 下扩展的 console.error 不落盘也不可见，错误必须走 ui.notify */
+function notifyOnce(ctx: ExtensionContext, text: string): void {
+  if (text === lastErrorText) return;
+  lastErrorText = text;
+  if (ctx.mode === "tui") ctx.ui.notify(text, "warning");
+  else console.error(text);
+}
+
 /** 深合并：文件配置覆盖内置默认（全局 + provider + model 三级），部分配置文件也能生效 */
 function mergeConfig(base: DynamicConfig, override: DynamicConfig): DynamicConfig {
   const merged: DynamicConfig = { ...base, ...override };
@@ -442,23 +468,98 @@ function dimSkip(text: string, marker: string, theme: Theme): string {
   );
 }
 
+/** 本插件视角下的模型对象子集（pi 的 Model 结构兼容，cost 只看四段扁平价） */
+interface ModelLike {
+  provider: string;
+  id: string;
+  cost?: TierRates;
+}
+
 /** 读取当前模型的生效 provider 配置；未配置 / 未生效 → null。 */
 function getActiveModelPricing(
   cfg: DynamicConfig | null,
-  ctx: ExtensionContext | null,
+  model: ModelLike | undefined,
+  date: Date = new Date(),
 ): ProviderPricing | null {
-  const model = ctx?.model;
-  const provider = model && cfg?.providers?.[model.provider];
-  if (!model || !provider || !provider.models[model.id] || !isEffective(provider)) return null;
+  const provider = model ? cfg?.providers?.[model.provider] : undefined;
+  if (!model || !provider || !provider.models[model.id] || !isEffective(provider, date)) return null;
   return provider;
 }
 
-/** 计算档位标签；当前模型无动态计价配置 / 配置缺失 / 未生效 / 档位未配置标签 → null */
-function computeFooterLabel(cfg: DynamicConfig | null, ctx: ExtensionContext | null): FooterLabel | null {
-  const provider = getActiveModelPricing(cfg, ctx);
-  if (!provider) return null;
-  const tier = currentTier(provider);
-  return provider.labels?.[tier] ?? DEFAULT_LABELS[tier] ?? null;
+/**
+ * 费率比较：浮点相对容差 1e-9，避免 JSON round-trip 造成误判漂移。
+ * ponytail: 只比四段扁平价。pi 的 ModelCost 还有请求级 tiers（本插件不产出该字段），
+ * 若将来支持分档价，这里要按 tier 逐段比较。
+ */
+export function sameRates(a: TierRates, b: TierRates | undefined): boolean {
+  if (!b) return false;
+  return (["input", "output", "cacheRead", "cacheWrite"] as const).every((k) => {
+    const want = a[k];
+    const have = b[k] as number | undefined;
+    if (typeof have !== "number") return false;
+    return Math.abs(want - have) <= 1e-9 * Math.max(Math.abs(want), Math.abs(have), 1);
+  });
+}
+
+/** 计费与档位不符：期望 = 配置 × 当前档位，实际 = 会话模型对象的 cost（真正参与计费的那个） */
+interface CostDrift {
+  providerId: string;
+  tier: Tier;
+  want: TierRates;
+  have: TierRates;
+}
+
+/** 当前模型的档位状态（一次 currentTier 判定，标签 / 漂移 / 诊断共用） */
+interface PricingState {
+  /** 生效的 provider 配置；该模型不归本插件管 → null */
+  provider: ProviderPricing | null;
+  /** 期望档位；同上 → null */
+  tier: Tier | null;
+  /** 期望费率；该模型该档位未配价 → null（保持原价，不算漂移） */
+  rates: TierRates | null;
+  /** 实际计费 cost 与期望费率不符 → 详情 */
+  drift: CostDrift | null;
+}
+
+function pricingState(
+  cfg: DynamicConfig | null,
+  model: ModelLike | undefined,
+  date: Date = new Date(),
+): PricingState {
+  const provider = getActiveModelPricing(cfg, model, date);
+  if (!provider || !model) return { provider: null, tier: null, rates: null, drift: null };
+  const tier = currentTier(provider, date);
+  const want = provider.models[model.id]?.[tier];
+  const have = model.cost;
+  const drift = want && have && !sameRates(want, have)
+    ? { providerId: model.provider, tier, want, have }
+    : null;
+  return { provider, tier, rates: want ?? null, drift };
+}
+
+/** 当前模型应有的费率（provider 未配置 / 未生效 / 该模型或该档位缺价 → null） */
+export function expectedRatesForModel(
+  cfg: DynamicConfig | null,
+  model: ModelLike | undefined,
+  date: Date = new Date(),
+): { tier: Tier; rates: TierRates } | null {
+  const st = pricingState(cfg, model, date);
+  return st.tier && st.rates ? { tier: st.tier, rates: st.rates } : null;
+}
+
+/** 漂移判定：有期望费率且与实际 cost 不等 → 详情；一致或不归本插件管 → null */
+export function detectCostDrift(
+  cfg: DynamicConfig | null,
+  model: ModelLike | undefined,
+  date: Date = new Date(),
+): CostDrift | null {
+  return pricingState(cfg, model, date).drift;
+}
+
+/** 档位标签；当前模型无动态计价配置 / 该档位未配置标签 → null */
+function computeFooterLabel(st: PricingState): FooterLabel | null {
+  if (!st.provider || !st.tier) return null;
+  return st.provider.labels?.[st.tier] ?? DEFAULT_LABELS[st.tier] ?? null;
 }
 
 function installCustomFooter(ctx: ExtensionContext): void {
@@ -524,12 +625,21 @@ function installCustomFooter(ctx: ExtensionContext): void {
       // 订阅制 provider 无法从扩展读取 modelRuntime，退化为内置的特例
       const usingSubscription = c?.model?.provider === "kimi-coding";
       // 未配置模型只改 CH；金额符号和档位标签保持内置默认
-      const customPricing = getActiveModelPricing(currentCfg, c);
+      const customPricing = getActiveModelPricing(currentCfg, c?.model);
       const labelInfo = customPricing ? footerLabel : null;
-      const labelStr = labelInfo ? colorize(theme, labelInfo.color, labelInfo.text) : "";
+      // 漂移时标签转 warning 并加问号，金额前缀 ≈：宁可显示“可疑”，不静默显示错数字
+      const drift = customPricing ? footerDrift : false;
+      const labelText = labelInfo
+        ? drift
+          ? labelInfo.text.endsWith("）")
+            ? `${labelInfo.text.slice(0, -1)}?）`
+            : `${labelInfo.text}?`
+          : labelInfo.text
+        : "";
+      const labelStr = labelInfo ? colorize(theme, drift ? "warning" : labelInfo.color, labelText) : "";
       if (totals.cost || usingSubscription) {
         const symbol = customPricing?.currencySymbol ?? "$";
-        const costStr = `${symbol}${totals.cost.toFixed(3)}${usingSubscription ? " (sub)" : ""}`;
+        const costStr = `${drift ? "≈" : ""}${symbol}${totals.cost.toFixed(3)}${usingSubscription ? " (sub)" : ""}`;
         statsParts.push(costStr + labelStr);
       }
       statsParts.push(contextPercentStr);
@@ -595,29 +705,61 @@ function apply(pi: ExtensionAPI, ctx: ExtensionContext): void {
   activeCtx = ctx;
   const cfg = loadDynamicConfig();
   currentCfg = cfg;
-  footerLabel = computeFooterLabel(cfg, ctx);
-  if (!cfg) return;
+  if (!cfg) {
+    footerLabel = null;
+    footerDrift = false;
+    return;
+  }
+  // 核心：以“实际计费 cost”为准，而不是以“档位字符串有没有变”为准
+  // 注：不能就地把费率写回 ctx.model.cost —— pi 的 model 对象被 Object.freeze（实测
+  // "Cannot assign to read only property"），且事件 ctx 无 setModel（只读 model）。
+  // 写进“参与计费的那个对象”的唯一通道就是 registerProvider → _refreshCurrentModelFromRegistry()。
+  const state = pricingState(cfg, ctx.model);
+  footerLabel = computeFooterLabel(state);
+  footerDrift = state.drift !== null;
+  const drift = state.drift;
   const providerTiers = Object.entries(cfg.providers || {}).map(([providerId, provider]) => {
     const tier = isEffective(provider) ? currentTier(provider) : null;
     return [providerId, tier] as const;
   });
   const key = `${JSON.stringify(cfg)}|${JSON.stringify(providerTiers)}`;
-  if (key === lastAppliedKey) return;
-  // 动态计价对所有配置了 providers 的模型生效（不限定 deepseek）
+  const keyChanged = key !== appliedKey;
+  // 预算复位：配置/档位变了（新状态）或已一致（收敛）
+  if (keyChanged || !drift) driftRetries = 0;
+  if (!keyChanged) {
+    if (!drift) return; // 正常路径：档位未变且计费一致 → 零开销
+    // 同一漂移状态重试用尽 → 不再每轮抖，转由 footer 告警
+    if (driftRetries >= DRIFT_RETRY_LIMIT) return;
+  }
+  // 只统计“漂移驱动”的重试：key 变化那轮本就要全量重注册，不该占预算
+  if (!keyChanged && drift) driftRetries++;
+
+  let allOk = true;
+  const touched: string[] = [];
   for (const [providerId, tier] of providerTiers) {
     if (!tier) continue;
+    // key 变了：全部重注册（与旧版一致）；只有漂移：只重注册当前模型的 provider
+    if (!keyChanged && providerId !== drift?.providerId) continue;
     try {
       const config = patchProviderConfig(cfg, providerId, tier);
+      // provider 不在 models.json 是永久配置态，不是瞬时失败，不阻塞守卫
       if (!config) continue;
       pi.registerProvider(providerId, config);
+      touched.push(`${providerId} → ${tier}`);
       console.log(`[pi-better-cost-display-footer] ${providerId} → ${tier}`);
     } catch (err) {
-      console.error(
-        `[pi-better-cost-display-footer] ${providerId} 注册失败: ${err instanceof Error ? err.message : String(err)}`,
-      );
+      allOk = false;
+      const msg = `${providerId} 注册失败: ${err instanceof Error ? err.message : String(err)}`;
+      lastRegisterResult = msg;
+      notifyOnce(ctx, `[pi-better-cost-display-footer] ${msg}`);
     }
   }
-  lastAppliedKey = key;
+  if (touched.length) lastRegisterResult = `${touched.join(", ")} @ 成功`;
+  // 失败不写守卫 → 下一轮自动重试（修 A）；恢复后允许同类故障再次告警
+  if (allOk) {
+    appliedKey = key;
+    lastErrorText = null; // 恢复后允许同类故障再次告警
+  }
 }
 
 export default function (pi: ExtensionAPI): void {
@@ -661,6 +803,38 @@ export default function (pi: ExtensionAPI): void {
       );
     },
   });
+  // 只读诊断。本次排障最大成本是“扩展 stderr 无落盘 + 状态不可观测”，命令把它们摊开
+  pi.registerCommand("cost-tier", {
+    description: "打印当前模型的计价档位诊断：期望费率 vs 实际计费费率、是否漂移、最近注册结果",
+    handler: async (_args, ctx) => {
+      const cfg = currentCfg ?? loadDynamicConfig();
+      const m = ctx.model;
+      const st = pricingState(cfg, m);
+      const fmt = (r: TierRates | null | undefined) =>
+        r ? `${r.input}/${r.output}/${r.cacheRead}/${r.cacheWrite}` : "—";
+      const fnSrc = m ? cfg?.providers?.[m.provider]?.tierFn : undefined;
+      const fnStatus = !fnSrc
+        ? "未配置（走 peakWindows / defaultTier）"
+        : tierFnCache.get(fnSrc) === null
+          ? "编译失败 → 回退 defaultTier"
+          : "OK";
+      const lines = [
+        `模型: ${m ? `${m.provider}/${m.id}` : "无"}`,
+        `配置: ${existsSync(CONFIG_FILE) ? CONFIG_FILE : `${CONFIG_FILE}（不存在，用内置默认）`}`,
+        `档位: ${st.tier ?? "不归本插件管（未配置 / 未生效 / 该模型无价）"}`,
+        `期望费率 in/out/cR/cW: ${fmt(st.rates)}`,
+        `实际 cost  in/out/cR/cW: ${fmt(m?.cost)}`,
+        `漂移: ${st.drift ? "是（计费与档位不符）" : m?.cost ? "否" : "—"}`,
+        `tierFn: ${fnStatus}`,
+        `守卫: ${appliedKey ? "已置位（配置与档位未变则不重注册）" : "未置位（下一轮重注册）"}`,
+        `漂移重试: ${driftRetries}/${DRIFT_RETRY_LIMIT}`,
+        `最近注册: ${lastRegisterResult}`,
+      ];
+      const text = `[pi-better-cost-display-footer]\n${lines.join("\n")}`;
+      if (ctx.mode === "tui") ctx.ui.notify(text, "info");
+      else console.log(text);
+    },
+  });
   pi.on("session_start", (_e, ctx) => {
     installCustomFooter(ctx);
     apply(pi, ctx);
@@ -672,4 +846,7 @@ export default function (pi: ExtensionAPI): void {
   pi.on("input", (_e, ctx) => apply(pi, ctx));
   // 切换模型时刷新档位标签（深色/深绿 ⇄ 其他模型）
   pi.on("model_select", (_e, ctx) => apply(pi, ctx));
+  // 每个 LLM 调用（含工具循环内的每次调用）开始前重算档位，对齐按调用点计费；
+  // 守卫 = 配置/档位 key 未变 且 实际 cost 与期望一致 → O(1) 返回，保证正常路径零开销。
+  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.1.0",
+  "version": "2.1.2",
   "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",