NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-13 15:14:50 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix: add missing kwargs to db session service

Browse Source 
Copybara import of the project:

--
bbd21e72e46227d5bbcaef6601f4a81724e7829f by Sanchit Rk <sanchitrrk@gmail.com>:

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
This commit is contained in:
Sanchit Rk 2025-05-28 13:27:23 -07:00 committed by Copybara-Service
parent 2a07eefc27
commit 3cd4cd3ecf
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/google/adk/sessions/database_session_service.py
View File

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