Skip to content

Commit d9f6c58

Browse files
Tongzhou-Jiangcopybara-github
authored andcommitted
fix: Add support for app in _init_session
PiperOrigin-RevId: 834389695
1 parent 26b7e51 commit d9f6c58

File tree

1 file changed

+3
-2
lines changed
  • vertexai/agent_engines/templates

1 file changed

+3
-2
lines changed

vertexai/agent_engines/templates/adk.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -610,8 +610,9 @@ async def _init_session(
610610
auth = _Authorization(**auth)
611611
session_state[auth_id] = auth.access_token
612612

613+
app = self._tmpl_attrs.get("app")
613614
session = await session_service.create_session(
614-
app_name=self._tmpl_attrs.get("app_name"),
615+
app_name=app.name if app else self._tmpl_attrs.get("app_name"),
615616
user_id=request.user_id,
616617
state=session_state,
617618
)
@@ -628,7 +629,7 @@ async def _init_session(
628629
):
629630
version_data = _ArtifactVersion(**version_data)
630631
saved_version = await artifact_service.save_artifact(
631-
app_name=self._tmpl_attrs.get("app_name"),
632+
app_name=app.name if app else self._tmpl_attrs.get("app_name"),
632633
user_id=request.user_id,
633634
session_id=session.id,
634635
filename=artifact.file_name,

0 commit comments

Comments
 (0)