Skip to content

Commit 77265af

Browse files
committed
add litellm error to the default retry exceptions
1 parent a30aed9 commit 77265af

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

eval_protocol/pytest/exception_config.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99
import backoff
1010

11-
11+
import litellm
1212
import requests
1313
import httpx
1414

@@ -28,8 +28,14 @@
2828
httpx.TimeoutException,
2929
httpx.NetworkError,
3030
httpx.RemoteProtocolError,
31-
}
3231

32+
litellm.exceptions.RateLimitError,
33+
litellm.exceptions.InternalServerError,
34+
litellm.exceptions.Timeout,
35+
litellm.exceptions.NotFoundError,
36+
litellm.exceptions.BadRequestError,
37+
litellm.exceptions.ServiceUnavailableError,
38+
}
3339

3440
@dataclass
3541
class BackoffConfig:

0 commit comments

Comments
 (0)