from pydantic import BaseModel
from typing import Any

class EncryptRequest(BaseModel):
    payload: Any  # Accepts any JSON serializable type

class EncryptResponse(BaseModel):
    success: bool = True
    token: str

class DecryptRequest(BaseModel):
    token: str

class DecryptResponse(BaseModel):
    success: bool = True
    payload: Any