phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix: fix linting

Browse source
This commit is contained in:
Krrish Dholakia 2024-05-10 20:23:33 -07:00
parent cd1a470b9a
commit c12af219af
2 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
.circleci/config.yml
View file

@ -80,7 +80,7 @@ jobs:
name: Linting Testing name: Linting Testing
command: | command: |
cd litellm 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 if ! python -m mypy . --ignore-missing-imports; then
echo "mypy detected errors" echo "mypy detected errors"
exit 1 exit 1

3
litellm/llms/vertex_ai.py
View file

@ -417,10 +417,9 @@ def completion(
from google.cloud import aiplatform # type: ignore from google.cloud import aiplatform # type: ignore
from google.protobuf import json_format # type: ignore from google.protobuf import json_format # type: ignore
from google.protobuf.struct_pb2 import Value # 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 from google.cloud.aiplatform_v1beta1.types import content as gapic_content_types # type: ignore
import google.auth # 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 ## Load credentials with the correct quota project ref: https://github.com/googleapis/python-aiplatform/issues/2557#issuecomment-1709284744
print_verbose( print_verbose(