forked from phoenix/litellm-mirror
(test) mistral api - safe_mode, random_seed
This commit is contained in:
parent
b52ffe1bdf
commit
e34e4d3b71
1 changed files with 4 additions and 1 deletions
|
@ -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)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue