mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-08 03:08:45 +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
Reference in a new issue