Ver código fonte

check dataset is none (#10682)

Jyong 5 meses atrás
pai
commit
70b9e4caf5
1 arquivos alterados com 3 adições e 1 exclusões
  1. 3 1
      api/tasks/document_indexing_task.py

+ 3 - 1
api/tasks/document_indexing_task.py

@@ -25,7 +25,9 @@ def document_indexing_task(dataset_id: str, document_ids: list):
     start_at = time.perf_counter()
 
     dataset = db.session.query(Dataset).filter(Dataset.id == dataset_id).first()
-
+    if not dataset:
+        logging.info(click.style("Dataset is not found: {}".format(dataset_id), fg="yellow"))
+        return
     # check document limit
     features = FeatureService.get_features(dataset.tenant_id)
     try: