Added deepseek completion test

This commit is contained in:
Paul Gauthier 2024-05-07 11:58:05 -07:00
parent 9162f9c2c5
commit 82a4c68e60

View file

@ -1061,16 +1061,16 @@ def test_completion_perplexity_api_2():
######### HUGGING FACE TESTS ######################## ######### HUGGING FACE TESTS ########################
##################################################### #####################################################
""" """
HF Tests we should pass HF Tests we should pass
- TGI: - TGI:
- Pro Inference API - Pro Inference API
- Deployed Endpoint - Deployed Endpoint
- Coversational - Coversational
- Free Inference API - Free Inference API
- Deployed Endpoint - Deployed Endpoint
- Neither TGI or Coversational - Neither TGI or Coversational
- Free Inference API - Free Inference API
- Deployed Endpoint - Deployed Endpoint
""" """
@ -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():