Fix: add missing db.session.close() to ensure proper session cleanup (#23122)

2025-07-30 10:34:24 +08:00
parent 0ea010d7ee
commit f17ca26b10
3 changed files with 5 additions and 0 deletions
--- a/api/tasks/add_document_to_index_task.py
+++ b/api/tasks/add_document_to_index_task.py
@@ -32,6 +32,7 @@ def add_document_to_index_task(dataset_document_id: str):
        return

    if dataset_document.indexing_status != "completed":
+        db.session.close()
        return

    indexing_cache_key = f"document_{dataset_document.id}_indexing"
@@ -112,3 +113,4 @@ def add_document_to_index_task(dataset_document_id: str):
        db.session.commit()
    finally:
        redis_client.delete(indexing_cache_key)
+        db.session.close()