mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-27 03:34:10 +00:00
test: handle unstable tests
This commit is contained in:
parent
4e38940694
commit
89c8dfc644
2 changed files with 10 additions and 5 deletions
|
@ -730,6 +730,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.InternalServerError:
|
||||
pytest.skip("Model is overloaded")
|
||||
except litellm.RateLimitError:
|
||||
pass
|
||||
except Exception as e:
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue