from datetime import datetime
from sqlmodel import SQLModel, Field

class ChatMessage(SQLModel, table=True):
    id: int = Field(default=None, primary_key=True)
    user: str
    prompt: str
    response: str
    timestamp: datetime = Field(default_factory=datetime.utcnow)