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

test(test_completion.py): handle async watsonx call fail

Browse source
This commit is contained in:
Krrish Dholakia 2024-05-13 18:40:51 -07:00
parent d4123951d9
commit 724d880a45
2 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/main.py
View file

@ -15,7 +15,6 @@ import dotenv, traceback, random, asyncio, time, contextvars
from copy import deepcopy from copy import deepcopy
import httpx import httpx
import litellm import litellm
from ._logging import verbose_logger from ._logging import verbose_logger
from litellm import ( # type: ignore from litellm import ( # type: ignore
client, client,

2
litellm/tests/test_completion.py
View file

@ -3427,6 +3427,8 @@ async def test_acompletion_watsonx():
) )
# Add any assertions here to check the response # Add any assertions here to check the response
print(response) print(response)
except litellm.RateLimitError as e:
pass
except Exception as e: except Exception as e:
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")