pi-sessions 0.3.0 → 0.3.2

package/README.md

@@ -20,6 +20,8 @@
 
 ## Install
 
+Requires Pi `0.78.1` or newer.
+
 **From npm** (recommended):
 
 ```bash

package/extensions/session-auto-title/command.ts

@@ -1,5 +1,6 @@
 import type { ExtensionCommandContext } from "@earendil-works/pi-coding-agent";
 import type { AutocompleteItem } from "@earendil-works/pi-tui";
+import { isTuiMode } from "../shared/pi-mode.js";
 
 export const TITLE_USAGE = "Usage: /title [this|folder|pi] [-f]";
 
@@ -51,7 +52,7 @@ export function createSessionAutoTitleCommandHandler(
   ) => Promise<RetitleCommandOutcome>,
 ): (args: string, ctx: ExtensionCommandContext) => Promise<void> {
   return async (args: string, ctx: ExtensionCommandContext): Promise<void> => {
-    const parsed = parseRetitleCommand(args, ctx.hasUI);
+    const parsed = parseRetitleCommand(args, isTuiMode(ctx));
     if (parsed.kind === "error") {
       ctx.ui.notify(parsed.message, "error");
       return;
@@ -76,10 +77,10 @@ export function getRetitleArgumentCompletions(argumentPrefix: string): Autocompl
   return filtered.length > 0 ? filtered : null;
 }
 
-export function parseRetitleCommand(args: string, hasUI: boolean): RetitleCommandParseResult {
+export function parseRetitleCommand(args: string, hasTui: boolean): RetitleCommandParseResult {
   const trimmedArgs = args.trim();
   if (!trimmedArgs) {
-    return hasUI ? { kind: "open-pane" } : { kind: "run", scope: "this" };
+    return hasTui ? { kind: "open-pane" } : { kind: "run", scope: "this" };
   }
 
   const tokens = trimmedArgs.split(/\s+/);

package/extensions/session-auto-title.ts

@@ -25,6 +25,7 @@ import {
   runRetitlePlan,
 } from "./session-auto-title/retitle.js";
 import { showRetitleWizard } from "./session-auto-title/wizard.js";
+import { isTuiMode } from "./shared/pi-mode.js";
 import { loadSettings } from "./shared/settings.js";
 
 export {
@@ -160,7 +161,7 @@ async function handleTitleInvocation(
   };
 
   if (invocation.kind === "open-pane") {
-    if (!ctx.hasUI) {
+    if (!isTuiMode(ctx)) {
       return retitleCurrentSession();
     }
 
@@ -171,7 +172,7 @@ async function handleTitleInvocation(
     return runWithInFlightTracking(state, retitleCurrentSession);
   }
 
-  if (ctx.hasUI && !invocation.force) {
+  if (isTuiMode(ctx) && !invocation.force) {
     return showRetitleWizard(
       pi,
       state.controller,

package/extensions/session-handoff/extract.ts

@@ -1,7 +1,9 @@
+import type { ThinkingLevel } from "@earendil-works/pi-agent-core";
 import {
   type AssistantMessage,
-  complete,
+  completeSimple,
   type Message,
+  type SimpleStreamOptions,
   type TextContent,
   type ThinkingContent,
   type Tool,
@@ -86,6 +88,7 @@ export interface HandoffDraftResult {
 export async function generateHandoffDraft(
   ctx: ExtensionContext,
   goal: string,
+  thinkingLevel: ThinkingLevel | undefined,
   signal?: AbortSignal,
 ): Promise<HandoffDraftResult | undefined> {
   if (!ctx.model) {
@@ -117,25 +120,23 @@ export async function generateHandoffDraft(
     timestamp: Date.now(),
   };
 
-  const response = await complete(
+  const requestOptions: SimpleStreamOptions = {
+    apiKey: auth.apiKey,
+    ...(auth.headers ? { headers: auth.headers } : {}),
+    ...(signal ? { signal } : {}),
+    ...(ctx.model.reasoning && thinkingLevel && thinkingLevel !== "off"
+      ? { reasoning: thinkingLevel }
+      : {}),
+  };
+
+  const response = await completeSimple(
     ctx.model,
     {
       systemPrompt: HANDOFF_SYSTEM_PROMPT,
       messages: [userMessage],
       tools: [HANDOFF_EXTRACTION_TOOL],
     },
-    signal
-      ? {
-          apiKey: auth.apiKey,
-          ...(auth.headers ? { headers: auth.headers } : {}),
-          signal,
-          toolChoice: "any",
-        }
-      : {
-          apiKey: auth.apiKey,
-          ...(auth.headers ? { headers: auth.headers } : {}),
-          toolChoice: "any",
-        },
+    requestOptions,
   );
 
   if (response.stopReason === "aborted") {

package/extensions/session-handoff.ts

@@ -23,6 +23,7 @@ import {
   launchSplitHandoffSession,
   validateSplitHandoffPrerequisites,
 } from "./session-handoff/spawn.js";
+import { isTuiMode } from "./shared/pi-mode.js";
 import { loadSettings } from "./shared/settings.js";
 
 const HANDOFF_USAGE = "Usage: /handoff [--left|--right|--up|--down] <goal for new thread>";
@@ -38,7 +39,7 @@ export default function sessionHandoffExtension(pi: ExtensionAPI): void {
   pi.registerCommand("handoff", {
     description: "Transfer context to a new focused session",
     handler: async (args: string, ctx: ExtensionCommandContext): Promise<void> => {
-      if (!ctx.hasUI) {
+      if (!isTuiMode(ctx)) {
         ctx.ui.notify("handoff requires interactive mode", "error");
         return;
       }
@@ -76,7 +77,8 @@ export default function sessionHandoffExtension(pi: ExtensionAPI): void {
         generatedDraft = await runWithLoader(
           ctx,
           "Generating handoff draft...",
-          async (signal: AbortSignal) => generateHandoffDraft(ctx, parsedArgs.goal, signal),
+          async (signal: AbortSignal) =>
+            generateHandoffDraft(ctx, parsedArgs.goal, pi.getThinkingLevel(), signal),
         );
       } catch (error) {
         ctx.ui.notify(formatHandoffError(error), "error");
@@ -157,7 +159,7 @@ export default function sessionHandoffExtension(pi: ExtensionAPI): void {
   pi.registerShortcut(settings.handoff.pickerShortcut, {
     description: "Open the session reference picker",
     handler: async (ctx) => {
-      if (!ctx.hasUI) {
+      if (!isTuiMode(ctx)) {
         return;
       }
 

package/extensions/session-index.ts

@@ -6,6 +6,7 @@ import {
 } from "@earendil-works/pi-coding-agent";
 import { type Focusable, matchesKey, visibleWidth } from "@earendil-works/pi-tui";
 import { type ReindexResult, rebuildSessionIndex } from "./session-search/reindex.js";
+import { isTuiMode } from "./shared/pi-mode.js";
 import { getIndexStatus, type SessionIndexStatus } from "./shared/session-index/index.js";
 import { loadSettings } from "./shared/settings.js";
 
@@ -17,7 +18,7 @@ export default function sessionIndexExtension(pi: ExtensionAPI): void {
   pi.registerCommand("session-index", {
     description: "Open the session index control panel",
     handler: async (_args, ctx) => {
-      if (!ctx.hasUI) {
+      if (!isTuiMode(ctx)) {
         ctx.ui.notify("/session-index requires interactive mode.", "warning");
         return;
       }

package/extensions/session-search/reindex.ts

@@ -1,4 +1,4 @@
-import { renameSync } from "node:fs";
+import { renameSync, rmSync } from "node:fs";
 import { SessionManager } from "@earendil-works/pi-coding-agent";
 import {
   createTempIndexPath,
@@ -34,11 +34,16 @@ export async function rebuildSessionIndex(options: ReindexOptions): Promise<Rein
   try {
     initializeSchema(db);
     ({ sessionCount, chunkCount } = indexSessionFiles(db, sessionFiles));
+    // Fold the WAL into the main file so the single-file rename below is
+    // self-contained. bun:sqlite does not checkpoint on close.
+    db.exec("PRAGMA wal_checkpoint(TRUNCATE)");
   } finally {
     db.close();
   }
 
   renameSync(tempIndexPath, finalIndexPath);
+  rmSync(`${tempIndexPath}-wal`, { force: true });
+  rmSync(`${tempIndexPath}-shm`, { force: true });
   return { sessionCount, chunkCount, indexPath: finalIndexPath };
 }
 

package/extensions/shared/pi-mode.ts

@@ -0,0 +1,5 @@
+import type { ExtensionContext } from "@earendil-works/pi-coding-agent";
+
+export function isTuiMode(ctx: Pick<ExtensionContext, "mode">): boolean {
+  return ctx.mode === "tui";
+}

package/extensions/shared/session-index/common.ts

@@ -1,7 +1,7 @@
-import type Database from "better-sqlite3";
 import { Type } from "typebox";
 import type { FileTouchOp, FileTouchSource, PathScope } from "../../session-search/normalize.js";
 import { safeParseTypeBoxJson } from "../typebox.js";
+import type { SqliteDatabase } from "./sqlite.js";
 
 export const INDEX_SCHEMA_VERSION = 7;
 
@@ -119,7 +119,7 @@ export interface SessionIndexStatus {
   lastFullReindexAt?: string | undefined;
 }
 
-export type SessionIndexDatabase = Database.Database;
+export type SessionIndexDatabase = SqliteDatabase;
 
 export const NULLABLE_STRING_SCHEMA = Type.Union([Type.String(), Type.Null()]);
 export const SESSION_ORIGIN_SCHEMA = Type.Union([

package/extensions/shared/session-index/lineage.ts

@@ -113,7 +113,7 @@ export function getSessionById(
     )
     .get(sessionId);
 
-  if (row === undefined) {
+  if (row === undefined || row === null) {
     return undefined;
   }
 
@@ -140,7 +140,7 @@ export function getSessionByPath(
     )
     .get(sessionPath);
 
-  if (row === undefined) {
+  if (row === undefined || row === null) {
     return undefined;
   }
 

package/extensions/shared/session-index/schema.ts

@@ -1,6 +1,5 @@
 import { existsSync, mkdirSync } from "node:fs";
 import path from "node:path";
-import Database from "better-sqlite3";
 import { parseTypeBoxValue } from "../typebox.js";
 import {
   INDEX_SCHEMA_VERSION,
@@ -9,6 +8,7 @@ import {
   type SessionIndexDatabase,
   type SessionIndexStatus,
 } from "./common.js";
+import { openSqlite } from "./sqlite.js";
 
 export function ensureIndexDir(dir: string): string {
   if (!existsSync(dir)) {
@@ -29,16 +29,7 @@ export function openIndexDatabase(
   options?: { create?: boolean; timeoutMs?: number },
 ): SessionIndexDatabase {
   const create = options?.create ?? true;
-  const db = new Database(
-    dbPath,
-    options?.timeoutMs === undefined
-      ? { fileMustExist: !create }
-      : { fileMustExist: !create, timeout: options.timeoutMs },
-  );
-  db.pragma("journal_mode = WAL");
-  db.pragma("synchronous = NORMAL");
-  db.pragma("foreign_keys = ON");
-  return db;
+  return openSqlite(dbPath, { create, timeoutMs: options?.timeoutMs });
 }
 
 export function initializeSchema(db: SessionIndexDatabase): void {
@@ -148,7 +139,7 @@ export function setMetadata(db: SessionIndexDatabase, key: string, value: string
 
 export function getMetadata(db: SessionIndexDatabase, key: string): string | undefined {
   const row = db.prepare(`SELECT value FROM metadata WHERE key = ?`).get(key);
-  if (row === undefined) {
+  if (row === undefined || row === null) {
     return undefined;
   }
 

package/extensions/shared/session-index/sqlite.ts

@@ -0,0 +1,65 @@
+import { createRequire } from "node:module";
+
+const requireModule = createRequire(import.meta.url);
+const isBun = Boolean(process.versions.bun);
+
+export type SqliteBindValue = string | number | bigint | null;
+
+export interface SqliteRunResult {
+  changes: number;
+  lastInsertRowid: number | bigint;
+}
+
+export interface SqliteStatement {
+  get(...params: SqliteBindValue[]): unknown;
+  all(...params: SqliteBindValue[]): unknown[];
+  run(...params: SqliteBindValue[]): SqliteRunResult;
+}
+
+export interface SqliteDatabase {
+  readonly inTransaction: boolean;
+  prepare(sql: string): SqliteStatement;
+  exec(sql: string): void;
+  transaction<Args extends unknown[], Result>(
+    fn: (...args: Args) => Result,
+  ): (...args: Args) => Result;
+  close(): void;
+}
+
+interface SqliteConstructor {
+  new (path: string, options?: Record<string, unknown>): SqliteDatabase;
+}
+
+// The driver is resolved at runtime so Bun never loads the native better-sqlite3
+// addon (unsupported) and Node never loads the bun:sqlite builtin. The require
+// result is untyped, so the cast to our structural contract is unavoidable here.
+function loadDatabaseConstructor(): SqliteConstructor {
+  if (isBun) {
+    return (requireModule("bun:sqlite") as { Database: SqliteConstructor }).Database;
+  }
+  return requireModule("better-sqlite3") as SqliteConstructor;
+}
+
+export function openSqlite(
+  dbPath: string,
+  options: { create: boolean; timeoutMs?: number | undefined },
+): SqliteDatabase {
+  const Database = loadDatabaseConstructor();
+  const db = isBun
+    ? new Database(dbPath, { create: options.create, readwrite: true })
+    : new Database(
+        dbPath,
+        options.timeoutMs === undefined
+          ? { fileMustExist: !options.create }
+          : { fileMustExist: !options.create, timeout: options.timeoutMs },
+      );
+
+  db.exec("PRAGMA journal_mode = WAL");
+  db.exec("PRAGMA synchronous = NORMAL");
+  db.exec("PRAGMA foreign_keys = ON");
+  if (isBun && options.timeoutMs !== undefined) {
+    db.exec(`PRAGMA busy_timeout = ${options.timeoutMs}`);
+  }
+
+  return db;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-sessions",
-  "version": "0.3.0",
+  "version": "0.3.2",
   "type": "module",
   "description": "Pi session search, ask, handoff, auto-titling, and indexing tools",
   "license": "MIT",
@@ -38,15 +38,14 @@
     "typecheck": "tsc --noEmit",
     "test": "vitest run",
     "check": "biome check . && tsc --noEmit && vitest run",
-    "prepare": "husky"
+    "prepare": "husky || true"
   },
   "devDependencies": {
     "@biomejs/biome": "^2.4.14",
-    "@earendil-works/pi-agent-core": "^0.78.0",
-    "@earendil-works/pi-ai": "^0.78.0",
-    "@earendil-works/pi-coding-agent": "^0.78.0",
-    "@earendil-works/pi-tui": "^0.78.0",
-    "@types/better-sqlite3": "^7.6.13",
+    "@earendil-works/pi-agent-core": "^0.78.1",
+    "@earendil-works/pi-ai": "^0.78.1",
+    "@earendil-works/pi-coding-agent": "^0.78.1",
+    "@earendil-works/pi-tui": "^0.78.1",
     "@types/node": "^25.6.2",
     "husky": "^9.1.7",
     "lint-staged": "^17.0.3",
@@ -55,16 +54,16 @@
     "vitest": "^4.1.5"
   },
   "peerDependencies": {
-    "@earendil-works/pi-agent-core": ">=0.78.0",
-    "@earendil-works/pi-ai": ">=0.78.0",
-    "@earendil-works/pi-coding-agent": ">=0.78.0",
-    "@earendil-works/pi-tui": ">=0.78.0",
+    "@earendil-works/pi-agent-core": ">=0.78.1",
+    "@earendil-works/pi-ai": ">=0.78.1",
+    "@earendil-works/pi-coding-agent": ">=0.78.1",
+    "@earendil-works/pi-tui": ">=0.78.1",
     "typebox": ">=1.1.24"
   },
   "lint-staged": {
     "*.{js,cjs,mjs,jsx,ts,tsx,json,jsonc}": "biome check --write --no-errors-on-unmatched"
   },
-  "dependencies": {
+  "optionalDependencies": {
     "better-sqlite3": "^12.9.0"
   }
 }