from typing import Any, Optional
from fastapi.responses import JSONResponse
from pydantic import BaseModel

class ResponseModel(BaseModel):
    code: int = 200
    message: str = "success"
    data: Optional[Any] = None

def success_response(*, data: Any = None, message: str = "success") -> dict:
    return ResponseModel(
        code=200,
        message=message,
        data=data
    ).model_dump()

def error_response(*, code: int = 400, message: str) -> dict:
    return ResponseModel(
        code=code,
        message=message
    ).model_dump()