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()