Files
2026-07-13 13:39:12 +08:00

548 lines
16 KiB
TypeScript

/**
* Audio Provider Registry
*
* Defines providers that support audio endpoints:
* - /v1/audio/transcriptions (Whisper API)
* - /v1/audio/translations (Whisper translate-to-English API)
* - /v1/audio/speech (TTS API)
*/
interface AudioModel {
id: string;
name: string;
}
export interface AudioProvider {
id: string;
baseUrl: string;
authType: string;
authHeader: string;
format?: string;
supportedFormats?: string[];
async?: boolean;
models: AudioModel[];
}
export const AUDIO_TRANSCRIPTION_PROVIDERS: Record<string, AudioProvider> = {
vertex: {
id: "vertex",
baseUrl: "https://us-central1-aiplatform.googleapis.com/v1",
authType: "apikey",
authHeader: "bearer",
format: "vertex-gemini",
models: [
{ id: "gemini-2.5-flash", name: "Gemini 2.5 Flash (Vertex Transcribe)" },
{ id: "gemini-2.5-pro", name: "Gemini 2.5 Pro (Vertex Transcribe)" },
{ id: "gemini-2.0-flash", name: "Gemini 2.0 Flash (Vertex Transcribe)" },
],
},
openai: {
id: "openai",
baseUrl: "https://api.openai.com/v1/audio/transcriptions",
authType: "apikey",
authHeader: "bearer",
models: [
{ id: "whisper-1", name: "Whisper 1" },
{ id: "gpt-4o-transcription", name: "GPT-4o Transcription" },
],
},
cohere: {
id: "cohere",
baseUrl: "https://api.cohere.com/v2/audio/transcriptions",
authType: "apikey",
authHeader: "bearer",
models: [{ id: "cohere-transcribe-03-2026", name: "Cohere Transcribe 2026-03" }],
},
groq: {
id: "groq",
baseUrl: "https://api.groq.com/openai/v1/audio/transcriptions",
authType: "apikey",
authHeader: "bearer",
models: [
{ id: "whisper-large-v3", name: "Whisper Large v3" },
{ id: "whisper-large-v3-turbo", name: "Whisper Large v3 Turbo" },
{ id: "distil-whisper-large-v3-en", name: "Distil Whisper Large v3 EN" },
],
},
deepgram: {
id: "deepgram",
baseUrl: "https://api.deepgram.com/v1/listen",
authType: "apikey",
authHeader: "token",
format: "deepgram",
models: [
{ id: "nova-3", name: "Nova 3" },
{ id: "nova-2", name: "Nova 2" },
{ id: "whisper-large", name: "Whisper Large" },
],
},
pollinations: {
id: "pollinations",
baseUrl: "https://gen.pollinations.ai/v1/audio/transcriptions",
authType: "apikey",
authHeader: "bearer",
format: "openai",
models: [{ id: "whisper", name: "Pollinations Whisper (Free)" }],
},
together: {
id: "together",
baseUrl: "https://api.together.xyz/v1/audio/transcriptions",
authType: "apikey",
authHeader: "bearer",
format: "openai",
models: [
{ id: "openai/whisper-large-v3", name: "Whisper Large v3" },
{ id: "openai/whisper-large-v3-turbo", name: "Whisper Large v3 Turbo" },
],
},
assemblyai: {
id: "assemblyai",
baseUrl: "https://api.assemblyai.com/v2/transcript",
authType: "apikey",
authHeader: "bearer",
async: true,
format: "assemblyai",
models: [
{ id: "universal-3-pro", name: "Universal 3 Pro" },
{ id: "universal-2", name: "Universal 2" },
],
},
nvidia: {
id: "nvidia",
baseUrl: "https://integrate.api.nvidia.com/v1/audio/transcriptions",
authType: "apikey",
authHeader: "bearer",
format: "nvidia-asr",
models: [
{ id: "nvidia/parakeet-ctc-1.1b-asr", name: "Parakeet CTC 1.1B" },
{ id: "openai/whisper-large-v3", name: "Whisper Large v3 (NVIDIA)" },
],
},
huggingface: {
id: "huggingface",
baseUrl: "https://api-inference.huggingface.co/models",
authType: "apikey",
authHeader: "bearer",
format: "huggingface-asr",
models: [
{ id: "openai/whisper-large-v3-turbo", name: "Whisper Large v3 Turbo (HF)" },
{ id: "openai/whisper-large-v3", name: "Whisper Large v3 (HF)" },
],
},
qwen: {
id: "qwen",
baseUrl: "http://localhost:8000/v1/audio/transcriptions",
authType: "none",
authHeader: "none",
format: "openai",
models: [{ id: "qwen3-asr", name: "Qwen3 ASR" }],
},
kie: {
id: "kie",
baseUrl: "https://api.kie.ai",
authType: "apikey",
authHeader: "bearer",
format: "kie-audio",
models: [
{ id: "elevenlabs/speech-to-text", name: "ElevenLabs STT" },
{ id: "elevenlabs/audio-isolation", name: "ElevenLabs Audio Isolation" },
],
},
};
/**
* Providers that expose an OpenAI-Whisper-compatible /audio/translations
* endpoint (translate-to-English). This is a narrower surface than
* transcription: only Whisper-family models support it, and there is no
* `language` input — output is always English regardless of source audio.
*/
export const AUDIO_TRANSLATION_PROVIDERS: Record<string, AudioProvider> = {
openai: {
id: "openai",
baseUrl: "https://api.openai.com/v1/audio/translations",
authType: "apikey",
authHeader: "bearer",
models: [{ id: "whisper-1", name: "Whisper 1" }],
},
groq: {
id: "groq",
baseUrl: "https://api.groq.com/openai/v1/audio/translations",
authType: "apikey",
authHeader: "bearer",
models: [{ id: "whisper-large-v3", name: "Whisper Large v3" }],
},
};
export const AUDIO_SPEECH_PROVIDERS: Record<string, AudioProvider> = {
vertex: {
id: "vertex",
baseUrl: "https://us-central1-aiplatform.googleapis.com/v1",
authType: "apikey",
authHeader: "bearer",
format: "vertex-gemini-tts",
models: [
{ id: "gemini-3.1-flash-tts-preview", name: "Gemini 3.1 Flash TTS (Vertex)" },
{ id: "gemini-2.5-flash-preview-tts", name: "Gemini 2.5 Flash TTS (Vertex)" },
{ id: "gemini-2.5-pro-preview-tts", name: "Gemini 2.5 Pro TTS (Vertex)" },
],
},
openai: {
id: "openai",
baseUrl: "https://api.openai.com/v1/audio/speech",
authType: "apikey",
authHeader: "bearer",
models: [
{ id: "tts-1-hd", name: "TTS 1 HD" },
{ id: "tts-1", name: "TTS 1" },
{ id: "gpt-4o-mini-tts", name: "GPT-4o Mini TTS" },
],
},
hyperbolic: {
id: "hyperbolic",
baseUrl: "https://api.hyperbolic.xyz/v1/audio/generation",
authType: "apikey",
authHeader: "bearer",
format: "hyperbolic",
models: [{ id: "melo-tts", name: "Melo TTS" }],
},
deepgram: {
id: "deepgram",
baseUrl: "https://api.deepgram.com/v1/speak",
authType: "apikey",
authHeader: "token",
format: "deepgram",
models: [
{ id: "aura-asteria-en", name: "Aura Asteria (EN)" },
{ id: "aura-luna-en", name: "Aura Luna (EN)" },
{ id: "aura-stella-en", name: "Aura Stella (EN)" },
],
},
nvidia: {
id: "nvidia",
baseUrl: "https://integrate.api.nvidia.com/v1/audio/speech",
authType: "apikey",
authHeader: "bearer",
format: "nvidia-tts",
models: [
{ id: "nvidia/fastpitch", name: "FastPitch" },
{ id: "nvidia/tacotron2", name: "Tacotron2" },
],
},
elevenlabs: {
id: "elevenlabs",
baseUrl: "https://api.elevenlabs.io/v1/text-to-speech",
authType: "apikey",
authHeader: "xi-api-key",
format: "elevenlabs",
models: [
{ id: "eleven_multilingual_v2", name: "Eleven Multilingual v2" },
{ id: "eleven_turbo_v2_5", name: "Eleven Turbo v2.5" },
],
},
huggingface: {
id: "huggingface",
baseUrl: "https://api-inference.huggingface.co/models",
authType: "apikey",
authHeader: "bearer",
format: "huggingface-tts",
models: [
{ id: "canopylabs/orpheus-3b-0.1-ft", name: "Orpheus 3B" },
{ id: "ResembleAI/chatterbox", name: "Chatterbox" },
{ id: "hexgrad/Kokoro-82M", name: "Kokoro TTS" },
],
},
coqui: {
id: "coqui",
baseUrl: "http://localhost:5002/api/tts",
authType: "none",
authHeader: "none",
format: "coqui",
models: [{ id: "tts_models/en/ljspeech/tacotron2-DDC", name: "Tacotron2 DDC (LJSpeech)" }],
},
tortoise: {
id: "tortoise",
baseUrl: "http://localhost:5000/api/tts",
authType: "none",
authHeader: "none",
format: "tortoise",
models: [{ id: "tortoise-v2", name: "Tortoise v2" }],
},
qwen: {
id: "qwen",
baseUrl: "http://localhost:8000/v1/audio/speech",
authType: "none",
authHeader: "none",
format: "openai",
models: [{ id: "qwen3-tts", name: "Qwen3 TTS" }],
},
// ── Cloud TTS Providers (#248) ────────────────────────────────────────────
inworld: {
id: "inworld",
// POST https://api.inworld.ai/tts/v1/voice
// Auth: Authorization: Basic <api-key>
// Response: JSON { audioContent: "<base64>", contentType, sampleRateHertz }
baseUrl: "https://api.inworld.ai/tts/v1/voice",
authType: "apikey",
authHeader: "basic",
format: "inworld",
supportedFormats: ["mp3", "wav", "opus", "pcm"],
models: [
{ id: "inworld-tts-2", name: "Inworld TTS 2" },
{ id: "inworld-tts-1.5-mini", name: "Inworld TTS 1.5 Mini" },
],
},
cartesia: {
id: "cartesia",
// POST https://api.cartesia.ai/tts/bytes
// Auth: X-API-Key header, Cartesia-Version: 2024-06-10
// Response: binary audio bytes
baseUrl: "https://api.cartesia.ai/tts/bytes",
authType: "apikey",
authHeader: "x-api-key",
format: "cartesia",
models: [
{ id: "sonic-3", name: "Sonic 3" },
{ id: "sonic-2", name: "Sonic 2" },
],
},
playht: {
id: "playht",
// POST https://api.play.ht/api/v2/tts/stream
// Auth: X-USER-ID + Authorization: Bearer <api-key>
// Response: audio stream (mp3/wav)
baseUrl: "https://api.play.ht/api/v2/tts/stream",
authType: "apikey",
authHeader: "playht",
format: "playht",
models: [
{ id: "PlayDialog", name: "PlayDialog" },
{ id: "Play3.0-mini", name: "Play3.0 Mini" },
],
},
kie: {
id: "kie",
baseUrl: "https://api.kie.ai",
authType: "apikey",
authHeader: "bearer",
format: "kie-audio",
models: [
{ id: "elevenlabs/text-to-speech-multilingual-v2", name: "ElevenLabs TTS v2" },
{ id: "elevenlabs/text-to-speech-turbo-2-5", name: "ElevenLabs TTS Turbo 2.5" },
{ id: "elevenlabs/text-to-dialogue-v3", name: "ElevenLabs Text to Dialogue v3" },
{ id: "elevenlabs/sound-effect-v2", name: "ElevenLabs Sound Effect v2" },
],
},
"aws-polly": {
id: "aws-polly",
// POST https://polly.{region}.amazonaws.com/v1/speech
// Auth: AWS SigV4. The provider apiKey stores Secret Access Key; PSD stores accessKeyId/region.
baseUrl: "https://polly.us-east-1.amazonaws.com",
authType: "apikey",
authHeader: "aws-sigv4",
format: "aws-polly",
models: [
{ id: "standard", name: "Polly Standard" },
{ id: "neural", name: "Polly Neural" },
{ id: "long-form", name: "Polly Long-Form" },
{ id: "generative", name: "Polly Generative" },
],
},
pollinations: {
id: "pollinations",
baseUrl: "https://gen.pollinations.ai/v1/audio/speech",
authType: "apikey",
authHeader: "bearer",
format: "openai",
models: [{ id: "default", name: "Pollinations TTS (Free)" }],
},
minimax: {
id: "minimax",
baseUrl: "https://api.minimax.io/v1/t2a_v2",
authType: "apikey",
authHeader: "bearer",
format: "minimax-tts",
models: [{ id: "speech-2.8-hd", name: "Speech 2.8 HD" }],
},
together: {
id: "together",
baseUrl: "https://api.together.xyz/v1/audio/speech",
authType: "apikey",
authHeader: "bearer",
format: "openai",
models: [
{ id: "cartesia/sonic-2", name: "Cartesia Sonic 2" },
{ id: "hexgrad/Kokoro-82M", name: "Kokoro 82M" },
{ id: "canopylabs/orpheus-3b-0.1-ft", name: "Orpheus 3B" },
],
},
"xiaomi-mimo": {
id: "xiaomi-mimo",
baseUrl: "https://api.xiaomimimo.com/v1/chat/completions",
authType: "apikey",
authHeader: "bearer",
format: "xiaomi-mimo-tts",
supportedFormats: ["mp3", "wav"],
models: [
{ id: "mimo-v2.5-tts", name: "MiMo V2.5 TTS" },
{ id: "mimo-v2.5-tts-voicedesign", name: "MiMo V2.5 Voice Design" },
{ id: "mimo-v2.5-tts-voiceclone", name: "MiMo V2.5 Voice Clone" },
],
},
};
/**
* Get transcription provider config by ID
*/
export function getTranscriptionProvider(providerId: string): AudioProvider | null {
return AUDIO_TRANSCRIPTION_PROVIDERS[providerId] || null;
}
/**
* Get translation provider config by ID
*/
export function getTranslationProvider(providerId: string): AudioProvider | null {
return AUDIO_TRANSLATION_PROVIDERS[providerId] || null;
}
/**
* Get speech provider config by ID
*/
export function getSpeechProvider(providerId: string): AudioProvider | null {
return AUDIO_SPEECH_PROVIDERS[providerId] || null;
}
export interface ProviderNodeRow {
prefix: string;
name: string;
baseUrl: string;
apiType?: string;
}
/**
* Build a dynamic AudioProvider from a provider_node DB entry.
* Only used for local providers (localhost/127.0.0.1) — remote nodes are
* excluded by the caller to prevent auth bypass and SSRF.
*/
export function buildDynamicAudioProvider(node: ProviderNodeRow, audioPath: string): AudioProvider {
if (!node.prefix || !node.baseUrl) {
throw new Error(`Invalid provider_node: missing prefix or baseUrl`);
}
const baseUrl = node.baseUrl.replace(/\/+$/, "");
return {
id: node.prefix,
baseUrl: `${baseUrl}${audioPath}`,
authType: "none",
authHeader: "none",
models: [],
};
}
function parseAudioModel(
modelStr: string | null,
registry: Record<string, AudioProvider>,
dynamicProviders?: AudioProvider[]
): { provider: string | null; model: string | null } {
if (!modelStr) return { provider: null, model: null };
// Phase 1: prefix match in hardcoded registry
for (const [providerId] of Object.entries(registry)) {
if (modelStr.startsWith(providerId + "/")) {
return { provider: providerId, model: modelStr.slice(providerId.length + 1) };
}
}
// Phase 2: bare model lookup in hardcoded registry
for (const [providerId, config] of Object.entries(registry)) {
if (config.models.some((m) => m.id === modelStr)) {
return { provider: providerId, model: modelStr };
}
}
// Phase 3: prefix match in dynamic providers (provider_nodes)
if (dynamicProviders) {
for (const dp of dynamicProviders) {
if (modelStr.startsWith(dp.id + "/")) {
return { provider: dp.id, model: modelStr.slice(dp.id.length + 1) };
}
}
}
return { provider: null, model: modelStr };
}
export function parseTranscriptionModel(
modelStr: string | null,
dynamicProviders?: AudioProvider[]
) {
return parseAudioModel(modelStr, AUDIO_TRANSCRIPTION_PROVIDERS, dynamicProviders);
}
export function parseSpeechModel(modelStr: string | null, dynamicProviders?: AudioProvider[]) {
return parseAudioModel(modelStr, AUDIO_SPEECH_PROVIDERS, dynamicProviders);
}
export function parseTranslationModel(modelStr: string | null, dynamicProviders?: AudioProvider[]) {
return parseAudioModel(modelStr, AUDIO_TRANSLATION_PROVIDERS, dynamicProviders);
}
/**
* Get all audio models as a flat list
*/
export function getAllAudioModels() {
const models = [];
for (const [providerId, config] of Object.entries(AUDIO_TRANSCRIPTION_PROVIDERS)) {
for (const model of config.models) {
models.push({
id: model.id.startsWith(`${providerId}/`) ? model.id : `${providerId}/${model.id}`,
name: model.name,
provider: providerId,
subtype: "transcription",
});
}
}
for (const [providerId, config] of Object.entries(AUDIO_SPEECH_PROVIDERS)) {
for (const model of config.models) {
models.push({
id: model.id.startsWith(`${providerId}/`) ? model.id : `${providerId}/${model.id}`,
name: model.name,
provider: providerId,
subtype: "speech",
});
}
}
return models;
}