Co-authored-by: Clément Drouin <clement.drouin@mistral.ai>
Co-authored-by: Michel Thomazo <51709227+michelTho@users.noreply.github.com>
Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai>
Co-authored-by: Mistral Vibe <vibe@mistral.ai>
This commit is contained in:
Mathias Gesbert 2026-05-11 11:44:53 +02:00 committed by GitHub
parent b23f49e5f4
commit 626f905186
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
59 changed files with 702 additions and 183 deletions

View file

@ -432,12 +432,12 @@ class TestMistralRetry:
with patch("vibe.core.llm.backend.mistral.Mistral") as mock_mistral_class:
mock_mistral_class.return_value = MagicMock()
backend._get_client()
mock_mistral_class.assert_called_once_with(
api_key=backend._api_key,
server_url=backend._server_url,
timeout_ms=720000,
retry_config=backend._retry_config,
)
call_kwargs = mock_mistral_class.call_args.kwargs
assert call_kwargs["api_key"] == backend._api_key
assert call_kwargs["server_url"] == backend._server_url
assert call_kwargs["timeout_ms"] == 720000
assert call_kwargs["retry_config"] is backend._retry_config
assert "async_client" in call_kwargs
class TestMistralMapperPrepareMessage: