@livekit/agents-plugin-openai 0.9.3 → 1.0.0-next.0

package/src/tts.ts

@@ -1,9 +1,8 @@
 // SPDX-FileCopyrightText: 2024 LiveKit, Inc.
 //
 // SPDX-License-Identifier: Apache-2.0
-import { AudioByteStream, tts } from '@livekit/agents';
+import { AudioByteStream, shortuuid, tts } from '@livekit/agents';
 import type { AudioFrame } from '@livekit/rtc-node';
-import { randomUUID } from 'crypto';
 import { OpenAI } from 'openai';
 import type { TTSModels, TTSVoices } from './models.js';
 
@@ -81,16 +80,17 @@ export class TTS extends tts.TTS {
 
 export class ChunkedStream extends tts.ChunkedStream {
   label = 'openai.ChunkedStream';
+  private stream: Promise<any>;
 
   // set Promise<T> to any because OpenAI returns an annoying Response type
   constructor(tts: TTS, text: string, stream: Promise<any>) {
     super(text, tts);
-    this.#run(stream);
+    this.stream = stream;
   }
 
-  async #run(stream: Promise<Response>) {
-    const buffer = await stream.then((r) => r.arrayBuffer());
-    const requestId = randomUUID();
+  protected async run() {
+    const buffer = await this.stream.then((r) => r.arrayBuffer());
+    const requestId = shortuuid();
     const audioByteStream = new AudioByteStream(OPENAI_TTS_SAMPLE_RATE, OPENAI_TTS_CHANNELS);
     const frames = audioByteStream.write(buffer);