getcompanion-ai/co-mono
1
0
Fork 0
mirror of https://github.com/getcompanion-ai/co-mono.git synced 2026-04-15 08:03:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Rename immediate -> allowDuringStreaming for hook commands

Browse source
This commit is contained in:
Mario Zechner 2025-12-28 15:00:52 +01:00
parent 454dd919be
commit f8352bb7d7
4 changed files with 15 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

8
.pi/hooks/test-command.ts
View file

@ -29,6 +29,14 @@ export default function (pi: HookAPI) {
return box;
});
pi.registerCommand("no-stream", {
description: "Send a message without streaming",
handler: async (ctx) => {
ctx.ui.notify("Sending message after streaming is done...");
},
allowDuringStreaming: true,
})
// Register /test-msg command
pi.registerCommand("test-msg", {
description: "Send a test custom message",

2
packages/coding-agent/examples/hooks/snake.ts
View file

@ -309,7 +309,7 @@ const SNAKE_SAVE_TYPE = "snake-save";
export default function (pi: HookAPI) {
pi.registerCommand("snake", {
description: "Play Snake!",
immediate: true, // Run immediately, even during streaming
allowDuringStreaming: true, // Run even during streaming, not queued
handler: async (ctx) => {
if (!ctx.hasUI) {
ctx.ui.notify("Snake requires interactive mode", "error");

6
packages/coding-agent/src/core/hooks/types.ts
View file

@ -442,8 +442,8 @@ export interface HookCommandContext {
export interface RegisteredCommand {
name: string;
description?: string;
/** If true, command runs immediately even during streaming (doesn't get queued) */
immediate?: boolean;
/** If true, command runs during streaming instead of being queued */
allowDuringStreaming?: boolean;
handler: (ctx: HookCommandContext) => Promise<void>;
}
@ -529,7 +529,7 @@ export interface HookAPI {
*/
registerCommand(
name: string,
options: { description?: string; immediate?: boolean; handler: RegisteredCommand["handler"] },
options: { description?: string; allowDuringStreaming?: boolean; handler: RegisteredCommand["handler"] },
): void;
/**

6
packages/coding-agent/src/modes/interactive/interactive-mode.ts
View file

@ -744,13 +744,13 @@ export class InteractiveMode {
return;
}
// Check if this is an immediate hook command (runs even during streaming)
// Check if this hook command can run during streaming (not queued)
if (text.startsWith("/") && this.session.hookRunner && this.session.isStreaming) {
const spaceIndex = text.indexOf(" ");
const commandName = spaceIndex === -1 ? text.slice(1) : text.slice(1, spaceIndex);
const command = this.session.hookRunner.getCommand(commandName);
if (command?.immediate) {
// Execute immediate hook command right away
if (command?.allowDuringStreaming) {
// Execute hook command right away
this.editor.addToHistory(text);
this.editor.setText("");
await this.session.prompt(text);