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

test(test_completion.py): fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-03 21:30:45 -08:00
parent ef0171e063
commit 3a19c8b600
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/tests/test_completion.py
View file

@ -37,11 +37,11 @@ def test_completion_custom_provider_model_name():
try:
litellm.cache = None
response = completion(
model="together_ai/mistralai/Mistral-7B-Instruct-v0.1",
model="together_ai/mistralai/Mixtral-8x7B-Instruct-v0.1",
messages=messages,
logger_fn=logger_fn,
)
# Add any assertions here to check the response
# Add any assertions here to check the, response
print(response)
print(response["choices"][0]["finish_reason"])
except Exception as e:
@ -1369,7 +1369,7 @@ def test_customprompt_together_ai():
print(litellm.success_callback)
print(litellm._async_success_callback)
response = completion(
model="together_ai/mistralai/Mistral-7B-Instruct-v0.1",
model="together_ai/mistralai/Mixtral-8x7B-Instruct-v0.1",
messages=messages,
roles={
"system": {
@ -1998,7 +1998,7 @@ def test_completion_together_ai_stream():
messages = [{"content": user_message, "role": "user"}]
try:
response = completion(
model="together_ai/mistralai/Mistral-7B-Instruct-v0.1",
model="together_ai/mistralai/Mixtral-8x7B-Instruct-v0.1",
messages=messages,
stream=True,
max_tokens=5,