浏览代码

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

xueyiming 2 月之前
父节点
当前提交
98a9d6eadd
共有 2 个文件被更改,包括 4 次插入4 次删除
  1. 1 2
      pqai_agent/agents/simple_chat_agent.py
  2. 3 2
      pqai_agent_server/agent_task_server.py

+ 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):
         """批量保存子任务到数据库"""