diff --git a/.circleci/config.yml b/.circleci/config.yml index 910f9804d..35707dbff 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -80,7 +80,7 @@ jobs: name: Linting Testing command: | cd litellm - python -m pip install types-requests types-setuptools types-redis types-PyYAML grpcio + python -m pip install types-requests types-setuptools types-redis types-PyYAML if ! python -m mypy . --ignore-missing-imports; then echo "mypy detected errors" exit 1 diff --git a/litellm/llms/vertex_ai.py b/litellm/llms/vertex_ai.py index cc4db1f9d..a61c07df0 100644 --- a/litellm/llms/vertex_ai.py +++ b/litellm/llms/vertex_ai.py @@ -417,10 +417,9 @@ def completion( from google.cloud import aiplatform # type: ignore from google.protobuf import json_format # type: ignore from google.protobuf.struct_pb2 import Value # type: ignore - from google.protobuf.json_format import MessageToDict from google.cloud.aiplatform_v1beta1.types import content as gapic_content_types # type: ignore import google.auth # type: ignore - import proto + import proto # type: ignore ## Load credentials with the correct quota project ref: https://github.com/googleapis/python-aiplatform/issues/2557#issuecomment-1709284744 print_verbose(