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>
This commit is contained in:
Mathias Gesbert 2026-03-23 18:45:21 +01:00 committed by GitHub
parent 5103019b01
commit eb580209d4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
180 changed files with 11136 additions and 1030 deletions

View file

@ -0,0 +1,107 @@
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