The wrong URLs don't exist and return an HTML error page. Actual: Request sent to https://us-aiplatform.googleapis.com → HTML error page

TL;DR

Update the get_vertex_base_url function to correctly handle multi-region endpoints for Google Vertex AI.

Guidance

• Verify the issue by checking the generated URL for us and eu locations using the get_vertex_base_url function.
• Update the get_vertex_base_url function to use a dictionary-based approach to map multi-region locations to their correct URLs, as shown in the proposed fix.
• Test the updated function with different locations, including us, eu, and global, to ensure correct URL generation.
• Consider adding additional logging or error handling to detect and handle any future URL generation issues.

Example

_MULTI_REGION_URLS = {
    "us": "https://aiplatform.us.rep.googleapis.com",
    "eu": "https://aiplatform.eu.rep.googleapis.com",
}

def get_vertex_base_url(vertex_location):
    if vertex_location == "global":
        return "https://aiplatform.googleapis.com"
    if vertex_location in _MULTI_REGION_URLS:
        return _MULTI_REGION_URLS[vertex_location]
    return f"https://{vertex_location}-aiplatform.googleapis.com"

Notes

The proposed fix assumes that the us and eu locations are the only multi-region endpoints. If additional multi-region endpoints are introduced, the _MULTI_REGION_URLS dictionary will need to be updated accordingly.

Recommendation

Apply the proposed fix to update the get_vertex_base_url function, as it correctly handles multi-region endpoints and provides a scalable solution for future additions.