raise error when process_rule is required but missing (#20599)
This commit is contained in:

committed by
GitHub

parent
01d500db14
commit
006496f24e
@@ -206,12 +206,16 @@ class DocumentAddByFileApi(DatasetApiResource):
|
|||||||
knowledge_config = KnowledgeConfig(**args)
|
knowledge_config = KnowledgeConfig(**args)
|
||||||
DocumentService.document_create_args_validate(knowledge_config)
|
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:
|
try:
|
||||||
documents, batch = DocumentService.save_document_with_dataset_id(
|
documents, batch = DocumentService.save_document_with_dataset_id(
|
||||||
dataset=dataset,
|
dataset=dataset,
|
||||||
knowledge_config=knowledge_config,
|
knowledge_config=knowledge_config,
|
||||||
account=dataset.created_by_account,
|
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",
|
created_from="api",
|
||||||
)
|
)
|
||||||
except ProviderTokenNotInitError as ex:
|
except ProviderTokenNotInitError as ex:
|
||||||
|
Reference in New Issue
Block a user