Browse Source

恢复任务增加查询状态

xueyiming 2 days ago
parent
commit
1d262b18cd
1 changed files with 4 additions and 1 deletions
  1. 4 1
      pqai_agent_server/task_server.py

+ 4 - 1
pqai_agent_server/task_server.py

@@ -190,7 +190,10 @@ class TaskManager:
     def get_in_progress_task(self):
         """获取执行中任务"""
         with self.session_maker() as session:
-            return session.query(AgentTestTask).filter(AgentTestTask.status == TestTaskStatus.IN_PROGRESS.value).all()
+            return session.query(AgentTestTask).filter(AgentTestTask.status.in_([
+                TestTaskStatus.NOT_STARTED.value,
+                TestTaskStatus.IN_PROGRESS.value
+            ])).all()
 
     def get_creating_task(self):
         """获取执行中任务"""