Retry 503 OpenAI errors

Signed-off-by: Merwane Hamadi <merwanehamadi@gmail.com>
Significant-Gravitas · Jun 19, 2023 · c5514df · c5514df
1 parent abb397e
commit c5514df
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/autogpt/llm/utils/__init__.py b/autogpt/llm/utils/__init__.py
@@ -97,7 +97,7 @@ def _wrapped(*args, **kwargs):
                         user_warned = True
 
                 except APIError as e:
-                    if (e.http_status not in [502, 429]) or (attempt == num_attempts):
+                    if (e.http_status not in [429, 502, 503]) or (attempt == num_attempts):
                         raise
 
                 backoff = backoff_base ** (attempt + 2)