fix: completion error when dataset was deleted (#819)

3 years ago · 2f179d61dc
parent 7457550673
commit 2f179d61dc
1 changed files with 3 additions and 0 deletions
--- 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