feat(vertex): add support for US multi-region endpoint

Author: TomerAberbach

src/anthropic/lib/vertex/_client.py

@@ -119,6 +119,8 @@ def __init__(
             if base_url is None:
                 if region == "global":
                     base_url = "https://aiplatform.googleapis.com/v1"
+                elif region == "us":
+                    base_url = "https://aiplatform.us.rep.googleapis.com/v1"
                 else:
                     base_url = f"https://{region}-aiplatform.googleapis.com/v1"
 

tests/lib/test_vertex.py

@@ -124,6 +124,11 @@ def test_global_region_base_url(self) -> None:
         client = AnthropicVertex(region="global", project_id="test-project", access_token="fake-token")
         assert str(client.base_url).rstrip("/") == "https://aiplatform.googleapis.com/v1"
 
+    def test_us_region_base_url(self) -> None:
+        """Test that us region uses the correct base URL."""
+        client = AnthropicVertex(region="us", project_id="test-project", access_token="fake-token")
+        assert str(client.base_url).rstrip("/") == "https://aiplatform.us.rep.googleapis.com/v1"
+
     @pytest.mark.parametrize("region", ["us-central1", "europe-west1", "asia-southeast1"])
     def test_regional_base_url(self, region: str) -> None:
         """Test that regional endpoints use the correct base URL format."""