add pool_pre_ping to sqlalchemy engine creation to ensure the connection liveness (#1206)

2024-11-16 12:28:20 -08:00
parent 19fcd4065f
commit 8c653b461a
1 changed files with 1 additions and 1 deletions
--- 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(