fix: dataset doc-form compatible (#24177)

Co-authored-by: huangzhuo <huangzhuo1@xiaomi.com>
This commit is contained in:
huangzhuo1949 2025-08-20 23:48:56 +08:00 committed by GitHub
parent 6b1606f4f4
commit 1caeac56f2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 0 deletions

View File

@ -250,6 +250,11 @@ class DatasetService:
dataset: Optional[Dataset] = db.session.query(Dataset).filter_by(id=dataset_id).first()
return dataset
@staticmethod
def check_doc_form(dataset: Dataset, doc_form: str):
if dataset.doc_form and doc_form != dataset.doc_form:
raise ValueError("doc_form is different from the dataset doc_form.")
@staticmethod
def check_dataset_model_setting(dataset):
if dataset.indexing_technique == "high_quality":
@ -1085,6 +1090,8 @@ class DocumentService:
dataset_process_rule: Optional[DatasetProcessRule] = None,
created_from: str = "web",
):
# check doc_form
DatasetService.check_doc_form(dataset, knowledge_config.doc_form)
# check document limit
features = FeatureService.get_features(current_user.current_tenant_id)