phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(test) mistral ai api

Browse source
This commit is contained in:
ishaan-jaff 2023-12-14 18:19:45 +05:30
parent 7945664e61
commit 0ef7bf237e
1 changed files with 17 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

17
litellm/tests/test_completion.py
View file

@ -68,6 +68,23 @@ def test_completion_claude():
# 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(): def test_completion_claude2_1():
try: try:
print("claude2.1 test request") print("claude2.1 test request")