Set 'us-central1' by default on GoogleProvider (#2031)

Author: Kludex

pydantic_ai_slim/pydantic_ai/providers/google.py

@@ -104,7 +104,12 @@ def __init__(
                 self._client = genai.Client(
                     vertexai=vertexai,
                     project=project or os.environ.get('GOOGLE_CLOUD_PROJECT'),
-                    location=location or os.environ.get('GOOGLE_CLOUD_LOCATION'),
+                    # From https://github.com/pydantic/pydantic-ai/pull/2031/files#r2169682149:
+                    # Currently `us-central1` supports the most models by far of any region including `global`, but not
+                    # all of them. `us-central1` has all google models but is missing some Anthropic partner models,
+                    # which use `us-east5` instead. `global` has fewer models but higher availability.
+                    # For more details, check: https://cloud.google.com/vertex-ai/generative-ai/docs/learn/locations#available-regions
+                    location=location or os.environ.get('GOOGLE_CLOUD_LOCATION') or 'us-central1',
                     credentials=credentials,
                     http_options={'headers': {'User-Agent': get_user_agent()}},
                 )

tests/conftest.py

@@ -8,6 +8,7 @@
 import sys
 from collections.abc import AsyncIterator, Iterator
 from contextlib import contextmanager
+from dataclasses import dataclass
 from datetime import datetime
 from pathlib import Path
 from types import ModuleType
@@ -175,8 +176,8 @@ def check_import() -> bool:
         import_success = True
 
 
-@pytest.fixture(autouse=True)
-def set_event_loop(anyio_backend: str) -> Iterator[None]:
+@pytest.fixture(scope='session', autouse=True)
+def event_loop() -> Iterator[None]:
     new_loop = asyncio.new_event_loop()
     asyncio.set_event_loop(new_loop)
     yield
@@ -315,6 +316,55 @@ def bedrock_provider():
         pytest.skip('boto3 is not installed')
 
 
+@pytest.fixture(autouse=True)
+def vertex_provider_auth(mocker: MockerFixture) -> None:  # pragma: lax no cover
+    # Locally, we authenticate via `gcloud` CLI, so we don't need to patch anything.
+    if not os.getenv('CI', False):
+        return  # pragma: lax no cover
+
+    try:
+        from google.genai import _api_client
+    except ImportError:
+        pytest.skip('google is not installed')
+
+    @dataclass
+    class NoOpCredentials:
+        token = 'my-token'
+        quota_project_id = 'pydantic-ai'
+
+        def refresh(self, request: httpx.Request): ...
+
+        def expired(self) -> bool:
+            return False
+
+    return_value = (NoOpCredentials(), 'pydantic-ai')
+    mocker.patch.object(_api_client, '_load_auth', return_value=return_value)
+
+
+@pytest.fixture()
+async def vertex_provider():
+    # NOTE: You need to comment out this line to rewrite the cassettes locally.
+    if not os.getenv('CI', False):  # pragma: lax no cover
+        pytest.skip('Requires properly configured local google vertex config to pass')
+
+    try:
+        from google import genai
+
+        from pydantic_ai.providers.google import GoogleProvider
+    except ImportError:  # pragma: lax no cover
+        pytest.skip('google is not installed')
+
+    project = os.getenv('GOOGLE_PROJECT', 'pydantic-ai')
+    location = os.getenv('GOOGLE_LOCATION', 'us-central1')
+    client = genai.Client(vertexai=True, project=project, location=location)
+
+    try:
+        yield GoogleProvider(client=client)
+    finally:
+        client.aio._api_client._httpx_client.close()  # type: ignore
+        await client.aio._api_client._async_httpx_client.aclose()  # type: ignore
+
+
 @pytest.fixture()
 def model(
     request: pytest.FixtureRequest,

tests/models/cassettes/test_google/test_google_model.yaml

@@ -30,11 +30,11 @@ interactions:
       alt-svc:
       - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
       content-length:
-      - '644'
+      - '687'
       content-type:
       - application/json; charset=UTF-8
       server-timing:
-      - gfet4t7; dur=322
+      - gfet4t7; dur=531
       transfer-encoding:
       - chunked
       vary:
@@ -43,14 +43,15 @@ interactions:
       - Referer
     parsed_body:
       candidates:
-      - avgLogprobs: -0.0009223055941137401
+      - avgLogprobs: -0.0007684422995556484
         content:
           parts:
           - text: |
               Hello there! How can I help you today?
           role: model
         finishReason: STOP
       modelVersion: gemini-1.5-flash
+      responseId: wVpeaPOdAaXTnvgPlIeSoQg
       usageMetadata:
         candidatesTokenCount: 11
         candidatesTokensDetails:

tests/models/cassettes/test_google/test_google_model_document_url_input.yaml

@@ -252,7 +252,7 @@ interactions:
       accept-ranges:
       - bytes
       age:
-      - '427557'
+      - '264068'
       alt-svc:
       - h3=":443"; ma=86400
       cache-control:
@@ -268,7 +268,7 @@ interactions:
       etag:
       - '"33d0-438b181451e00"'
       expires:
-      - Tue, 13 May 2025 21:19:22 GMT
+      - Tue, 24 Jun 2025 13:27:15 GMT
       last-modified:
       - Mon, 27 Aug 2007 17:15:36 GMT
       strict-transport-security:
@@ -312,11 +312,11 @@ interactions:
       alt-svc:
       - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
       content-length:
-      - '751'
+      - '776'
       content-type:
       - application/json; charset=UTF-8
       server-timing:
-      - gfet4t7; dur=1704
+      - gfet4t7; dur=1228
       transfer-encoding:
       - chunked
       vary:
@@ -325,26 +325,27 @@ interactions:
       - Referer
     parsed_body:
       candidates:
-      - avgLogprobs: -0.27702528635660806
+      - avgLogprobs: -0.28572704394658405
         content:
           parts:
           - text: |
-              The main content of the document is the phrase "Dummy PDF file".
+              The document appears to be a "Dummy PDF file".
           role: model
         finishReason: STOP
       modelVersion: gemini-2.0-flash
+      responseId: 4FpeaJWYOLq3nvgP0vasuQk
       usageMetadata:
-        candidatesTokenCount: 15
+        candidatesTokenCount: 12
         candidatesTokensDetails:
         - modality: TEXT
-          tokenCount: 15
+          tokenCount: 12
         promptTokenCount: 1305
         promptTokensDetails:
         - modality: TEXT
           tokenCount: 15
         - modality: DOCUMENT
           tokenCount: 1290
-        totalTokenCount: 1320
+        totalTokenCount: 1317
     status:
       code: 200
       message: OK

tests/models/cassettes/test_google/test_google_model_empty_user_prompt.yaml

@@ -30,11 +30,11 @@ interactions:
       alt-svc:
       - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
       content-length:
-      - '736'
+      - '701'
       content-type:
       - application/json; charset=UTF-8
       server-timing:
-      - gfet4t7; dur=361
+      - gfet4t7; dur=444
       transfer-encoding:
       - chunked
       vary:
@@ -43,25 +43,25 @@ interactions:
       - Referer
     parsed_body:
       candidates:
-      - avgLogprobs: -0.20726837430681502
+      - avgLogprobs: -0.11437161529765409
         content:
           parts:
           - text: |
-              Please provide me with a question or task. I need some information to be able to help you.
+              I'm ready to assist you.  Please tell me what you need.
           role: model
         finishReason: STOP
       modelVersion: gemini-1.5-flash
-      responseId: y5BBaPCQELGqmecP68OBiAc
+      responseId: 5lpeaN7wJ___698Pv8HGgAg
       usageMetadata:
-        candidatesTokenCount: 21
+        candidatesTokenCount: 17
         candidatesTokensDetails:
         - modality: TEXT
-          tokenCount: 21
+          tokenCount: 17
         promptTokenCount: 7
         promptTokensDetails:
         - modality: TEXT
           tokenCount: 7
-        totalTokenCount: 28
+        totalTokenCount: 24
     status:
       code: 200
       message: OK

tests/models/cassettes/test_google/test_google_model_image_as_binary_content_input.yaml

@@ -33,11 +33,11 @@ interactions:
       alt-svc:
       - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
       content-length:
-      - '713'
+      - '756'
       content-type:
       - application/json; charset=UTF-8
       server-timing:
-      - gfet4t7; dur=4099
+      - gfet4t7; dur=2659
       transfer-encoding:
       - chunked
       vary:
@@ -53,17 +53,18 @@ interactions:
           role: model
         finishReason: STOP
       modelVersion: gemini-2.0-flash
+      responseId: 2VpeaPm3DaHp1PIPwK-EmAM
       usageMetadata:
         candidatesTokenCount: 9
         candidatesTokensDetails:
         - modality: TEXT
           tokenCount: 9
         promptTokenCount: 3367
         promptTokensDetails:
-        - modality: IMAGE
-          tokenCount: 3354
         - modality: TEXT
           tokenCount: 13
+        - modality: IMAGE
+          tokenCount: 3354
         totalTokenCount: 3376
     status:
       code: 200

tests/models/cassettes/test_google/test_google_model_image_url_input.yaml

@@ -580,7 +580,7 @@ interactions:
       access-control-allow-origin:
       - '*'
       age:
-      - '432185'
+      - '1386476'
       cache-control:
       - public, max-age=31536000
       connection:
@@ -632,11 +632,11 @@ interactions:
       alt-svc:
       - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
       content-length:
-      - '696'
+      - '740'
       content-type:
       - application/json; charset=UTF-8
       server-timing:
-      - gfet4t7; dur=979
+      - gfet4t7; dur=1424
       transfer-encoding:
       - chunked
       vary:
@@ -645,26 +645,27 @@ interactions:
       - Referer
     parsed_body:
       candidates:
-      - avgLogprobs: -0.2194784709385463
+      - avgLogprobs: -0.1821905771891276
         content:
           parts:
           - text: |
-              That's a potato!
+              That is a potato.
           role: model
         finishReason: STOP
       modelVersion: gemini-2.0-flash
+      responseId: 3VpeaPexBLq3nvgP0vasuQk
       usageMetadata:
-        candidatesTokenCount: 7
+        candidatesTokenCount: 6
         candidatesTokensDetails:
         - modality: TEXT
-          tokenCount: 7
+          tokenCount: 6
         promptTokenCount: 1817
         promptTokensDetails:
         - modality: TEXT
           tokenCount: 11
         - modality: IMAGE
           tokenCount: 1806
-        totalTokenCount: 1824
+        totalTokenCount: 1823
     status:
       code: 200
       message: OK

tests/models/cassettes/test_google/test_google_model_instructions.yaml

@@ -30,11 +30,11 @@ interactions:
       alt-svc:
       - h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
       content-length:
-      - '637'
+      - '680'
       content-type:
       - application/json; charset=UTF-8
       server-timing:
-      - gfet4t7; dur=449
+      - gfet4t7; dur=444
       transfer-encoding:
       - chunked
       vary:
@@ -43,14 +43,15 @@ interactions:
       - Referer
     parsed_body:
       candidates:
-      - avgLogprobs: -0.00021161907352507114
+      - avgLogprobs: -0.0001856483577284962
         content:
           parts:
           - text: |
               The capital of France is Paris.
           role: model
         finishReason: STOP
       modelVersion: gemini-2.0-flash
+      responseId: 41peaK-wOMSenvgPh-vRiAY
       usageMetadata:
         candidatesTokenCount: 8
         candidatesTokensDetails: