@tmustier/pi-usage-extension 0.10.0 → 0.10.1

package/CHANGELOG.md

@@ -1,5 +1,10 @@
 # Changelog
 
+## [0.10.1] - 2026-10-28
+
+### Fixed
+- Read sessions from the directory Pi uses. `/usage` always read `<agentDir>/sessions`, so with a custom session directory it showed sessions Pi no longer used and missed the ones it did. It now follows Pi's order: `--session-dir`, then `PI_CODING_AGENT_SESSION_DIR`, then the `sessionDir` setting (project over global). Thanks to @JeremieAlcaraz for the report ([#98](https://github.com/tmustier/pi-extensions/issues/98)).
+
 ## [0.10.0] - 2026-10-26
 
 ### Added

package/README.md

@@ -194,7 +194,7 @@ On narrow terminals, `/usage` automatically switches to a compact table instead
 
 ## Performance & Caching
 
-`/usage` builds its stats from every session JSONL file under `<agentDir>/sessions`. To keep opens fast on large histories (multi-GB, thousands of files):
+`/usage` builds its stats from every session JSONL file in Pi's session directory (see [Data Source](#data-source)). To keep opens fast on large histories (multi-GB, thousands of files):
 
 - **On-disk cache.** Per-file extraction results are cached in `<agentDir>/usage-extension-cache.json` (respects `PI_CODING_AGENT_DIR`), keyed by file size + mtime. Warm opens only re-parse session files that changed since the last run — on a 5.2 GB / 3,310-file corpus that takes the open from ~17 s to ~0.3 s.
 - **First open** after install (or after deleting the cache) does a one-off full build, showing the usual cancellable loader. Cancelling saves partial progress, so the next open resumes where it left off.
@@ -226,11 +226,11 @@ The "Cache" column combines both read and write tokens.
 
 ## Data Source
 
-Statistics are parsed recursively from session files in `~/.pi/agent/sessions/`, including nested subagent runs such as `run-0/` directories. Each session is a JSONL file containing assistant messages and, on Pi 0.81.0+, optional usage on tool-result, compaction, and branch-summary entries.
+Statistics are parsed recursively from session files in Pi's session directory, including nested subagent runs such as `run-0/` directories. Each session is a JSONL file containing assistant messages and, on Pi 0.81.0+, optional usage on tool-result, compaction, and branch-summary entries.
 
 Assistant messages duplicated across branched session files are deduplicated by timestamp + total tokens. Auxiliary usage is deduplicated by its stable session entry id, which avoids collapsing parallel tools that happen to report identical usage. Nested-agent reports are suppressed when their child session files are part of the scan (resolved across all copies of the parent entry, since branch copies change runId-derived paths); reports whose children are missing stay under `Tools / summaries`. Tool and summary usage contributes cost and tokens to totals, graphs, project/session mix, and burn trend, but does not count as an assistant message or distort conversation context/cache insights.
 
-Respects the `PI_CODING_AGENT_DIR` environment variable if set.
+The session directory is the one Pi uses: `--session-dir`, then the `PI_CODING_AGENT_SESSION_DIR` environment variable, then the `sessionDir` setting (project `.pi/settings.json` over the global `settings.json`), and otherwise `~/.pi/agent/sessions/`. The agent directory respects the `PI_CODING_AGENT_DIR` environment variable if set.
 
 ## Changelog
 

package/data.ts

@@ -11,6 +11,7 @@
  *   re-parses files that changed since the last run.
  */
 
+import { readFileSync } from "node:fs";
 import { readdir, readFile, rename, stat, writeFile } from "node:fs/promises";
 import { homedir } from "node:os";
 import { basename, dirname, isAbsolute, join, resolve } from "node:path";
@@ -239,8 +240,24 @@ export function getAgentDir(): string {
 	return process.env.PI_CODING_AGENT_DIR || join(homedir(), ".pi", "agent");
 }
 
-export function getSessionsDir(): string {
-	return join(getAgentDir(), "sessions");
+function sessionDirSetting(settingsPath: string): string | undefined {
+	try {
+		const { sessionDir } = JSON.parse(readFileSync(settingsPath, "utf8"));
+		return typeof sessionDir === "string" ? sessionDir : undefined;
+	} catch {
+		return undefined; // No settings file, or one Pi cannot read either.
+	}
+}
+
+/** Where Pi keeps sessions: `--session-dir`, `PI_CODING_AGENT_SESSION_DIR`, then `sessionDir` from project over global settings. */
+function getSessionsDir(): string {
+	const flag = process.argv.lastIndexOf("--session-dir");
+	const configured =
+		(flag >= 0 ? process.argv[flag + 1] : undefined) ||
+		process.env.PI_CODING_AGENT_SESSION_DIR ||
+		sessionDirSetting(join(".pi", "settings.json")) ||
+		sessionDirSetting(join(getAgentDir(), "settings.json"));
+	return configured ? resolve(configured.replace(/^~(?=\/|$)/, homedir())) : join(getAgentDir(), "sessions");
 }
 
 export function getDefaultCachePath(): string {
@@ -1387,7 +1404,7 @@ export interface CollectUsageOptions {
 	signal?: AbortSignal;
 	/** Called once before parsing begins and periodically while files are parsed. */
 	onProgress?: (progress: CollectProgress) => void;
-	/** Defaults to `<agentDir>/sessions`. */
+	/** Defaults to the directory where Pi keeps sessions. */
 	sessionsDir?: string;
 	/** Defaults to `<agentDir>/usage-extension-cache.json`. Pass `null` to disable the on-disk cache. */
 	cachePath?: string | null;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tmustier/pi-usage-extension",
-  "version": "0.10.0",
+  "version": "0.10.1",
   "description": "Usage statistics dashboard for Pi sessions.",
   "license": "MIT",
   "author": "Thomas Mustier",