test: handle watsonx rate limit error

This commit is contained in:
Krrish Dholakia 2024-05-13 18:27:39 -07:00
parent f8e1b1db2e
commit 87a21115c5
4 changed files with 9 additions and 1 deletions

View file

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