style: Fix linter warnings in models.py

This commit is contained in:
Paul Gauthier (aider) 2025-03-05 16:57:56 -08:00
parent 67b12d4416
commit bcc8b1917a

View file

@ -777,16 +777,16 @@ def sanity_check_model(io, model):
check_pip_install_extra( check_pip_install_extra(
io, "boto3", "AWS Bedrock models require the boto3 package.", ["boto3"] io, "boto3", "AWS Bedrock models require the boto3 package.", ["boto3"]
) )
# Check if this is a Vertex AI model and ensure google-cloud-aiplatform is installed # Check if this is a Vertex AI model and ensure google-cloud-aiplatform is installed
if model.name.startswith("vertex_ai/"): if model.name.startswith("vertex_ai/"):
from aider.utils import check_pip_install_extra from aider.utils import check_pip_install_extra
check_pip_install_extra( check_pip_install_extra(
io, io,
"google.cloud.aiplatform", "google.cloud.aiplatform",
"Google Vertex AI models require the google-cloud-aiplatform package.", "Google Vertex AI models require the google-cloud-aiplatform package.",
["google-cloud-aiplatform"] ["google-cloud-aiplatform"],
) )
if not model.info: if not model.info: