From 0ef7bf237ec45d7b1fdceda9c39df1c4bd69880c Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 14 Dec 2023 18:19:45 +0530 Subject: [PATCH] (test) mistral ai api --- litellm/tests/test_completion.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index d6bed46aa..78bbb94eb 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -68,6 +68,23 @@ def test_completion_claude(): # test_completion_claude() +def test_completion_mistral_api(): + try: + response = completion( + model="mistral/mistral-tiny", + messages=[ + { + "role": "user", + "content": "Hey, how's it going?", + } + ] + ) + # Add any assertions here to check the response + print(response) + except Exception as e: + pytest.fail(f"Error occurred: {e}") +# test_completion_mistral_api() + def test_completion_claude2_1(): try: print("claude2.1 test request")