handle API timeouts (#3024)

This commit is contained in:
Soheil Sam Yasrebi
2023-04-24 00:26:14 -07:00
committed by GitHub
parent 89125376ba
commit 794a164098

View File

@@ -5,7 +5,7 @@ from typing import List, Optional
import openai
from colorama import Fore, Style
from openai.error import APIError, RateLimitError
from openai.error import APIError, RateLimitError, Timeout
from autogpt.api_manager import api_manager
from autogpt.config import Config
@@ -123,7 +123,7 @@ def create_chat_completion(
+ f"You can read more here: {Fore.CYAN}https://github.com/Significant-Gravitas/Auto-GPT#openai-api-keys-configuration{Fore.RESET}"
)
warned_user = True
except APIError as e:
except (APIError, Timeout) as e:
if e.http_status != 502:
raise
if attempt == num_retries - 1:
@@ -172,7 +172,7 @@ def create_embedding_with_ada(text) -> list:
)
except RateLimitError:
pass
except APIError as e:
except (APIError, Timeout) as e:
if e.http_status != 502:
raise
if attempt == num_retries - 1: