phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(test) native perplexity ai api test

Browse source
This commit is contained in:
ishaan-jaff 2023-10-20 14:29:05 -07:00
parent 61dd4f167f
commit 66206aea3a
1 changed files with 25 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

27
litellm/tests/test_completion.py
View file

@ -110,7 +110,7 @@ def test_completion_with_litellm_call_id():
def test_completion_perplexity_api(): def test_completion_perplexity_api():
try: try:
litellm.set_verbose=True # litellm.set_verbose=True
messages=[{ messages=[{
"role": "system", "role": "system",
"content": "You're a good bot" "content": "You're a good bot"
@ -129,7 +129,30 @@ def test_completion_perplexity_api():
except Exception as e: except Exception as e:
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
# test_completion_perplexity_api() test_completion_perplexity_api()
def test_completion_perplexity_api_2():
try:
# litellm.set_verbose=True
messages=[{
"role": "system",
"content": "You're a good bot"
},{
"role": "user",
"content": "Hey",
},{
"role": "user",
"content": "Hey",
}]
response = completion(
model="perplexity/mistral-7b-instruct",
messages=messages
)
print(response)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_completion_perplexity_api_2()
# commenting out as this is a flaky test on circle ci # commenting out as this is a flaky test on circle ci
# def test_completion_nlp_cloud(): # def test_completion_nlp_cloud():
# try: # try: