UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-16 13:40:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: cascade shared chat delete

Browse Source
This commit is contained in:
Timothy J. Baek 2024-04-02 08:00:26 -07:00
parent 95505bc4c1
commit f61869d90d
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
backend/apps/web/models/chats.py
View File

@ -230,10 +230,13 @@ class ChatTable:
def delete_chats_by_user_id(self, user_id: str) -> bool: def delete_chats_by_user_id(self, user_id: str) -> bool:
try: try:
self.delete_shared_chats_by_user_id(user_id)
query = Chat.delete().where(Chat.user_id == user_id) query = Chat.delete().where(Chat.user_id == user_id)
query.execute() # Remove the rows, return number of rows removed. query.execute() # Remove the rows, return number of rows removed.
return True and self.delete_shared_chats_by_user_id(user_id) return True
except: except:
return False return False
@ -244,6 +247,8 @@ class ChatTable:
for chat in Chat.select().where(Chat.user_id == user_id) for chat in Chat.select().where(Chat.user_id == user_id)
] ]
print(shared_chat_ids)
query = Chat.delete().where(Chat.user_id << shared_chat_ids) query = Chat.delete().where(Chat.user_id << shared_chat_ids)
query.execute() # Remove the rows, return number of rows removed. query.execute() # Remove the rows, return number of rows removed.