Spaces:
Sleeping
Sleeping
File size: 566 Bytes
50553ea |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
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 |