forked from phoenix/litellm-mirror
Added deepseek completion test
This commit is contained in:
parent
9162f9c2c5
commit
82a4c68e60
1 changed files with 23 additions and 9 deletions
|
@ -1061,16 +1061,16 @@ def test_completion_perplexity_api_2():
|
|||
######### HUGGING FACE TESTS ########################
|
||||
#####################################################
|
||||
"""
|
||||
HF Tests we should pass
|
||||
- TGI:
|
||||
- Pro Inference API
|
||||
- Deployed Endpoint
|
||||
- Coversational
|
||||
- Free Inference API
|
||||
- Deployed Endpoint
|
||||
HF Tests we should pass
|
||||
- TGI:
|
||||
- Pro Inference API
|
||||
- Deployed Endpoint
|
||||
- Coversational
|
||||
- Free Inference API
|
||||
- Deployed Endpoint
|
||||
- Neither TGI or Coversational
|
||||
- Free Inference API
|
||||
- Deployed Endpoint
|
||||
- Free Inference API
|
||||
- Deployed Endpoint
|
||||
"""
|
||||
|
||||
|
||||
|
@ -3016,6 +3016,20 @@ async def test_acompletion_gemini():
|
|||
else:
|
||||
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
|
||||
def test_completion_palm():
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue