retry on APIConnectionError

This commit is contained in:
Paul Gauthier 2023-07-26 07:40:36 -03:00
parent 86309f336c
commit 3e4b4d1b0d

View file

@ -4,7 +4,13 @@ import json
import backoff
import openai
import requests
from openai.error import APIError, RateLimitError, ServiceUnavailableError, Timeout
from openai.error import (
APIConnectionError,
APIError,
RateLimitError,
ServiceUnavailableError,
Timeout,
)
@backoff.on_exception(
@ -14,6 +20,7 @@ from openai.error import APIError, RateLimitError, ServiceUnavailableError, Time
APIError,
ServiceUnavailableError,
RateLimitError,
APIConnectionError,
requests.exceptions.ConnectionError,
),
max_tries=10,