mirror of
https://github.com/QuivrHQ/quivr.git
synced 2024-12-15 01:21:48 +03:00
303ba72028
* add sqlalchemy models * add neon settings * add insert brain * abstract supabase from Brain class * abstract supabase from Brain class * abstract supabase from /models * update Database to Repository * update neon_tables to pg_tables * update chat, api-key and message * update vector class * update settings * update env vars for test * Update backend-tests.yml * fix test * fix fetch_user_requests_count() * fix fetch_user_requests_count() * fix increment_user_request_count * fix increment_user_request_count * fix asset upload_response message * fix pyright * fix brain_subscription * fix brain_subscription * fix brain_subscription * fix get user request stat * update create_brain_user * add delete brain vector and user * add delete brain vector and user * correctly call function --------- Co-authored-by: Noé Pion <noe.pion@onfido.com> Co-authored-by: raoufchebri <raouf@chebri.com> Co-authored-by: Stan Girard <girard.stanislas@gmail.com>
72 lines
2.1 KiB
Python
72 lines
2.1 KiB
Python
from datetime import datetime
|
|
from uuid import uuid4
|
|
|
|
from sqlalchemy import Boolean, Column, DateTime, ForeignKey, Integer, String
|
|
from sqlalchemy.ext.declarative import declarative_base
|
|
from sqlalchemy.orm import relationship
|
|
|
|
Base = declarative_base()
|
|
|
|
|
|
class User(Base):
|
|
__tablename__ = 'users'
|
|
|
|
user_id = Column(String, primary_key=True)
|
|
email = Column(String)
|
|
date = Column(DateTime)
|
|
requests_count = Column(Integer)
|
|
|
|
|
|
class Brain(Base):
|
|
__tablename__ = 'brains'
|
|
|
|
brain_id = Column(Integer, primary_key=True)
|
|
name = Column(String)
|
|
users = relationship('BrainUser', back_populates='brain')
|
|
vectors = relationship('BrainVector', back_populates='brain')
|
|
|
|
|
|
class BrainUser(Base):
|
|
__tablename__ = 'brains_users'
|
|
|
|
id = Column(Integer, primary_key=True)
|
|
user_id = Column(Integer, ForeignKey('users.user_id'))
|
|
brain_id = Column(Integer, ForeignKey('brains.brain_id'))
|
|
rights = Column(String)
|
|
|
|
user = relationship('User')
|
|
brain = relationship('Brain', back_populates='users')
|
|
|
|
|
|
class BrainVector(Base):
|
|
__tablename__ = 'brains_vectors'
|
|
|
|
vector_id = Column(String, primary_key=True, default=lambda: str(uuid4()))
|
|
brain_id = Column(Integer, ForeignKey('brains.brain_id'))
|
|
file_sha1 = Column(String)
|
|
|
|
brain = relationship('Brain', back_populates='vectors')
|
|
|
|
class BrainSubscriptionInvitation(Base):
|
|
__tablename__ = 'brain_subscription_invitations'
|
|
|
|
id = Column(Integer, primary_key=True) # Assuming an integer primary key named 'id'
|
|
brain_id = Column(String, ForeignKey('brains.brain_id'))
|
|
email = Column(String, ForeignKey('users.email'))
|
|
rights = Column(String)
|
|
|
|
brain = relationship('Brain')
|
|
user = relationship('User', foreign_keys=[email])
|
|
|
|
class ApiKey(Base):
|
|
__tablename__ = 'api_keys'
|
|
|
|
key_id = Column(String, primary_key=True, default=lambda: str(uuid4()))
|
|
user_id = Column(Integer, ForeignKey('users.user_id'))
|
|
api_key = Column(String, unique=True)
|
|
creation_time = Column(DateTime, default=datetime.utcnow)
|
|
is_active = Column(Boolean, default=True)
|
|
deleted_time = Column(DateTime, nullable=True)
|
|
|
|
user = relationship('User')
|