alexion
4def357d09
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: d981de1229ef899957bbe968bc8dcda02a21f477
52 lines
1.3 KiB
TypeScript
52 lines
1.3 KiB
TypeScript
/**
|
|
* Prompt Templates
|
|
*
|
|
* File-based templates that inject content when invoked with /templatename.
|
|
*/
|
|
|
|
import {
|
|
createAgentSession,
|
|
createSyntheticSourceInfo,
|
|
DefaultResourceLoader,
|
|
getAgentDir,
|
|
type PromptTemplate,
|
|
SessionManager,
|
|
} from "@earendil-works/pi-coding-agent";
|
|
|
|
// Define custom templates
|
|
const deployTemplate: PromptTemplate = {
|
|
name: "deploy",
|
|
description: "Deploy the application",
|
|
filePath: "/virtual/prompts/deploy.md",
|
|
sourceInfo: createSyntheticSourceInfo("/virtual/prompts/deploy.md", { source: "sdk" }),
|
|
content: `# Deploy Instructions
|
|
|
|
1. Build: npm run build
|
|
2. Test: npm test
|
|
3. Deploy: npm run deploy`,
|
|
};
|
|
|
|
const loader = new DefaultResourceLoader({
|
|
cwd: process.cwd(),
|
|
agentDir: getAgentDir(),
|
|
promptsOverride: (current) => ({
|
|
prompts: [...current.prompts, deployTemplate],
|
|
diagnostics: current.diagnostics,
|
|
}),
|
|
});
|
|
await loader.reload();
|
|
|
|
// Discover templates from cwd/.pi/prompts/ and ~/.pi/agent/prompts/
|
|
const discovered = loader.getPrompts().prompts;
|
|
console.log("Discovered prompt templates:");
|
|
for (const template of discovered) {
|
|
console.log(` /${template.name}: ${template.description}`);
|
|
}
|
|
|
|
const { session } = await createAgentSession({
|
|
resourceLoader: loader,
|
|
sessionManager: SessionManager.inMemory(),
|
|
});
|
|
console.log(`Session created with ${discovered.length + 1} prompt templates`);
|
|
session.dispose();
|