models.py•1.01 kB
from sqlalchemy import Column, Integer, String, DateTime, ForeignKey, Text
from sqlalchemy.orm import relationship
from datetime import datetime
from database import Base
class User(Base):
__tablename__ = "users"
id = Column(Integer, primary_key=True, index=True)
email = Column(String, unique=True, index=True)
name = Column(String)
subscription_tier = Column(String, default="free")
created_at = Column(DateTime, default=datetime.utcnow)
# Relationships
chat_messages = relationship("ChatMessage", back_populates="user")
class ChatMessage(Base):
__tablename__ = "chat_messages"
id = Column(Integer, primary_key=True, index=True)
user_id = Column(Integer, ForeignKey("users.id"))
message = Column(Text)
response = Column(Text)
context = Column(Text) # Store as JSON string
timestamp = Column(DateTime, default=datetime.utcnow)
# Relationships
user = relationship("User", back_populates="chat_messages")
