feat(companion): add context warning stream helpers

2026-02-16 19:12:02 -08:00
parent 43968f830a
commit 636f24016f
4 changed files with 65 additions and 1 deletions
@@ -403,6 +403,12 @@ export class CompanionRuntimeClient {
    return this.subscribeEvent<TData>(COMPANION_EVENT_NAMES.agentTyping, handler);
  }

+  subscribeContextWarning<TData = unknown>(
+    handler: CompanionTypedEventHandler<TData>,
+  ): () => void {
+    return this.subscribeEvent<TData>(COMPANION_EVENT_NAMES.contextWarning, handler);
+  }
+
  waitForEvent<TData = unknown>(
    eventName: string,
    options?: {
@@ -475,6 +481,14 @@ export class CompanionRuntimeClient {
    return this.waitForEvent<TData>(COMPANION_EVENT_NAMES.agentTyping, options);
  }

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