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 backoff
import openai import openai
import requests import requests
from openai.error import APIError, RateLimitError, ServiceUnavailableError, Timeout from openai.error import (
APIConnectionError,
APIError,
RateLimitError,
ServiceUnavailableError,
Timeout,
)
@backoff.on_exception( @backoff.on_exception(
@ -14,6 +20,7 @@ from openai.error import APIError, RateLimitError, ServiceUnavailableError, Time
APIError, APIError,
ServiceUnavailableError, ServiceUnavailableError,
RateLimitError, RateLimitError,
APIConnectionError,
requests.exceptions.ConnectionError, requests.exceptions.ConnectionError,
), ),
max_tries=10, max_tries=10,