bugfix

applications/service/response.py

@@ -15,5 +15,5 @@ class TaskScheduleResponse:
         return {"code": error_code, "status": "error", "message": error_message}
 
     @classmethod
-    async def success_response(cls, data):
-        return {"code": 0, "status": "task execute successfully", "data": data}
+    async def success_response(cls, task_name,  data):
+        return {"code": 0, "status": "task execute successfully", "data": data, "task_name": task_name}

routes/blueprint.py

@@ -18,6 +18,7 @@ def server_routes(pools):
     async def run_task():
         data = await request.get_json()
         task_scheduler = TaskScheduler(data)
-        return jsonify(await task_scheduler.deal())
+        response = await task_scheduler.deal()
+        return jsonify(response)
 
     return server_blueprint

tasks/task_scheduler.py

@@ -17,7 +17,7 @@ class TaskScheduler:
         match task_name:
             case "check_kimi_balance":
                 response = await check_kimi_balance()
-                return task_schedule_response.success_response(response)
+                return await task_schedule_response.success_response(task_name=task_name, data=response)
             case _:
                 return await task_schedule_response.fail_response(
                     error_code="4001",