pi-sessions 0.7.2 → 0.8.0

package/README.md

@@ -190,6 +190,7 @@ To change auto-titling settings, edit `~/.pi/agent/settings.json`:
   "sessions": {
     "autoTitle": {
       "refreshTurns": 4,
+      "timeoutSecs": 15,
       "model": "anthropic/claude-haiku-4-5",
       "prompt": "Custom prompt that overrides the default."
     }

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

@@ -7,7 +7,6 @@ import type { ExtensionContext } from "@earendil-works/pi-coding-agent";
 import type { AutoTitleContext } from "./context.ts";
 import type { AutoTitleTrigger } from "./state.ts";
 
-const AUTO_TITLE_REQUEST_TIMEOUT_MS = 15_000;
 const AUTO_TITLE_MAX_TOKENS = 64;
 const AUTO_TITLE_CHAR_MAX = 80;
 
@@ -34,6 +33,7 @@ export async function generateAutoTitle(
   context: AutoTitleContext,
   trigger: AutoTitleTrigger,
   systemPrompt: string,
+  timeoutMs: number,
 ): Promise<AutoTitleGenerationResult> {
   if (!context.conversationText) {
     return {
@@ -64,7 +64,7 @@ export async function generateAutoTitle(
   };
 
   const abortController = new AbortController();
-  const timeoutId = setTimeout(() => abortController.abort(), AUTO_TITLE_REQUEST_TIMEOUT_MS);
+  const timeoutId = setTimeout(() => abortController.abort(), timeoutMs);
 
   try {
     const requestContext = {

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

@@ -58,6 +58,7 @@ export async function runBulkRetitle(
   mode: RetitleMode,
   getSessionEpoch: () => number,
   systemPrompt: string,
+  timeoutMs: number,
 ): Promise<BulkRetitleResult> {
   const result: BulkRetitleResult = {
     attempted: 0,
@@ -85,6 +86,7 @@ export async function runBulkRetitle(
         model,
         isManual: true,
         systemPrompt,
+        timeoutMs,
         getSessionEpoch,
         notifyOnSuccess: false,
       });
@@ -92,7 +94,7 @@ export async function runBulkRetitle(
       continue;
     }
 
-    const outcome = await retitleStoredSession(ctx, model, session.path, systemPrompt);
+    const outcome = await retitleStoredSession(ctx, model, session.path, systemPrompt, timeoutMs);
     result[outcome] += 1;
   }
 
@@ -165,6 +167,7 @@ export interface RetitlePlanOptions {
   model: Model<Api> | undefined;
   isManual: boolean;
   systemPrompt: string;
+  timeoutMs: number;
   existingPlan?: AutoTitleTriggerPlan;
   getSessionEpoch?: () => number;
   notifyOnSuccess?: boolean;
@@ -173,8 +176,17 @@ export interface RetitlePlanOptions {
 export async function runRetitlePlan(
   options: RetitlePlanOptions,
 ): Promise<AutoTitleGenerationResult> {
-  const { pi, controller, ctx, model, isManual, systemPrompt, existingPlan, getSessionEpoch } =
-    options;
+  const {
+    pi,
+    controller,
+    ctx,
+    model,
+    isManual,
+    systemPrompt,
+    timeoutMs,
+    existingPlan,
+    getSessionEpoch,
+  } = options;
   const notifyOnSuccess = options.notifyOnSuccess ?? isManual;
 
   const plan = existingPlan ?? controller.handleManualRetitle(ctx);
@@ -211,6 +223,7 @@ export async function runRetitlePlan(
     titleContext,
     plan.reason,
     systemPrompt,
+    timeoutMs,
   );
   if (!generatedTitle.ok) {
     return generatedTitle;
@@ -257,6 +270,7 @@ async function retitleStoredSession(
   model: Model<Api>,
   sessionPath: string,
   systemPrompt: string,
+  timeoutMs: number,
 ): Promise<"retitled" | "unchanged" | "failed"> {
   const sessionManager = SessionManager.open(sessionPath);
   const currentTitle = sessionManager.getSessionName();
@@ -279,6 +293,7 @@ async function retitleStoredSession(
     titleContext,
     plan.reason,
     systemPrompt,
+    timeoutMs,
   );
   if (!generatedTitle.ok) {
     return "failed";

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

@@ -57,6 +57,7 @@ export async function showRetitleWizard(
   model: Model<Api> | undefined,
   getSessionEpoch: () => number,
   systemPrompt: string,
+  timeoutMs: number,
   options?: RetitleWizardOptions,
 ): Promise<RetitleCommandOutcome> {
   return ctx.ui.custom<RetitleCommandOutcome>(
@@ -70,6 +71,7 @@ export async function showRetitleWizard(
         model,
         getSessionEpoch,
         systemPrompt,
+        timeoutMs,
         done,
         options,
       ),
@@ -99,6 +101,7 @@ class RetitleWizardPanel implements Focusable {
     private readonly model: Model<Api> | undefined,
     private readonly getSessionEpoch: () => number,
     private readonly systemPrompt: string,
+    private readonly timeoutMs: number,
     private readonly done: (result: RetitleCommandOutcome) => void,
     options?: RetitleWizardOptions,
   ) {
@@ -250,6 +253,7 @@ class RetitleWizardPanel implements Focusable {
       model: this.model,
       isManual: true,
       systemPrompt: this.systemPrompt,
+      timeoutMs: this.timeoutMs,
       getSessionEpoch: this.getSessionEpoch,
     });
     if (result.ok) {
@@ -310,6 +314,7 @@ class RetitleWizardPanel implements Focusable {
       mode,
       this.getSessionEpoch,
       this.systemPrompt,
+      this.timeoutMs,
     );
     notifyBulkRetitleResult(this.ctx, scan, mode, result);
     this.done(

package/extensions/session-auto-title.ts

@@ -75,6 +75,7 @@ export default function sessionAutoTitleExtension(pi: ExtensionAPI): void {
           model,
           invocation,
           settings.autoTitle.prompt,
+          settings.autoTitle.timeoutMs,
         );
       },
     ),
@@ -104,6 +105,7 @@ export default function sessionAutoTitleExtension(pi: ExtensionAPI): void {
       getSessionEpoch: () => sessionEpoch,
       notifyOnSuccess: false,
       systemPrompt: settings.autoTitle.prompt,
+      timeoutMs: settings.autoTitle.timeoutMs,
     })
       .then((outcome) => {
         if (outcome.ok) {
@@ -139,6 +141,7 @@ async function handleTitleInvocation(
   model: Model<Api> | undefined,
   invocation: RetitleCommandInvocation,
   systemPrompt: string,
+  timeoutMs: number,
 ): Promise<RetitleCommandOutcome> {
   const retitleOpts = {
     pi,
@@ -147,6 +150,7 @@ async function handleTitleInvocation(
     model,
     isManual: true,
     systemPrompt,
+    timeoutMs,
     getSessionEpoch: state.getSessionEpoch,
   };
 
@@ -165,7 +169,15 @@ async function handleTitleInvocation(
       return retitleCurrentSession();
     }
 
-    return showRetitleWizard(pi, state.controller, ctx, model, state.getSessionEpoch, systemPrompt);
+    return showRetitleWizard(
+      pi,
+      state.controller,
+      ctx,
+      model,
+      state.getSessionEpoch,
+      systemPrompt,
+      timeoutMs,
+    );
   }
 
   if (invocation.scope === "this") {
@@ -180,6 +192,7 @@ async function handleTitleInvocation(
       model,
       state.getSessionEpoch,
       systemPrompt,
+      timeoutMs,
       {
         initialInvocation: {
           scope: invocation.scope,
@@ -201,6 +214,7 @@ async function handleTitleInvocation(
       mode,
       state.getSessionEpoch,
       systemPrompt,
+      timeoutMs,
     );
     notifyBulkRetitleResult(ctx, scan, mode, result);
     return "success";

package/extensions/shared/settings.ts

@@ -7,6 +7,7 @@ import { type Static, Type } from "typebox";
 import { parseTypeBoxValue } from "./typebox.ts";
 
 export const DEFAULT_AUTO_TITLE_REFRESH_TURNS = 4;
+export const DEFAULT_AUTO_TITLE_TIMEOUT_SECONDS = 15;
 export const DEFAULT_AUTO_TITLE_PROMPT = `Name this coding session (under 80 chars). Be specific to what is being discussed. Your exact output will be displayed to the user, so make sure that it contains ONLY the title itself and nothing else.`;
 const SESSION_FILE_SETTINGS_SCHEMA = Type.Object({
   handoff: Type.Optional(
@@ -22,6 +23,7 @@ const SESSION_FILE_SETTINGS_SCHEMA = Type.Object({
   autoTitle: Type.Optional(
     Type.Object({
       refreshTurns: Type.Optional(Type.Integer({ minimum: 1 })),
+      timeoutSecs: Type.Optional(Type.Integer({ minimum: 1 })),
       model: Type.Optional(Type.String()),
       thinkingLevel: Type.Optional(Type.String()),
       prompt: Type.Optional(Type.String()),
@@ -57,6 +59,7 @@ export interface AgentModelSettings {
 
 export interface AutoTitleSettings extends AgentModelSettings {
   refreshTurns: number;
+  timeoutMs: number;
   prompt: string;
 }
 
@@ -193,6 +196,7 @@ function resolveSessionSettings(fileSettings: SessionFileSettings): SessionSetti
     autoTitle: {
       ...resolveAgentModelSettings(fileSettings.autoTitle),
       refreshTurns: fileSettings.autoTitle?.refreshTurns ?? DEFAULT_AUTO_TITLE_REFRESH_TURNS,
+      timeoutMs: (fileSettings.autoTitle?.timeoutSecs ?? DEFAULT_AUTO_TITLE_TIMEOUT_SECONDS) * 1000,
       prompt: normalizeAutoTitlePrompt(fileSettings.autoTitle?.prompt),
     },
     ask: {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-sessions",
-  "version": "0.7.2",
+  "version": "0.8.0",
   "type": "module",
   "description": "Pi session search, ask, handoff, messaging, auto-titling, and indexing tools",
   "license": "MIT",