소스 검색

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:
         for segment in segments:
             db.session.delete(segment)
             db.session.delete(segment)
-
+        db.session.commit()
         end_at = time.perf_counter()
         end_at = time.perf_counter()
         logging.info(
         logging.info(
             click.style('Cleaned document when document deleted: {} latency: {}'.format(document_id, end_at - start_at), fg='green'))
             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:
         for segment in segments:
             db.session.delete(segment)
             db.session.delete(segment)
-
+        db.session.commit()
         end_at = time.perf_counter()
         end_at = time.perf_counter()
         logging.info(
         logging.info(
             click.style('Cleaned document when document update data source or process rule: {} latency: {}'.format(document_id, end_at - start_at), fg='green'))
             click.style('Cleaned document when document update data source or process rule: {} latency: {}'.format(document_id, end_at - start_at), fg='green'))