File tree Expand file tree Collapse file tree 1 file changed +8
-1
lines changed Expand file tree Collapse file tree 1 file changed +8
-1
lines changed Original file line number Diff line number Diff line change 27
27
from typing import Any
28
28
from typing import List
29
29
from typing import Literal
30
+ from typing import Mapping
30
31
from typing import Optional
31
32
32
33
import click
@@ -218,6 +219,7 @@ def get_fast_api_app(
218
219
* ,
219
220
agents_dir : str ,
220
221
session_service_uri : Optional [str ] = None ,
222
+ session_db_kwargs : Optional [Mapping [str , Any ]] = None ,
221
223
artifact_service_uri : Optional [str ] = None ,
222
224
memory_service_uri : Optional [str ] = None ,
223
225
eval_storage_uri : Optional [str ] = None ,
@@ -369,7 +371,12 @@ def _parse_agent_engine_resource_name(agent_engine_id_or_resource_name):
369
371
else :
370
372
from ..sessions .database_session_service import DatabaseSessionService
371
373
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
+ )
373
380
else :
374
381
session_service = InMemorySessionService ()
375
382
You can’t perform that action at this time.
0 commit comments