39 lines
801 B
Python
39 lines
801 B
Python
|
from pydantic import BaseModel
|
||
|
|
from typing import List, Optional
|
||
|
|
from datetime import datetime
|
||
|
|
|
||
|
|
class MessageBase(BaseModel):
|
||
|
|
content: str
|
||
|
|
role: str
|
||
|
|
|
||
|
|
class MessageCreate(MessageBase):
|
||
|
|
chat_id: int
|
||
|
|
|
||
|
|
class MessageResponse(MessageBase):
|
||
|
|
id: int
|
||
|
|
chat_id: int
|
||
|
|
created_at: datetime
|
||
|
|
updated_at: datetime
|
||
|
|
|
||
|
|
class Config:
|
||
|
|
from_attributes = True
|
||
|
|
|
||
|
|
class ChatBase(BaseModel):
|
||
|
|
title: str
|
||
|
|
|
||
|
|
class ChatCreate(ChatBase):
|
||
|
|
knowledge_base_ids: List[int]
|
||
|
|
|
||
|
|
class ChatUpdate(ChatBase):
|
||
|
|
knowledge_base_ids: Optional[List[int]] = None
|
||
|
|
|
||
|
|
class ChatResponse(ChatBase):
|
||
|
|
id: int
|
||
|
|
user_id: int
|
||
|
|
created_at: datetime
|
||
|
|
updated_at: datetime
|
||
|
|
messages: List[MessageResponse] = []
|
||
|
|
knowledge_base_ids: List[int] = []
|
||
|
|
|
||
|
|
class Config:
|
||
|
|
from_attributes = True
|