Parcourir la source

raise error when process_rule is required but missing (#20599)

tags/1.4.2
Abdullah AlOsaimi il y a 5 mois
Parent
révision
006496f24e
Aucun compte lié à l'adresse e-mail de l'auteur
1 fichiers modifiés avec 5 ajouts et 1 suppressions
  1. 5
    1
      api/controllers/service_api/dataset/document.py

+ 5
- 1
api/controllers/service_api/dataset/document.py Voir le fichier

@@ -206,12 +206,16 @@ class DocumentAddByFileApi(DatasetApiResource):
knowledge_config = KnowledgeConfig(**args)
DocumentService.document_create_args_validate(knowledge_config)

dataset_process_rule = dataset.latest_process_rule if "process_rule" not in args else None
if not knowledge_config.original_document_id and not dataset_process_rule and not knowledge_config.process_rule:
raise ValueError("process_rule is required.")

try:
documents, batch = DocumentService.save_document_with_dataset_id(
dataset=dataset,
knowledge_config=knowledge_config,
account=dataset.created_by_account,
dataset_process_rule=dataset.latest_process_rule if "process_rule" not in args else None,
dataset_process_rule=dataset_process_rule,
created_from="api",
)
except ProviderTokenNotInitError as ex:

Chargement…
Annuler
Enregistrer