fix: catch google-auth wrapped errors

PiperOrigin-RevId: 905215860

Author: yinghsienwu

google/genai/_api_client.py

@@ -46,6 +46,7 @@
 from google.auth.credentials import Credentials
 from google.auth.transport import mtls
 from google.auth.transport.requests import AuthorizedSession
+from google.auth import exceptions as auth_exceptions
 import httpx
 from pydantic import BaseModel
 from pydantic import ValidationError
@@ -1402,6 +1403,7 @@ async def _async_request_once(
             aiohttp.ClientConnectorDNSError,
             aiohttp.ClientOSError,
             aiohttp.ServerDisconnectedError,
+            auth_exceptions.TransportError,
         ) as e:
           await asyncio.sleep(1 + random.randint(0, 9))
           logger.info('Retrying due to aiohttp error: %s' % e)
@@ -1479,6 +1481,7 @@ async def _async_request_once(
             aiohttp.ClientConnectorDNSError,
             aiohttp.ClientOSError,
             aiohttp.ServerDisconnectedError,
+            auth_exceptions.TransportError,
         ) as e:
           await asyncio.sleep(1 + random.randint(0, 9))
           logger.info('Retrying due to aiohttp error: %s' % e)

google/genai/tests/client/test_client_close.py

@@ -153,10 +153,19 @@ async def run():
           )
       )
       assert async_client._api_client._aiohttp_session is not None
-      assert not async_client._api_client._aiohttp_session.closed
+      if hasattr(async_client._api_client._aiohttp_session, 'closed'):
+        assert not async_client._api_client._aiohttp_session.closed
       # Close the client and check that the session is closed.
       await async_client.aclose()
-      assert async_client._api_client._aiohttp_session.closed
+      if hasattr(async_client._api_client._aiohttp_session, 'closed'):
+        assert async_client._api_client._aiohttp_session.closed
+      else:
+        from google.auth.aio.transport.sessions import AsyncAuthorizedSession
+
+        if isinstance(
+            async_client._api_client._aiohttp_session, AsyncAuthorizedSession
+        ):
+          assert async_client._api_client._aiohttp_session._auth_request._closed
 
   asyncio.run(run())
 
@@ -197,8 +206,17 @@ async def run():
             )
         )
         assert async_client._api_client._aiohttp_session is not None
-        assert not async_client._api_client._aiohttp_session.closed
-
-      assert async_client._api_client._aiohttp_session.closed
+        if hasattr(async_client._api_client._aiohttp_session, 'closed'):
+          assert not async_client._api_client._aiohttp_session.closed
+
+      if hasattr(async_client._api_client._aiohttp_session, 'closed'):
+        assert async_client._api_client._aiohttp_session.closed
+      else:
+        from google.auth.aio.transport.sessions import AsyncAuthorizedSession
+
+        if isinstance(
+            async_client._api_client._aiohttp_session, AsyncAuthorizedSession
+        ):
+          assert async_client._api_client._aiohttp_session._auth_request._closed
 
   asyncio.run(run())