diff --git a/src/backend/langflow/api/v1/users.py b/src/backend/langflow/api/v1/users.py index 9541bb2d4..aef639b0a 100644 --- a/src/backend/langflow/api/v1/users.py +++ b/src/backend/langflow/api/v1/users.py @@ -67,7 +67,7 @@ def read_current_user( def read_all_users( skip: int = 0, limit: int = 10, - current_user: Session = Depends(get_current_active_superuser), + _: Session = Depends(get_current_active_superuser), session: Session = Depends(get_session), ) -> UsersResponse: """ 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 5d5bab0f4..0f0d2f889 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 @@ -22,8 +22,12 @@ class ApiKey(ApiKeyBase, table=True): api_key: str = Field(index=True, unique=True) # User relationship + # Delete API keys when user is deleted user_id: UUID = Field(index=True, foreign_key="user.id") - user: "User" = Relationship(back_populates="api_keys") + user: "User" = Relationship( + back_populates="api_keys", + sa_relationship_kwargs={"cascade": "delete"}, + ) class ApiKeyCreate(ApiKeyBase):