Armin Ronacher
9e05370b29
This change makes system prompt text and tool changes part of the transcript rather than silently rewriting its starting conditions. This lets Pi record when instructions changed or tools became available, restore that state after resuming or navigating branches, and preserve cached prompt prefixes where the upstream supports it.
50 lines
1.7 KiB
TypeScript
50 lines
1.7 KiB
TypeScript
/**
|
|
* Prompt Customizer Extension
|
|
*
|
|
* Demonstrates using systemPromptOptions to add context-aware prompt sections
|
|
* without replacing or reparsing the complete rendered prompt.
|
|
*
|
|
* Usage:
|
|
* 1. Copy this file to ~/.pi/agent/extensions/ or your project's .pi/extensions/
|
|
* 2. Use the extension — it automatically adapts to your active tools and skills
|
|
*/
|
|
|
|
import type { BuildSystemPromptOptions, ExtensionAPI } from "@earendil-works/pi-coding-agent";
|
|
|
|
function buildToolGuidance(options: BuildSystemPromptOptions): string {
|
|
const hasTool = (name: string) => options.selectedTools?.includes(name) ?? false;
|
|
const rules: string[] = [];
|
|
|
|
if (hasTool("read")) {
|
|
rules.push(
|
|
"- Use `read` for file contents; it supports text and images.",
|
|
"- For large files, use `offset` and `limit` to read in chunks.",
|
|
);
|
|
}
|
|
if (hasTool("bash")) {
|
|
rules.push("- Use `bash` for file operations such as `ls`, `find`, and `grep`.");
|
|
}
|
|
if (hasTool("edit")) {
|
|
rules.push("- Use `edit` for precise text replacements that match existing content exactly.");
|
|
}
|
|
if (hasTool("write")) {
|
|
rules.push("- Use `write` to create new files or replace existing files completely.");
|
|
}
|
|
if (options.skills && options.skills.length > 0) {
|
|
rules.push(`- Available skills: ${options.skills.map((skill) => skill.name).join(", ")}.`);
|
|
}
|
|
|
|
return rules.join("\n");
|
|
}
|
|
|
|
export default function promptCustomizer(pi: ExtensionAPI) {
|
|
pi.on("before_agent_start", (event) => {
|
|
const guidance = buildToolGuidance(event.systemPromptOptions);
|
|
if (guidance) {
|
|
event.systemPromptOptions.sections.tool_guidance = guidance;
|
|
} else {
|
|
delete event.systemPromptOptions.sections.tool_guidance;
|
|
}
|
|
});
|
|
}
|