@braintrust/pi-extension 0.2.0 → 0.3.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@braintrust/pi-extension",
-  "version": "0.2.0",
+  "version": "0.3.1",
   "description": "Braintrust extension for pi. Includes automatic tracing for pi sessions, turns, LLM calls, and tool executions to Braintrust.",
   "keywords": [
     "braintrust",
@@ -24,17 +24,16 @@
     "access": "public"
   },
   "dependencies": {
-    "braintrust": "^3.7.0",
+    "braintrust": "^3.8.0",
     "valibot": "^1.3.1"
   },
   "devDependencies": {
-    "@mariozechner/pi-ai": "^0.64.0",
-    "@mariozechner/pi-coding-agent": "^0.64.0",
-    "@types/node": "^25.5.0",
-    "tsx": "^4.21.0",
+    "@mariozechner/pi-ai": "^0.67.2",
+    "@mariozechner/pi-coding-agent": "^0.67.2",
+    "@types/node": "^25.6.0",
     "typescript": "^6.0.2",
-    "vite-plus": "^0.1.14",
-    "vitest": "^4.1.2"
+    "vite-plus": "^0.1.16",
+    "vitest": "^4.1.4"
   },
   "peerDependencies": {
     "@mariozechner/pi-coding-agent": "*"
@@ -66,6 +65,6 @@
     "test:integration": "vitest run src/index.integration.test.ts",
     "test:watch": "vitest",
     "typecheck": "vp check",
-    "smoke": "tsx -e \"import('./src/index.ts')\""
+    "smoke": "vp exec node -e \"import('./src/index.ts')\""
   }
 }

package/src/index.integration.test.ts

@@ -433,6 +433,9 @@ async function createHarness(options?: {
     return { agentDir, cwd, session, stateDir };
   }
 
+  // TODO: Remove this legacy fallback once our supported pi compatibility window no
+  // longer includes pi <0.65.0, which introduced the session runtime API and the
+  // session_start-only post-transition model.
   const resourceLoader = new DefaultResourceLoader({
     cwd,
     agentDir,
@@ -447,26 +450,32 @@ async function createHarness(options?: {
     resourceLoader,
     sessionManager,
   });
+  const legacyRuntimeSession = legacySession as typeof legacySession & {
+    newSession(): Promise<boolean>;
+    switchSession(sessionPath: string): Promise<boolean>;
+    fork(entryId: string): Promise<{ cancelled: boolean; selectedText?: string }>;
+    sessionManager: SessionManager;
+  };
 
   const session: TestSessionController = {
-    prompt: (text) => legacySession.prompt(text),
-    newSession: () => legacySession.newSession(),
-    switchSession: (sessionPath) => legacySession.switchSession(sessionPath),
+    prompt: (text) => legacyRuntimeSession.prompt(text),
+    newSession: () => legacyRuntimeSession.newSession(),
+    switchSession: (sessionPath) => legacyRuntimeSession.switchSession(sessionPath),
     fork: async (entryId) => {
-      const result = await legacySession.fork(entryId);
+      const result = await legacyRuntimeSession.fork(entryId);
       return {
         cancelled: result.cancelled,
-        selectedText: result.selectedText,
+        selectedText: result.selectedText ?? "",
       };
     },
     dispose: async () => {
-      legacySession.dispose();
+      legacyRuntimeSession.dispose();
     },
     get sessionFile() {
-      return legacySession.sessionFile;
+      return legacyRuntimeSession.sessionFile;
     },
     get sessionManager() {
-      return legacySession.sessionManager;
+      return legacyRuntimeSession.sessionManager;
     },
   };
 

package/src/index.ts

@@ -622,12 +622,21 @@ export default function braintrustPiExtension(pi: ExtensionAPI): void {
     });
   });
 
-  pi.on("session_switch", async (event, ctx) => {
+  // TODO: Remove these legacy transition listeners once our compatibility window
+  // no longer includes pi <0.65.0.
+  const legacyPi = pi as ExtensionAPI & {
+    on(
+      event: "session_switch" | "session_fork",
+      handler: (event: unknown, ctx: ExtensionContext) => Promise<void> | void,
+    ): void;
+  };
+
+  legacyPi.on("session_switch", async (event, ctx) => {
     refreshTracingUi(ctx);
     await rolloverSession(ctx, "session_switch", getPreviousSessionFile(event));
   });
 
-  pi.on("session_fork", async (event, ctx) => {
+  legacyPi.on("session_fork", async (event, ctx) => {
     refreshTracingUi(ctx);
     await rolloverSession(ctx, "session_fork", getPreviousSessionFile(event));
   });