Kaynağa Gözat

fix: dataset doc-form compatible (#24177)

Co-authored-by: huangzhuo <huangzhuo1@xiaomi.com>
tags/1.8.0
huangzhuo1949 2 ay önce
ebeveyn
işleme
1caeac56f2
No account linked to committer's email address
1 değiştirilmiş dosya ile 7 ekleme ve 0 silme
  1. 7
    0
      api/services/dataset_service.py

+ 7
- 0
api/services/dataset_service.py Dosyayı Görüntüle

@@ -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…
İptal
Kaydet