Skip to content

Commit 2a73067

Browse files
fix(web): support ltm in web and fix bugs (#214)
1 parent def374b commit 2a73067

File tree

1 file changed

+15
-2
lines changed

1 file changed

+15
-2
lines changed

veadk/cli/cli_web.py

Lines changed: 15 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)