1 files changed, 11 insertions, 2 deletions
diff --git a/app/models/integrations.py b/app/models/integrations.py
index 4a1ddb9..f3eaca4 100644
--- a/app/models/integrations.py
+++ b/app/models/integrations.py
@@ -1,7 +1,12 @@
+from typing import TYPE_CHECKING
+
 from sqlalchemy import ForeignKey, Integer, String
-from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+
+from app.utils.db import Base
 
-Base = DeclarativeBase()
+if TYPE_CHECKING:
+    from .user import User
 
 
 class UserIntegration(Base):
@@ -17,3 +22,7 @@ class UserIntegration(Base):
     x: Mapped[str | None] = mapped_column(String(255), nullable=True)  # Twitter/X
     behance: Mapped[str | None] = mapped_column(String(255), nullable=True)
     instagram: Mapped[str | None] = mapped_column(String(255), nullable=True)
+
+    user: Mapped["User"] = relationship(
+        "User", back_populates="integrations", uselist=False
+    )