wzj/dify-1.72
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix: dataset doc-form compatible (#24177)

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
api/services/dataset_service.py
View File

@ -250,6 +250,11 @@ class DatasetService:
dataset: Optional[Dataset] = db.session.query(Dataset).filter_by(id=dataset_id).first() dataset: Optional[Dataset] = db.session.query(Dataset).filter_by(id=dataset_id).first()
return dataset 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 @staticmethod
def check_dataset_model_setting(dataset): def check_dataset_model_setting(dataset):
if dataset.indexing_technique == "high_quality": if dataset.indexing_technique == "high_quality":
@ -1085,6 +1090,8 @@ class DocumentService:
dataset_process_rule: Optional[DatasetProcessRule] = None, dataset_process_rule: Optional[DatasetProcessRule] = None,
created_from: str = "web", created_from: str = "web",
): ):
# check doc_form
DatasetService.check_doc_form(dataset, knowledge_config.doc_form)
# check document limit # check document limit
features = FeatureService.get_features(current_user.current_tenant_id) features = FeatureService.get_features(current_user.current_tenant_id)