From 1c9169e563ce31132f40e73529683960bd7dd1f9 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 8 Sep 2023 15:26:09 -0700 Subject: [PATCH] raise cohere, vertex, tenacity import errors --- litellm/main.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/litellm/main.py b/litellm/main.py index ad2e4b99a..008d37362 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -484,7 +484,7 @@ def completion( try: import cohere except: - Exception("Cohere import failed please run `pip install cohere`") + raise Exception("Cohere import failed please run `pip install cohere`") cohere_key = ( api_key @@ -587,7 +587,7 @@ def completion( try: import vertexai except: - Exception("vertexai import failed please run `pip install google-cloud-aiplatform`") + raise Exception("vertexai import failed please run `pip install google-cloud-aiplatform`") from vertexai.preview.language_models import ChatModel, InputOutputTextPair vertexai.init( @@ -618,7 +618,7 @@ def completion( try: import vertexai except: - Exception("vertexai import failed please run `pip install google-cloud-aiplatform`") + raise Exception("vertexai import failed please run `pip install google-cloud-aiplatform`") from vertexai.language_models import TextGenerationModel vertexai.init( @@ -835,7 +835,7 @@ def completion_with_retries(*args, **kwargs): try: import tenacity except: - Exception("tenacity import failed please run `pip install tenacity`") + raise Exception("tenacity import failed please run `pip install tenacity`") retryer = tenacity.Retrying(stop=tenacity.stop_after_attempt(3), reraise=True) return retryer(completion, *args, **kwargs)