tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Projects 1 Releases 2 Packages 1 Activity Actions

refac: shared chat user id

Browse source
This commit is contained in:
Timothy J. Baek 2024-04-02 06:33:59 -07:00
parent 488301f150
commit 48e7376336
1 changed files with 10 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

11
backend/apps/web/models/chats.py
View file

@ -108,7 +108,7 @@ class ChatTable:
shared_chat = ChatModel( shared_chat = ChatModel(
**{ **{
"id": str(uuid.uuid4()), "id": str(uuid.uuid4()),
"user_id": "shared", "user_id": f"shared-{chat_id}",
"title": chat.title, "title": chat.title,
"chat": chat.chat, "chat": chat.chat,
"timestamp": int(time.time()), "timestamp": int(time.time()),
@ -122,6 +122,15 @@ class ChatTable:
return shared_chat if (shared_result and result) else None return shared_chat if (shared_result and result) else None
def delete_shared_chat_by_chat_id(self, chat_id: str) -> bool:
try:
query = Chat.delete().where(Chat.user_id == f"shared-{chat_id}")
query.execute() # Remove the rows, return number of rows removed.
return True
except:
return False
def update_chat_share_id_by_id( def update_chat_share_id_by_id(
self, od: str, share_id: Optional[str] self, od: str, share_id: Optional[str]
) -> Optional[ChatModel]: ) -> Optional[ChatModel]: