Add complete list of errors to MakeMeSay utils (#1436)

This closes issue #1432. The only meaningful change is the list of errors; other modified lines are from `pre-commit`.
openai · Dec 20, 2023 · 4fcbed2 · 4fcbed2
1 parent f4dc762
commit 4fcbed2
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/evals/elsuite/make_me_say/utils.py b/evals/elsuite/make_me_say/utils.py
@@ -1,13 +1,11 @@
-import os
 import functools
+import os
 from typing import Callable, Union
 
 import backoff
-import openai
-from openai import OpenAI
-
 import openai
 import urllib3.exceptions
+from openai import OpenAI
 
 from evals.api import CompletionResult
 
@@ -16,6 +14,9 @@
  backoff.expo,
  (
  openai.RateLimitError,
+ openai.APIConnectionError,
+ openai.APITimeoutError,
+ openai.InternalServerError,
  urllib3.exceptions.TimeoutError,
  ),
 )