瀏覽代碼

delete segment not commit (#309)

Jyong 1 年之前
父節點
當前提交
35a2beb195
共有 2 個文件被更改,包括 2 次插入2 次删除
  1. 1 1
      api/tasks/clean_document_task.py
  2. 1 1
      api/tasks/document_indexing_update_task.py

+ 1 - 1
api/tasks/clean_document_task.py

@@ -43,7 +43,7 @@ def clean_document_task(document_id: str, dataset_id: str):
 
         for segment in segments:
             db.session.delete(segment)
-
+        db.session.commit()
         end_at = time.perf_counter()
         logging.info(
             click.style('Cleaned document when document deleted: {} latency: {}'.format(document_id, end_at - start_at), fg='green'))

+ 1 - 1
api/tasks/document_indexing_update_task.py

@@ -59,7 +59,7 @@ def document_indexing_update_task(dataset_id: str, document_id: str):
 
         for segment in segments:
             db.session.delete(segment)
-
+        db.session.commit()
         end_at = time.perf_counter()
         logging.info(
             click.style('Cleaned document when document update data source or process rule: {} latency: {}'.format(document_id, end_at - start_at), fg='green'))