aljaz/Auto-GPT
1
0
Fork 0
mirror of https://github.com/aljazceru/Auto-GPT.git synced 2025-12-22 00:14:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

handle API timeouts (#3024)

Browse Source
This commit is contained in:
Soheil Sam Yasrebi
2023-04-24 00:26:14 -07:00
committed by GitHub
parent 89125376ba
commit 794a164098
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
autogpt/llm_utils.py
View File

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