address edge

backend/alembic/versions/d575d732e4ba_update_user_file_cc_pair_constraint_to_.py

@@ -0,0 +1,44 @@
+"""update_user_file_cc_pair_constraint_to_cascade
+
+Revision ID: d575d732e4ba
+Revises: cf90764725d8
+Create Date: 2025-04-09 11:15:59.490947
+
+"""
+from alembic import op
+
+
+# revision identifiers, used by Alembic.
+revision = "d575d732e4ba"
+down_revision = "cf90764725d8"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    # Drop the existing foreign key constraint
+    op.drop_constraint("user_file_cc_pair_id_fkey", "user_file", type_="foreignkey")
+
+    # Recreate it with ON DELETE CASCADE
+    op.create_foreign_key(
+        "user_file_cc_pair_id_fkey",
+        "user_file",
+        "connector_credential_pair",
+        ["cc_pair_id"],
+        ["id"],
+        ondelete="CASCADE",
+    )
+
+
+def downgrade() -> None:
+    # Drop the CASCADE constraint
+    op.drop_constraint("user_file_cc_pair_id_fkey", "user_file", type_="foreignkey")
+
+    # Recreate the original constraint without CASCADE
+    op.create_foreign_key(
+        "user_file_cc_pair_id_fkey",
+        "user_file",
+        "connector_credential_pair",
+        ["cc_pair_id"],
+        ["id"],
+    )

backend/onyx/db/models.py

@@ -2336,7 +2336,7 @@ class User__ExternalUserGroupId(Base):
     # These group ids have been prefixed by the source type
     external_user_group_id: Mapped[str] = mapped_column(String, primary_key=True)
     cc_pair_id: Mapped[int] = mapped_column(
-        ForeignKey("connector_credential_pair.id"), primary_key=True
+        ForeignKey("connector_credential_pair.id", ondelete="CASCADE"), primary_key=True
     )
 
 
@@ -2443,7 +2443,9 @@ class UserFile(Base):
     token_count: Mapped[int | None] = mapped_column(Integer, nullable=True)
 
     cc_pair_id: Mapped[int | None] = mapped_column(
-        ForeignKey("connector_credential_pair.id"), nullable=True, unique=True
+        ForeignKey("connector_credential_pair.id", ondelete="CASCADE"),
+        nullable=True,
+        unique=True,
     )
     cc_pair: Mapped["ConnectorCredentialPair"] = relationship(
         "ConnectorCredentialPair", back_populates="user_file"