From 3cd4cd3ecfb1c279254d9993b2d6d6b30133fa7d Mon Sep 17 00:00:00 2001 From: Sanchit Rk Date: Wed, 28 May 2025 13:27:23 -0700 Subject: [PATCH] Fix: add missing kwargs to db session service Copybara import of the project: -- bbd21e72e46227d5bbcaef6601f4a81724e7829f by Sanchit Rk : Fix: add missing kwargs to db session service COPYBARA_INTEGRATE_REVIEW=https://github.com/google/adk-python/pull/946 from sanchitrk:fix/missing-kwargs-db-session-service ebb699b04d8420ad14244cf3d43a2982b01d6b7f PiperOrigin-RevId: 764392251 --- src/google/adk/sessions/database_session_service.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/google/adk/sessions/database_session_service.py b/src/google/adk/sessions/database_session_service.py index 34ced22..8770d93 100644 --- a/src/google/adk/sessions/database_session_service.py +++ b/src/google/adk/sessions/database_session_service.py @@ -240,14 +240,14 @@ class StorageUserState(Base): class DatabaseSessionService(BaseSessionService): """A session service that uses a database for storage.""" - def __init__(self, db_url: str): + def __init__(self, db_url: str, **kwargs: Any): """Initializes the database session service with a database URL.""" # 1. Create DB engine for db connection # 2. Create all tables based on schema # 3. Initialize all properties try: - db_engine = create_engine(db_url) + db_engine = create_engine(db_url, **kwargs) except Exception as e: if isinstance(e, ArgumentError): raise ValueError(