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

fix(utils.py): add param mapping for perplexity, anyscale, deepinfra

Browse source 
n

n
This commit is contained in:
Krrish Dholakia 2023-11-22 10:04:27 -08:00
parent e7bb4a0cbd
commit 10fe16c965
5 changed files with 141 additions and 83 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/tests/test_completion.py
View file

@ -164,7 +164,9 @@ def test_completion_gpt4_vision():
def test_completion_perplexity_api():
try:
# litellm.set_verbose=True
litellm.set_verbose=True
litellm.num_retries = 0
litellm.drop_params = True
messages=[{
"role": "system",
"content": "You're a good bot"
@ -178,12 +180,12 @@ def test_completion_perplexity_api():
response = completion(
model="mistral-7b-instruct",
messages=messages,
api_base="https://api.perplexity.ai")
api_base="https://api.perplexity.ai", stop="Hello")
print(response)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_completion_perplexity_api()
test_completion_perplexity_api()
def test_completion_perplexity_api_2():
try: