test_mistral_api.py

importasyncio
importos
importsys
importtraceback

fromdotenvimportload_dotenv

importlitellm.types
importlitellm.types.utils
fromlitellm.llms.anthropic.chatimportModelResponseIterator

load_dotenv()
importio
importos

sys.path.insert(
0, os.path.abspath("../..")
) # Adds the parent directory to the system path
fromtypingimportOptional
fromunittest.mockimportMagicMock, patch

importpytest

importlitellm

fromlitellm.llms.anthropic.common_utilsimportprocess_anthropic_headers
fromhttpximportHeaders
frombase_llm_unit_testsimportBaseLLMChatTest


classTestMistralCompletion(BaseLLMChatTest):
defget_base_completion_call_args(self) ->dict:
litellm.set_verbose=True
return {"model": "mistral/mistral-small-latest"}

deftest_tool_call_no_arguments(self, tool_call_no_arguments):
"""Test that tool calls with no arguments is translated correctly. Relevant issue: https://github.com/BerriAI/litellm/issues/6833"""
pass

deftest_multilingual_requests(self):
"""
 Mistral API raises a 400 BadRequest error when the request contains invalid utf-8 sequences.
 """
pass