瀏覽代碼

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

takatost 1 年之前
父節點
當前提交
2f179d61dc
共有 1 個文件被更改,包括 3 次插入0 次删除
  1. 3 0
      api/core/orchestrator_rule_parser.py

+ 3 - 0
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