fix raise better error message on reaching failed vertex import

2024-06-27 12:02:19 -07:00 · 2024-06-27 12:02:19 -07:00 · 80d8bf5d8f
commit 80d8bf5d8f
parent 9d50fc1f2a
1 changed files with 1 additions and 1 deletions
--- a/litellm/llms/vertex_ai.py
+++ b/litellm/llms/vertex_ai.py
@ -437,7 +437,7 @@ def completion(
    except:
        raise VertexAIError(
            status_code=400,
-            message="vertexai import failed please run `pip install google-cloud-aiplatform`",
+            message="vertexai import failed please run `pip install google-cloud-aiplatform`. This is required for the 'vertex_ai/' route on LiteLLM",
        )

    if not (