Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 3 additions & 2 deletions google/genai/live.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@

import google.auth
import pydantic
from google.auth.transport import requests
from websockets import ConnectionClosed

from . import _api_module
Expand Down Expand Up @@ -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

Expand Down Expand Up @@ -1162,4 +1163,4 @@ async def _t_live_connect_config(
stacklevel=4,
)

return parameter_model_copy
return parameter_model_copy