"""投顾 Agent 与工作台之间的响应协议。"""
from __future__ import annotations

from typing import Any

from common.common_const import ERR_CODE_OK


def agent_success(data: Any = None, *, trace_id: str | None = None) -> dict:
    return {
        "code": ERR_CODE_OK,
        "message": "success",
        "data": data,
        "trace_id": trace_id,
    }


def agent_failure(
    code: int,
    message: str,
    data: Any = None,
    *,
    trace_id: str | None = None,
) -> dict:
    return {
        "code": code,
        "message": message,
        "data": data,
        "trace_id": trace_id,
    }