mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-24 10:14:26 +00:00
test: handle service unavailable error
This commit is contained in:
parent
620a0f4805
commit
2adb2fc6a5
1 changed files with 2 additions and 0 deletions
|
@ -947,6 +947,8 @@ class BaseLLMChatTest(ABC):
|
|||
second_response.choices[0].message.content is not None
|
||||
or second_response.choices[0].message.tool_calls is not None
|
||||
)
|
||||
except litellm.ServiceUnavailableError:
|
||||
pytest.skip("Model is overloaded")
|
||||
except litellm.InternalServerError:
|
||||
pytest.skip("Model is overloaded")
|
||||
except litellm.RateLimitError:
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue