From 2f179d61dcfbcc634cf7ffa31a2a639004bfe5f4 Mon Sep 17 00:00:00 2001 From: takatost Date: Sun, 13 Aug 2023 00:25:05 +0800 Subject: [PATCH] fix: completion error when dataset was deleted (#819) --- api/core/orchestrator_rule_parser.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/api/core/orchestrator_rule_parser.py b/api/core/orchestrator_rule_parser.py index 021f8c935f..4755588ba5 100644 --- a/api/core/orchestrator_rule_parser.py +++ b/api/core/orchestrator_rule_parser.py @@ -188,6 +188,9 @@ class OrchestratorRuleParser: Dataset.id == tool_config.get("id") ).first() + if not dataset: + return None + if dataset and dataset.available_document_count == 0 and dataset.available_document_count == 0: return None