From 70b9e4caf560c537ffcf8f605fa6250bd301188e Mon Sep 17 00:00:00 2001
From: Jyong <76649700+JohnJyong@users.noreply.github.com>
Date: Thu, 14 Nov 2024 14:07:19 +0800
Subject: [PATCH] check dataset is none (#10682)

---
 api/tasks/document_indexing_task.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/api/tasks/document_indexing_task.py b/api/tasks/document_indexing_task.py
index 72c4674e0f..df1177d578 100644
--- 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: