v2.5.0 (#495)

Co-authored-by: Quentin Torroba <quentin.torroba@mistral.ai>
Co-authored-by: Clement Sirieix <clem.sirieix@gmail.com>
Co-authored-by: Kim-Adeline Miguel <kimadeline.miguel@mistral.ai>
Co-authored-by: Simon Van de Kerckhove <simon.vandekerckhove@mistral.ai>
Co-authored-by: Vincent Guilloux <vincent.guilloux@mistral.ai>
Co-authored-by: Michel Thomazo <michel.thomazo@mistral.ai>
Co-authored-by: Mistral Vibe <vibe@mistral.ai>

tests/core/test_transcribe_config.py

@@ -0,0 +1,115 @@
+from __future__ import annotations
+
+import pytest
+
+from tests.conftest import build_test_vibe_config
+from vibe.core.config import (
+    DEFAULT_TRANSCRIBE_MODELS,
+    DEFAULT_TRANSCRIBE_PROVIDERS,
+    TranscribeClient,
+    TranscribeModelConfig,
+    TranscribeProviderConfig,
+)
+
+
+class TestTranscribeConfigDefaults:
+    def test_default_transcribe_providers_loaded(self) -> None:
+        config = build_test_vibe_config()
+        assert len(config.transcribe_providers) == len(DEFAULT_TRANSCRIBE_PROVIDERS)
+        assert config.transcribe_providers[0].name == "mistral"
+        assert config.transcribe_providers[0].api_base == "wss://api.mistral.ai"
+
+    def test_default_transcribe_models_loaded(self) -> None:
+        config = build_test_vibe_config()
+        assert len(config.transcribe_models) == len(DEFAULT_TRANSCRIBE_MODELS)
+        assert config.transcribe_models[0].alias == "voxtral-realtime"
+        assert (
+            config.transcribe_models[0].name == "voxtral-mini-transcribe-realtime-2602"
+        )
+
+    def test_default_active_transcribe_model(self) -> None:
+        config = build_test_vibe_config()
+        assert config.active_transcribe_model == "voxtral-realtime"
+
+
+class TestGetActiveTranscribeModel:
+    def test_resolves_by_alias(self) -> None:
+        config = build_test_vibe_config()
+        model = config.get_active_transcribe_model()
+        assert model.alias == "voxtral-realtime"
+        assert model.name == "voxtral-mini-transcribe-realtime-2602"
+
+    def test_raises_for_unknown_alias(self) -> None:
+        config = build_test_vibe_config(active_transcribe_model="nonexistent")
+        with pytest.raises(ValueError, match="not found in configuration"):
+            config.get_active_transcribe_model()
+
+
+class TestGetTranscribeProviderForModel:
+    def test_resolves_by_name(self) -> None:
+        config = build_test_vibe_config()
+        model = config.get_active_transcribe_model()
+        provider = config.get_transcribe_provider_for_model(model)
+        assert provider.name == "mistral"
+        assert provider.api_base == "wss://api.mistral.ai"
+
+    def test_raises_for_unknown_provider(self) -> None:
+        config = build_test_vibe_config(
+            transcribe_models=[
+                TranscribeModelConfig(
+                    name="test-model", provider="nonexistent", alias="test"
+                )
+            ],
+            active_transcribe_model="test",
+        )
+        model = config.get_active_transcribe_model()
+        with pytest.raises(ValueError, match="not found in configuration"):
+            config.get_transcribe_provider_for_model(model)
+
+
+class TestTranscribeModelUniqueness:
+    def test_duplicate_aliases_raise(self) -> None:
+        with pytest.raises(ValueError, match="Duplicate transcribe model alias"):
+            build_test_vibe_config(
+                transcribe_models=[
+                    TranscribeModelConfig(
+                        name="model-a", provider="mistral", alias="same-alias"
+                    ),
+                    TranscribeModelConfig(
+                        name="model-b", provider="mistral", alias="same-alias"
+                    ),
+                ],
+                active_transcribe_model="same-alias",
+            )
+
+
+class TestTranscribeModelConfig:
+    def test_alias_defaults_to_name(self) -> None:
+        model = TranscribeModelConfig.model_validate({
+            "name": "my-model",
+            "provider": "mistral",
+        })
+        assert model.alias == "my-model"
+
+    def test_explicit_alias(self) -> None:
+        model = TranscribeModelConfig(
+            name="my-model", provider="mistral", alias="custom-alias"
+        )
+        assert model.alias == "custom-alias"
+
+    def test_default_values(self) -> None:
+        model = TranscribeModelConfig(
+            name="my-model", provider="mistral", alias="my-model"
+        )
+        assert model.sample_rate == 16000
+        assert model.encoding == "pcm_s16le"
+        assert model.language == "en"
+        assert model.target_streaming_delay_ms == 500
+
+
+class TestTranscribeProviderConfig:
+    def test_default_values(self) -> None:
+        provider = TranscribeProviderConfig(name="test")
+        assert provider.api_base == "wss://api.mistral.ai"
+        assert provider.api_key_env_var == ""
+        assert provider.client == TranscribeClient.MISTRAL