eb580209d4
Co-authored-by: Clément Drouin <clement.drouin@mistral.ai> Co-authored-by: Clément Sirieix <clement.sirieix@mistral.ai> Co-authored-by: Gauthier Guinet <43207538+Gguinet@users.noreply.github.com> Co-authored-by: Kim-Adeline Miguel <kimadeline.miguel@mistral.ai> Co-authored-by: Michel Thomazo <51709227+michelTho@users.noreply.github.com> Co-authored-by: Quentin <torroba.q@gmail.com> Co-authored-by: Simon <80467011+sorgfresser@users.noreply.github.com> Co-authored-by: Simon Van de Kerckhove <simon.vandekerckhove@mistral.ai> Co-authored-by: Vincent G <10739306+VinceOPS@users.noreply.github.com> Co-authored-by: angelapopopo <angele.lenglemetz@mistral.ai> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
107 lines
3.8 KiB
Python
107 lines
3.8 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from tests.conftest import build_test_vibe_config
|
|
from vibe.core.config import (
|
|
DEFAULT_TTS_MODELS,
|
|
DEFAULT_TTS_PROVIDERS,
|
|
TTSClient,
|
|
TTSModelConfig,
|
|
TTSProviderConfig,
|
|
)
|
|
|
|
|
|
class TestTTSConfigDefaults:
|
|
def test_default_tts_providers_loaded(self) -> None:
|
|
config = build_test_vibe_config()
|
|
assert len(config.tts_providers) == len(DEFAULT_TTS_PROVIDERS)
|
|
assert config.tts_providers[0].name == "mistral"
|
|
assert config.tts_providers[0].api_base == "https://api.mistral.ai"
|
|
|
|
def test_default_tts_models_loaded(self) -> None:
|
|
config = build_test_vibe_config()
|
|
assert len(config.tts_models) == len(DEFAULT_TTS_MODELS)
|
|
assert config.tts_models[0].alias == "voxtral-tts"
|
|
assert config.tts_models[0].name == "voxtral-mini-tts-latest"
|
|
|
|
def test_default_active_tts_model(self) -> None:
|
|
config = build_test_vibe_config()
|
|
assert config.active_tts_model == "voxtral-tts"
|
|
|
|
|
|
class TestGetActiveTTSModel:
|
|
def test_resolves_by_alias(self) -> None:
|
|
config = build_test_vibe_config()
|
|
model = config.get_active_tts_model()
|
|
assert model.alias == "voxtral-tts"
|
|
assert model.name == "voxtral-mini-tts-latest"
|
|
|
|
def test_raises_for_unknown_alias(self) -> None:
|
|
config = build_test_vibe_config(active_tts_model="nonexistent")
|
|
with pytest.raises(ValueError, match="not found in configuration"):
|
|
config.get_active_tts_model()
|
|
|
|
|
|
class TestGetTTSProviderForModel:
|
|
def test_resolves_by_name(self) -> None:
|
|
config = build_test_vibe_config()
|
|
model = config.get_active_tts_model()
|
|
provider = config.get_tts_provider_for_model(model)
|
|
assert provider.name == "mistral"
|
|
assert provider.api_base == "https://api.mistral.ai"
|
|
|
|
def test_raises_for_unknown_provider(self) -> None:
|
|
config = build_test_vibe_config(
|
|
tts_models=[
|
|
TTSModelConfig(name="test-model", provider="nonexistent", alias="test")
|
|
],
|
|
active_tts_model="test",
|
|
)
|
|
model = config.get_active_tts_model()
|
|
with pytest.raises(ValueError, match="not found in configuration"):
|
|
config.get_tts_provider_for_model(model)
|
|
|
|
|
|
class TestTTSModelUniqueness:
|
|
def test_duplicate_aliases_raise(self) -> None:
|
|
with pytest.raises(ValueError, match="Duplicate TTS model alias"):
|
|
build_test_vibe_config(
|
|
tts_models=[
|
|
TTSModelConfig(
|
|
name="model-a", provider="mistral", alias="same-alias"
|
|
),
|
|
TTSModelConfig(
|
|
name="model-b", provider="mistral", alias="same-alias"
|
|
),
|
|
],
|
|
active_tts_model="same-alias",
|
|
)
|
|
|
|
|
|
class TestTTSModelConfig:
|
|
def test_alias_defaults_to_name(self) -> None:
|
|
model = TTSModelConfig.model_validate({
|
|
"name": "my-model",
|
|
"provider": "mistral",
|
|
})
|
|
assert model.alias == "my-model"
|
|
|
|
def test_explicit_alias(self) -> None:
|
|
model = TTSModelConfig(
|
|
name="my-model", provider="mistral", alias="custom-alias"
|
|
)
|
|
assert model.alias == "custom-alias"
|
|
|
|
def test_default_values(self) -> None:
|
|
model = TTSModelConfig(name="my-model", provider="mistral", alias="my-model")
|
|
assert model.voice == "gb_jane_neutral"
|
|
assert model.response_format == "wav"
|
|
|
|
|
|
class TestTTSProviderConfig:
|
|
def test_default_values(self) -> None:
|
|
provider = TTSProviderConfig(name="test")
|
|
assert provider.api_base == "https://api.mistral.ai"
|
|
assert provider.api_key_env_var == ""
|
|
assert provider.client == TTSClient.MISTRAL
|