deal dataset metadata and metadata binding (#17544)

This commit is contained in:
Jyong
2025-04-07 17:38:06 +08:00
committed by GitHub
parent e7f1d0deea
commit e1439c4a5b
2 changed files with 12 additions and 2 deletions

View File

@@ -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(