|
@@ -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()
|
|
|
|
|
|
# 重新执行任务
|