فهرست منبع

增加更新时间

xueyiming 4 روز پیش
والد
کامیت
f177905187
1فایلهای تغییر یافته به همراه2 افزوده شده و 3 حذف شده
  1. 2 3
      pqai_agent_server/task_server.py

+ 2 - 3
pqai_agent_server/task_server.py

@@ -277,14 +277,13 @@ class TaskManager:
                     except Exception as e:
                         logger.info(f"Task {task_id} stopped with exception: {e}")
 
-
         with self.session_maker() as session:
             with session.begin():
                 session.query(AgentTestTask).filter(AgentTestTask.id == task_id).update(
                     {"status": TestTaskStatus.CANCELLED.value, "update_user": user, "update_time": datetime.now()})
                 session.query(AgentTestTaskConversations).filter(AgentTestTaskConversations.task_id == task_id).filter(
                     AgentTestTaskConversations.status == TestTaskConversationsStatus.PENDING.value).update(
-                    {"status": TestTaskConversationsStatus.CANCELLED.value})
+                    {"status": TestTaskConversationsStatus.CANCELLED.value, "update_time": datetime.now()})
                 session.commit()
 
         self._cleanup_task_resources(task_id)
@@ -301,7 +300,7 @@ class TaskManager:
                     {"status": TestTaskStatus.NOT_STARTED.value, "update_user": user, "update_time": datetime.now()})
                 session.query(AgentTestTaskConversations).filter(AgentTestTaskConversations.task_id == task_id).filter(
                     AgentTestTaskConversations.status == TestTaskConversationsStatus.CANCELLED.value).update(
-                    {"status": TestTaskConversationsStatus.PENDING.value})
+                    {"status": TestTaskConversationsStatus.PENDING.value, "update_time": datetime.now()})
                 session.commit()
 
         # 重新执行任务