check dataset is none (langgenius#10682)

ZuzooVn · Nov 14, 2024 · 70b9e4c · 70b9e4c
1 parent 317ae92
commit 70b9e4c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/api/tasks/document_indexing_task.py b/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: