style(main.py): fix linting issues

This commit is contained in:
Krrish Dholakia 2023-10-31 19:23:14 -07:00
parent 125642563c
commit f3efd566c9

View file

@ -1332,10 +1332,9 @@ def completion(
) )
except Exception as e: except Exception as e:
if num_retries: if num_retries:
if (isinstance(e, openai.APIError) if (isinstance(e, openai.error.APIError)
or isinstance(e, openai.Timeout) or isinstance(e, openai.error.Timeout)
or isinstance(e, openai.Timeout) or isinstance(e, openai.error.ServiceUnavailableError)):
or isinstance(e, openai.ServiceUnavailableError)):
return completion_with_retries(num_retries=num_retries, **args) return completion_with_retries(num_retries=num_retries, **args)
else: else:
raise e raise e