alexion
6081e3e325
Imported as a linear vendor commit so forge rebases do not replay the upstream root tree. git-subtree-dir: third_party/pi git-subtree-split: 71023aec1546d6d677a65669c5665c0927ce0dd0
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;
|
|
}
|
|
});
|
|
}
|