2f832b190c
* fix: remove groups from UserRead model * fix: add missing system dependencies for postgres * fix: change vector db provider environment variable name * fix: WeaviateAdapter retrieve bug * fix: correctly return data point objects from retrieve method * fix: align graph object properties * feat: add node example
35 lines
962 B
Python
35 lines
962 B
Python
from uuid import UUID as uuid_UUID
|
|
from sqlalchemy import ForeignKey, Column
|
|
from sqlalchemy.orm import relationship, Mapped
|
|
from fastapi_users.db import SQLAlchemyBaseUserTableUUID
|
|
from cognee.infrastructure.databases.relational import UUID
|
|
from .Principal import Principal
|
|
from .UserGroup import UserGroup
|
|
|
|
class User(SQLAlchemyBaseUserTableUUID, Principal):
|
|
__tablename__ = "users"
|
|
|
|
id = Column(UUID, ForeignKey("principals.id"), primary_key = True)
|
|
|
|
groups: Mapped[list["Group"]] = relationship(
|
|
secondary = UserGroup.__tablename__,
|
|
back_populates = "users",
|
|
)
|
|
|
|
__mapper_args__ = {
|
|
"polymorphic_identity": "user",
|
|
}
|
|
|
|
|
|
# Keep these schemas in sync with User model
|
|
from fastapi_users import schemas
|
|
|
|
class UserRead(schemas.BaseUser[uuid_UUID]):
|
|
# groups: list[uuid_UUID] # Add groups attribute
|
|
pass
|
|
|
|
class UserCreate(schemas.BaseUserCreate):
|
|
pass
|
|
|
|
class UserUpdate(schemas.BaseUserUpdate):
|
|
pass
|