Mercurial
diff love/poppy/utils/models.py @ 38:cf9caa4abc3e
[Love] FE and BE. Can chat and render images. Also created MCP for powerpoint generations.
| author | MrJuneJune <me@mrjunejune.com> |
|---|---|
| date | Mon, 01 Dec 2025 20:35:56 -0800 |
| parents | |
| children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/love/poppy/utils/models.py Mon Dec 01 20:35:56 2025 -0800 @@ -0,0 +1,21 @@ +from sqlmodel import SQLModel, Field, Relationship +from typing import Optional, List +from datetime import datetime + + +class Message(SQLModel, table=True): + id: Optional[int] = Field(default=None, primary_key=True) + chat_id: int = Field(foreign_key="chat.id") + role: str = Field(index=True) # "user" | "assistant" + content: str + created_at: datetime = Field(default_factory=datetime.utcnow) + + chat: Optional["Chat"] = Relationship(back_populates="messages") + + +class Chat(SQLModel, table=True): + id: Optional[int] = Field(default=None, primary_key=True) + title: str = Field(default="New Chat") + created_at: datetime = Field(default_factory=datetime.utcnow) + + messages: List[Message] = Relationship(back_populates="chat")