소스 검색

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