mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-27 03:34:10 +00:00
test: mark flaky tests
This commit is contained in:
parent
a260afb74d
commit
2486a106f4
2 changed files with 7 additions and 0 deletions
|
@ -311,6 +311,9 @@ def exception_type( # type: ignore # noqa: PLR0915
|
|||
elif (
|
||||
"invalid_request_error" in error_str
|
||||
and "content_policy_violation" in error_str
|
||||
) or (
|
||||
"Invalid prompt" in error_str
|
||||
and "violating our usage policy" in error_str
|
||||
):
|
||||
exception_mapping_worked = True
|
||||
raise ContentPolicyViolationError(
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue