f8182c97df
Implement logic to resolve and inject Replicate Kokoro model-specific language codes based on language tag and voice, using new mapping utilities. Update `buildReplicateInput` to set `language_code` for the built-in Kokoro model. Expand language utilities with code mapping and expose `resolveReplicateKokoroLanguageCode`. Enhance tests to cover language code resolution and input mapping for Replicate Kokoro, and verify language input key discovery for built-in models.
109 lines
3.4 KiB
TypeScript
109 lines
3.4 KiB
TypeScript
import { describe, expect, test } from 'vitest';
|
|
|
|
import {
|
|
buildReplicateInput,
|
|
buildTTSCacheKey,
|
|
extractReplicateAudioUrl,
|
|
resolveReplicateLanguageValue,
|
|
} from '../../src/lib/server/tts/generate';
|
|
import { REPLICATE_KOKORO_82M_VERSIONED_MODEL } from '../../src/lib/shared/tts-provider-catalog';
|
|
|
|
describe('replicate output URL extraction', () => {
|
|
test('returns direct URL string output', () => {
|
|
expect(extractReplicateAudioUrl('https://replicate.delivery/audio.mp3')).toBe(
|
|
'https://replicate.delivery/audio.mp3'
|
|
);
|
|
});
|
|
|
|
test('extracts URL from FileOutput-like objects', () => {
|
|
const output = {
|
|
url: () => new URL('https://replicate.delivery/file.wav'),
|
|
};
|
|
expect(extractReplicateAudioUrl(output)).toBe('https://replicate.delivery/file.wav');
|
|
});
|
|
|
|
test('extracts first URL from array outputs', () => {
|
|
const output: unknown[] = [
|
|
{ value: 'not-a-url' },
|
|
{ toString: () => 'https://replicate.delivery/chunk-0.mp3' },
|
|
'https://replicate.delivery/chunk-1.mp3',
|
|
];
|
|
expect(extractReplicateAudioUrl(output)).toBe('https://replicate.delivery/chunk-0.mp3');
|
|
});
|
|
|
|
test('extracts nested URL from object outputs', () => {
|
|
const output = {
|
|
output: {
|
|
audio: {
|
|
url: 'https://replicate.delivery/nested.mp3',
|
|
},
|
|
},
|
|
};
|
|
expect(extractReplicateAudioUrl(output)).toBe('https://replicate.delivery/nested.mp3');
|
|
});
|
|
|
|
test('returns null for non-url outputs', () => {
|
|
const output = { status: 'ok', value: 123 };
|
|
expect(extractReplicateAudioUrl(output)).toBeNull();
|
|
});
|
|
});
|
|
|
|
describe('TTS upstream cache identity', () => {
|
|
test('includes language in the upstream audio cache key', () => {
|
|
const request = {
|
|
text: '同じ文章です。',
|
|
voice: 'jf_alpha',
|
|
speed: 1,
|
|
format: 'mp3',
|
|
model: 'kokoro',
|
|
provider: 'custom-openai',
|
|
apiKey: 'test',
|
|
};
|
|
|
|
expect(buildTTSCacheKey({ ...request, language: 'ja' }))
|
|
.not.toBe(buildTTSCacheKey({ ...request, language: 'en' }));
|
|
});
|
|
});
|
|
|
|
describe('Replicate language schema values', () => {
|
|
test('uses language codes or advertised display names without a model table', () => {
|
|
expect(resolveReplicateLanguageValue('ja-JP', [])).toBe('ja-JP');
|
|
expect(resolveReplicateLanguageValue('ja-JP', ['en', 'ja', 'zh'])).toBe('ja');
|
|
expect(resolveReplicateLanguageValue('ja-JP', ['English', 'Japanese'])).toBe('Japanese');
|
|
expect(resolveReplicateLanguageValue('ja-JP', ['English', 'French'])).toBeNull();
|
|
});
|
|
|
|
test('includes language_code for the built-in Replicate Kokoro model', async () => {
|
|
await expect(buildReplicateInput({
|
|
text: 'Hello world',
|
|
voice: 'af_sarah',
|
|
speed: 1,
|
|
format: 'mp3',
|
|
model: REPLICATE_KOKORO_82M_VERSIONED_MODEL,
|
|
language: 'en',
|
|
provider: 'replicate',
|
|
apiKey: 'r8_token',
|
|
testNamespace: null,
|
|
})).resolves.toEqual({
|
|
text: 'Hello world',
|
|
voice: 'af_sarah',
|
|
language_code: 'a',
|
|
});
|
|
|
|
await expect(buildReplicateInput({
|
|
text: 'Hello world',
|
|
voice: 'bf_emma',
|
|
speed: 1,
|
|
format: 'mp3',
|
|
model: REPLICATE_KOKORO_82M_VERSIONED_MODEL,
|
|
language: 'en',
|
|
provider: 'replicate',
|
|
apiKey: 'r8_token',
|
|
testNamespace: null,
|
|
})).resolves.toEqual({
|
|
text: 'Hello world',
|
|
voice: 'bf_emma',
|
|
language_code: 'b',
|
|
});
|
|
});
|
|
});
|