From 62a4e94c71c0dae350e3df24c859b795ad0889a7 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 25 Sep 2023 19:58:21 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20fix(api=5Fkey.py):=20remove=20un?= =?UTF-8?q?necessary=20sa=5Frelationship=5Fkwargs=20to=20improve=20code=20?= =?UTF-8?q?readability=20and=20maintainability=20=F0=9F=94=A7=20fix(user.p?= =?UTF-8?q?y):=20add=20sa=5Frelationship=5Fkwargs=20to=20api=5Fkeys=20rela?= =?UTF-8?q?tionship=20to=20enable=20cascading=20delete=20when=20deleting?= =?UTF-8?q?=20a=20user?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../langflow/services/database/models/api_key/api_key.py | 1 - src/backend/langflow/services/database/models/user/user.py | 5 ++++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/services/database/models/api_key/api_key.py b/src/backend/langflow/services/database/models/api_key/api_key.py index 0f0d2f889..35aa6c7a9 100644 --- a/src/backend/langflow/services/database/models/api_key/api_key.py +++ b/src/backend/langflow/services/database/models/api_key/api_key.py @@ -26,7 +26,6 @@ class ApiKey(ApiKeyBase, table=True): user_id: UUID = Field(index=True, foreign_key="user.id") user: "User" = Relationship( back_populates="api_keys", - sa_relationship_kwargs={"cascade": "delete"}, ) diff --git a/src/backend/langflow/services/database/models/user/user.py b/src/backend/langflow/services/database/models/user/user.py index 5d751d080..5f2f3e1c4 100644 --- a/src/backend/langflow/services/database/models/user/user.py +++ b/src/backend/langflow/services/database/models/user/user.py @@ -21,7 +21,10 @@ class User(SQLModelSerializable, table=True): create_at: datetime = Field(default_factory=datetime.utcnow) updated_at: datetime = Field(default_factory=datetime.utcnow) last_login_at: Optional[datetime] = Field() - api_keys: list["ApiKey"] = Relationship(back_populates="user") + api_keys: list["ApiKey"] = Relationship( + back_populates="user", + sa_relationship_kwargs={"cascade": "delete"}, + ) flows: list["Flow"] = Relationship(back_populates="user")