agent单元测试 增加模块id

pqai_agent_server/api_server.py

@@ -567,9 +567,12 @@ def create_test_task():
     """
     req_data = request.json
     agent_id = req_data.get('agentId', None)
+    model_id = req_data.get('modelId', None)
     if not agent_id:
-        return wrap_response(400, msg='agent id is required')
-    app.task_manager.create_task(agent_id)
+        return wrap_response(404, msg='agent id is required')
+    if not model_id:
+        return wrap_response(404, msg='model id is required')
+    app.task_manager.create_task(agent_id, model_id)
     return wrap_response(200)
 
 

pqai_agent_server/task_server.py

@@ -195,7 +195,7 @@ class TaskManager:
             "list": response_data,
         }
 
-    def create_task(self, agent_id: int) -> Dict:
+    def create_task(self, agent_id: int, model_id: int) -> Dict:
         """创建新任务"""
 
         conn = self.db.get_connection()