feat(companion): add typed wait helpers for stream events

2026-02-16 19:11:10 -08:00
parent 1b69970065
commit 43968f830a
4 changed files with 64 additions and 1 deletions
@@ -459,6 +459,22 @@ export class CompanionRuntimeClient {
    });
  }

+  waitForAgentStream<TData = unknown>(options?: {
+    timeoutMs?: number;
+    predicate?: CompanionEventPredicate<TData>;
+    signal?: AbortSignal;
+  }): Promise<TData> {
+    return this.waitForEvent<TData>(COMPANION_EVENT_NAMES.agentStream, options);
+  }
+
+  waitForAgentTyping<TData = unknown>(options?: {
+    timeoutMs?: number;
+    predicate?: CompanionEventPredicate<TData>;
+    signal?: AbortSignal;
+  }): Promise<TData> {
+    return this.waitForEvent<TData>(COMPANION_EVENT_NAMES.agentTyping, options);
+  }
+
  async call<T>(method: string, params?: Record<string, unknown>): Promise<T> {
    if (!this.connected) {
      if (!this.autoConnect) {