File size: 1,932 Bytes
67f8819 2c9444f 67f8819 | 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 56 57 58 59 60 61 62 63 64 65 66 67 68 69 | """
Conversation model representing a user's chat session with AI.
Per @specs/001-chatbot-mcp/data-model.md
"""
from sqlmodel import SQLModel, Field, Relationship
from typing import TYPE_CHECKING, Optional
from datetime import datetime
from uuid import UUID, uuid4
if TYPE_CHECKING:
from models.user import UserTable
from models.message import MessageTable
class ConversationTable(SQLModel, table=True):
"""
A chat session between a user and the AI assistant.
All conversations MUST be scoped to a single user to ensure data isolation
per constitutional principle III (User Isolation via JWT).
"""
__tablename__ = "conversations"
# Primary key
id: UUID = Field(
default_factory=uuid4,
primary_key=True,
index=True,
description="Unique conversation identifier"
)
# Foreign key to User
user_id: UUID = Field(
foreign_key="users.id",
index=True,
nullable=False,
description="ID of the user who owns this conversation"
)
# Conversation attributes
title: Optional[str] = Field(
default=None,
max_length=255,
description="Auto-generated title from first message (e.g., 'Grocery shopping')"
)
# Timestamps
created_at: datetime = Field(
default_factory=datetime.utcnow,
nullable=False,
description="Timestamp when conversation was created"
)
updated_at: datetime = Field(
default_factory=datetime.utcnow,
sa_column_kwargs={"onupdate": datetime.utcnow},
nullable=False,
index=True,
description="Timestamp of last message in conversation"
)
# Relationships
user: "UserTable" = Relationship(back_populates="conversations")
messages: list["MessageTable"] = Relationship(
back_populates="conversation",
sa_relationship_kwargs={"cascade": "all, delete-orphan"}
)
|