long_articles_agentic_supply/supply/infra/shared/response.py

from typing import Any, Dict


class TaskScheduleResponse:
    """任务调度专用响应格式"""

    @classmethod
    def fail(cls, error_code: str, error_message: str) -> Dict[str, Any]:
        return {"code": error_code, "status": "error", "message": error_message}

    @classmethod
    def success(cls, task_name: str, data: Any) -> Dict[str, Any]:
        return {
            "code": 0,
            "status": "task execute successfully",
            "data": data,
            "task_name": task_name,
        }