@pi-claudian/auto-save-to-markdown 0.1.0 → 0.1.1

package/README.md

@@ -67,10 +67,12 @@ cost: 0.023401
 tokens: 18745
 tokens_input: 15230
 tokens_output: 3515
+tokens_cache_read: 0
+tokens_cache_write: 0
 messages: 8
 created: "2026-08-29T05:05:12.000Z"
 updated: "2026-08-29T05:42:10.000Z"
-cwd: "/Users/me/project"
+project_root: "/Users/me/project"
 session_file: "~/.pi/agent/sessions/--Users-me-project-20260829-050500_ab12.jsonl"
 ---
 
@@ -103,6 +105,12 @@ kept in a collapsible `<details>` block) and summarizes each tool call and
 result in one line, so the file stays readable while still showing what the
 agent did.
 
+`cost` and the token fields cover the whole saved branch and include cached
+tokens (priced at the provider's cache rates), so the totals are comparable
+with provider-side accounting (e.g. OpenRouter activity). Requests that never
+landed in the session tree (failed retries, other sessions sharing the same
+API key) are necessarily excluded.
+
 ## Branch behavior
 
 Pi sessions are trees: `/tree` navigates to an earlier point and a new prompt

package/README.zh.md

@@ -57,10 +57,12 @@ cost: 0.023401
 tokens: 18745
 tokens_input: 15230
 tokens_output: 3515
+tokens_cache_read: 0
+tokens_cache_write: 0
 messages: 8
 created: "2026-08-29T05:05:12.000Z"
 updated: "2026-08-29T05:42:10.000Z"
-cwd: "/Users/me/project"
+project_root: "/Users/me/project"
 session_file: "~/.pi/agent/sessions/--Users-me-project-20260829-050500_ab12.jsonl"
 ---
 
@@ -92,6 +94,10 @@ auth 重构之后登录页一直重定向死循环……
 `<details>` 块中），每个工具调用和结果各压缩成一行摘要，既可读又能看出
 agent 做了什么。
 
+`cost` 和 token 字段统计整条已保存分支，且包含缓存 token（按供应商缓存
+价格计费），因此总计可与供应商侧账单（如 OpenRouter Activity）对照。未
+进入会话树的请求（失败重试、共用同一 API key 的其他会话）不在其中。
+
 ## 分支行为
 
 Pi 会话是树：`/tree` 导航到更早的位置后再提问就分出新的分支。每个

package/index.ts

@@ -18,8 +18,8 @@
  *   of the deepest message entry at file creation, and <time> is the local
  *   file-creation timestamp (YYYYMMDD-HHmmss).
  * - Frontmatter: title, session id, tree (branch key), model, provider,
- *   cumulative cost and tokens, message count, created/updated timestamps,
- *   cwd and session file.
+ *   cumulative cost and tokens (input, output, cache read/write), message
+ *   count, created/updated timestamps, project root and session file.
  * - Branching: each file records exactly ONE branch (the root→leaf path
  *   returned by sessionManager.getBranch()). State is persisted via
  *   `pi.appendEntry()` custom entries, which are part of the session tree
@@ -114,10 +114,12 @@ interface BranchMeta {
   cost: number;
   tokensInput: number;
   tokensOutput: number;
+  tokensCacheRead: number;
+  tokensCacheWrite: number;
   messages: number;
   created: string;
   updated: string;
-  cwd: string;
+  projectRoot: string;
 }
 
 export default function (pi: ExtensionAPI) {
@@ -221,23 +223,32 @@ export default function (pi: ExtensionAPI) {
   // ---------- markdown rendering ----------
 
   function renderAssistant(m: AssistantMessage, t: string): string {
-    const texts: string[] = [];
+    // Render blocks in their original chronological order: thinking always
+    // precedes the text it produced, instead of being grouped after the fact.
+    const header = `## Assistant · ${t}${m.model ? ` · ${m.model}` : ""}`;
+    const parts: string[] = [];
     const thinkings: string[] = [];
+    const flushThinking = () => {
+      if (thinkings.length) {
+        parts.push(
+          `<details>\n<summary>Thinking</summary>\n\n${thinkings.join("\n\n")}\n\n</details>`,
+        );
+        thinkings.length = 0;
+      }
+    };
     const calls: string[] = [];
     for (const b of m.content) {
-      if (b.type === "text") texts.push(b.text);
-      else if (b.type === "thinking") thinkings.push(b.thinking);
-      else if (b.type === "toolCall") calls.push(`- \`${b.name}\` — ${previewArgs(b.arguments)}`);
-    }
-
-    const header = `## Assistant · ${t}${m.model ? ` · ${m.model}` : ""}`;
-    const parts: string[] = [];
-    if (texts.length) parts.push(texts.join("\n\n"));
-    if (thinkings.length) {
-      parts.push(
-        `<details>\n<summary>Thinking</summary>\n\n${thinkings.join("\n\n")}\n\n</details>`,
-      );
+      if (b.type === "text") {
+        flushThinking();
+        parts.push(b.text);
+      } else if (b.type === "thinking") {
+        thinkings.push(b.thinking);
+      } else if (b.type === "toolCall") {
+        flushThinking();
+        calls.push(`- \`${b.name}\` — ${previewArgs(b.arguments)}`);
+      }
     }
+    flushThinking();
     if (calls.length) parts.push(`**Tool calls**\n\n${calls.join("\n")}`);
     if (m.errorMessage) parts.push(`> Error: ${m.errorMessage.replace(/\s+/g, " ").trim()}`);
     if (parts.length === 0) parts.push("_(empty response)_");
@@ -292,13 +303,19 @@ export default function (pi: ExtensionAPI) {
     if (meta.model) lines.push(`model: ${yamlQuote(meta.model)}`);
     if (meta.provider) lines.push(`provider: ${yamlQuote(meta.provider)}`);
     lines.push(`cost: ${meta.cost.toFixed(6)}`);
-    lines.push(`tokens: ${meta.tokensInput + meta.tokensOutput}`);
+    // `tokens` counts everything billed, including cached tokens, so it is
+    // comparable with provider-side token totals (e.g. OpenRouter activity).
+    lines.push(
+      `tokens: ${meta.tokensInput + meta.tokensOutput + meta.tokensCacheRead + meta.tokensCacheWrite}`,
+    );
     lines.push(`tokens_input: ${meta.tokensInput}`);
     lines.push(`tokens_output: ${meta.tokensOutput}`);
+    lines.push(`tokens_cache_read: ${meta.tokensCacheRead}`);
+    lines.push(`tokens_cache_write: ${meta.tokensCacheWrite}`);
     lines.push(`messages: ${meta.messages}`);
     lines.push(`created: ${yamlQuote(meta.created)}`);
     lines.push(`updated: ${yamlQuote(meta.updated)}`);
-    lines.push(`cwd: ${yamlQuote(meta.cwd)}`);
+    lines.push(`project_root: ${yamlQuote(meta.projectRoot)}`);
     if (meta.sessionFile) lines.push(`session_file: ${yamlQuote(meta.sessionFile)}`);
     lines.push("---");
     return lines.join("\n");
@@ -323,18 +340,21 @@ export default function (pi: ExtensionAPI) {
     let cost = 0;
     let tokensInput = 0;
     let tokensOutput = 0;
+    let tokensCacheRead = 0;
+    let tokensCacheWrite = 0;
     for (const e of pathMessages) {
       const m = e.message;
+      const usage = m.role === "assistant" ? m.usage : m.role === "toolResult" ? m.usage : null;
       if (m.role === "assistant") {
         model = m.model;
         provider = m.provider;
-        cost += m.usage?.cost?.total ?? 0;
-        tokensInput += m.usage?.input ?? 0;
-        tokensOutput += m.usage?.output ?? 0;
-      } else if (m.role === "toolResult" && m.usage) {
-        cost += m.usage.cost?.total ?? 0;
-        tokensInput += m.usage.input ?? 0;
-        tokensOutput += m.usage.output ?? 0;
+      }
+      if (usage) {
+        cost += usage.cost?.total ?? 0;
+        tokensInput += usage.input ?? 0;
+        tokensOutput += usage.output ?? 0;
+        tokensCacheRead += usage.cacheRead ?? 0;
+        tokensCacheWrite += usage.cacheWrite ?? 0;
       }
     }
     const now = new Date().toISOString();
@@ -348,10 +368,12 @@ export default function (pi: ExtensionAPI) {
       cost,
       tokensInput,
       tokensOutput,
+      tokensCacheRead,
+      tokensCacheWrite,
       messages: pathMessages.length,
       created: created ?? now,
       updated: now,
-      cwd: ctx.cwd,
+      projectRoot: ctx.cwd,
     };
   }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi-claudian/auto-save-to-markdown",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "Pi extension that automatically saves each completed conversation turn as a markdown file with YAML frontmatter, one file per session-tree branch.",
   "type": "module",
   "license": "MIT",