Skip to content

Commit fece486

Browse files
Tongzhou-Jiangcopybara-github
authored andcommitted
fix: investigate missing auth_id for _init_session
PiperOrigin-RevId: 828119747
1 parent d02a7da commit fece486

File tree

1 file changed

+7
-0
lines changed
  • vertexai/preview/reasoning_engines/templates

1 file changed

+7
-0
lines changed

vertexai/preview/reasoning_engines/templates/adk.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -551,8 +551,13 @@ async def _init_session(
551551
"""Initializes the session, and returns the session id."""
552552
from google.adk.events.event import Event
553553
import random
554+
from google.cloud.aiplatform import base
555+
556+
_LOGGER = base.Logger(__name__)
554557

555558
session_state = None
559+
_LOGGER.info("authorizations: %s", request.authorizations)
560+
print("authorizations: %s", request.authorizations)
556561
if request.authorizations:
557562
session_state = {}
558563
for auth_id, auth in request.authorizations.items():
@@ -563,6 +568,8 @@ async def _init_session(
563568
session_id = request.session_id
564569
else:
565570
session_id = f"temp_session_{random.randbytes(8).hex()}"
571+
_LOGGER.info("session_state: %s", session_state)
572+
print("session_state: %s", session_state)
566573
session = await session_service.create_session(
567574
app_name=self._tmpl_attrs.get("app_name"),
568575
user_id=request.user_id,

0 commit comments

Comments
 (0)