pi-sessions 0.2.1 → 0.3.0

package/README.md

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

package/extensions/session-ask.ts

@@ -1,7 +1,7 @@
-import { complete, type Message } from "@mariozechner/pi-ai";
-import type { ExtensionAPI, Theme } from "@mariozechner/pi-coding-agent";
-import { Text } from "@mariozechner/pi-tui";
-import { Type } from "@sinclair/typebox";
+import { complete, type Message } from "@earendil-works/pi-ai";
+import type { ExtensionAPI, Theme } from "@earendil-works/pi-coding-agent";
+import { Text } from "@earendil-works/pi-tui";
+import { Type } from "typebox";
 import { type RenderedSessionTree, renderSessionTreeMarkdown } from "./session-search/extract.js";
 import {
   getIndexStatus,

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

@@ -1,5 +1,5 @@
-import type { ExtensionCommandContext } from "@mariozechner/pi-coding-agent";
-import type { AutocompleteItem } from "@mariozechner/pi-tui";
+import type { ExtensionCommandContext } from "@earendil-works/pi-coding-agent";
+import type { AutocompleteItem } from "@earendil-works/pi-tui";
 
 export const TITLE_USAGE = "Usage: /title [this|folder|pi] [-f]";
 

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

@@ -3,7 +3,7 @@ import {
   convertToLlm,
   type SessionEntry,
   serializeConversation,
-} from "@mariozechner/pi-coding-agent";
+} from "@earendil-works/pi-coding-agent";
 
 export interface AutoTitleContext {
   cwd: string | undefined;

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

@@ -1,5 +1,6 @@
-import { buildSessionContext, type ExtensionContext } from "@mariozechner/pi-coding-agent";
+import { buildSessionContext, type ExtensionContext } from "@earendil-works/pi-coding-agent";
 import type { AutoTitleSettings } from "../shared/settings.js";
+import type { AutoTitleFailure } from "./generate.js";
 import {
   type AutoTitleMode,
   type AutoTitlePersistedState,
@@ -8,20 +9,6 @@ import {
   getLatestAutoTitleState,
 } from "./state.js";
 
-export interface AutoTitleFailure {
-  at: string;
-  trigger: AutoTitleTrigger;
-  model: string;
-  message: string;
-  status?: number;
-}
-
-export function formatAutoTitleFailureSummary(failure: AutoTitleFailure): string {
-  return failure.status === undefined
-    ? failure.message
-    : `HTTP ${failure.status} · ${failure.message}`;
-}
-
 export interface SessionAutoTitleStateSnapshot {
   currentSessionFile: string | undefined;
   mode: AutoTitleMode;
@@ -294,5 +281,5 @@ function countUserTurns(messages: Array<{ role: string }>): number {
 }
 
 function formatFailureKey(failure: AutoTitleFailure): string {
-  return `${failure.model}\n${failure.status ?? ""}\n${failure.message}`;
+  return `${failure.model}\n${failure.message}`;
 }

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

@@ -0,0 +1,256 @@
+import { appendFileSync, mkdirSync } from "node:fs";
+import { homedir } from "node:os";
+import { join } from "node:path";
+import {
+  type Api,
+  completeSimple,
+  type Model,
+  type TextContent,
+  type UserMessage,
+} from "@earendil-works/pi-ai";
+import type { ExtensionContext } from "@earendil-works/pi-coding-agent";
+import type { AutoTitleContext } from "./context.js";
+import type { AutoTitleTrigger } from "./state.js";
+
+const AUTO_TITLE_REQUEST_TIMEOUT_MS = 15_000;
+const AUTO_TITLE_MAX_TOKENS = 64;
+const AUTO_TITLE_CHAR_MAX = 80;
+
+export interface AutoTitleFailure {
+  at: string;
+  trigger: AutoTitleTrigger;
+  model: string;
+  message: string;
+}
+
+export type AutoTitleGenerationResult =
+  | {
+      ok: true;
+      title: string;
+    }
+  | {
+      ok: false;
+      failure: AutoTitleFailure;
+    };
+
+export async function generateAutoTitle(
+  ctx: ExtensionContext,
+  model: Model<Api>,
+  context: AutoTitleContext,
+  trigger: AutoTitleTrigger,
+  systemPrompt: string,
+): Promise<AutoTitleGenerationResult> {
+  if (!context.conversationText) {
+    return {
+      ok: false,
+      failure: createAutoTitleFailure(
+        trigger,
+        model,
+        "No conversation available for auto-title generation.",
+      ),
+    };
+  }
+
+  const auth = await ctx.modelRegistry.getApiKeyAndHeaders(model);
+  if (!auth.ok) {
+    return {
+      ok: false,
+      failure: createAutoTitleFailure(trigger, model, "Failed to authenticate auto-title model."),
+    };
+  }
+
+  const shouldPreserveTitle = trigger === "periodic" && Boolean(context.currentTitle?.trim());
+  const resolvedSystemPrompt = buildAutoTitleSystemPrompt(systemPrompt, shouldPreserveTitle);
+  const userPrompt = buildAutoTitlePrompt(context, resolvedSystemPrompt, shouldPreserveTitle);
+  const message: UserMessage = {
+    role: "user",
+    content: [{ type: "text", text: userPrompt }],
+    timestamp: Date.now(),
+  };
+
+  const abortController = new AbortController();
+  const timeoutId = setTimeout(() => abortController.abort(), AUTO_TITLE_REQUEST_TIMEOUT_MS);
+
+  try {
+    const requestContext = {
+      systemPrompt: resolvedSystemPrompt,
+      messages: [message],
+    };
+    writeAutoTitleDebugRequest(model, trigger, requestContext);
+    const response = await completeSimple(model, requestContext, {
+      ...(auth.apiKey && { apiKey: auth.apiKey }),
+      ...(auth.headers && { headers: auth.headers }),
+      maxTokens: AUTO_TITLE_MAX_TOKENS,
+      signal: abortController.signal,
+    });
+    writeAutoTitleDebugResponse(model, trigger, response);
+
+    if (response.stopReason === "error" || response.stopReason === "aborted") {
+      const fallbackMessage =
+        response.stopReason === "aborted" ? "Request was aborted." : "Provider returned an error.";
+      return {
+        ok: false,
+        failure: createAutoTitleFailure(trigger, model, response.errorMessage || fallbackMessage),
+      };
+    }
+
+    const normalizedTitle = normalizeGeneratedAutoTitle(extractResponseText(response.content));
+    if (!normalizedTitle) {
+      return {
+        ok: false,
+        failure: createAutoTitleFailure(trigger, model, "Model returned an empty title."),
+      };
+    }
+
+    return {
+      ok: true,
+      title: normalizedTitle,
+    };
+  } catch (error) {
+    return {
+      ok: false,
+      failure: createAutoTitleFailure(trigger, model, extractErrorMessage(error)),
+    };
+  } finally {
+    clearTimeout(timeoutId);
+  }
+}
+
+export function createAutoTitleFailure(
+  trigger: AutoTitleTrigger,
+  model: Model<Api> | undefined,
+  message: string,
+): AutoTitleFailure {
+  return {
+    at: new Date().toISOString(),
+    trigger,
+    model: formatModelLabel(model),
+    message,
+  };
+}
+
+function buildAutoTitleSystemPrompt(systemPrompt: string, shouldPreserveTitle: boolean): string {
+  if (!shouldPreserveTitle) {
+    return systemPrompt;
+  }
+
+  return `${systemPrompt}\n\nPreserve the current title unless the conversation has meaningfully shifted.`;
+}
+
+function buildAutoTitlePrompt(
+  context: AutoTitleContext,
+  titleInstructions: string,
+  shouldPreserveTitle: boolean,
+): string {
+  const sections = ["Generate the title from this session context.", "<session_context>"];
+
+  if (context.cwd) {
+    sections.push(`<cwd>${context.cwd}</cwd>`);
+  }
+
+  sections.push(`<counts>\n${formatCounts(context)}\n</counts>`);
+
+  if (shouldPreserveTitle) {
+    sections.push(`<current_title>${context.currentTitle ?? ""}</current_title>`);
+  }
+
+  sections.push(`<conversation>\n${context.conversationText || "(none)"}\n</conversation>`);
+  sections.push("</session_context>");
+  sections.push(`<title_instructions>\n${titleInstructions}\n</title_instructions>`);
+
+  return sections.join("\n\n");
+}
+
+function normalizeGeneratedAutoTitle(value: string): string | undefined {
+  const withoutQuotes = value.trim().replace(/^["'`]+|["'`]+$/g, "");
+  const collapsed = withoutQuotes
+    .replace(/\s+/g, " ")
+    .replace(/[.!?]+$/g, "")
+    .trim();
+  if (!collapsed) {
+    return undefined;
+  }
+
+  const truncated = collapsed.slice(0, AUTO_TITLE_CHAR_MAX).trim();
+  return truncated || undefined;
+}
+
+function writeAutoTitleDebugRequest(
+  model: Model<Api>,
+  trigger: AutoTitleTrigger,
+  request: unknown,
+): void {
+  writeAutoTitleDebugEntry({
+    phase: "request",
+    trigger,
+    model: formatModelLabel(model),
+    request,
+  });
+}
+
+function writeAutoTitleDebugResponse(
+  model: Model<Api>,
+  trigger: AutoTitleTrigger,
+  response: unknown,
+): void {
+  writeAutoTitleDebugEntry({
+    phase: "response",
+    trigger,
+    model: formatModelLabel(model),
+    response,
+  });
+}
+
+function writeAutoTitleDebugEntry(entry: Record<string, unknown>): void {
+  try {
+    const dir = join(homedir(), ".pi", "agent", "pi-sessions");
+    mkdirSync(dir, { recursive: true });
+    appendFileSync(
+      join(dir, "auto-title-debug.jsonl"),
+      `${JSON.stringify({ at: new Date().toISOString(), ...entry })}\n`,
+    );
+  } catch {
+    // Debug logging must not affect title generation.
+  }
+}
+
+function extractResponseText(content: unknown[]): string {
+  return content
+    .filter(
+      (part): part is TextContent =>
+        isObject(part) && part.type === "text" && typeof part.text === "string",
+    )
+    .map((part) => part.text)
+    .join("\n");
+}
+
+function formatCounts(context: AutoTitleContext): string {
+  return [
+    `user_turns: ${context.userTurnCount}`,
+    `assistant_turns: ${context.assistantTurnCount}`,
+  ].join("\n");
+}
+
+function formatModelLabel(model: Model<Api> | undefined): string {
+  if (!model) {
+    return "(no model resolved)";
+  }
+
+  return `${model.provider}/${model.id}`;
+}
+
+function extractErrorMessage(error: unknown): string {
+  if (error instanceof Error && error.message.trim()) {
+    return error.message.trim();
+  }
+
+  if (typeof error === "string" && error.trim()) {
+    return error.trim();
+  }
+
+  return "Unknown provider error.";
+}
+
+function isObject(value: unknown): value is Record<string, unknown> {
+  return typeof value === "object" && value !== null;
+}

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

@@ -1,5 +1,5 @@
-import type { Api, Model } from "@mariozechner/pi-ai";
-import type { ExtensionContext } from "@mariozechner/pi-coding-agent";
+import type { Api, Model } from "@earendil-works/pi-ai";
+import type { ExtensionContext } from "@earendil-works/pi-coding-agent";
 import { ModelReference } from "../shared/settings.js";
 
 const DEFAULT_AUTO_TITLE_FALLBACK_MODELS: readonly ModelReference[] = [