diff --git a/rag/svr/task_executor.py b/rag/svr/task_executor.py index febfc7699..3e86573c6 100644 --- a/rag/svr/task_executor.py +++ b/rag/svr/task_executor.py @@ -118,7 +118,10 @@ def collect(): cron_logger.info("Task {} has been canceled.".format(msg["id"])) return pd.DataFrame() tasks = TaskService.get_tasks(msg["id"]) - assert tasks, "{} empty task!".format(msg["id"]) + if not tasks: + cron_logger.warn("{} empty task!".format(msg["id"])) + return [] + tasks = pd.DataFrame(tasks) if msg.get("type", "") == "raptor": tasks["task_type"] = "raptor"