File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed
vertexai/agent_engines/templates Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff 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 ,
You can’t perform that action at this time.
0 commit comments