ai/src/realtime/realtime-attempt.ts

AI SDK by Vercel - build apps like ChatGPT, Claude, Gemini, and more with a single interface for any model using the Vercel AI Gateway or go direct to OpenAI, Anthropic, Google, or any other model provider.

/** Cancellation and settlement belong to one connection attempt. */
export class RealtimeAttempt {
  readonly abort = new AbortController();
  readonly reportedTransportErrors = new WeakSet<Error>();
  private timers = new Map<string, ReturnType<typeof setTimeout>>();
  private settleClose?: () => void;
  closePromise?: Promise<void>;
  active = true;
  ready = false;
  closing = false;
  transportClosing = false;
  cause?: Error;

  timer(name: string, duration: number, callback: () => void): void {
    this.clearTimer(name);
    this.timers.set(
      name,
      setTimeout(() => {
        this.timers.delete(name);
        if (this.active) callback();
      }, duration),
    );
  }

  clearTimer(name: string): void {
    clearTimeout(this.timers.get(name));
    this.timers.delete(name);
  }

  beginClose(): Promise<void> {
    this.closing = true;
    return (this.closePromise ??= new Promise(resolve => {
      this.settleClose = resolve;
    }));
  }

  retire(): void {
    if (!this.active) return;
    this.active = false;
    this.abort.abort();
    for (const timer of this.timers.values()) clearTimeout(timer);
    this.timers.clear();
    this.settleClose?.();
    this.settleClose = undefined;
  }
}