can1357
41 lines
1.2 KiB
TypeScript
41 lines
1.2 KiB
TypeScript
/**
|
|
* Tiny-model UI labels for spawned subagents.
|
|
*/
|
|
import { logger, prompt } from "@oh-my-pi/pi-utils";
|
|
import type { ModelRegistry } from "../config/model-registry";
|
|
import type { Settings } from "../config/settings";
|
|
import taskLabelSystemPrompt from "../prompts/system/task-label.md" with { type: "text" };
|
|
import { generateSessionTitle } from "../utils/title-generator";
|
|
|
|
const TASK_LABEL_SYSTEM_PROMPT = prompt.render(taskLabelSystemPrompt);
|
|
|
|
/** Compresses a delegated assignment into a one-sentence UI label via the tiny title model — fired by the executor spawn path because the task wire schema no longer carries a `description`; null on empty input or failure. */
|
|
export async function generateTaskLabel(
|
|
assignment: string,
|
|
registry: ModelRegistry,
|
|
settings: Settings,
|
|
sessionId?: string,
|
|
signal?: AbortSignal,
|
|
): Promise<string | null> {
|
|
const text = assignment.trim();
|
|
if (!text) return null;
|
|
try {
|
|
return await generateSessionTitle(
|
|
text,
|
|
registry,
|
|
settings,
|
|
sessionId,
|
|
undefined,
|
|
undefined,
|
|
TASK_LABEL_SYSTEM_PROMPT,
|
|
signal,
|
|
);
|
|
} catch (err) {
|
|
logger.debug("task-label: generation failed", {
|
|
sessionId,
|
|
error: err instanceof Error ? err.message : String(err),
|
|
});
|
|
return null;
|
|
}
|
|
}
|