From f3efd566c9935dc732e4289114f3bd7be20f04e5 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 31 Oct 2023 19:23:14 -0700 Subject: [PATCH] style(main.py): fix linting issues --- litellm/main.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/litellm/main.py b/litellm/main.py index f2e1f7e2998..e3df4f80892 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -1332,10 +1332,9 @@ def completion( ) except Exception as e: if num_retries: - if (isinstance(e, openai.APIError) - or isinstance(e, openai.Timeout) - or isinstance(e, openai.Timeout) - or isinstance(e, openai.ServiceUnavailableError)): + if (isinstance(e, openai.error.APIError) + or isinstance(e, openai.error.Timeout) + or isinstance(e, openai.error.ServiceUnavailableError)): return completion_with_retries(num_retries=num_retries, **args) else: raise e