Add session_db_kwargs to FastAPI App

The get_fast_api_app function supports a session_db_url argument that allows constructing the session service with a DatabaseSessionService.

adk-python/src/google/adk/cli/fast_api.py

Lines 193 to 202 in 484b33e

    
           def get_fast_api_app( 
        
               *, 
        
               agents_dir: str, 
        
               session_db_url: str = "", 
        
               artifact_storage_uri: Optional[str] = None, 
        
               allow_origins: Optional[list[str]] = None, 
        
               web: bool, 
        
               trace_to_cloud: bool = False, 
        
               lifespan: Optional[Lifespan[FastAPI]] = None, 
        
           ) -> FastAPI:

adk-python/src/google/adk/cli/fast_api.py

Line 276 in 484b33e

session_service = DatabaseSessionService(db_url=session_db_url)

However, it does not support passing additional DB kwargs required for full control or advanced use cases that production resources require. DB kwargs are where users can configure their connection pool settings like pool size, timeouts, etc.

The DatabaseSessionService supports database engine kwargs and these should be configurable using the the FastAPI app.

adk-python/src/google/adk/sessions/database_session_service.py

Lines 304 to 314 in 484b33e

    
           class DatabaseSessionService(BaseSessionService): 
        
             """A session service that uses a database for storage.""" 
        
             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, **kwargs)

This is critical for creating DatabaseSessionServices's with Cloud SQL and AlloyDB as the recommended best practice is to use an AlloyDB Python Connector which thousands of customers use today.

The Python Connectors for Cloud SQL and AlloyDB require the creator SQLAlchemy kwarg to work. They can not be used with ADK via the FastAPI app until this is fixed.

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Metadata

Assignees

Labels

Type

Projects

Milestone

Relationships

Development

	def get_fast_api_app(
	*,
	agents_dir: str,
	session_db_url: str = "",
	artifact_storage_uri: Optional[str] = None,
	allow_origins: Optional[list[str]] = None,
	web: bool,
	trace_to_cloud: bool = False,
	lifespan: Optional[Lifespan[FastAPI]] = None,
	) -> FastAPI:

	class DatabaseSessionService(BaseSessionService):
	"""A session service that uses a database for storage."""

	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, **kwargs)

Description

Metadata

Metadata

Assignees

Labels

Type

Projects

Milestone

Relationships

Development

Issue actions