Skip to content

Commit f841334

Browse files
committed
retry on openai too
1 parent d3ef46c commit f841334

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

eval_protocol/pytest/exception_config.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
import litellm
1212
import requests
1313
import httpx
14+
import openai
1415

1516
# Default exceptions that should be retried with backoff
1617
DEFAULT_RETRYABLE_EXCEPTIONS: Set[Type[Exception]] = {
@@ -34,6 +35,8 @@
3435
litellm.exceptions.NotFoundError,
3536
litellm.exceptions.BadRequestError, # remove this once we have a long term solution
3637
litellm.exceptions.ServiceUnavailableError,
38+
openai.NotFoundError,
39+
openai.BadRequestError, # remove this once we have a long term solution
3740
}
3841

3942

0 commit comments

Comments
 (0)