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/delete_segment_from_index_task.py
+++ b/api/tasks/delete_segment_from_index_task.py
@ -41,3 +41,5 @@ def delete_segment_from_index_task(index_node_ids: list, dataset_id: str, docume
        logging.info(click.style("Segment deleted from index latency: {}".format(end_at - start_at), fg="green"))
    except Exception:
        logging.exception("delete segment from index failed")
+    finally:
+        db.session.close()