v2.9.6 (#682)
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:
parent
b23f49e5f4
commit
626f905186
59 changed files with 702 additions and 183 deletions
|
|
@ -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:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue