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 ########################
|
######### 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():
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue