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

Add immediate flag to hook commands for non-queued execution

Browse source 
Commands with immediate: true run right away even during streaming.
Used for UI-only commands like /snake that don't interact with LLM.
This commit is contained in:
Mario Zechner 2025-12-27 03:22:55 +01:00
parent 165fb58b39
commit 818196d2c3
3 changed files with 21 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -405,6 +405,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;
handler: (ctx: HookCommandContext) => Promise<void>;
}
@ -478,7 +480,10 @@ export interface HookAPI {
* Register a custom slash command.
* Handler receives HookCommandContext.
*/
registerCommand(name: string, options: { description?: string; handler: RegisteredCommand["handler"] }): void;
registerCommand(
name: string,
options: { description?: string; immediate?: boolean; handler: RegisteredCommand["handler"] },
): void;
/**
* Execute a shell command and return stdout/stderr/code.

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

@ -752,6 +752,20 @@ export class InteractiveMode {
return;
}
// Check if this is an immediate hook command (runs even during streaming)
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
this.editor.addToHistory(text);
this.editor.setText("");
await this.session.prompt(text);
return;
}
}
// Queue message if agent is streaming
if (this.session.isStreaming) {
await this.session.queueMessage(text);