Retry 503 OpenAI errors

Signed-off-by: Merwane Hamadi <merwanehamadi@gmail.com>
Significant-Gravitas · Jun 19, 2023 · 1fead30 · 1fead30
1 parent abb397e
commit 1fead30
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/autogpt/llm/utils/__init__.py b/autogpt/llm/utils/__init__.py
@@ -97,7 +97,9 @@ 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)