Преглед на файлове

fix: dataset doc-form compatible (#24177)

Co-authored-by: huangzhuo <huangzhuo1@xiaomi.com>
tags/1.8.0
huangzhuo1949 преди 2 месеца
родител
ревизия
1caeac56f2
No account linked to committer's email address
променени са 1 файла, в които са добавени 7 реда и са изтрити 0 реда
  1. 7
    0
      api/services/dataset_service.py

+ 7
- 0
api/services/dataset_service.py Целия файл

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


Loading…
Отказ
Запис