AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-14 13:35:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

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"