From 219470ed681334a4f5dc1ebd28b18b31d8d5671b Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 21 Nov 2023 09:53:47 -0800 Subject: [PATCH] (test) add claude-2.1 --- litellm/tests/test_completion.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) 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(