Procházet zdrojové kódy

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

takatost před 1 rokem
rodič
revize
2f179d61dc
1 změnil soubory, kde provedl 3 přidání a 0 odebrání
  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