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

feat: update shared chat

Browse source
This commit is contained in:
Timothy J. Baek 2024-04-02 07:42:37 -07:00
parent 0b823f90e6
commit 9975cb17a9
3 changed files with 23 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -98,7 +98,7 @@ class ChatTable:
except:
return None
def insert_shared_chat(self, chat_id: str) -> Optional[ChatModel]:
def insert_shared_chat_by_chat_id(self, chat_id: str) -> Optional[ChatModel]:
# Get the existing chat to share
chat = Chat.get(Chat.id == chat_id)
# Check if the chat is already shared
@ -122,6 +122,24 @@ class ChatTable:
return shared_chat if (shared_result and result) else None
def update_shared_chat_by_chat_id(self, chat_id: str) -> Optional[ChatModel]:
try:
print("update_shared_chat_by_id")
chat = Chat.get(Chat.id == chat_id)
print(chat)
query = Chat.update(
title=chat.title,
chat=chat.chat,
).where(Chat.id == chat.share_id)
query.execute()
chat = Chat.get(Chat.id == chat.share_id)
return ChatModel(**model_to_dict(chat))
except:
return 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}")