Sfoglia il codice sorgente

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

xueyiming 3 giorni fa
parent
commit
ce98a6c88f
1 ha cambiato i file con 15 aggiunte e 14 eliminazioni
  1. 15 14
      pqai_agent_server/agent_task_server.py

+ 15 - 14
pqai_agent_server/agent_task_server.py

@@ -183,20 +183,21 @@ class AgentTaskManager:
     def get_agent_task_detail(self, agent_task_id):
         agent_task = self.get_agent_task(agent_task_id)
         agent_task_details = self.get_agent_task_details(agent_task_id)
-        agent_task_detail_datas = []
-        for agent_task_detail in agent_task_details:
-            data = {}
-            data["id"] = agent_task_detail.id
-            data["executorType"] = agent_task_detail.executor_type
-            data["status"] = get_agent_task_detail_status_desc(agent_task_detail.status)
-            data["inputData"] = agent_task_detail.input_data
-            data["executorName"] = agent_task_detail.executor_name
-            data["reasoning"] = agent_task_detail.reasoning
-            data["outputData"] = agent_task_detail.output_data
-            data["errorMessage"] = agent_task_detail.error_message
-            data["createTime"]: agent_task_detail.create_time.strftime("%Y-%m-%d %H:%M:%S")
-            data["updateTime"]: agent_task_detail.update_time.strftime("%Y-%m-%d %H:%M:%S")
-            agent_task_detail_datas.append(data)
+        agent_task_detail_datas = [
+            {
+                "id": agent_task_detail.id,
+                "executorType": agent_task_detail.executor_type,
+                "statusName": get_agent_task_detail_status_desc(agent_task_detail.status),
+                "inputData": agent_task_detail.input_data,
+                "executorName": agent_task_detail.executor_name,
+                "reasoning": agent_task_detail.reasoning,
+                "outputData": agent_task_detail.output_data,
+                "errorMessage": agent_task_detail.error_message,
+                "createTime": agent_task_detail.create_time.strftime("%Y-%m-%d %H:%M:%S"),
+                "updateTime": agent_task_detail.update_time.strftime("%Y-%m-%d %H:%M:%S")
+            }
+            for agent_task_detail in agent_task_details
+        ]
         return {
             "input": agent_task.input,
             "tools": agent_task.tools,