From 3299b5142b27226fc05de79b81d65c14bb848b7c Mon Sep 17 00:00:00 2001 From: miko-garrido <69011999+miko-garrido@users.noreply.github.com> Date: Sun, 6 Jul 2025 09:12:01 +0800 Subject: [PATCH] feat: Add session_service_db_kwargs to get_fast_api_app params and DatabaseSessionService init to allow more configuration for session service in production --- src/google/adk/cli/fast_api.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/google/adk/cli/fast_api.py b/src/google/adk/cli/fast_api.py index 136051479..4aeb3053c 100644 --- a/src/google/adk/cli/fast_api.py +++ b/src/google/adk/cli/fast_api.py @@ -196,6 +196,7 @@ def get_fast_api_app( *, agents_dir: str, session_service_uri: Optional[str] = None, + session_service_db_kwargs: Optional[dict[str, Any]] = None, artifact_service_uri: Optional[str] = None, memory_service_uri: Optional[str] = None, eval_storage_uri: Optional[str] = None, @@ -314,7 +315,9 @@ async def internal_lifespan(app: FastAPI): agent_engine_id=agent_engine_id, ) else: - session_service = DatabaseSessionService(db_url=session_service_uri) + session_service = DatabaseSessionService( + db_url=session_service_uri, **(session_service_db_kwargs or {}) + ) else: session_service = InMemorySessionService()