self-constructing_graph/agent/libs/response.py

from fastapi import status
from fastapi.responses import JSONResponse, Response, ORJSONResponse  # , ORJSONResponse
from typing import Union
from datetime import datetime
from json import JSONEncoder

SUCCESS_CODE = 200
SUCCESS_MESSAGE = "Operation successful"

    
# 注意有个 * 号 不是笔误， 意思是调用的时候要指定参数 e.g.resp_200（data=xxxx)
def resp_200(*, data: Union[list, dict, str]) -> Response:
    return ORJSONResponse(
        status_code=status.HTTP_200_OK,
        content={
            'code': 200,
            'message': SUCCESS_MESSAGE,
            'data': data,
        }
    )
    
def resp_400(*, data: str = None, message: str="BAD REQUEST") -> Response:
    return ORJSONResponse(
        status_code=status.HTTP_200_OK,
        content={
            'code': 400,
            'message': message,
            'data': data,
        }
    )