phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +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

7
litellm/tests/test_router.py
View file

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

2
litellm/tests/test_text_completion.py
View file

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