gmakstutis/ragflow
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

alter way of alerting info of empty task (#2094)

Browse source 
### What problem does this PR solve?

#2043

### Type of change

- [x] Refactoring
This commit is contained in:
Kevin Hu 2024-08-26 13:45:26 +08:00 • committed by GitHub
parent a6d85c6c2f
commit e9202999cb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
rag/svr/task_executor.py
View file

@ -118,7 +118,10 @@ def collect():
cron_logger.info("Task {} has been canceled.".format(msg["id"])) cron_logger.info("Task {} has been canceled.".format(msg["id"]))
return pd.DataFrame() return pd.DataFrame()
tasks = TaskService.get_tasks(msg["id"]) 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) tasks = pd.DataFrame(tasks)
if msg.get("type", "") == "raptor": if msg.get("type", "") == "raptor":
tasks["task_type"] = "raptor" tasks["task_type"] = "raptor"