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

Added deepseek completion test

Browse source
This commit is contained in:
Paul Gauthier 2024-05-07 11:58:05 -07:00
parent 9162f9c2c5
commit 82a4c68e60
1 changed files with 23 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

14
litellm/tests/test_completion.py
View file

@ -3016,6 +3016,20 @@ async def test_acompletion_gemini():
else: else:
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
# Deepseek tests
def test_completion_deepseek():
litellm.set_verbose = True
model_name = "deepseek/deepseek-chat"
messages = [{"role": "user", "content": "Hey, how's it going?"}]
try:
response = completion(model=model_name, messages=messages)
# Add any assertions here to check the response
print(response)
except litellm.APIError as e:
pass
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# Palm tests # Palm tests
def test_completion_palm(): def test_completion_palm():