@tekmidian/pai 1.0.3 → 1.0.4

package/dist/session-usage-CPvXvSkw.mjs

@@ -0,0 +1,222 @@
+import { createReadStream, existsSync, readFileSync } from "node:fs";
+import { createInterface } from "node:readline";
+
+//#region src/audit/session-usage.ts
+/**
+* session-usage.ts — parse the usage numbers out of a Claude Code session
+* (or subagent, or pai-worker event-mirror) JSONL transcript.
+*
+* All three log shapes share the same assistant-message envelope
+* ({ type: "assistant", message: { id, model, usage } }), so one parser
+* covers `pai audit tokens session` and the per-log readings feeding
+* `pai audit tokens spawn`.
+*
+* Streaming writes one JSONL line per content block of the same logical
+* turn, repeating message.id and usage each time — summing every line would
+* multiply usage by the block count. Keeping only the first line seen per
+* message.id is what the reference script (and this parser) count instead.
+*/
+const USAGE_KEYS = [
+	"cache_read_input_tokens",
+	"cache_creation_input_tokens",
+	"input_tokens",
+	"output_tokens"
+];
+function emptyTotals() {
+	return {
+		cache_read_input_tokens: 0,
+		cache_creation_input_tokens: 0,
+		input_tokens: 0,
+		output_tokens: 0
+	};
+}
+/**
+* `type:"system"` `subtype:"compact_boundary"` lines mark a compaction.
+* turnIndex is the count of assistant turns already folded when it fired,
+* so it lines up with the turn numbering the text/JSON report prints.
+*/
+function isCompactBoundary(line) {
+	return line.type === "system" && line.subtype === "compact_boundary";
+}
+function parseCompactionEvent(line, turnIndex) {
+	const meta = line.compactMetadata;
+	if (!meta || typeof meta.preTokens !== "number") return null;
+	return {
+		trigger: meta.trigger ?? "unknown",
+		preTokens: meta.preTokens,
+		turnIndex
+	};
+}
+/**
+* `type:"system"` `subtype:"model_refusal_fallback"` lines mark an
+* automatic model switch fired by an API safety refusal (e.g. "cyber"
+* category), not a user or config choice — worth flagging separately since
+* it can rebuild the whole prompt cache mid-session.
+*/
+function isModelFallback(line) {
+	return line.type === "system" && line.subtype === "model_refusal_fallback";
+}
+function parseFallbackEvent(line, turnIndex) {
+	return {
+		turnIndex,
+		from: line.originalModel ?? "unknown",
+		to: line.fallbackModel ?? "unknown",
+		category: line.apiRefusalCategory ?? "unknown",
+		scope: line.scope ?? "unknown"
+	};
+}
+/**
+* Fold one already-parsed JSONL line into a report being accumulated.
+* Exported separately so both the streaming file reader below and tests
+* (which build fixtures as arrays of objects, not files) share one path.
+* Returns the turn's context value when a new turn was counted, else null
+* (non-assistant line, no usage, or a duplicate message.id already seen).
+*/
+function foldAssistantLine(report, seenIds, line, threshold) {
+	if (line.type !== "assistant") return null;
+	const message = line.message;
+	const usage = message?.usage;
+	if (!usage) return null;
+	const id = message?.id ?? line.uuid;
+	if (id) {
+		if (seenIds.has(id)) return null;
+		seenIds.add(id);
+	}
+	report.turns++;
+	for (const key of USAGE_KEYS) {
+		const v = usage[key];
+		if (typeof v === "number") report.totals[key] += v;
+	}
+	const context = (Number(usage.cache_read_input_tokens) || 0) + (Number(usage.cache_creation_input_tokens) || 0) + (Number(usage.input_tokens) || 0);
+	if (report.firstTurnContext === null) {
+		report.firstTurnContext = context;
+		report.firstTurnAt = line.timestamp ?? null;
+	}
+	const atMs = line.timestamp ? Date.parse(line.timestamp) : NaN;
+	if (!Number.isNaN(atMs)) {
+		if (report.lastTurnAtMs !== null && atMs - report.lastTurnAtMs > 3600 * 1e3) report.idleGapsOver60min++;
+		report.lastTurnAtMs = atMs;
+	}
+	report.lastTurnContext = context;
+	report.maxContext = report.maxContext === null ? context : Math.max(report.maxContext, context);
+	if (context > threshold) report.turnsAboveThreshold++;
+	if ((Number(usage.cache_creation_input_tokens) || 0) > 2e4) report.cacheRebuildTurns++;
+	const model = message?.model ?? "unknown";
+	report.models[model] = (report.models[model] ?? 0) + 1;
+	if (report.lastModel !== null && report.lastModel !== model) report.modelSwitches.push({
+		turnIndex: report.turns,
+		from: report.lastModel,
+		to: model,
+		cacheRead: Number(usage.cache_read_input_tokens) || 0,
+		cacheCreation: Number(usage.cache_creation_input_tokens) || 0
+	});
+	report.lastModel = model;
+	const split = usage.cache_creation;
+	if (split) {
+		report.cacheCreationSplit.ephemeral5m += Number(split.ephemeral_5m_input_tokens) || 0;
+		report.cacheCreationSplit.ephemeral1h += Number(split.ephemeral_1h_input_tokens) || 0;
+	}
+	return context;
+}
+/**
+* True for a real human-authored `type:"user"` prompt line: content is a
+* plain string, or a content array with no `tool_result` block. Claude Code
+* encodes tool results as `type:"user"` messages whose content array is
+* entirely (or partly) tool_result blocks — those must not count as prompts.
+*/
+function isRealUserPrompt(line) {
+	if (line.type !== "user") return false;
+	if (line.isMeta) return false;
+	if (line.origin?.kind && line.origin.kind !== "human") return false;
+	const content = line.message?.content;
+	if (typeof content === "string") return true;
+	if (Array.isArray(content)) return !content.some((block) => block?.type === "tool_result");
+	return false;
+}
+/** Plain text of a real user prompt: the string content, or its first text block. */
+function extractUserPromptText(line) {
+	const content = line.message?.content;
+	if (typeof content === "string") return content;
+	if (Array.isArray(content)) return content.find((b) => b?.type === "text")?.text ?? "";
+	return "";
+}
+/**
+* Parse a session/subagent/worker-event JSONL file into a usage report.
+* `keepaliveWord`, when given, counts real user prompts whose text exactly
+* matches it (trimmed) — the sessions.cacheKeepalive beat prompt — into
+* `keepaliveBeats`; omitted, `keepaliveBeats` stays null.
+*/
+async function parseSessionUsage(path, threshold = 2e5, keepaliveWord) {
+	const report = {
+		path,
+		sizeBytes: existsSync(path) ? readFileSync(path).byteLength : 0,
+		turns: 0,
+		totals: emptyTotals(),
+		models: {},
+		firstTurnContext: null,
+		lastTurnContext: null,
+		cacheCreationSplit: {
+			ephemeral5m: 0,
+			ephemeral1h: 0
+		},
+		avgContext: null,
+		maxContext: null,
+		turnsAboveThreshold: 0,
+		cacheRebuildTurns: 0,
+		userPrompts: 0,
+		promptExposure: 0,
+		compactions: [],
+		firstTurnAt: null,
+		lastModel: null,
+		modelSwitches: [],
+		fallbacks: [],
+		lastTurnAtMs: null,
+		idleGapsOver60min: 0,
+		keepaliveBeats: keepaliveWord ? 0 : null
+	};
+	const seenIds = /* @__PURE__ */ new Set();
+	let contextSum = 0;
+	const rl = createInterface({
+		input: createReadStream(path, "utf8"),
+		crlfDelay: Infinity
+	});
+	for await (const raw of rl) {
+		if (!raw.trim()) continue;
+		let obj;
+		try {
+			obj = JSON.parse(raw);
+		} catch {
+			continue;
+		}
+		if (obj.type === "user") {
+			if (isRealUserPrompt(obj)) {
+				report.userPrompts++;
+				if (keepaliveWord && extractUserPromptText(obj).trim() === keepaliveWord) report.keepaliveBeats = (report.keepaliveBeats ?? 0) + 1;
+			}
+			continue;
+		}
+		if (isCompactBoundary(obj)) {
+			const event = parseCompactionEvent(obj, report.turns);
+			if (event) report.compactions.push(event);
+			continue;
+		}
+		if (isModelFallback(obj)) {
+			report.fallbacks.push(parseFallbackEvent(obj, report.turns));
+			continue;
+		}
+		const context = foldAssistantLine(report, seenIds, obj, threshold);
+		if (context !== null) {
+			contextSum += context;
+			report.promptExposure += report.userPrompts;
+		}
+	}
+	report.avgContext = report.turns > 0 ? Math.round(contextSum / report.turns) : null;
+	return report;
+}
+function totalUsageTokens(totals) {
+	return totals.cache_read_input_tokens + totals.cache_creation_input_tokens + totals.input_tokens + totals.output_tokens;
+}
+
+//#endregion
+export { totalUsageTokens as i, isRealUserPrompt as n, parseSessionUsage as r, extractUserPromptText as t };
+//# sourceMappingURL=session-usage-CPvXvSkw.mjs.map

