from pydantic import BaseModel

from ocr.api.message.models import MessageModel
from ocr.api.report.dto import Paging
from ocr.core.wrappers import OcrResponseWrapper


class CreateMessageRequest(BaseModel):
    text: str

class MessageWrapper(OcrResponseWrapper[MessageModel]):
    pass


class AllMessageResponse(BaseModel):
    paging: Paging
    data: list[MessageModel]


class AllMessageWrapper(OcrResponseWrapper[AllMessageResponse]):
    pass