Add session activity dock and auto naming

This commit is contained in:
Federico Jaramillo Martinez
2026-05-08 14:53:06 +02:00
parent f64ea8d15d
commit 1bbad2e89f
9 changed files with 156 additions and 19 deletions
@@ -0,0 +1,66 @@
import { getApiProvider, type Api, type AssistantMessage, type Model } from "@earendil-works/pi-ai";
import type { ModelRegistry } from "@earendil-works/pi-coding-agent";
const SESSION_NAME_TIMEOUT_MS = 10_000;
const SESSION_NAME_MAX_INPUT_CHARS = 4_000;
const SESSION_NAME_MAX_LENGTH = 60;
export async function generateShortSessionName<TApi extends Api>(modelRegistry: ModelRegistry, model: Model<TApi>, firstMessage: string): Promise<string | undefined> {
const provider = getApiProvider(model.api);
if (provider === undefined) return undefined;
const auth = await modelRegistry.getApiKeyAndHeaders(model);
if (!auth.ok) return undefined;
const stream = provider.streamSimple(
model,
{
systemPrompt: "Generate a concise title for a coding-agent chat session. Return only the title, with no quotes or punctuation wrapper.",
messages: [{
role: "user",
content: `Create a 2-6 word title for this request:\n\n${truncateInput(firstMessage)}`,
timestamp: Date.now(),
}],
},
{
temperature: 0.2,
maxTokens: 24,
reasoning: "minimal",
signal: AbortSignal.timeout(SESSION_NAME_TIMEOUT_MS),
...(auth.apiKey === undefined ? {} : { apiKey: auth.apiKey }),
...(auth.headers === undefined ? {} : { headers: auth.headers }),
},
);
let streamedText = "";
let finalMessage: AssistantMessage | undefined;
for await (const event of stream) {
if (event.type === "text_delta") streamedText += event.delta;
if (event.type === "done") finalMessage = event.message;
if (event.type === "error") return undefined;
}
return cleanSessionName(finalMessage === undefined ? streamedText : textFromAssistant(finalMessage));
}
export function cleanSessionName(value: string): string | undefined {
const title = (value.split("\n", 1)[0] ?? "")
.replace(/^\s*["'`]+|["'`.]+\s*$/g, "")
.replace(/^(title|session title)\s*:\s*/i, "")
.replace(/\s+/g, " ")
.trim()
.slice(0, SESSION_NAME_MAX_LENGTH)
.trim();
return title === "" ? undefined : title;
}
function textFromAssistant(message: AssistantMessage): string {
return message.content
.filter((part) => part.type === "text")
.map((part) => part.text)
.join("");
}
function truncateInput(value: string): string {
return value.length <= SESSION_NAME_MAX_INPUT_CHARS ? value : `${value.slice(0, SESSION_NAME_MAX_INPUT_CHARS)}`;
}