Spaces:
Running
Running
File size: 1,283 Bytes
b82f276 0a7afc4 b82f276 | 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 | from sqlalchemy import Column, Integer, String, Text, DateTime, ForeignKey
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)
username = Column(String, unique=True, index=True)
email = Column(String, unique=True, index=True)
hashed_password = Column(String)
role = Column(String)
mobile_number = Column(String, nullable=True)
reset_token = Column(String, nullable=True)
question_count = Column(Integer, default=0)
# Relationship to chat interactions
interactions = relationship("ChatInteraction", back_populates="user")
class ChatInteraction(Base):
__tablename__ = "chat_interactions"
id = Column(Integer, primary_key=True, index=True)
user_id = Column(Integer, ForeignKey("users.id"), nullable=True)
case_id = Column(String, nullable=True)
role = Column(String, index=True, nullable=False) # Role context for this interaction
query = Column(Text)
response = Column(Text)
created_at = Column(DateTime, default=datetime.utcnow)
# Relationship to user
user = relationship("User", back_populates="interactions")
|