From a93506df186c1e6cc15e8f70779ee2ed8dd99a1e Mon Sep 17 00:00:00 2001 From: Jyong <76649700+JohnJyong@users.noreply.github.com> Date: Sat, 8 Jul 2023 17:29:56 +0800 Subject: [PATCH] Fix/dataset clean task (#534) --- api/tasks/clean_dataset_task.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/tasks/clean_dataset_task.py b/api/tasks/clean_dataset_task.py index 1232a8df1..14e17220b 100644 --- a/api/tasks/clean_dataset_task.py +++ b/api/tasks/clean_dataset_task.py @@ -7,7 +7,7 @@ from celery import shared_task from core.index.index import IndexBuilder from extensions.ext_database import db from models.dataset import DocumentSegment, Dataset, DatasetKeywordTable, DatasetQuery, DatasetProcessRule, \ - AppDatasetJoin + AppDatasetJoin, Document @shared_task @@ -32,7 +32,7 @@ def clean_dataset_task(dataset_id: str, tenant_id: str, indexing_technique: str, index_struct=index_struct ) - documents = db.session.query(DocumentSegment).filter(DocumentSegment.dataset_id == dataset_id).all() + documents = db.session.query(Document).filter(Document.dataset_id == dataset_id).all() segments = db.session.query(DocumentSegment).filter(DocumentSegment.dataset_id == dataset_id).all() vector_index = IndexBuilder.get_index(dataset, 'high_quality')