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

test: remove duplicate test

Browse source
This commit is contained in:
Krrish Dholakia 2024-11-20 05:53:43 +05:30
parent f2e6c9c9d8
commit 4ddbf1395f
1 changed files with 0 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

26
tests/local_testing/test_completion.py
View file

@ -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: