|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
from typing import Union, Optional, List
|
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field, model_validator
|
|
|
|
|
from module_admin.annotation.pydantic_annotation import as_query
|
|
|
|
|
from pydantic.alias_generators import to_camel
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class CrudChatModel(BaseModel):
|
|
|
|
|
is_success: bool
|
|
|
|
|
message: str
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@as_query
|
|
|
|
|
class AiListQuery(BaseModel):
|
|
|
|
|
model_config = ConfigDict(alias_generator=to_camel, from_attributes=True)
|
|
|
|
|
session_id: str
|
|
|
|
|
page_num: int
|
|
|
|
|
page_size: int
|
|
|
|
|
|
|
|
|
|
class AiChatModel(BaseModel):
|
|
|
|
|
"""
|
|
|
|
|
菜单表对应pydantic模型
|
|
|
|
|
"""
|
|
|
|
|
chatId: Optional[str] = None
|
|
|
|
|
sessionId: Optional[str] = None
|
|
|
|
|
sessionName: Optional[str] = None
|
|
|
|
|
type: Optional[str] = None
|
|
|
|
|
isEnd: Optional[bool] = None
|
|
|
|
|
isStop: Optional[bool] = None
|
|
|
|
|
user: Optional[int] = None
|
|
|
|
|
time: Optional[str] = None
|
|
|
|
|
content: Optional[str] = None
|
|
|
|
|
operate: Optional[str] = None
|
|
|
|
|
thumbDownReason: Optional[str] = None
|
|
|
|
|
file: Optional[str] = None
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class ThumbOperateModel(BaseModel):
|
|
|
|
|
chatId: str
|
|
|
|
|
operate: str
|
|
|
|
|
thumbDownReason: str
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class DownloadFile(BaseModel):
|
|
|
|
|
file: str
|
|
|
|
|
bucket: str
|
|
|
|
|
sessionId: str
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class AiChatListModel(BaseModel):
|
|
|
|
|
chats: List[Union[AiChatModel, None]]
|
