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

test(test_router.py): fix router test

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-09 11:08:35 +05:30
parent 83835f10f6
commit 11ff324ff7
2 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/tests/test_router.py
View file

@ -456,6 +456,9 @@ async def test_aimg_gen_on_router():
router.reset() router.reset()
except Exception as e: except Exception as e:
if "Your task failed as a result of our safety system." in str(e):
pass
else:
traceback.print_exc() traceback.print_exc()
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")

2
litellm/tests/test_text_completion.py
View file

@ -2756,7 +2756,7 @@ def test_completion_chatgpt_prompt():
def test_text_completion_basic(): def test_text_completion_basic():
try: try:
print("\n test 003 with echo and logprobs \n") print("\n test 003 with logprobs \n")
litellm.set_verbose = False litellm.set_verbose = False
response = text_completion( response = text_completion(
model="gpt-3.5-turbo-instruct", model="gpt-3.5-turbo-instruct",