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

fix testing

Browse source
This commit is contained in:
Krrish Dholakia 2023-09-30 08:50:51 -07:00
parent 905a3f08ee
commit 1d8b56cc74
2 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_budget_manager.py
View file

@ -67,8 +67,8 @@ def test_save_user_budget():
if response["status"] == "error":
raise Exception(f"An error occurred - {json.dumps(response)}")
print(response)
except:
pytest.fail(f"An error occurred")
except Exception as e:
pytest.fail(f"An error occurred: {str(e)}")
## Scenario 4: Getting list of users
def test_get_users():

2
litellm/tests/test_completion.py
View file

@ -713,7 +713,7 @@ def test_completion_replicate_vicuna():
######## Test TogetherAI ########
def test_completion_together_ai():
model_name = "togethercomputer/llama-2-70b-chat"
model_name = "together_ai/togethercomputer/llama-2-70b-chat"
try:
response = completion(model=model_name, messages=messages, max_tokens=256, logger_fn=logger_fn)
# Add any assertions here to check the response