@braintrust/pi-extension 0.5.0 → 0.5.1

package/README.md

@@ -2,7 +2,7 @@
 
 [![npm version](https://img.shields.io/npm/v/%40braintrust%2Fpi-extension)](https://www.npmjs.com/package/@braintrust/pi-extension)
 
-Braintrust extension for [pi](https://github.com/mariozechner/pi-coding-agent).
+Braintrust extension for [pi](https://github.com/earendil-works/pi-coding-agent).
 
 Today this extension automatically traces pi sessions, turns, model calls, and tool executions to Braintrust.
 
@@ -47,7 +47,7 @@ pi -e .
 
 ## Compatibility
 
-This package supports the **latest patch release from each of the last six stable pi minor versions**, currently excluding pi versions before `0.65.0`.
+This package supports the **latest patch release from each of the last five stable pi minor versions**, currently excluding pi versions before `0.65.0`.
 
 Our GitHub Actions compatibility job automatically resolves and tests that compatibility window, so new pi releases are picked up without manually updating the matrix.
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@braintrust/pi-extension",
-  "version": "0.5.0",
+  "version": "0.5.1",
   "description": "Braintrust extension for pi. Includes automatic tracing for pi sessions, turns, LLM calls, and tool executions to Braintrust.",
   "keywords": [
     "braintrust",
@@ -28,15 +28,15 @@
     "valibot": "^1.3.1"
   },
   "devDependencies": {
-    "@mariozechner/pi-ai": "^0.71.1",
-    "@mariozechner/pi-coding-agent": "^0.71.1",
+    "@earendil-works/pi-ai": "^0.74.0",
+    "@earendil-works/pi-coding-agent": "^0.74.0",
     "@types/node": "^25.6.0",
     "typescript": "^6.0.2",
     "vite-plus": "^0.1.16",
     "vitest": "^4.1.4"
   },
   "peerDependencies": {
-    "@mariozechner/pi-coding-agent": "*"
+    "@earendil-works/pi-coding-agent": "*"
   },
   "devEngines": {
     "packageManager": {
@@ -57,10 +57,11 @@
   },
   "scripts": {
     "preinstall": "node -e \"const userAgent = process.env.npm_config_user_agent || ''; if (process.env.INIT_CWD === process.cwd() && !userAgent.includes('pnpm/')) { console.error('Use pnpm in this repo.'); process.exit(1); }\"",
-    "check": "vp check",
+    "check": "node scripts/sync-version.mjs --check && vp check",
     "fmt": "vp fmt",
     "lint": "vp lint",
     "pack": "vp pack",
+    "sync:version": "node scripts/sync-version.mjs",
     "test": "vitest run",
     "test:integration": "vitest run src/index.integration.test.ts",
     "test:watch": "vitest",

package/src/config.test.ts

@@ -324,9 +324,8 @@ describe("loadConfig", () => {
 });
 
 describe("createLogger", () => {
-  it("writes json log lines to the default log file when debug is enabled", async () => {
-    const stateDir = makeTempDir("pi-extension-state-");
-    const config: TraceConfig = {
+  function makeLoggerConfig(overrides: Partial<TraceConfig> = {}): TraceConfig {
+    return {
       enabled: true,
       apiKey: "key",
       apiUrl: undefined,
@@ -335,14 +334,19 @@ describe("createLogger", () => {
       projectName: "pi",
       debug: true,
       logFile: undefined,
-      stateDir,
+      stateDir: makeTempDir("pi-extension-state-"),
       additionalMetadata: {},
       parentSpanId: undefined,
       rootSpanId: undefined,
       showUi: true,
       showTraceLink: true,
       configIssues: [],
+      ...overrides,
     };
+  }
+
+  it("writes json log lines to the default log file when debug is enabled", async () => {
+    const config = makeLoggerConfig();
 
     const logger = createLogger(config);
     logger.debug("debug message", { nested: { value: 1 } });
@@ -361,4 +365,27 @@ describe("createLogger", () => {
       message: "warn message",
     });
   });
+
+  it("writes warnings and errors to the default log file when debug is disabled", async () => {
+    const config = makeLoggerConfig({ debug: false, logFile: undefined });
+    const logger = createLogger(config);
+
+    logger.debug("debug message");
+    logger.info("info message");
+    logger.warn("warn message", { reason: "test" });
+    logger.error("error message");
+    await logger.flush();
+
+    const lines = readFileSync(logger.filePath, "utf8").trim().split("\n");
+    expect(lines).toHaveLength(2);
+    expect(JSON.parse(lines[0])).toMatchObject({
+      level: "warn",
+      message: "warn message",
+      data: { reason: "test" },
+    });
+    expect(JSON.parse(lines[1])).toMatchObject({
+      level: "error",
+      message: "error message",
+    });
+  });
 });

package/src/config.ts

@@ -450,14 +450,25 @@ export function createLogger(config: TraceConfig): Logger {
     config.logFile && config.logFile !== "true" && config.logFile !== "auto"
       ? config.logFile
       : join(config.stateDir, "braintrust-pi-extension.log");
-  const loggingEnabled = config.debug || Boolean(config.logFile);
+  const infoLoggingEnabled = config.debug || Boolean(config.logFile);
 
-  if (loggingEnabled) ensureDir(dirname(explicitLogFile));
+  let logDirEnsured = false;
+
+  function shouldLog(level: LogLevel): boolean {
+    return level === "warn" || level === "error" || infoLoggingEnabled;
+  }
+
+  function ensureLogDir(): void {
+    if (logDirEnsured) return;
+    ensureDir(dirname(explicitLogFile));
+    logDirEnsured = true;
+  }
 
   let pendingWrite = Promise.resolve();
 
   function emit(level: LogLevel, message: string, data?: unknown): void {
-    if (!loggingEnabled) return;
+    if (!shouldLog(level)) return;
+    ensureLogDir();
     pendingWrite = pendingWrite
       .catch(() => {})
       .then(async () => {

package/src/index.integration.test.ts

@@ -11,14 +11,14 @@ import {
   type Model,
   type SimpleStreamOptions,
   type ToolCall,
-} from "@mariozechner/pi-ai";
-import * as piCodingAgent from "@mariozechner/pi-coding-agent";
+} from "@earendil-works/pi-ai";
+import * as piCodingAgent from "@earendil-works/pi-coding-agent";
 import {
   createAgentSession,
   DefaultResourceLoader,
   SessionManager,
   type ExtensionAPI,
-} from "@mariozechner/pi-coding-agent";
+} from "@earendil-works/pi-coding-agent";
 import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
 import braintrustPiExtension from "./index.ts";
 

package/src/index.test.ts

@@ -1,5 +1,9 @@
+import { readFileSync } from "node:fs";
 import { beforeEach, describe, expect, it, vi } from "vitest";
 
+const packageVersion = (JSON.parse(readFileSync("package.json", "utf8")) as { version: string })
+  .version;
+
 const mockState = vi.hoisted(() => ({
   startSpans: [] as Array<Record<string, unknown>>,
   logSpans: [] as Array<Record<string, unknown>>,
@@ -193,6 +197,7 @@ describe("braintrustPiExtension", () => {
     expect(mockState.widgets.at(-1)?.content?.[1]).toBe(
       "braintrust.dev/app/test-org/p/pi/logs?oid=trace-row-1",
     );
+    expect(mockState.startSpans[0]?.metadata).toMatchObject({ extension_version: packageVersion });
 
     await emit("session_shutdown");
 

package/src/index.ts

@@ -1,6 +1,10 @@
-import { basename, resolve } from "node:path";
 import { hostname, userInfo } from "node:os";
-import type { AgentEndEvent, ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
+import { basename, resolve } from "node:path";
+import type {
+  AgentEndEvent,
+  ExtensionAPI,
+  ExtensionContext,
+} from "@earendil-works/pi-coding-agent";
 import { BraintrustClient, type BraintrustSpanHandle } from "./client.ts";
 import { createLogger, loadConfig } from "./config.ts";
 import { createStateStore } from "./state.ts";
@@ -13,6 +17,7 @@ import type {
   NormalizedAssistantMessage,
   TraceConfig,
 } from "./types.ts";
+import { EXTENSION_VERSION } from "./version.ts";
 import {
   buildTurnInput,
   extractErrorText,
@@ -29,7 +34,6 @@ import {
   toUnixSeconds,
 } from "./utils.ts";
 
-const EXTENSION_VERSION = "0.1.0";
 const TRACING_STATUS_KEY = "braintrust-tracing";
 const TRACING_WIDGET_KEY = "braintrust-trace-link";
 

package/src/version.ts

@@ -0,0 +1,2 @@
+// This file is generated by scripts/sync-version.mjs. Do not edit by hand.
+export const EXTENSION_VERSION = "0.5.1";