pi-cache-optimizer 2.7.0 → 2.8.0

package/README.md

@@ -68,10 +68,10 @@ This extension is validated against Pi 0.83.0 and remains designed for Pi 0.82+.
 | `/cache-optimizer compat` | Shows copyable compat advice for the active model, if applicable. |
 | `/cache-optimizer stats` | Shows today's local provider/model counters and recent trend for the active model. |
 | `/cache-optimizer reset` | Resets local footer stats for the active provider/model; upstream provider cache is not modified. |
-| `/cache-optimizer config footer-mode session\|total` | Persist the footer stats mode. Persistent command configuration overrides the environment variable. |
+| `/cache-optimizer config footer-mode total\|session\|process` | Persist the footer stats mode. Persistent command configuration overrides the environment variable. |
 | `/cache-optimizer fix` | Auto-repairs safe compat issues for the active model (adaptive thinking, DeepSeek reasoning, OpenAI proxy session affinity). Shows preview + risk warning, requires confirmation. **Only modifies `models.json` after explicit user approval.** |
 
-`enable` / `disable` are current-process switches. For a persistent opt-out, use environment variables below.
+The interactive `/cache-optimizer` menu includes `Footer mode`, where you can choose `total`, `session`, or `process`. `enable` / `disable` are current-process switches. For a persistent opt-out, use environment variables below.
 
 ## Persistent opt-out
 
@@ -84,21 +84,24 @@ This extension is validated against Pi 0.83.0 and remains designed for Pi 0.82+.
 
 ## Footer cache stats mode
 
-**v2.7.0+** supports both daily cumulative and current-session footer scopes. The footer defaults to `total`, which shows the provider/model's local counters for the current day across Pi sessions and process restarts. Use either the command or environment variable to select the scope:
+**v2.7.0+** supports daily cumulative and conversation-session footer scopes. **v2.7.1** adds the Footer mode option to the interactive `/cache-optimizer` menu. **v2.8.0+** also supports process-scoped counters. The footer defaults to `total`, which shows the provider/model's local counters for the current day across Pi sessions and process restarts. Use either the command or environment variable to select the scope:
 
 | Value | Effect |
 |---|---|
 | `total` (default) | Show today's restart-persistent provider/model totals across sessions. Local day rollover resets the counters. |
-| `session` | Show only the current hashed Pi session's counters. A fresh session starts at `0/0`; `/reload` in the same session restores that session bucket. |
+| `session` | Show only the current hashed Pi conversation session's counters. A fresh conversation session starts at `0/0`; restarting Pi while resuming the same session restores that session bucket. |
+
+| `process` | Show only counters collected by the current Pi process. It starts at `0/0` after Pi restart or extension reload and is never restored from disk. |
 
 Persistent command configuration takes precedence over the environment variable:
 
 ```text
-/cache-optimizer config footer-mode session
 /cache-optimizer config footer-mode total
+/cache-optimizer config footer-mode session
+/cache-optimizer config footer-mode process
 ```
 
-The explicit setting is stored in `pi-cache-optimizer-config.json` under Pi's agent directory. If no command override exists, `PI_CACHE_OPTIMIZER_FOOTER_MODE=session|total` is used; values are case-insensitive, and missing or invalid values fall back to `total`. To return an existing installation to environment-controlled behavior, manually delete `pi-cache-optimizer-config.json` and run `/reload`.
+The explicit setting is stored in `pi-cache-optimizer-config.json` under Pi's agent directory. If no command override exists, `PI_CACHE_OPTIMIZER_FOOTER_MODE=total|session|process` is used; values are case-insensitive, and missing or invalid values fall back to `total`. To return an existing installation to environment-controlled behavior, manually delete `pi-cache-optimizer-config.json` and run `/reload`.
 
 ## OpenAI-compatible proxy setup
 
@@ -271,7 +274,7 @@ If only one model should change, use `modelOverrides`:
 
 ## Footer stats
 
