feat(companion): add typed agent stream subscription helpers
This commit is contained in:
@@ -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?: {
|
||||
|
||||
Reference in New Issue
Block a user