From 37fa3ce353ffe34195828ba383a6575c0dbb955d Mon Sep 17 00:00:00 2001 From: Tongzhou Jiang Date: Tue, 4 Nov 2025 17:50:01 -0800 Subject: [PATCH] fix: populate missing auth_id for _init_session into adk session.state PiperOrigin-RevId: 828207341 --- vertexai/agent_engines/templates/adk.py | 2 +- vertexai/preview/reasoning_engines/templates/adk.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/vertexai/agent_engines/templates/adk.py b/vertexai/agent_engines/templates/adk.py index 7e21e6c680..feea1a6715 100644 --- a/vertexai/agent_engines/templates/adk.py +++ b/vertexai/agent_engines/templates/adk.py @@ -609,7 +609,7 @@ async def _init_session( session_state = {} for auth_id, auth in request.authorizations.items(): auth = _Authorization(**auth) - session_state[f"temp:{auth_id}"] = auth.access_token + session_state[auth_id] = auth.access_token if request.session_id: session_id = request.session_id diff --git a/vertexai/preview/reasoning_engines/templates/adk.py b/vertexai/preview/reasoning_engines/templates/adk.py index a3220228ac..07c3ee44c3 100644 --- a/vertexai/preview/reasoning_engines/templates/adk.py +++ b/vertexai/preview/reasoning_engines/templates/adk.py @@ -557,7 +557,7 @@ async def _init_session( session_state = {} for auth_id, auth in request.authorizations.items(): auth = _Authorization(**auth) - session_state[f"temp:{auth_id}"] = auth.access_token + session_state[auth_id] = auth.access_token if request.session_id: session_id = request.session_id