UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-01-19 09:16:44 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2024-12-31 13:04:27 -08:00
parent 3d69eaf34f
commit d905202afb
1 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
backend/open_webui/models/messages.py
View File

@ -251,6 +251,18 @@ class MessageTable:
db.commit()
return True
def delete_reactions_by_id(self, id: str) -> bool:
with get_db() as db:
db.query(MessageReaction).filter_by(message_id=id).delete()
db.commit()
return True
def delete_replies_by_id(self, id: str) -> bool:
with get_db() as db:
db.query(Message).filter_by(parent_id=id).delete()
db.commit()
return True
def delete_message_by_id(self, id: str) -> bool:
with get_db() as db:
db.query(Message).filter_by(id=id).delete()
@ -258,6 +270,16 @@ class MessageTable:
# Delete all reactions to this message
db.query(MessageReaction).filter_by(message_id=id).delete()
# List all replies to this message
replies = db.query(Message).filter_by(parent_id=id).all()
# Delete all reactions to each reply
for reply in replies:
db.query(MessageReaction).filter_by(message_id=reply.id).delete()
# Delete all replies to this message
db.query(Message).filter_by(parent_id=id).delete()
db.commit()
return True