mirror of
https://github.com/getcompanion-ai/co-mono.git
synced 2026-04-15 17:00:59 +00:00
6f7c10e323
- Add setEditorText() and getEditorText() to HookUIContext for prompt generator pattern - custom() now accepts async factories for fire-and-forget work - Add CancellableLoader component to tui package - Add BorderedLoader component for hooks with cancel UI - Export HookAPI, HookContext, HookFactory from main package - Update all examples to import from packages instead of relative paths - Update hooks.md and custom-tools.md documentation fixes #350
44 lines
1.3 KiB
TypeScript
44 lines
1.3 KiB
TypeScript
/**
|
|
* Custom System Prompt
|
|
*
|
|
* Shows how to replace or modify the default system prompt.
|
|
*/
|
|
|
|
import { createAgentSession, SessionManager } from "@mariozechner/pi-coding-agent";
|
|
|
|
// Option 1: Replace prompt entirely
|
|
const { session: session1 } = await createAgentSession({
|
|
systemPrompt: `You are a helpful assistant that speaks like a pirate.
|
|
Always end responses with "Arrr!"`,
|
|
sessionManager: SessionManager.inMemory(),
|
|
});
|
|
|
|
session1.subscribe((event) => {
|
|
if (event.type === "message_update" && event.assistantMessageEvent.type === "text_delta") {
|
|
process.stdout.write(event.assistantMessageEvent.delta);
|
|
}
|
|
});
|
|
|
|
console.log("=== Replace prompt ===");
|
|
await session1.prompt("What is 2 + 2?");
|
|
console.log("\n");
|
|
|
|
// Option 2: Modify default prompt (receives default, returns modified)
|
|
const { session: session2 } = await createAgentSession({
|
|
systemPrompt: (defaultPrompt) => `${defaultPrompt}
|
|
|
|
## Additional Instructions
|
|
- Always be concise
|
|
- Use bullet points when listing things`,
|
|
sessionManager: SessionManager.inMemory(),
|
|
});
|
|
|
|
session2.subscribe((event) => {
|
|
if (event.type === "message_update" && event.assistantMessageEvent.type === "text_delta") {
|
|
process.stdout.write(event.assistantMessageEvent.delta);
|
|
}
|
|
});
|
|
|
|
console.log("=== Modify prompt ===");
|
|
await session2.prompt("List 3 benefits of TypeScript.");
|
|
console.log();
|