Skip to content

Commit 6cc3d9d

Browse files
Colstuwjxcopybara-github
authored andcommitted
chore: follow-up #946 to allow session db kwargs in fast api entrance
Merge #1042 COPYBARA_INTEGRATE_REVIEW=#1042 from Colstuwjx:chore/expose-database-args-in-fastapi 83a20cf PiperOrigin-RevId: 785714547
1 parent 083dcb4 commit 6cc3d9d

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

src/google/adk/cli/fast_api.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
from typing import Any
2828
from typing import List
2929
from typing import Literal
30+
from typing import Mapping
3031
from typing import Optional
3132

3233
import click
@@ -218,6 +219,7 @@ def get_fast_api_app(
218219
*,
219220
agents_dir: str,
220221
session_service_uri: Optional[str] = None,
222+
session_db_kwargs: Optional[Mapping[str, Any]] = None,
221223
artifact_service_uri: Optional[str] = None,
222224
memory_service_uri: Optional[str] = None,
223225
eval_storage_uri: Optional[str] = None,
@@ -369,7 +371,12 @@ def _parse_agent_engine_resource_name(agent_engine_id_or_resource_name):
369371
else:
370372
from ..sessions.database_session_service import DatabaseSessionService
371373

372-
session_service = DatabaseSessionService(db_url=session_service_uri)
374+
# Database session additional settings
375+
if session_db_kwargs is None:
376+
session_db_kwargs = {}
377+
session_service = DatabaseSessionService(
378+
db_url=session_service_uri, **session_db_kwargs
379+
)
373380
else:
374381
session_service = InMemorySessionService()
375382

0 commit comments

Comments
 (0)