diff --git a/tests/tools/test_transcription_tools.py b/tests/tools/test_transcription_tools.py index 9f5fab6294..f781c32bd4 100644 --- a/tests/tools/test_transcription_tools.py +++ b/tests/tools/test_transcription_tools.py @@ -875,7 +875,7 @@ def mock_mistral_module(): mock_mistral_cls = MagicMock(return_value=mock_client) fake_module = MagicMock() fake_module.Mistral = mock_mistral_cls - with patch.dict("sys.modules", {"mistralai": fake_module}): + with patch.dict("sys.modules", {"mistralai": fake_module, "mistralai.client": fake_module}): yield mock_client diff --git a/tools/transcription_tools.py b/tools/transcription_tools.py index 296c74a21b..d4f9145c2d 100644 --- a/tools/transcription_tools.py +++ b/tools/transcription_tools.py @@ -546,7 +546,7 @@ def _transcribe_mistral(file_path: str, model_name: str) -> Dict[str, Any]: return {"success": False, "transcript": "", "error": "MISTRAL_API_KEY not set"} try: - from mistralai import Mistral + from mistralai.client import Mistral with Mistral(api_key=api_key) as client: with open(file_path, "rb") as audio_file: