assemblyai/src/services/streaming/factory.ts

The AssemblyAI JavaScript SDK provides an easy-to-use interface for interacting with the AssemblyAI API, which supports async and real-time transcription, as well as the latest LeMUR models.

import {
  BaseServiceParams,
  StreamingTokenParams,
  StreamingTranscriberParams,
  StreamingTemporaryTokenResponse,
} from "../..";
import { StreamingTranscriber } from "./service";
import { BaseService } from "../base";

export class StreamingTranscriberFactory extends BaseService {
  private baseServiceParams: BaseServiceParams;

  constructor(params: BaseServiceParams) {
    super(params);
    this.baseServiceParams = params;
  }

  transcriber(params: StreamingTranscriberParams): StreamingTranscriber {
    const serviceParams = { ...params };

    if (!serviceParams.token && !serviceParams.apiKey) {
      serviceParams.apiKey = this.baseServiceParams.apiKey;
    }

    return new StreamingTranscriber(serviceParams);
  }

  async createTemporaryToken(params: StreamingTokenParams) {
    const searchParams = new URLSearchParams();

    // Add each param to the search params
    Object.entries(params).forEach(([key, value]) => {
      if (value !== undefined && value !== null) {
        searchParams.append(key, String(value));
      }
    });

    const queryString = searchParams.toString();
    const url = queryString ? `/v3/token?${queryString}` : "/v3/token";

    const data = await this.fetchJson<StreamingTemporaryTokenResponse>(url, {
      method: "GET",
    });
    return data.token;
  }
}

export class StreamingServiceFactory extends StreamingTranscriberFactory {}