|
|
|
|
@ -9,7 +9,7 @@ from core.rag.index_processor.index_processor_factory import IndexProcessorFacto
|
|
|
|
|
from core.tools.utils.rag_web_reader import get_image_upload_file_ids
|
|
|
|
|
from extensions.ext_database import db
|
|
|
|
|
from extensions.ext_storage import storage
|
|
|
|
|
from models.dataset import Dataset, DocumentSegment
|
|
|
|
|
from models.dataset import Dataset, DatasetMetadataBinding, DocumentSegment
|
|
|
|
|
from models.model import UploadFile
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@ -67,6 +67,12 @@ def clean_document_task(document_id: str, dataset_id: str, doc_form: str, file_i
|
|
|
|
|
db.session.delete(file)
|
|
|
|
|
db.session.commit()
|
|
|
|
|
|
|
|
|
|
# delete dataset metadata binding
|
|
|
|
|
db.session.query(DatasetMetadataBinding).filter(
|
|
|
|
|
DatasetMetadataBinding.dataset_id == dataset_id,
|
|
|
|
|
DatasetMetadataBinding.document_id == document_id,
|
|
|
|
|
).delete()
|
|
|
|
|
|
|
|
|
|
end_at = time.perf_counter()
|
|
|
|
|
logging.info(
|
|
|
|
|
click.style(
|
|
|
|
|
|