pi-sessions 0.3.1 → 0.3.2

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

@@ -77,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");

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/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.1",
+  "version": "0.3.2",
   "type": "module",
   "description": "Pi session search, ask, handoff, auto-titling, and indexing tools",
   "license": "MIT",
@@ -38,7 +38,7 @@
     "typecheck": "tsc --noEmit",
     "test": "vitest run",
     "check": "biome check . && tsc --noEmit && vitest run",
-    "prepare": "husky"
+    "prepare": "husky || true"
   },
   "devDependencies": {
     "@biomejs/biome": "^2.4.14",
@@ -46,7 +46,6 @@
     "@earendil-works/pi-ai": "^0.78.1",
     "@earendil-works/pi-coding-agent": "^0.78.1",
     "@earendil-works/pi-tui": "^0.78.1",
-    "@types/better-sqlite3": "^7.6.13",
     "@types/node": "^25.6.2",
     "husky": "^9.1.7",
     "lint-staged": "^17.0.3",
@@ -64,7 +63,7 @@
   "lint-staged": {
     "*.{js,cjs,mjs,jsx,ts,tsx,json,jsonc}": "biome check --write --no-errors-on-unmatched"
   },
-  "dependencies": {
+  "optionalDependencies": {
     "better-sqlite3": "^12.9.0"
   }
 }