mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-04 02:31:27 +00:00
fix(vertex_ai_anthropic.py): fix import
This commit is contained in:
parent
3b2c34d587
commit
f662dac5c9
2 changed files with 4 additions and 1 deletions
|
|
@ -148,6 +148,7 @@ def completion(
|
|||
):
|
||||
try:
|
||||
import vertexai
|
||||
from anthropic import AnthropicVertex
|
||||
except:
|
||||
raise VertexAIError(
|
||||
status_code=400,
|
||||
|
|
@ -162,7 +163,6 @@ def completion(
|
|||
message="""Upgrade vertex ai. Run `pip install "google-cloud-aiplatform>=1.38"`""",
|
||||
)
|
||||
try:
|
||||
from anthropic import AnthropicVertex
|
||||
|
||||
## Load Config
|
||||
config = litellm.VertexAIAnthropicConfig.get_config()
|
||||
|
|
|
|||
|
|
@ -85,6 +85,9 @@ async def get_response():
|
|||
pytest.fail(f"An error occurred - {str(e)}")
|
||||
|
||||
|
||||
@pytest.mark.skip(
|
||||
reason="Local test. Vertex AI Quota is low. Leads to rate limit errors on ci/cd."
|
||||
)
|
||||
def test_vertex_ai_anthropic():
|
||||
load_vertex_ai_credentials()
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue