style: Fix linting issues in models.py

This commit is contained in:
Paul Gauthier (aider) 2025-03-05 17:06:39 -08:00
parent 74e60e98b7
commit 207a631a65

View file

@ -791,7 +791,7 @@ def sanity_check_model(io, model):
def check_for_dependencies(io, model_name):
"""
Check for model-specific dependencies and install them if needed.
Args:
io: The IO object for user interaction
model_name: The name of the model to check dependencies for
@ -803,14 +803,14 @@ def check_for_dependencies(io, model_name):
check_pip_install_extra(
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
elif model_name.startswith("vertex_ai/"):
check_pip_install_extra(
io,
"google.cloud.aiplatform",
"Google Vertex AI models require the google-cloud-aiplatform package.",
["google-cloud-aiplatform"]
io,
"google.cloud.aiplatform",
"Google Vertex AI models require the google-cloud-aiplatform package.",
["google-cloud-aiplatform"],
)