Spaces:
Running
Running
File size: 2,492 Bytes
a63c61f | 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 | from sqlalchemy import Column, Integer, String, DateTime, Text, ForeignKey, CheckConstraint, Boolean, Enum
from sqlalchemy.orm import declarative_base, relationship
from datetime import datetime
import enum
Base = declarative_base()
class UserRole(str, enum.Enum):
super_admin = "super_admin"
user = "user"
class User(Base):
__tablename__ = "users"
id = Column(Integer, primary_key=True, index=True)
username = Column(String, unique=True, index=True, nullable=False)
email = Column(String, unique=True, index=True, nullable=False)
hashed_password = Column(String, nullable=False)
role = Column(Enum(UserRole), default=UserRole.user, nullable=False)
is_active = Column(Boolean, default=True, nullable=False)
full_name = Column(String, nullable=True)
created_at = Column(DateTime, default=datetime.utcnow)
updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
refresh_tokens = relationship("RefreshToken", back_populates="user", cascade="all, delete-orphan")
class RefreshToken(Base):
__tablename__ = "refresh_tokens"
id = Column(Integer, primary_key=True, index=True)
token = Column(String, unique=True, index=True, nullable=False)
user_id = Column(Integer, ForeignKey("users.id"), nullable=False)
expires_at = Column(DateTime, nullable=False)
revoked = Column(Boolean, default=False, nullable=False)
created_at = Column(DateTime, default=datetime.utcnow)
user = relationship("User", back_populates="refresh_tokens")
class ChatHistory(Base):
__tablename__ = "chat_history"
id = Column(Integer, primary_key=True, index=True)
session_id = Column(String, index=True, nullable=False)
user_id = Column(Integer, ForeignKey("users.id"), nullable=True)
role = Column(String, nullable=False)
content = Column(Text, nullable=False)
retrieved_doc_ids = Column(Text, nullable=True)
pinned = Column(Boolean, default=False, nullable=False)
timestamp = Column(DateTime, default=datetime.utcnow)
class Feedback(Base):
__tablename__ = "feedback"
id = Column(Integer, primary_key=True, index=True)
session_id = Column(String, index=True, nullable=False)
message_id = Column(Integer, nullable=False)
user_id = Column(Integer, ForeignKey("users.id"), nullable=True)
rating = Column(Integer, CheckConstraint('rating IN (1, -1)'))
comment = Column(Text, nullable=True)
timestamp = Column(DateTime, default=datetime.utcnow)
|