Jelajahi Sumber

agent单元测试

xueyiming 1 Minggu lalu
induk
melakukan
c96f4839dd
1 mengubah file dengan 2 tambahan dan 1 penghapusan
  1. 2 1
      pqai_agent_server/task_server.py

+ 2 - 1
pqai_agent_server/task_server.py

@@ -201,7 +201,7 @@ class TaskManager:
         conn = self.db.get_connection()
         try:
             conn.begin()
-            # 插入任务
+            # TODO 插入任务
             with conn.cursor() as cursor:
                 cursor.execute(
                     f"""INSERT INTO {self.test_task_table} (agent_id, status, create_user, update_user) VALUES (%s, %s, %s, %s)""",
@@ -209,6 +209,7 @@ class TaskManager:
                 )
             task_id = cursor.lastrowid
             task_conversations = []
+            # TODO 具体的数据集信息
             i = 0
             for _ in range(30):
                 i = i + 1