-Stats are read-only local counters stored in Pi's agent directory (default: `~/.pi/agent/pi-cache-optimizer-stats.json`; custom agent dirs use `PI_CODING_AGENT_DIR`). Both today's provider/model totals and hashed session buckets are maintained. The footer shows daily totals by default, or the current session bucket when `session` mode is selected. The stats file contains only dates and numeric counters — no API keys, prompts, payloads, headers, responses, or model output. Footer mode configuration is stored separately in `pi-cache-optimizer-config.json`.
+Stats are read-only local counters stored in Pi's agent directory (default: `~/.pi/agent/pi-cache-optimizer-stats.json`; custom agent dirs use `PI_CODING_AGENT_DIR`). Both today's provider/model totals and hashed session buckets are maintained. The footer shows daily totals by default, the conversation-session bucket in `session` mode, or the in-memory process bucket in `process` mode. The stats file contains only dates and numeric counters — no API keys, prompts, payloads, headers, responses, or model output. Footer mode configuration is stored separately in `pi-cache-optimizer-config.json`. Process-mode counters are memory-only and are intentionally absent from that file.
 
 Pi 0.79+ also includes a built-in footer `CH` marker for the latest prompt cache hit rate. This extension complements that marker with persisted provider/model counters plus proxy compat diagnostics.
 

package/README.zh-CN.md

@@ -68,10 +68,10 @@ Pi 0.79.7 及之后，`pi update` 默认只更新 Pi 本体。若要更新已安
 | `/cache-optimizer compat` | 对当前模型显示可复制的 compat 建议（如适用）。 |
 | `/cache-optimizer stats` | 显示当前模型今天的本地 provider/model 统计和近期趋势。 |
 | `/cache-optimizer reset` | 重置当前 provider/model 的本地 footer 统计；不会修改上游 provider 缓存。 |
-| `/cache-optimizer config footer-mode session\|total` | 持久设置 footer 统计模式；持久命令配置优先于环境变量。 |
+| `/cache-optimizer config footer-mode total\|session\|process` | 持久设置 footer 统计模式；持久命令配置优先于环境变量。 |
 | `/cache-optimizer fix` | 为当前模型自动修复安全的 compat 问题（adaptive thinking、DeepSeek reasoning、OpenAI proxy session affinity）。展示预览 + 风险提示，需要用户确认。**仅在用户明确批准后才修改 `models.json`。** |
 
-`enable` / `disable` 是当前进程内开关。若要持久关闭某些能力，请使用下面的环境变量。
+交互式 `/cache-optimizer` 菜单包含 `Footer mode`，可以选择 `total`、`session` 或 `process`。`enable` / `disable` 是当前进程内开关。若要持久关闭某些能力，请使用下面的环境变量。
 
 ## 持久 Opt-out
 
@@ -84,21 +84,23 @@ Pi 0.79.7 及之后，`pi update` 默认只更新 Pi 本体。若要更新已安
 
 ## Footer 缓存统计模式
 
-**v2.7.0+** 同时支持每日累计与当前 session 两种 footer 统计范围。Footer 默认使用 `total`，显示当前本地日期内、跨 Pi session 和进程重启延续的 provider/model 统计。可以通过命令或环境变量切换显示范围：
+**v2.7.0+** 支持每日累计和 conversation session 两种 footer 统计范围。**v2.7.1** 在交互式 `/cache-optimizer` 菜单中加入了 Footer mode 选项。**v2.8.0+** 另外支持进程级统计。Footer 默认使用 `total`，显示当前本地日期内、跨 Pi session 和进程重启延续的 provider/model 统计。可以通过命令或环境变量切换显示范围：
 
 | 值 | 作用 |
 |---|---|
 | `total`（默认） | 显示今天跨 session 持久延续的 provider/model 累计统计；本地日期切换时重置。 |
-| `session` | 仅显示当前 hashed Pi session 的统计。新 session 从 `0/0` 开始；同一 session 内 `/reload` 会恢复该 session 桶。 |
+| `session` | 仅显示当前 hashed Pi conversation session 的统计。新 conversation session 从 `0/0` 开始；重启 Pi 后如果恢复同一 session，会恢复该 session 桶。 |
+| `process` | 仅显示当前 Pi 进程采集的统计。Pi 重启或 extension reload 后从 `0/0` 开始，且不会从磁盘恢复。 |
 
 持久命令配置优先于环境变量：
 
 ```text
-/cache-optimizer config footer-mode session
 /cache-optimizer config footer-mode total
+/cache-optimizer config footer-mode session
+/cache-optimizer config footer-mode process
 ```
 
