dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-20 04:14:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: Add Vertex AI model package check for google-cloud-aiplatform

Browse source
This commit is contained in:
Paul Gauthier (aider) 2025-03-05 16:57:51 -08:00
parent dc02daecee
commit 67b12d4416
1 changed files with 11 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

11
aider/models.py
View file

@ -777,6 +777,17 @@ def sanity_check_model(io, model):
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
if model.name.startswith("vertex_ai/"):
from aider.utils import check_pip_install_extra
check_pip_install_extra(
io,
"google.cloud.aiplatform",
"Google Vertex AI models require the google-cloud-aiplatform package.",
["google-cloud-aiplatform"]
)
if not model.info:
show = True