From 1d8b56cc74ef05d1c2b3167c6d340c91fd9edfd6 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 30 Sep 2023 08:50:51 -0700 Subject: [PATCH] fix testing --- litellm/tests/test_budget_manager.py | 4 ++-- litellm/tests/test_completion.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/litellm/tests/test_budget_manager.py b/litellm/tests/test_budget_manager.py index e47682d9b4e..230cc60eb06 100644 --- a/litellm/tests/test_budget_manager.py +++ b/litellm/tests/test_budget_manager.py @@ -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(): diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index 0b55e868ae1..ad4b33f311f 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -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