浏览代码

fix: dataset doc-form compatible (#24177)

Co-authored-by: huangzhuo <huangzhuo1@xiaomi.com>
tags/1.8.0
huangzhuo1949 2 个月前
父节点
当前提交
1caeac56f2
没有帐户链接到提交者的电子邮件
共有 1 个文件被更改,包括 7 次插入0 次删除
  1. 7
    0
      api/services/dataset_service.py

+ 7
- 0
api/services/dataset_service.py 查看文件

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":
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)



正在加载...
取消
保存