37 lines
520 B
Python
37 lines
520 B
Python
from pydantic import BaseModel
|
|
|
|
|
|
class MessageBase(BaseModel):
|
|
content: str
|
|
message_type: str
|
|
|
|
|
|
class MessageCreate(MessageBase):
|
|
pass
|
|
|
|
|
|
class Message(MessageBase):
|
|
session_id: int
|
|
id: int
|
|
|
|
class Config:
|
|
orm_mode = True
|
|
|
|
|
|
class SessionBase(BaseModel):
|
|
pass
|
|
|
|
|
|
class SessionCreate(SessionBase):
|
|
location_id: str
|
|
session_data: dict | None = None
|
|
|
|
|
|
class Session(SessionBase):
|
|
id: int
|
|
messages: list[Message]
|
|
is_active: bool
|
|
|
|
class Config:
|
|
orm_mode = True
|