Spaces:
Sleeping
Sleeping
File size: 1,449 Bytes
b93a6a5 | 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 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | """
Message model for AI chatbot.
Represents a single message in a conversation (either from user or assistant).
"""
from sqlmodel import SQLModel, Field, Relationship
from datetime import datetime
from typing import Optional, Literal, TYPE_CHECKING
if TYPE_CHECKING:
from src.models.conversation import Conversation
from src.models.user import User
class Message(SQLModel, table=True):
"""
Message model representing a single message in a conversation.
Messages are immutable once created (no editing).
Role must be either 'user' or 'assistant'.
"""
__tablename__ = "messages"
id: Optional[int] = Field(default=None, primary_key=True)
conversation_id: int = Field(
foreign_key="conversations.id",
index=True,
nullable=False
)
user_id: int = Field(
foreign_key="users.id",
index=True,
nullable=False
)
role: str = Field(
max_length=20,
nullable=False,
description="Message sender role: 'user' or 'assistant'"
)
content: str = Field(
max_length=10000,
nullable=False,
description="Message text content"
)
created_at: datetime = Field(
default_factory=datetime.utcnow,
nullable=False,
index=True
)
# Relationships
conversation: Optional["Conversation"] = Relationship(back_populates="messages")
user: Optional["User"] = Relationship()
|