forked from phoenix/litellm-mirror
(test) mistral ai api
This commit is contained in:
parent
7945664e61
commit
0ef7bf237e
1 changed files with 17 additions and 0 deletions
|
@ -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")
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue