浏览代码

增加agent提交执行和查询任务功能

xueyiming 2 月之前
父节点
当前提交
c664d02a64
共有 1 个文件被更改,包括 6 次插入2 次删除
  1. 6 2
      pqai_agent_server/agent_task_server.py

+ 6 - 2
pqai_agent_server/agent_task_server.py

@@ -96,9 +96,13 @@ class AgentTaskManager:
                 {
                     "id": agent_task.id,
                     "agentName": agent_configuration.name,
-                    "createUser": agent_task.create_user,
-                    "updateUser": agent_task.update_user,
                     "statusName": get_agent_task_status_desc(agent_task.status),
+                    "startTime": agent_task.start_time.strftime("%Y-%m-%d %H:%M:%S"),
+                    "end_time": agent_task.start_time.strftime(
+                        "%Y-%m-%d %H:%M:%S") if agent_task.start_time is not None else None,
+                    "createUser": agent_task.create_user,
+                    "input": agent_task.input,
+                    "output": agent_task.output,
                     "createTime": agent_task.create_time.strftime("%Y-%m-%d %H:%M:%S"),
                     "updateTime": agent_task.update_time.strftime("%Y-%m-%d %H:%M:%S")
                 }