@agnishc/edb-auto-name-session 0.4.0

package/CHANGELOG.md

@@ -0,0 +1,8 @@
+# Changelog
+
+## [0.2.0] - 2026-04-29
+
+### Added
+- Initial release: replaces Pi's default first-message session label with a generated session title
+- Uses the `opencode/big-pickle` model to generate a concise session title after the first user message
+- Shows interactive notifications while auto-naming runs and when the session title is set

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Agnish Chakraborty
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,49 @@
+# @agnishc/edb-auto-name-session
+
+A Pi CLI extension that replaces Pi's default first-message session label with a short generated title.
+
+By default, Pi already shows the first user message in the session picker when no explicit session name exists. This extension watches the first real user prompt in a new unnamed session, sends that prompt to `opencode/big-pickle`, and then stores a cleaner display name with `pi.setSessionName()`.
+
+## Install
+
+```bash
+pi install npm:@agnishc/edb-auto-name-session
+```
+
+## Use case
+
+Without this extension:
+- Pi falls back to the raw first user message as the session label
+
+With this extension:
+- Pi keeps the same session flow
+- after the first user message, the fallback label is replaced with a concise generated title
+
+Example:
+
+- First prompt: `The next extension to build is auto-name-session. This will use the opencode big pickle model.`
+- Generated session name: `Build Auto Name Session`
+
+## Behavior
+
+- Runs once per fresh unnamed session
+- Waits until the first user message is actually recorded
+- Uses `opencode/big-pickle` to generate a concise title
+- Shows an interactive notification while auto-naming runs, then confirms the final title
+- Calls `pi.setSessionName()` only if the session is still unnamed
+- Leaves already named, resumed, or forked sessions alone
+
+## Requirements
+
+The extension uses the `opencode` provider and the `big-pickle` model.
+Configure OpenCode access with either:
+
+```bash
+export OPENCODE_API_KEY=...
+```
+
+or `/login` in Pi if you store provider credentials there.
+
+## License
+
+[MIT](LICENSE) © Agnish Chakraborty

package/package.json

@@ -0,0 +1,43 @@
+{
+	"name": "@agnishc/edb-auto-name-session",
+	"version": "0.4.0",
+	"description": "Pi extension: replace Pi's first-message session label with a generated title",
+	"keywords": [
+		"pi-package",
+		"pi-extension",
+		"edb"
+	],
+	"type": "module",
+	"license": "MIT",
+	"author": "Agnish Chakraborty",
+	"repository": {
+		"type": "git",
+		"url": "git+https://github.com/agnishcc/pi-extention-monorepo.git",
+		"directory": "packages/edb-auto-name-session"
+	},
+	"homepage": "https://github.com/agnishcc/pi-extention-monorepo/tree/main/packages/edb-auto-name-session#readme",
+	"bugs": {
+		"url": "https://github.com/agnishcc/pi-extention-monorepo/issues"
+	},
+	"publishConfig": {
+		"access": "public"
+	},
+	"scripts": {
+		"test": "vitest run"
+	},
+	"files": [
+		"src",
+		"README.md",
+		"LICENSE",
+		"CHANGELOG.md"
+	],
+	"pi": {
+		"extensions": [
+			"./src/index.ts"
+		]
+	},
+	"peerDependencies": {
+		"@mariozechner/pi-ai": "*",
+		"@mariozechner/pi-coding-agent": "*"
+	}
+}

package/src/index.ts

@@ -0,0 +1,102 @@
+import { complete } from "@mariozechner/pi-ai";
+import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
+import { extractUserText, sanitizeSessionName, shouldArmAutoNaming } from "./title";
+
+const MODEL_PROVIDER = "opencode";
+const MODEL_ID = "big-pickle";
+
+const SYSTEM_PROMPT = `You create short session titles for coding and technical work.
+Return exactly one title based only on the user's first message.
+Rules:
+- Prefer 2 to 6 words
+- Use Title Case
+- Mention the task, feature, bug, or file focus when clear
+- No quotes
+- No markdown
+- No labels like Title:
+- No trailing punctuation
+- Maximum 60 characters`;
+
+export default function autoNameSessionExtension(pi: ExtensionAPI): void {
+	let sessionToken = 0;
+	let armed = false;
+	let pending = false;
+
+	pi.on("session_start", async (_event, ctx) => {
+		sessionToken += 1;
+		armed = shouldArmAutoNaming(ctx.sessionManager.getBranch(), pi.getSessionName());
+		pending = false;
+	});
+
+	pi.on("session_shutdown", async () => {
+		sessionToken += 1;
+		armed = false;
+		pending = false;
+	});
+
+	pi.on("message_end", async (event, ctx) => {
+		if (!armed || pending || pi.getSessionName()) return;
+		if (event.message.role !== "user") return;
+
+		const prompt = extractUserText(event.message.content);
+		armed = false;
+		if (!prompt) return;
+
+		pending = true;
+		const token = sessionToken;
+
+		if (ctx.hasUI) {
+			ctx.ui.notify("Auto-naming session…", "info");
+		}
+
+		try {
+			const name = await generateSessionName(prompt, ctx);
+			if (!name) return;
+			if (token !== sessionToken) return;
+			if (pi.getSessionName()) return;
+			pi.setSessionName(name);
+			if (ctx.hasUI) {
+				ctx.ui.notify(`Session named: ${name}`, "info");
+			}
+		} catch (error: unknown) {
+			console.error("[edb-auto-name-session] Failed to generate session name:", error);
+		} finally {
+			if (token === sessionToken) pending = false;
+		}
+	});
+}
+
+async function generateSessionName(prompt: string, ctx: ExtensionContext): Promise<string | undefined> {
+	const model = ctx.modelRegistry.find(MODEL_PROVIDER, MODEL_ID);
+	if (!model) return undefined;
+
+	const auth = await ctx.modelRegistry.getApiKeyAndHeaders(model);
+	if (!auth.ok || !auth.apiKey) return undefined;
+
+	const response = await complete(
+		model,
+		{
+			systemPrompt: SYSTEM_PROMPT,
+			messages: [
+				{
+					role: "user",
+					content: [{ type: "text", text: prompt }],
+					timestamp: Date.now(),
+				},
+			],
+		},
+		{
+			apiKey: auth.apiKey,
+			headers: auth.headers,
+			maxTokens: 256,
+		},
+	);
+
+	const text = response.content
+		.filter((part): part is { type: "text"; text: string } => part.type === "text")
+		.map((part) => part.text)
+		.join("\n")
+		.trim();
+
+	return sanitizeSessionName(text);
+}

