diff --git a/tests/local_testing/test_completion.py b/tests/local_testing/test_completion.py index 936d2edc9..12f7b65cc 100644 --- a/tests/local_testing/test_completion.py +++ b/tests/local_testing/test_completion.py @@ -1222,32 +1222,6 @@ def test_completion_mistral_api_modified_input(): pytest.fail(f"Error occurred: {e}") -def test_completion_claude2_1(): - try: - litellm.set_verbose = True - print("claude2.1 test request") - messages = [ - { - "role": "system", - "content": "Your goal is generate a joke on the topic user gives.", - }, - {"role": "user", "content": "Generate a 3 liner joke for me"}, - ] - # test without max tokens - response = completion(model="claude-2.1", messages=messages) - # Add any assertions here to check the response - print(response) - print(response.usage) - print(response.usage.completion_tokens) - print(response["usage"]["completion_tokens"]) - # print("new cost tracking") - except Exception as e: - pytest.fail(f"Error occurred: {e}") - - -# test_completion_claude2_1() - - @pytest.mark.asyncio async def test_acompletion_claude2_1(): try: