can1357
2be354322d
- Renamed npm package scope from @mariozechner to @oh-my-pi across all packages for consistent branding. - Removed packages/pods directory and all related pod management functionality. - Updated import paths and module declarations throughout codebase to reflect new package names. - Reformatted code with consistent trailing comma removal and ternary operator alignment.
62 lines
1.6 KiB
TypeScript
62 lines
1.6 KiB
TypeScript
/**
|
|
* Hooks Configuration
|
|
*
|
|
* Hooks intercept agent events for logging, blocking, or modification.
|
|
*/
|
|
|
|
import { createAgentSession, type HookFactory, SessionManager } from "@oh-my-pi/pi-coding-agent";
|
|
|
|
// Logging hook
|
|
const loggingHook: HookFactory = (api) => {
|
|
api.on("agent_start", async () => {
|
|
console.log("[Hook] Agent starting");
|
|
});
|
|
|
|
api.on("tool_call", async (event) => {
|
|
console.log(`[Hook] Tool: ${event.toolName}`);
|
|
return undefined; // Don't block
|
|
});
|
|
|
|
api.on("agent_end", async (event) => {
|
|
console.log(`[Hook] Done, ${event.messages.length} messages`);
|
|
});
|
|
};
|
|
|
|
// Blocking hook (returns { block: true, reason: "..." })
|
|
const safetyHook: HookFactory = (api) => {
|
|
api.on("tool_call", async (event) => {
|
|
if (event.toolName === "bash") {
|
|
const cmd = (event.input as { command?: string }).command ?? "";
|
|
if (cmd.includes("rm -rf")) {
|
|
return { block: true, reason: "Dangerous command blocked" };
|
|
}
|
|
}
|
|
return undefined;
|
|
});
|
|
};
|
|
|
|
// Use inline hooks
|
|
const { session } = await createAgentSession({
|
|
hooks: [{ factory: loggingHook }, { factory: safetyHook }],
|
|
sessionManager: SessionManager.inMemory(),
|
|
});
|
|
|
|
session.subscribe((event) => {
|
|
if (event.type === "message_update" && event.assistantMessageEvent.type === "text_delta") {
|
|
process.stdout.write(event.assistantMessageEvent.delta);
|
|
}
|
|
});
|
|
|
|
await session.prompt("List files in the current directory.");
|
|
console.log();
|
|
|
|
// Disable all hooks:
|
|
// hooks: []
|
|
|
|
// Merge with discovered hooks:
|
|
// const discovered = await discoverHooks();
|
|
// hooks: [...discovered, { factory: myHook }]
|
|
|
|
// Add paths without replacing discovery:
|
|
// additionalHookPaths: ["/extra/hooks"]
|