-显式设置保存在 Pi agent 目录下的 `pi-cache-optimizer-config.json`。没有命令覆盖时，读取 `PI_CACHE_OPTIMIZER_FOOTER_MODE=session|total`；值不区分大小写，缺失或非法值均回退到 `total`。如需让已有安装重新由环境变量控制，请手动删除 `pi-cache-optimizer-config.json`，然后运行 `/reload`。
+显式设置保存在 Pi agent 目录下的 `pi-cache-optimizer-config.json`。没有命令覆盖时，读取 `PI_CACHE_OPTIMIZER_FOOTER_MODE=total|session|process`；值不区分大小写，缺失或非法值均回退到 `total`。如需让已有安装重新由环境变量控制，请手动删除 `pi-cache-optimizer-config.json`，然后运行 `/reload`。
 
 ## OpenAI-compatible 代理配置
 
@@ -271,7 +273,7 @@ Provider 级最小 override：
 
 ## Footer 统计
 
-统计是只读本地计数，保存在 Pi agent 目录（默认 `~/.pi/agent/pi-cache-optimizer-stats.json`；自定义 agent 目录使用 `PI_CODING_AGENT_DIR`）。扩展同时维护当天的 provider/model totals 和 hashed session buckets。Footer 默认显示每日 totals；选择 `session` 模式后显示当前 session 桶。统计文件只包含日期和数字计数，不包含 API key、prompt、payload、headers、响应或模型输出。Footer 模式配置单独保存在 `pi-cache-optimizer-config.json`。
+统计是只读本地计数，保存在 Pi agent 目录（默认 `~/.pi/agent/pi-cache-optimizer-stats.json`；自定义 agent 目录使用 `PI_CODING_AGENT_DIR`）。扩展同时维护当天的 provider/model totals 和 hashed session buckets。Footer 默认显示每日 totals；选择 `session` 模式后显示当前 conversation session 桶，选择 `process` 模式后显示当前进程内存桶。统计文件只包含日期和数字计数，不包含 API key、prompt、payload、headers、响应或模型输出。Footer 模式配置单独保存在 `pi-cache-optimizer-config.json`。Process 模式统计只存在内存中，不会写入该文件。
 
 Pi 0.79+ 已内置 footer `CH` 标记，用于显示最近一次 prompt cache hit rate。本扩展在此基础上补充持久化的 provider/model 计数，以及代理 compat 诊断。
 

package/index.ts

@@ -103,7 +103,7 @@ const OPENAI_PROMPT_CACHE_KEY_MAX_LENGTH = 64;
 const NO_SKILL_COMPRESSION_ENV = "PI_CACHE_OPTIMIZER_NO_SKILL_COMPRESSION";
 const NO_PROMPT_REWRITE_ENV = "PI_CACHE_OPTIMIZER_NO_PROMPT_REWRITE";
 const FOOTER_MODE_ENV = "PI_CACHE_OPTIMIZER_FOOTER_MODE";
