deal db session in celery worker (#17549)

2025-04-07 20:31:26 +08:00 · 2025-04-07 20:31:26 +08:00 · c8145ce581
commit c8145ce581
parent da7f8ad936
26 changed files with 104 additions and 25 deletions
--- a/api/tasks/annotation/batch_import_annotations_task.py
+++ b/api/tasks/annotation/batch_import_annotations_task.py
@ -88,3 +88,5 @@ def batch_import_annotations_task(job_id: str, content_list: list[dict], app_id:
            indexing_error_msg_key = "app_annotation_batch_import_error_msg_{}".format(str(job_id))
            redis_client.setex(indexing_error_msg_key, 600, str(e))
            logging.exception("Build index for batch import annotations failed")
+        finally:
+            db.session.close()