Pārlūkot izejas kodu

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

xueyiming 2 dienas atpakaļ
vecāks
revīzija
98a9d6eadd

+ 1 - 2
pqai_agent/agents/simple_chat_agent.py

@@ -66,8 +66,7 @@ class SimpleOpenAICompatibleChatAgent:
                     self.agent_task_details.append(agent_task_detail)
 
                     if function_name in self.tool_map:
-                        # result = self.tool_map[function_name](**arguments)
-                        result = "success"
+                        result = self.tool_map[function_name](**arguments)
                         messages.append({
                             "role": "tool",
                             "tool_call_id": tool_call.id,

+ 3 - 2
pqai_agent_server/agent_task_server.py

@@ -53,9 +53,10 @@ class AgentTaskManager:
             ])).all()
 
     def get_agent_task_details(self, task_id):
-        """更新任务状态"""
+        """获取任务详情"""
         with self.session_maker() as session:
-            return session.query(AgentTaskDetail).filter(AgentTaskDetail.agent_task_id == task_id).all()
+            return session.query(AgentTaskDetail).filter(AgentTaskDetail.agent_task_id == task_id).filter(
+                AgentTaskDetail.parent_execution_id == None).all()
 
     def save_agent_task_details_batch(self, agent_task_details: list, agent_task_id: int, message: str):
         """批量保存子任务到数据库"""