feat(companion): add typed agent stream subscription helpers

This commit is contained in:
William Valentin
2026-02-16 19:08:13 -08:00
parent fc0dd8b73a
commit 0a43abd531
5 changed files with 62 additions and 1 deletions
+18
View File
@@ -45,6 +45,12 @@ export type CompanionEventHandler = (event: string, data: unknown) => void;
export type CompanionTypedEventHandler<TData = unknown> = (data: TData) => void;
export type CompanionEventPredicate<TData = unknown> = (data: TData) => boolean;
export const COMPANION_EVENT_NAMES = {
agentStream: 'agent.stream',
agentTyping: 'agent.typing',
contextWarning: 'context_warning',
} as const;
export interface RegisterNodeInput {
nodeId: string;
role: string;
@@ -385,6 +391,18 @@ export class CompanionRuntimeClient {
});
}
subscribeAgentStream<TData = unknown>(
handler: CompanionTypedEventHandler<TData>,
): () => void {
return this.subscribeEvent<TData>(COMPANION_EVENT_NAMES.agentStream, handler);
}
subscribeAgentTyping<TData = unknown>(
handler: CompanionTypedEventHandler<TData>,
): () => void {
return this.subscribeEvent<TData>(COMPANION_EVENT_NAMES.agentTyping, handler);
}
waitForEvent<TData = unknown>(
eventName: string,
options?: {