mirror of
https://github.com/mistralai/mistral-vibe
synced 2026-04-25 17:14:55 +02:00
e1a25caa52
Co-authored-by: Bastien <bastien.baret@gmail.com> Co-authored-by: Clément Sirieix <clement.sirieix@mistral.ai> Co-authored-by: Julien Legrand <72564015+JulienLGRD@users.noreply.github.com> Co-authored-by: Kim-Adeline Miguel <51720070+kimadeline@users.noreply.github.com> Co-authored-by: Mathias Gesbert <mathias.gesbert@mistral.ai> Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai> Co-authored-by: Quentin <quentin.torroba@mistral.ai> Co-authored-by: Vincent G <10739306+VinceOPS@users.noreply.github.com> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
115 lines
3.6 KiB
Python
115 lines
3.6 KiB
Python
from __future__ import annotations
|
|
|
|
import base64
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from vibe.core.config import TTSModelConfig, TTSProviderConfig
|
|
from vibe.core.tts import MistralTTSClient, TTSResult
|
|
|
|
|
|
def _make_provider() -> TTSProviderConfig:
|
|
return TTSProviderConfig(
|
|
name="mistral",
|
|
api_base="https://api.mistral.ai",
|
|
api_key_env_var="MISTRAL_API_KEY",
|
|
)
|
|
|
|
|
|
def _make_model() -> TTSModelConfig:
|
|
return TTSModelConfig(
|
|
name="voxtral-mini-tts-latest",
|
|
alias="voxtral-tts",
|
|
provider="mistral",
|
|
voice="gb_jane_neutral",
|
|
)
|
|
|
|
|
|
class TestMistralTTSClientInit:
|
|
def test_lazy_client_creation(self, monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setenv("MISTRAL_API_KEY", "test-key")
|
|
client = MistralTTSClient(_make_provider(), _make_model())
|
|
assert client._client is None
|
|
|
|
def test_get_client_creates_mistral_instance(
|
|
self, monkeypatch: pytest.MonkeyPatch
|
|
) -> None:
|
|
monkeypatch.setenv("MISTRAL_API_KEY", "test-key")
|
|
client = MistralTTSClient(_make_provider(), _make_model())
|
|
sdk_client = client._get_client()
|
|
assert sdk_client is not None
|
|
assert client._client is sdk_client
|
|
assert client._get_client() is sdk_client
|
|
|
|
|
|
class TestMistralTTSClient:
|
|
@pytest.mark.asyncio
|
|
async def test_speak_returns_decoded_audio(
|
|
self, monkeypatch: pytest.MonkeyPatch
|
|
) -> None:
|
|
monkeypatch.setenv("MISTRAL_API_KEY", "test-key")
|
|
|
|
raw_audio = b"fake-audio-data-for-testing"
|
|
encoded_audio = base64.b64encode(raw_audio).decode()
|
|
|
|
mock_response = MagicMock()
|
|
mock_response.audio_data = encoded_audio
|
|
|
|
mock_complete_async = AsyncMock(return_value=mock_response)
|
|
|
|
client = MistralTTSClient(_make_provider(), _make_model())
|
|
|
|
with patch.object(
|
|
type(client._get_client().audio.speech),
|
|
"complete_async",
|
|
mock_complete_async,
|
|
):
|
|
result = await client.speak("Hello")
|
|
|
|
assert isinstance(result, TTSResult)
|
|
assert result.audio_data == raw_audio
|
|
mock_complete_async.assert_called_once_with(
|
|
model="voxtral-mini-tts-latest",
|
|
input="Hello",
|
|
voice_id="gb_jane_neutral",
|
|
response_format="wav",
|
|
)
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_speak_raises_on_sdk_error(
|
|
self, monkeypatch: pytest.MonkeyPatch
|
|
) -> None:
|
|
monkeypatch.setenv("MISTRAL_API_KEY", "test-key")
|
|
import httpx
|
|
from mistralai.client.errors import SDKError
|
|
|
|
fake_response = httpx.Response(
|
|
status_code=500,
|
|
request=httpx.Request("POST", "https://api.mistral.ai/v1/audio/speech"),
|
|
)
|
|
mock_complete_async = AsyncMock(
|
|
side_effect=SDKError("API error", fake_response)
|
|
)
|
|
|
|
client = MistralTTSClient(_make_provider(), _make_model())
|
|
|
|
with (
|
|
patch.object(
|
|
type(client._get_client().audio.speech),
|
|
"complete_async",
|
|
mock_complete_async,
|
|
),
|
|
pytest.raises(SDKError),
|
|
):
|
|
await client.speak("Hello")
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_close_resets_client(self, monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setenv("MISTRAL_API_KEY", "test-key")
|
|
client = MistralTTSClient(_make_provider(), _make_model())
|
|
client._get_client()
|
|
assert client._client is not None
|
|
await client.close()
|
|
assert client._client is None
|