From 8c653b461a84e392a56dd774fcd2fe0556da82f8 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Sat, 16 Nov 2024 12:28:20 -0800 Subject: [PATCH] add pool_pre_ping to sqlalchemy engine creation to ensure the connection liveness (#1206) --- skyvern/forge/sdk/db/client.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/skyvern/forge/sdk/db/client.py b/skyvern/forge/sdk/db/client.py index 63370907..76596713 100644 --- a/skyvern/forge/sdk/db/client.py +++ b/skyvern/forge/sdk/db/client.py @@ -79,7 +79,7 @@ class AgentDB: def __init__(self, database_string: str, debug_enabled: bool = False) -> None: super().__init__() self.debug_enabled = debug_enabled - self.engine = create_async_engine(database_string, json_serializer=_custom_json_serializer) + self.engine = create_async_engine(database_string, json_serializer=_custom_json_serializer, pool_pre_ping=True) self.Session = async_sessionmaker(bind=self.engine) async def create_task(