Skip to content

Commit 70b9e4c

Browse files
authored
check dataset is none (#10682)
1 parent 317ae92 commit 70b9e4c

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

api/tasks/document_indexing_task.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,9 @@ def document_indexing_task(dataset_id: str, document_ids: list):
2525
start_at = time.perf_counter()
2626

2727
dataset = db.session.query(Dataset).filter(Dataset.id == dataset_id).first()
28-
28+
if not dataset:
29+
logging.info(click.style("Dataset is not found: {}".format(dataset_id), fg="yellow"))
30+
return
2931
# check document limit
3032
features = FeatureService.get_features(dataset.tenant_id)
3133
try:

0 commit comments

Comments
 (0)