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

(test) try/pass APIError for TG AI

Browse source
This commit is contained in:
ishaan-jaff 2023-11-11 11:34:15 -08:00
parent 59c76db34e
commit 96bca9a836
1 changed files with 10 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

13
litellm/tests/test_completion.py
View file

@ -142,7 +142,7 @@ def test_completion_gpt4_vision():
pass pass
except Exception as e: except Exception as e:
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
test_completion_gpt4_vision() # test_completion_gpt4_vision()
def test_completion_perplexity_api(): def test_completion_perplexity_api():
try: try:
@ -876,10 +876,17 @@ def test_completion_together_ai():
def test_customprompt_together_ai(): def test_customprompt_together_ai():
try: try:
litellm.set_verbose = True litellm.set_verbose = True
response = completion(model="together_ai/OpenAssistant/llama2-70b-oasst-sft-v10", messages=messages, response = completion(
roles={"system":{"pre_message":"<|im_start|>system\n", "post_message":"<|im_end|>"}, "assistant":{"pre_message":"<|im_start|>assistant\n","post_message":"<|im_end|>"}, "user":{"pre_message":"<|im_start|>user\n","post_message":"<|im_end|>"}}) model="together_ai/OpenAssistant/llama2-70b-oasst-sft-v10",
messages=messages,
roles={"system":{"pre_message":"<|im_start|>system\n", "post_message":"<|im_end|>"}, "assistant":{"pre_message":"<|im_start|>assistant\n","post_message":"<|im_end|>"}, "user":{"pre_message":"<|im_start|>user\n","post_message":"<|im_end|>"}}
)
print(response) print(response)
except litellm.APIError as e:
pass
except Exception as e: except Exception as e:
print(type(e))
print(e)
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
# test_customprompt_together_ai() # test_customprompt_together_ai()