@maxiaochao/pi-toolkit 0.1.7 → 0.1.8

package/CHANGELOG.md

@@ -1,3 +1,7 @@
+## 2026-08-31 - v0.1.8
+
+- fix cache export branch attribution and update HTML artifact guidance
+
 ## 2026-08-21 - v0.1.7
 
 - add the html-artifact skill for interactive HTML explanations and codebase visualizations

package/extensions/cache-export/index.ts

@@ -63,7 +63,10 @@ export default function (pi) {
   pi.registerCommand("cache_export", {
     description: "Export a tau-style interactive cache dashboard (hit-rate charts, per-request table, events)",
     handler: async (args, ctx) => {
-      const entries = ctx.sessionManager.getEntries();
+      const sessionManager = ctx.sessionManager;
+      const entries = typeof sessionManager.getBranch === "function"
+        ? sessionManager.getBranch()
+        : sessionManager.getEntries();
       if (!entries || entries.length === 0) {
         ctx.ui.notify("No session entries to analyze.", "error");
         return;

package/extensions/cache-export/tests/run-tests.mjs

@@ -1,5 +1,9 @@
 // Deterministic test cases for analyzeStreaks / analyzeMisses / parseEntries.
 // Run: node --experimental-strip-types tests/run-tests.mjs
+import { mkdtempSync, readFileSync, rmSync } from "node:fs";
+import { join } from "node:path";
+import { tmpdir } from "node:os";
+import registerCacheExport from "../index.ts";
 import { parseEntries, analyzeStreaks, analyzeMisses, lineChartForTest } from "../render.ts";
 
 let pass = 0, fail = 0;
@@ -314,6 +318,53 @@ const hitOf = (r) => r.cached / (r.fresh + r.cached + r.cacheWrite);
     s.length === 2 && s.every((x) => x.hits.length === 3), JSON.stringify(s.map((x) => [x.from, x.to])));
 }
 
+// Q. subagent results remain visible as main-session tool input -----------
+{
+  const entries = [
+    req({ cached: 0, fresh: 10000, min: 1, content: [
+      { type: "toolCall", tool: "subagent", name: "subagent" },
+      { type: "toolCall", tool: "bash", name: "bash" },
+    ] }),
+    toolResult("delegated result", "subagent"),
+    toolResult("ordinary result", "bash"),
+    req({ cached: 9000, fresh: 10000, min: 2 }),
+  ];
+  const d = parseEntries(entries);
+  check("Q subagent invocation remains in main-session tool counts",
+    d.toolCalls.some(([name]) => name === "subagent") && d.toolCalls.some(([name]) => name === "bash"),
+    JSON.stringify(d.toolCalls));
+  check("Q subagent result remains available for main fresh-input attribution",
+    d.requests[1].gapTools.length === 2 && d.requests[1].gapTools.some(({ tool }) => tool === "subagent"),
+    JSON.stringify(d.requests[1].gapTools));
+}
+
+// R. command exports only the current main-session branch ----------------
+{
+  const root = mkdtempSync(join(tmpdir(), "cache-export-test-"));
+  const reportFile = join(root, "report.html");
+  const entry = (value, id, parentId) => ({ ...value, id, parentId });
+  const first = entry(req({ cached: 9000, fresh: 1000, min: 1, model: "main-first" }), "p1", null);
+  const abandoned = entry(req({ cached: 0, fresh: 9000, min: 2, model: "abandoned-model" }), "p2", "p1");
+  const active = entry(req({ cached: 8000, fresh: 2000, min: 3, model: "active-model" }), "p3", "p1");
+
+  let command;
+  registerCacheExport({ registerCommand: (name, definition) => { if (name === "cache_export") command = definition; } });
+  await command.handler(reportFile, {
+    sessionManager: {
+      getBranch: () => [first, active],
+      getEntries: () => [first, abandoned, active],
+      getSessionFile: () => join(root, "main.jsonl"),
+      getSessionId: () => "main",
+    },
+    ui: { notify: () => {} },
+  });
+  const html = readFileSync(reportFile, "utf8");
+  check("R current branch excludes abandoned requests",
+    html.includes("2 requests") && html.includes("active-model") && !html.includes("abandoned-model"));
+  check("R main-only report has no subagent dashboard", !html.includes("Subagent cache"));
+  rmSync(root, { recursive: true, force: true });
+}
+
 // K. degenerate inputs ---------------------------------------------------
 {
   check("K empty session -> no crash, no streaks", analyzeStreaks(parseEntries([]).requests).length === 0);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@maxiaochao/pi-toolkit",
-  "version": "0.1.7",
+  "version": "0.1.8",
   "description": "Personal pi coding agent toolkit: /todos extension + /cache_export interactive cache dashboard.",
   "type": "module",
   "keywords": [

package/skills/html-artifact/SKILL.md

@@ -1,12 +1,10 @@
 ---
 name: html-artifact
-description: Create self-contained, responsive HTML artifacts for complex explanations that benefit from visual structure or interaction, including workflows, code/module relationships, state machines, event/data flows, architecture maps, timelines, comparisons, dashboards, and interactive demonstrations. Use Markdown for ordinary prose, short answers, diffs, and focused code reviews.
+description: Create self-contained, responsive HTML artifacts for complex explanations that benefit from visual structure or interaction, including workflows, code/module relationships, state machines, event/data flows, architecture maps, timelines, comparisons, dashboards, and interactive demonstrations.
 ---
 
 # HTML Artifact
 
-Use this skill when the user needs to understand or explore a complex process, codebase relationship, system design, state transition, comparison, or other content where static Markdown would be hard to scan or interact with.
-
 ## Decide the format
 
 Use Markdown when the response is short, linear, text-first, or primarily a focused code review. Use an HTML artifact when visual layout, filtering, expanding details, tabs, highlighting, diagrams, timelines, or interactive state changes materially improve understanding.