-type FooterStatsMode = "session" | "total";
+type FooterStatsMode = "session" | "total" | "process";
 type FooterStatsModeSource = "config" | "env" | "default";
 type PersistedCacheOptimizerConfigV1 = {
   version: 1;
@@ -1143,7 +1143,7 @@ function isEnabledEnv(value: string | undefined): boolean {
 function parseFooterStatsMode(value: unknown): FooterStatsMode | undefined {
   if (typeof value !== "string") return undefined;
   const normalized = value.trim().toLowerCase();
-  return normalized === "session" || normalized === "total" ? normalized : undefined;
+  return normalized === "session" || normalized === "total" || normalized === "process" ? normalized : undefined;
 }
 
 function parsePersistedCacheOptimizerConfig(value: unknown): PersistedCacheOptimizerConfigV1 | undefined {
@@ -3897,8 +3897,10 @@ function selectFooterStatsForModel(
   statsByModel: Record<string, CacheStats>,
   totalsByModel: Record<string, CacheStats>,
   model: { provider: string; id: string },
+  processByModel: Record<string, CacheStats> = {},
 ): CacheStats | undefined {
   if (mode === "total") return totalsByModel[modelKey(model)];
+  if (mode === "process") return processByModel[modelKey(model)];
   if (!sessionHash) return undefined;
   return statsByModel[makeSessionModelKey(sessionHash, model.provider, model.id)];
 }
@@ -3909,6 +3911,7 @@ function buildExactRouterStatusEntry(
   lastRoutedModel: PersistedRoutedModelRef | undefined,
   totalsByModel: Record<string, CacheStats> = {},
   mode: FooterStatsMode = "total",
+  processByModel: Record<string, CacheStats> = {},
 ): { model: PiModel; adapter: CacheProviderAdapter; stats: CacheStats } | undefined {
   if (!sessionHash || !lastRoutedModel) return undefined;
 
@@ -3919,7 +3922,7 @@ function buildExactRouterStatusEntry(
   return {
     model,
     adapter,
-    stats: selectFooterStatsForModel(mode, sessionHash, statsByModel, totalsByModel, model) ?? emptyCacheStats(),
+    stats: selectFooterStatsForModel(mode, sessionHash, statsByModel, totalsByModel, model, processByModel) ?? emptyCacheStats(),
   };
 }
 
@@ -3928,9 +3931,12 @@ function findBestRouterModelStats(
   sessionHash: string | undefined,
   statsByModel: Record<string, CacheStats>,
   totalsByModel: Record<string, CacheStats>,
+  processByModel: Record<string, CacheStats> = {},
 ): { model: PiModel; adapter: CacheProviderAdapter; stats: CacheStats } | undefined {
   const entries = mode === "total"
     ? Object.entries(totalsByModel)
+    : mode === "process"
+      ? Object.entries(processByModel)
     : sessionHash
       ? Object.entries(statsByModel)
         .filter(([key]) => key.startsWith(`${sessionHash}:`))
@@ -6752,6 +6758,7 @@ export default function (pi: ExtensionAPI) {
   const openAISdkHeader403Models = new Set<string>();
   const warnedOpenAISdkHeader403Models = new Set<string>();
   let cacheStatsByModel: Record<string, CacheStats> = {};
+  let cacheStatsProcessByModel: Record<string, CacheStats> = {};
   let cacheStatsTotalsByModel: Record<string, CacheStats> = {};
   let cacheStatsLegacyFamily: Partial<Record<CacheProviderId, CacheStats>> = emptyAllCacheStats();
   let lastStatusText: string | undefined;
@@ -6920,6 +6927,17 @@ export default function (pi: ExtensionAPI) {
     return created;
   }
 
+  /** Get or create the current-process provider/model stats entry. */
+  function getOrCreateProcessStatsForModel(model: PiModel): CacheStats {
+    const key = modelKey(model);
+    const existing = cacheStatsProcessByModel[key];
+    if (existing) return existing;
+
+    const created = emptyCacheStats();
+    cacheStatsProcessByModel[key] = created;
+    return created;
+  }
+
   /** Get or create the cumulative provider/model stats entry shown in the footer. */
   function getOrCreateTotalStatsForModel(model: PiModel): CacheStats {
     const key = modelKey(model);
@@ -6937,6 +6955,7 @@ export default function (pi: ExtensionAPI) {
       if (modelKeyFromSessionScoped(key) === displayKey) delete cacheStatsByModel[key];
     }
     delete cacheStatsTotalsByModel[displayKey];
+    delete cacheStatsProcessByModel[displayKey];
     pendingDeletedTotalModelKeys.add(displayKey);
     for (const key of Array.from(recentSamplesByModelKey.keys())) {
       if (modelKeyFromSessionScoped(key) === displayKey) recentSamplesByModelKey.delete(key);
@@ -6949,6 +6968,7 @@ export default function (pi: ExtensionAPI) {
     for (const key of Object.keys(cacheStatsByModel)) {
       if (key.startsWith(prefix)) delete cacheStatsByModel[key];
     }
+    cacheStatsProcessByModel = {};
     cacheStatsTotalsByModel = {};
     replacePersistedTotalsOnNextWrite = true;
     for (const key of Array.from(recentSamplesByModelKey.keys())) {
@@ -7051,6 +7071,14 @@ export default function (pi: ExtensionAPI) {
     }
 
     // Roll over cumulative provider/model totals.
+    for (const key of Object.keys(cacheStatsProcessByModel)) {
+      const stats = cacheStatsProcessByModel[key];
+      if (stats && stats.day !== day) {
+        cacheStatsProcessByModel[key] = emptyCacheStats(day);
+        changed = true;
+      }
+    }
+
     for (const key of Object.keys(cacheStatsTotalsByModel)) {
       const stats = cacheStatsTotalsByModel[key];
       if (stats && stats.day !== day) {
@@ -7091,6 +7119,7 @@ export default function (pi: ExtensionAPI) {
         persisted,
         currentSessionHashSet ? currentSessionHash : undefined,
       );
+      cacheStatsProcessByModel = {};
       cacheStatsTotalsByModel = persisted?.totalsByModel ?? {};
       cacheStatsLegacyFamily = persisted?.legacyFamily ?? emptyAllCacheStats();
       lastActualRoutedModel = currentSessionHashSet
@@ -7109,6 +7138,7 @@ export default function (pi: ExtensionAPI) {
       persisted,
       currentSessionHashSet ? currentSessionHash : undefined,
     );
+    cacheStatsProcessByModel = {};
     cacheStatsTotalsByModel = persisted?.totalsByModel ?? {};
     cacheStatsLegacyFamily = persisted?.legacyFamily ?? emptyAllCacheStats();
     lastActualRoutedModel = currentSessionHashSet
@@ -7143,11 +7173,13 @@ export default function (pi: ExtensionAPI) {
         lastActualRoutedModel,
         cacheStatsTotalsByModel,
         mode,
+        cacheStatsProcessByModel,
       ) ?? findBestRouterModelStats(
         mode,
         sessionHash,
         cacheStatsByModel,
         cacheStatsTotalsByModel,
+        cacheStatsProcessByModel,
       );
       if (realEntry) {
         const statsText = formatCacheStats(realEntry.adapter, realEntry.stats);
@@ -7161,7 +7193,7 @@ export default function (pi: ExtensionAPI) {
       // Footer mode defaults to daily provider/model totals. Users may select
       // current-session counters through persistent command config or the env var.
       const stats = displayModel
-        ? selectFooterStatsForModel(mode, sessionHash, cacheStatsByModel, cacheStatsTotalsByModel, displayModel)
+        ? selectFooterStatsForModel(mode, sessionHash, cacheStatsByModel, cacheStatsTotalsByModel, displayModel, cacheStatsProcessByModel)
         : undefined;
       const statsText = formatCacheStats(adapter, stats ?? emptyCacheStats());
       statusText = runtimeOptimizerEnabled ? statsText : `Cache Optimizer disabled · ${statsText}`;
@@ -7584,11 +7616,12 @@ export default function (pi: ExtensionAPI) {
 
     await rollOverStatsIfNeeded(ctx);
 
-    // Update stats scoped to current session + actual routed model.
-    // Falls back to legacy family when no model is available.
+    // Update session, process-local, and cumulative buckets for the actual
+    // routed model. The process bucket is intentionally never persisted.
     if (statsModel) {
       const sk = sessionModelKey(statsModel);
       addUsageToCacheStats(getOrCreateStatsByModelKey(sk), usage);
+      addUsageToCacheStats(getOrCreateProcessStatsForModel(statsModel), usage);
       addUsageToCacheStats(getOrCreateTotalStatsForModel(statsModel), usage);
     } else {
       addUsageToCacheStats(getStatsForModel(undefined, adapter), usage);
@@ -7607,7 +7640,7 @@ export default function (pi: ExtensionAPI) {
   //             with low-hit diagnosis
   //   stats   — show active model stats bucket, recent trend, usage
   //   compat  — show compat suggestion with file path
-  //   config footer-mode session|total — persist footer mode override
+  //   config footer-mode total|session|process — persist footer mode override
   //   fix     — auto-fix compat issues (writes models.json, requires UI)
   //   reset   — reset current provider/model footer stats bucket (local only)
   //   (no args) — interactive menu (with UI) or help summary
@@ -7662,10 +7695,10 @@ export default function (pi: ExtensionAPI) {
       } else if (subcommand === "config") {
         const configKey = commandParts[1];
         const requestedMode = commandParts[2];
-        if (configKey !== "footer-mode" || !requestedMode || !["session", "total"].includes(requestedMode)) {
+        if (configKey !== "footer-mode" || !requestedMode || !["session", "total", "process"].includes(requestedMode)) {
           const resolved = resolveFooterStatsMode(persistedFooterStatsMode);
           cmdCtx.ui.notify(
-            `Usage: /cache-optimizer config footer-mode session|total\n` +
+            `Usage: /cache-optimizer config footer-mode total|session|process\n` +
             `Current footer mode: ${resolved.mode} (${resolved.source})`,
             "info",
           );
@@ -8074,6 +8107,7 @@ export default function (pi: ExtensionAPI) {
             "Stats — Show cache stats and trend",
             "Compat — Show compat suggestion",
             "Fix — Auto-fix compat issues (writes models.json)",
+            "Footer mode — Choose total, session, or process stats",
             "Reset — Reset local provider/model stats",
             "Cancel",
           ];
@@ -8256,6 +8290,33 @@ export default function (pi: ExtensionAPI) {
               );
             }
           } else if (choice === menuOptions[6]) {
+            const modeOptions = ["total — Daily cumulative totals (default)", "session — Current Pi conversation session", "process — Current Pi process only", "Cancel"];
+            const modeChoice = await cmdCtx.ui.select("Footer cache stats mode", modeOptions);
+            const nextMode = modeChoice === modeOptions[0]
+              ? "total"
+              : modeChoice === modeOptions[1]
+                ? "session"
+                : modeChoice === modeOptions[2]
+                  ? "process"
+                  : undefined;
+            if (nextMode) {
+              try {
+                await writePersistedFooterMode(nextMode);
+                persistedFooterStatsMode = nextMode;
+                lastStatusText = undefined;
+                await publishStatus(cmdCtx as unknown as ExtensionContext, model);
+                cmdCtx.ui.notify(
+                  `✅ Footer mode set to ${nextMode}. Persistent config overrides ${FOOTER_MODE_ENV}.`,
+                  "info",
+                );
+              } catch (error) {
+                cmdCtx.ui.notify(
+                  `❌ Could not update footer mode config: ${error instanceof Error ? error.message : String(error)}`,
+                  "error",
+                );
+              }
+            }
+          } else if (choice === menuOptions[7]) {
             if (!model) {
               cmdCtx.ui.notify("No active model selected. Select a model first with /model or pi --model.", "warning");
             } else {
@@ -8287,7 +8348,7 @@ export default function (pi: ExtensionAPI) {
         diagnosis.push("  doctor  — Show current model/provider/api/baseUrl/compat and low-hit diagnosis");
         diagnosis.push("  stats   — Show active model stats bucket and recent trend");
         diagnosis.push("  compat  — Show compat suggestion with edit location");
-        diagnosis.push("  config footer-mode session|total — Persist the footer stats mode");
+        diagnosis.push("  config footer-mode total|session|process — Persist the footer stats mode");
         diagnosis.push("  fix     — Auto-fix compat issues (writes models.json, requires UI)");
         diagnosis.push("  reset   — Reset local provider/model stats for current model (does not affect upstream)");
         diagnosis.push("");

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-cache-optimizer",
-  "version": "2.7.0",
+  "version": "2.8.0",
   "description": "Improve Pi prompt/KV cache hit rates with stable prompts, OpenAI-compatible cache keys, proxy compat warnings, and footer cache stats.",
   "keywords": [
     "pi-package",