Skip to content
Merged
5 changes: 3 additions & 2 deletions src/backend/base/langflow/services/database/service.py
Original file line number Diff line number Diff line change
Expand Up @@ -127,12 +127,13 @@ def _create_engine(self) -> AsyncEngine:

poolclass_key = kwargs.get("poolclass")
if poolclass_key is not None:
pool_class = getattr(sa, poolclass_key, None)
if pool_class and isinstance(pool_class(), sa.pool.Pool):
pool_class = getattr(sa.pool, poolclass_key, None)
if pool_class and issubclass(pool_class, sa.pool.Pool):
logger.debug(f"Using poolclass: {poolclass_key}.")
kwargs["poolclass"] = pool_class
else:
logger.error(f"Invalid poolclass '{poolclass_key}' specified. Using default pool class.")
kwargs.pop("poolclass", None)

return create_async_engine(
self.database_url,
Expand Down