fix: populate missing auth_id for _init_session into adk session.state

PiperOrigin-RevId: 828119747

Author: Tongzhou-Jiang

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

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