trauma/api/chat/schemas.py

from pydantic import BaseModel

from trauma.api.chat.dto import ModelType
from trauma.api.chat.model import ChatModel
from trauma.api.common.dto import Paging
from trauma.core.wrappers import TraumaResponseWrapper


class CreateChatRequest(BaseModel):
    model: ModelType = ModelType.gpt_4o_mini


class ChatWrapper(TraumaResponseWrapper[ChatModel]):
    pass


class AllChatResponse(BaseModel):
    paging: Paging
    data: list[ChatModel]


class AllChatWrapper(TraumaResponseWrapper[AllChatResponse]):
    pass


class ChatTitleRequest(BaseModel):
    title: str