@f5-sales-demo/xcsh-stats 19.105.7 → 20.0.1

package/package.json

@@ -1,7 +1,7 @@
 {
 	"type": "module",
 	"name": "@f5-sales-demo/xcsh-stats",
-	"version": "19.105.7",
+	"version": "20.0.1",
 	"description": "Local observability dashboard for pi AI usage statistics",
 	"homepage": "https://github.com/f5-sales-demo/xcsh",
 	"author": "Can Boluk",
@@ -33,12 +33,13 @@
 		"check": "biome check . && bun run check:types",
 		"check:types": "tsgo -p tsconfig.json --noEmit && tsgo -p tsconfig.client.json --noEmit",
 		"lint": "biome lint .",
+		"test": "bun test",
 		"fix": "biome check --write --unsafe .",
 		"fmt": "biome format --write ."
 	},
 	"dependencies": {
-		"@f5-sales-demo/pi-ai": "19.105.7",
-		"@f5-sales-demo/pi-utils": "19.105.7",
+		"@f5-sales-demo/pi-ai": "20.0.1",
+		"@f5-sales-demo/pi-utils": "20.0.1",
 		"@tailwindcss/node": "^4.2",
 		"chart.js": "^4.5",
 		"date-fns": "~4.4.0",

package/src/aggregator.ts

@@ -15,7 +15,7 @@ import {
 	insertMessageStats,
 	setFileOffset,
 } from "./db";
-import { getSessionEntry, listAllSessionFiles, parseSessionFile } from "./parser";
+import { getSessionMessageChain, listAllSessionFiles, parseSessionFile } from "./parser";
 import type { DashboardStats, MessageStats, RequestDetails } from "./types";
 
 /**
@@ -106,17 +106,14 @@ export async function getRequestDetails(id: number): Promise<RequestDetails | nu
 	const msg = getMessageById(id);
 	if (!msg) return null;
 
-	const entry = await getSessionEntry(msg.sessionFile, msg.entryId);
-	if (entry?.type !== "message") return null;
-
-	// TODO: Get parent/context messages?
-	// For now we return the single entry which contains the assistant response.
-	// The user prompt is likely the parent.
+	const entries = await getSessionMessageChain(msg.sessionFile, msg.entryId);
+	const entry = entries.at(-1);
+	if (entry?.message.role !== "assistant") return null;
 
 	return {
 		...msg,
-		messages: [entry],
-		output: (entry as any).message,
+		messages: entries.map(item => item.message),
+		output: entry.message,
 	};
 }
 

package/src/parser.ts

@@ -167,3 +167,41 @@ export async function getSessionEntry(sessionPath: string, entryId: string): Pro
 	}
 	return null;
 }
+
+/**
+ * Reconstruct the message ancestry ending at a session entry.
+ *
+ * Non-message tree entries are traversed but omitted from the returned conversation.
+ * Missing parents and malformed cycles terminate the walk without hiding the messages
+ * that were recovered successfully.
+ */
+export async function getSessionMessageChain(sessionPath: string, entryId: string): Promise<SessionMessageEntry[]> {
+	let bytes: Uint8Array;
+	try {
+		bytes = await Bun.file(sessionPath).bytes();
+	} catch (err) {
+		if (isEnoent(err)) return [];
+		throw err;
+	}
+
+	const { entries } = parseSessionEntriesLenient(bytes);
+	const byId = new Map<string, SessionMessageEntry | (SessionEntry & { id: string; parentId: string | null })>();
+	for (const entry of entries) {
+		if ("id" in entry && "parentId" in entry && typeof entry.id === "string") {
+			byId.set(entry.id, entry);
+		}
+	}
+
+	const messages: SessionMessageEntry[] = [];
+	const visited = new Set<string>();
+	let current = byId.get(entryId);
+	while (current && !visited.has(current.id)) {
+		visited.add(current.id);
+		if (current.type === "message" && "message" in current) {
+			messages.push(current as SessionMessageEntry);
+		}
+		current = current.parentId ? byId.get(current.parentId) : undefined;
+	}
+
+	return messages.reverse();
+}

package/src/types.ts

@@ -1,4 +1,4 @@
-import type { AssistantMessage, StopReason, Usage } from "@f5-sales-demo/pi-ai";
+import type { AssistantMessage, Message, StopReason, Usage } from "@f5-sales-demo/pi-ai";
 
 /**
  * Extracted stats from an assistant message.
@@ -36,8 +36,8 @@ export interface MessageStats {
  * Full details of a request, including content.
  */
 export interface RequestDetails extends MessageStats {
-	messages: any[]; // The full conversation history or just the last turn
-	output: any; // The model's response
+	messages: Message[];
+	output: AssistantMessage;
 }
 
 /**
@@ -169,7 +169,14 @@ export interface SessionMessageEntry {
 	id: string;
 	parentId: string | null;
 	timestamp: string;
-	message: AssistantMessage | { role: "user" | "toolResult" };
+	message: Message;
 }
 
-export type SessionEntry = SessionHeader | SessionMessageEntry | { type: string };
+export interface SessionTreeEntry {
+	type: string;
+	id: string;
+	parentId: string | null;
+	[key: string]: unknown;
+}
+
+export type SessionEntry = SessionHeader | SessionMessageEntry | SessionTreeEntry | { type: string };