feat: Add "eu" as a supported service location for Vertex AI platform.

google-genai-bot · copybara-github · commit 287733a0a50c · 2026-04-08T16:19:37.000-07:00
PiperOrigin-RevId: 896680569
diff --git a/google/genai/_api_client.py b/google/genai/_api_client.py
@@ -91,6 +91,8 @@
 INITIAL_RETRY_DELAY = 1  # second
 DELAY_MULTIPLIER = 2
 
+_MULTI_REGIONAL_LOCATIONS = {'us', 'eu'}
+
 
 class EphemeralTokenAPIKeyError(ValueError):
   """Error raised when the API key is invalid."""
@@ -702,7 +704,10 @@ def __init__(
           self.api_key or self.location == 'global'
       ) and not self.custom_base_url:
         self._http_options.base_url = f'https://aiplatform.googleapis.com/'
-      elif self.location == 'us' and not self.custom_base_url:
+      elif (
+          self.location in _MULTI_REGIONAL_LOCATIONS
+          and not self.custom_base_url
+      ):
         self._http_options.base_url = (
             f'https://aiplatform.{self.location}.rep.googleapis.com/'
         )
diff --git a/google/genai/tests/client/test_client_initialization.py b/google/genai/tests/client/test_client_initialization.py
@@ -532,6 +532,22 @@ def test_vertexai_location_us_routing(monkeypatch):
     )
 
 
+def test_vertexai_location_eu_routing(monkeypatch):
+  # Verify that location='eu' correctly routes to the eu.rep endpoint
+  project_id = "fake_project_id"
+  location = "eu"
+
+  with monkeypatch.context() as m:
+    m.delenv("GOOGLE_CLOUD_LOCATION", raising=False)
+    client = Client(vertexai=True, project=project_id, location=location)
+    assert client.models._api_client.location == location
+    assert client.models._api_client.project == project_id
+    assert (
+        client.models._api_client.get_read_only_http_options()["base_url"]
+        == "https://aiplatform.eu.rep.googleapis.com/"
+    )
+
+
 def test_vertexai_location_us_routing_base_url_override(monkeypatch):
   # Verify that base_url override takes precedence over location='us' routing
   project_id = "fake_project_id"
