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): def check_for_dependencies(io, model_name):
""" """
Check for model-specific dependencies and install them if needed. Check for model-specific dependencies and install them if needed.
Args: Args:
io: The IO object for user interaction io: The IO object for user interaction
model_name: The name of the model to check dependencies for 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( 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
elif model_name.startswith("vertex_ai/"): elif model_name.startswith("vertex_ai/"):
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"],
) )