Spaces:
Running
Running
from sqlalchemy import Boolean, Column, ForeignKey, Integer, String, Float, ARRAY, DateTime | |
from sqlalchemy.orm import relationship | |
from core.database import Base | |
from datetime import datetime | |
class User(Base): | |
__tablename__ = "users" | |
id = Column(Integer, primary_key=True, index=True) | |
email = Column(String, unique=True, index=True, nullable=False) | |
username = Column(String, unique=True, index=True, nullable=False) | |
hashed_password = Column(String, nullable=False) | |
first_name = Column(String, nullable=True) | |
last_name = Column(String, nullable=True) | |
age = Column(Integer, nullable=True) | |
preferences = Column(ARRAY(String), nullable=True) | |
is_active = Column(Boolean, default=True) | |
is_admin = Column(Boolean, default=False) | |
updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow) | |
created_at = Column(DateTime, default=datetime.utcnow) | |
orders = relationship("Order", back_populates="user") | |
embeddings = relationship("UserEmbeddings", back_populates="user", uselist=False) | |
class UserEmbeddings(Base): | |
__tablename__ = "user_embeddings" | |
id = Column(Integer, primary_key=True, index=True) | |
user_id = Column(Integer, ForeignKey("users.id"), unique=True, nullable=False) | |
embeddings = Column(ARRAY(Float), nullable=False) | |
user = relationship("User", back_populates="embeddings") | |