fix notion dataset rule not found (#21236)

pull/21310/head
Jyong 11 months ago committed by GitHub
parent fa70033438
commit 57f7368a0e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -976,12 +976,17 @@ class DocumentService:
process_rule = knowledge_config.process_rule process_rule = knowledge_config.process_rule
if process_rule: if process_rule:
if process_rule.mode in ("custom", "hierarchical"): if process_rule.mode in ("custom", "hierarchical"):
if process_rule.rules:
dataset_process_rule = DatasetProcessRule( dataset_process_rule = DatasetProcessRule(
dataset_id=dataset.id, dataset_id=dataset.id,
mode=process_rule.mode, mode=process_rule.mode,
rules=process_rule.rules.model_dump_json() if process_rule.rules else None, rules=process_rule.rules.model_dump_json() if process_rule.rules else None,
created_by=account.id, created_by=account.id,
) )
else:
dataset_process_rule = dataset.latest_process_rule
if not dataset_process_rule:
raise ValueError("No process rule found.")
elif process_rule.mode == "automatic": elif process_rule.mode == "automatic":
dataset_process_rule = DatasetProcessRule( dataset_process_rule = DatasetProcessRule(
dataset_id=dataset.id, dataset_id=dataset.id,

Loading…
Cancel
Save