diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index 306f0b2f6ed..b5a2b72e7a7 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -59,6 +59,26 @@ def test_completion_claude(): # test_completion_claude() +def test_completion_claude2_1(): + try: + print("claude2.1 test request") + # test without max tokens + response = completion( + model="claude-2.1", + messages=messages, + request_timeout=10, + max_tokens=10 + ) + # 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() + # def test_completion_oobabooga(): # try: # response = completion(