From ee7d26ac8b8d80d846466f4227538f3e8a346e6a Mon Sep 17 00:00:00 2001 From: Tomasz Zawiszowski Date: Fri, 21 Aug 2026 15:39:57 +0200 Subject: [PATCH] celery: dispose DB engines in worker_process_init --- python-ai-kit/app/integrations/celery/core.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/python-ai-kit/app/integrations/celery/core.py b/python-ai-kit/app/integrations/celery/core.py index 6bccb62..ca93b8c 100644 --- a/python-ai-kit/app/integrations/celery/core.py +++ b/python-ai-kit/app/integrations/celery/core.py @@ -1,7 +1,15 @@ from app.config import settings +from app.database import async_engine, sync_engine from celery import Celery from celery import current_app as current_celery_app from celery.schedules import crontab +from celery.signals import worker_process_init + + +@worker_process_init.connect +def _dispose_db_engines_after_fork(**kwargs: object) -> None: + sync_engine.dispose(close=False) + async_engine.sync_engine.dispose(close=False) def create_celery() -> Celery: