from pydantic import BaseModel

class PayloadRequest(BaseModel):
    group_name: str
    password: str
    data: str = None # Optional for retrieve, required for store

class RetrieveGroupRequest(BaseModel):
    group_name: str
    password: str
    burn: bool = False # <--- NEW FLAG
    limit: int = 100