bias-test-gpt-pairs

Sleeping

Rafal commited on Dec 5, 2023

Commit

a4cea1f

1 Parent(s): 2a7e3b8

Updated OpenAI error to handle new version of OpenAI

Files changed (1) hide show

openAI_manager.py CHANGED Viewed

@@ -48,9 +48,9 @@ Other attribute terms in this context are: {att_terms_str}. Use them for interpr
   #print(f"Instruction: {instruction}")
   # https://github.com/openai/openai-cookbook/blob/main/examples/How_to_handle_rate_limits.ipynb
-  @backoff.on_exception(backoff.expo, (openai.error.RateLimitError,
-                                       openai.error.APIError,
-                                       openai.error.ServiceUnavailableError,
                                     ConnectionResetError,
                                     json.decoder.JSONDecodeError))#,
                                     #max_time=300,

   #print(f"Instruction: {instruction}")
   # https://github.com/openai/openai-cookbook/blob/main/examples/How_to_handle_rate_limits.ipynb
+  @backoff.on_exception(backoff.expo, (openai.RateLimitError,
+                                       openai.APIError,
+                                       openai.ServiceUnavailableError,
                                     ConnectionResetError,
                                     json.decoder.JSONDecodeError))#,
                                     #max_time=300,