Skip to content

Commit c9c7858

Browse files
authored
add litellm error to the default retry exceptions (#291)
* add litellm error to the default retry exceptions * format * add * format
1 parent 09088fd commit c9c7858

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

eval_protocol/pytest/exception_config.py

Lines changed: 7 additions & 1 deletion
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,6 +28,12 @@
2828
httpx.TimeoutException,
2929
httpx.NetworkError,
3030
httpx.RemoteProtocolError,
31+
litellm.exceptions.RateLimitError,
32+
litellm.exceptions.InternalServerError,
33+
litellm.exceptions.Timeout,
34+
litellm.exceptions.NotFoundError,
35+
litellm.exceptions.BadRequestError, # remove this once we have a long term solution
36+
litellm.exceptions.ServiceUnavailableError,
3137
}
3238

3339

0 commit comments

Comments
 (0)