wehub-resource-sync
29 lines
965 B
JavaScript
29 lines
965 B
JavaScript
class OpenAiSTT {
|
|
constructor() {
|
|
if (!process.env.OPEN_AI_KEY) throw new Error("No OpenAI API key was set.");
|
|
const { OpenAI: OpenAIApi } = require("openai");
|
|
this.openai = new OpenAIApi({
|
|
apiKey: process.env.OPEN_AI_KEY,
|
|
});
|
|
this.model = process.env.STT_OPEN_AI_MODEL ?? "whisper-1";
|
|
}
|
|
|
|
/**
|
|
* Transcribes an audio buffer to text using the OpenAI STT service.
|
|
* @param {Buffer} audioBuffer - The audio buffer to be transcribed.
|
|
* @param {string} filename - Original filename, used to hint the audio container/codec to OpenAI.
|
|
* @returns {Promise<string>} The transcribed text.
|
|
*/
|
|
async transcribe(audioBuffer, filename = "audio.webm") {
|
|
const { toFile } = require("openai");
|
|
const file = await toFile(audioBuffer, filename);
|
|
const result = await this.openai.audio.transcriptions.create({
|
|
file,
|
|
model: this.model,
|
|
});
|
|
return result?.text ?? "";
|
|
}
|
|
}
|
|
|
|
module.exports = { OpenAiSTT };
|