(test) mistral api - safe_mode, random_seed

This commit is contained in:
ishaan-jaff 2023-12-14 18:42:43 +05:30
parent b52ffe1bdf
commit e34e4d3b71

View file

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