@@ -56,10 +56,14 @@ def _get_ltm_from_env() -> LongTermMemory | None:
5656 logger = get_logger (__name__ )
5757
5858 long_term_memory_backend = os .getenv ("LONG_TERM_MEMORY_BACKEND" )
59+ app_name = os .getenv ("VEADK_WEB_APP_NAME" , "" )
60+ user_id = os .getenv ("VEADK_WEB_USER_ID" , "" )
5961
6062 if long_term_memory_backend :
6163 logger .info (f"Long term memory: backend={ long_term_memory_backend } " )
62- return LongTermMemory (backend = long_term_memory_backend ) # type: ignore
64+ return LongTermMemory (
65+ backend = long_term_memory_backend , app_name = app_name , user_id = user_id
66+ ) # type: ignore
6367 else :
6468 logger .warning ("No long term memory backend settings detected." )
6569 return None
@@ -131,7 +135,13 @@ def wrapped_get_fast_api(self, *args, **kwargs):
131135
132136@click .command ()
133137@click .option ("--host" , default = "127.0.0.1" , help = "Host to run the web server on" )
134- def web (host : str ) -> None :
138+ @click .option (
139+ "--app_name" , default = "" , help = "The `app_name` for initializing long term memory"
140+ )
141+ @click .option (
142+ "--user_id" , default = "" , help = "The `user_id` for initializing long term memory"
143+ )
144+ def web (host : str , app_name : str , user_id : str ) -> None :
135145 """Launch web with long term and short term memory."""
136146 import os
137147 from typing import Any
@@ -175,6 +185,9 @@ def init_for_veadk(
175185 self .session_service = short_term_memory .session_service
176186 self .memory_service = long_term_memory
177187
188+ os .environ ["VEADK_WEB_APP_NAME" ] = app_name
189+ os .environ ["VEADK_WEB_USER_ID" ] = user_id
190+
178191 import google .adk .cli .adk_web_server
179192
180193 google .adk .cli .adk_web_server .AdkWebServer .__init__ = init_for_veadk
0 commit comments