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

(test) mistral api - safe_mode, random_seed

Browse source
This commit is contained in:
ishaan-jaff 2023-12-14 18:42:43 +05:30
parent b52ffe1bdf
commit e34e4d3b71
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_completion.py
View file

@ -70,6 +70,7 @@ def test_completion_claude():
def test_completion_mistral_api(): def test_completion_mistral_api():
try: try:
litellm.set_verbose=True
response = completion( response = completion(
model="mistral/mistral-tiny", model="mistral/mistral-tiny",
messages=[ messages=[
@ -77,7 +78,9 @@ def test_completion_mistral_api():
"role": "user", "role": "user",
"content": "Hey, how's it going?", "content": "Hey, how's it going?",
} }
] ],
random_seed=42,
safe_mode = True
) )
# Add any assertions here to check the response # Add any assertions here to check the response
print(response) print(response)