diff --git a/aider/models.py b/aider/models.py index 0c7e0ecc9..4d51bd0a9 100644 --- a/aider/models.py +++ b/aider/models.py @@ -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"], )