forked from open-webui/open-webui
fix: cascade shared chat delete
This commit is contained in:
parent
95505bc4c1
commit
f61869d90d
1 changed files with 6 additions and 1 deletions
|
@ -230,10 +230,13 @@ class ChatTable:
|
|||
|
||||
def delete_chats_by_user_id(self, user_id: str) -> bool:
|
||||
try:
|
||||
|
||||
self.delete_shared_chats_by_user_id(user_id)
|
||||
|
||||
query = Chat.delete().where(Chat.user_id == user_id)
|
||||
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:
|
||||
return False
|
||||
|
||||
|
@ -244,6 +247,8 @@ class ChatTable:
|
|||
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.execute() # Remove the rows, return number of rows removed.
|
||||
|
||||
|
|
Loading…
Reference in a new issue