Skip to content

Commit b5f6423

Browse files
Fix: hydrate session state in create_session for VertexAiSessionService
1 parent 73a19d5 commit b5f6423

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

src/google/adk/sessions/vertex_ai_session_service.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -138,6 +138,16 @@ async def create_session(
138138
logger.debug('Create session response: %s', api_response)
139139
get_session_response = api_response.response
140140
session_id = get_session_response.name.split('/')[-1]
141+
try:
142+
hydrated_session = await self.get_session(
143+
app_name=app_name,
144+
user_id=user_id,
145+
session_id=session_id,
146+
)
147+
if hydrated_session is not None:
148+
return hydrated_session
149+
except Exception:
150+
logger.debug("Hydration via get_session failed, falling back")
141151

142152
session = Session(
143153
app_name=app_name,

0 commit comments

Comments
 (0)