package/dist/{sources-ESUkZaWv.mjs → sources-CiE3rZek.mjs}

@@ -114,4 +114,4 @@ function tail(s, n) {
 
 //#endregion
 export { cmdMemorySources };
-//# sourceMappingURL=sources-ESUkZaWv.mjs.map
+//# sourceMappingURL=sources-CiE3rZek.mjs.map

package/dist/{viewer-CqozWW39.mjs → viewer-DEySE4c_.mjs}

@@ -11,12 +11,12 @@ import "./local-tiers-BE4jGaMa.mjs";
 import "./caller-identity-yLFb7CKq.mjs";
 import "./capacity-DEveQdFV.mjs";
 import "./local-candidate-DLdwQt0s.mjs";
-import "./controls-CPoeudou.mjs";
-import "./endreason-C577xIQz.mjs";
-import "./render-BYmQkuLd.mjs";
+import "./controls-JLm6g0yP.mjs";
+import "./endreason-BC-v05H_.mjs";
+import "./render-C84BqQRC.mjs";
 import "./operator-CyuB3aVw.mjs";
-import "./kill-tree-EWYaEk5v.mjs";
-import "./pane-audit-DKQt71s4.mjs";
-import { a as followTailOutput, c as inboxCounts, d as psOutput, f as replayOutput, g as workerEnded, h as strayWorkerProcesses, i as backfillLines, l as initialFollowState, m as statusLineOutput, n as FOLLOW_TAIL_ENTRIES, o as followWorkers, p as resumeArgv, r as applyEvent, s as gutterBody, t as BACKFILL_EVENTS, u as makeOperatorInput } from "./viewer-mDXUNHNy.mjs";
+import "./kill-tree-DleMNZX3.mjs";
+import "./pane-audit-Lqy9N_Ki.mjs";
+import { a as followTailOutput, c as inboxCounts, d as psOutput, f as replayOutput, g as workerEnded, h as strayWorkerProcesses, i as backfillLines, l as initialFollowState, m as statusLineOutput, n as FOLLOW_TAIL_ENTRIES, o as followWorkers, p as resumeArgv, r as applyEvent, s as gutterBody, t as BACKFILL_EVENTS, u as makeOperatorInput } from "./viewer-wfKnvnrI.mjs";
 
 export { applyEvent, initialFollowState };

package/dist/{viewer-mDXUNHNy.mjs → viewer-wfKnvnrI.mjs}

@@ -7,11 +7,11 @@ import { _ as lastCommand } from "./tool-errors-7E13StWd.mjs";
 import { M as waitingLine, O as parseWaitOn, i as isWaitPollCommand } from "./wait-ByV0OyMZ.mjs";
 import { a as buildsDir, l as listBuildsSync, p as renderBuildQueue, v as poolSlots } from "./caller-identity-yLFb7CKq.mjs";
 import { l as schedLine, o as readSchedFile } from "./scheduler-CtUyWJ3O.mjs";
-import { Bt as chatTickerRow, Ft as chatEnter, Ht as parseChatLine, It as chatInsertLine, K as renderPlanLines, Lt as chatLeave, Nt as CHAT_HELP, Pt as chatBlankRow, Rt as chatPromptRow, Ut as wrapText, V as loadPlanStatuses, Vt as holdAutoExit, _t as paneTitle, i as localIdleLine, n as markControlsHeld, qt as readInbox, s as readLocalIdle, t as clickrControlsArgv, tt as closeOwnPane, vt as writePaneTitle, yt as psTable, zt as chatScrollRegion } from "./controls-CPoeudou.mjs";
-import { a as endLabel } from "./endreason-C577xIQz.mjs";
-import { C as denialLines, _ as renderTable, b as tickerTool, c as headerLine, f as paneStatusRow, g as renderStatusLine, h as renderEvent, i as dayOf, l as intentOf, r as contextMeter, s as gutterFor, t as blankBetween, u as makeColor, y as tickerText } from "./render-BYmQkuLd.mjs";
+import { Bt as chatTickerRow, Ft as chatEnter, Ht as parseChatLine, It as chatInsertLine, K as renderPlanLines, Lt as chatLeave, Nt as CHAT_HELP, Pt as chatBlankRow, Rt as chatPromptRow, Ut as wrapText, V as loadPlanStatuses, Vt as holdAutoExit, _t as paneTitle, i as localIdleLine, n as markControlsHeld, qt as readInbox, s as readLocalIdle, t as clickrControlsArgv, tt as closeOwnPane, vt as writePaneTitle, yt as psTable, zt as chatScrollRegion } from "./controls-JLm6g0yP.mjs";
+import { a as endLabel } from "./endreason-BC-v05H_.mjs";
+import { C as denialLines, _ as renderTable, b as tickerTool, c as headerLine, f as paneStatusRow, g as renderStatusLine, h as renderEvent, i as dayOf, l as intentOf, r as contextMeter, s as gutterFor, t as blankBetween, u as makeColor, y as tickerText } from "./render-C84BqQRC.mjs";
 import { r as sayToWorker } from "./operator-CyuB3aVw.mjs";
-import { a as readAudit, d as loadPaneRunStatuses, p as renderPaneRunLines } from "./pane-audit-DKQt71s4.mjs";
+import { a as readAudit, d as loadPaneRunStatuses, p as renderPaneRunLines } from "./pane-audit-Lqy9N_Ki.mjs";
 import { closeSync, existsSync, openSync, readFileSync, readSync } from "node:fs";
 import { spawn } from "node:child_process";
 import { createInterface } from "node:readline";
@@ -909,4 +909,4 @@ function statusLineOutput(logDir, term, _cwd, claudeSession = "", now = /* @__PU
 
 //#endregion
 export { followTailOutput as a, inboxCounts as c, psOutput as d, replayOutput as f, workerEnded as g, strayWorkerProcesses as h, backfillLines as i, initialFollowState as l, statusLineOutput as m, FOLLOW_TAIL_ENTRIES as n, followWorkers as o, resumeArgv as p, applyEvent as r, gutterBody as s, BACKFILL_EVENTS as t, makeOperatorInput as u };
-//# sourceMappingURL=viewer-mDXUNHNy.mjs.map
+//# sourceMappingURL=viewer-wfKnvnrI.mjs.map

package/dist/{worker-config-DESkIjMF.mjs → worker-config-CO5FXLAT.mjs}

@@ -7,7 +7,7 @@ import "./status-ggJ7lgVg.mjs";
 import "./pool-Dh-q3iJi.mjs";
 import "./helpers-CRuBUas3.mjs";
 import "./scope-D-h6F6DN.mjs";
-import "./routing-BAAST9vT.mjs";
+import "./routing-DmkheCRH.mjs";
 import "./tool-errors-7E13StWd.mjs";
 import "./wait-ByV0OyMZ.mjs";
 import "./local-tiers-BE4jGaMa.mjs";
@@ -15,27 +15,27 @@ import "./caller-identity-yLFb7CKq.mjs";
 import "./capacity-DEveQdFV.mjs";
 import "./local-instances-DOxSgT_L.mjs";
 import "./local-candidate-DLdwQt0s.mjs";
-import "./factory-DjAAsqr_.mjs";
-import "./postgres-C6ti4haA.mjs";
+import "./factory-B7nv75rC.mjs";
+import "./postgres-DNIOQijl.mjs";
 import "./embedding-gate-6rZrWRXa.mjs";
 import "./embeddings-BVq3esBF.mjs";
 import "./router-CQSc3lk3.mjs";
-import "./launch-Jh5NVeNe.mjs";
-import "./health-Dnym-qcT.mjs";
-import "./run-CSC7mgPf.mjs";
-import "./controls-CPoeudou.mjs";
-import "./endreason-C577xIQz.mjs";
-import "./render-BYmQkuLd.mjs";
+import "./launch-Di7N-J5X.mjs";
+import "./health-C4FQmkJn.mjs";
+import "./run-BwZKBJZZ.mjs";
+import "./controls-JLm6g0yP.mjs";
+import "./endreason-BC-v05H_.mjs";
+import "./render-C84BqQRC.mjs";
 import "./operator-CyuB3aVw.mjs";
-import "./kill-tree-EWYaEk5v.mjs";
+import "./kill-tree-DleMNZX3.mjs";
 import "./pool-cli-BggU8GRp.mjs";
-import "./pane-audit-DKQt71s4.mjs";
-import "./providers-CblTFrhz.mjs";
-import { A as workerMcpCommand, C as workerBenchCommand, D as workerFallbackCommand, E as workerControlsCommand, F as workerProvidersCommand, I as workerProxyCommand, L as workerRolesCommand, M as workerOnCommand, N as workerPolicyCommand, O as workerInstallCommand, P as workerPoolCommand, R as workerRouteCommand, S as stopInstance, T as workerConfigCommand, _ as portOpen, a as collectEnv, b as startTier, c as currentInstances, d as isUp, f as killInstance, g as portListeners, h as parseCostTier, i as cliInflight, j as workerOffCommand, k as workerLocalCommand, l as fail, m as listServers, n as WORKER_CONFIG_COMMANDS, o as collectTags, p as launchDetached, r as buildClassTarget, s as controlsArgv, t as PROVIDER_EDIT_COMMANDS, u as healthOk, v as serverLine, w as workerClassesCommand, x as stopAll, y as startInstance } from "./worker-config-CJhEMk19.mjs";
-import "./main-config-ops-BEMpuhDz.mjs";
-import "./cockpit-actions-BBHeyEZI.mjs";
-import "./ipc-client-CaiGJnSM.mjs";
-import "./embed-pass-DIWjA68P.mjs";
-import "./fallback-DFlItw1Z.mjs";
+import "./pane-audit-Lqy9N_Ki.mjs";
+import "./fallback-CBWpp2F2.mjs";
+import "./providers-C9QC0TtO.mjs";
+import { A as workerMcpCommand, C as workerBenchCommand, D as workerFallbackCommand, E as workerControlsCommand, F as workerProvidersCommand, I as workerProxyCommand, L as workerRolesCommand, M as workerOnCommand, N as workerPolicyCommand, O as workerInstallCommand, P as workerPoolCommand, R as workerRouteCommand, S as stopInstance, T as workerConfigCommand, _ as portOpen, a as collectEnv, b as startTier, c as currentInstances, d as isUp, f as killInstance, g as portListeners, h as parseCostTier, i as cliInflight, j as workerOffCommand, k as workerLocalCommand, l as fail, m as listServers, n as WORKER_CONFIG_COMMANDS, o as collectTags, p as launchDetached, r as buildClassTarget, s as controlsArgv, t as PROVIDER_EDIT_COMMANDS, u as healthOk, v as serverLine, w as workerClassesCommand, x as stopAll, y as startInstance } from "./worker-config-eY5Qy4vl.mjs";
+import "./main-config-ops-kTXR8Pyv.mjs";
+import "./cockpit-actions-OodH_vx_.mjs";
+import "./ipc-client-CnYEF31R.mjs";
+import "./embed-pass-e64AwDCa.mjs";
 
 export { currentInstances, startInstance, startTier, stopAll, stopInstance };