diff --git a/google/genai/live.py b/google/genai/live.py index 0d8453fc5..26dd0a697 100644 --- a/google/genai/live.py +++ b/google/genai/live.py @@ -26,6 +26,7 @@ import google.auth import pydantic +from google.auth.transport import requests from websockets import ConnectionClosed from . import _api_module @@ -1027,7 +1028,7 @@ async def connect( # creds.valid is False, and creds.token is None # Need to refresh credentials to populate those if not (creds.token and creds.valid): - auth_req = google.auth.transport.requests.Request() # type: ignore + auth_req = requests.Request() # type: ignore creds.refresh(auth_req) bearer_token = creds.token @@ -1162,4 +1163,4 @@ async def _t_live_connect_config( stacklevel=4, ) - return parameter_model_copy + return parameter_model_copy \ No newline at end of file