package/src/title.test.ts

@@ -0,0 +1,33 @@
+import { describe, expect, it } from "vitest";
+import { countUserMessages, extractUserText, sanitizeSessionName, shouldArmAutoNaming } from "./title";
+
+describe("title helpers", () => {
+	it("extracts only text parts from user content", () => {
+		expect(
+			extractUserText([
+				{ type: "text", text: "Build an auto name session extension" },
+				{ type: "image", source: { type: "base64", data: "abc" } },
+				{ type: "text", text: "Use big pickle" },
+			]),
+		).toBe("Build an auto name session extension\nUse big pickle");
+	});
+
+	it("sanitizes labels, quotes, and punctuation", () => {
+		expect(sanitizeSessionName('Title: "Auto Name Session".')).toBe("Auto Name Session");
+	});
+
+	it("counts only user messages", () => {
+		const entries = [
+			{ type: "message", message: { role: "user", content: [{ type: "text", text: "one" }] } },
+			{ type: "message", message: { role: "assistant", content: [{ type: "text", text: "two" }] } },
+			{ type: "custom", customType: "x", data: {} },
+		] as any;
+
+		expect(countUserMessages(entries)).toBe(1);
+	});
+
+	it("arms only for empty unnamed sessions", () => {
+		expect(shouldArmAutoNaming([], undefined)).toBe(true);
+		expect(shouldArmAutoNaming([], "Already Named")).toBe(false);
+	});
+});

package/src/title.ts

@@ -0,0 +1,76 @@
+import type { SessionEntry } from "@mariozechner/pi-coding-agent";
+
+const MAX_PROMPT_CHARS = 4_000;
+const MAX_TITLE_CHARS = 60;
+
+type TextPart = {
+	type?: string;
+	text?: string;
+};
+
+export function shouldArmAutoNaming(entries: SessionEntry[], currentName: string | undefined): boolean {
+	return !currentName?.trim() && countUserMessages(entries) === 0;
+}
+
+export function countUserMessages(entries: SessionEntry[]): number {
+	return entries.filter(isUserMessageEntry).length;
+}
+
+export function extractUserText(content: unknown): string {
+	if (typeof content === "string") return normalizePrompt(content);
+	if (!Array.isArray(content)) return "";
+
+	return normalizePrompt(
+		content
+			.filter(isTextPart)
+			.map((part) => part.text)
+			.join("\n"),
+	);
+}
+
+export function sanitizeSessionName(value: string): string | undefined {
+	const firstLine = value
+		.replace(/^```[a-z0-9_-]*\s*/i, "")
+		.replace(/```$/g, "")
+		.split(/\r?\n/)
+		.map((line) => line.trim())
+		.find(Boolean);
+
+	if (!firstLine) return undefined;
+
+	let title = firstLine
+		.replace(/^(title|session name)\s*:\s*/i, "")
+		.replace(/^[-*]\s*/, "")
+		.replace(/[.?!:;,]+$/g, "")
+		.replace(/^['"`]+|['"`]+$/g, "")
+		.replace(/\s+/g, " ")
+		.trim();
+
+	if (!title) return undefined;
+	if (title.length <= MAX_TITLE_CHARS) return title;
+
+	title = title.slice(0, MAX_TITLE_CHARS).trimEnd();
+	const lastSpace = title.lastIndexOf(" ");
+	if (lastSpace > 20) title = title.slice(0, lastSpace);
+	return title.trim() || undefined;
+}
+
+function isUserMessageEntry(entry: SessionEntry): entry is SessionEntry & {
+	type: "message";
+	message: { role: "user"; content: unknown };
+} {
+	return entry.type === "message" && entry.message.role === "user";
+}
+
+function isTextPart(value: unknown): value is TextPart & { type: "text"; text: string } {
+	return Boolean(
+		value &&
+			typeof value === "object" &&
+			(value as TextPart).type === "text" &&
+			typeof (value as TextPart).text === "string",
+	);
+}
+
+function normalizePrompt(value: string): string {
+	return value.replace(/\r\n/g, "\n").trim().slice(0, MAX_PROMPT_CHARS);
+}