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

feat: wip chat route

Browse source
This commit is contained in:
Timothy J. Baek 2023-12-25 21:55:29 -08:00
parent 9174331025
commit 540b50e176
2 changed files with 13 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -88,7 +88,10 @@ class ChatTable:
) -> List[ChatModel]: ) -> List[ChatModel]:
return [ return [
ChatModel(**model_to_dict(chat)) ChatModel(**model_to_dict(chat))
for chat in Chat.select(Chat.user_id == user_id).limit(limit).offset(skip) for chat in Chat.select(Chat.id, Chat.title)
.where(Chat.user_id == user_id)
.limit(limit)
.offset(skip)
] ]
def get_chat_by_id_and_user_id(self, id: str, user_id: str) -> Optional[ChatModel]: def get_chat_by_id_and_user_id(self, id: str, user_id: str) -> Optional[ChatModel]:

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

@ -33,7 +33,7 @@ async def get_user_chats(skip: int = 0, limit: int = 50, cred=Depends(bearer_sch
user = Users.get_user_by_token(token) user = Users.get_user_by_token(token)
if user: if user:
return Chats.get_chat_titles_and_ids_by_user_id(user.id, skip, limit) return Chats.get_chat_lists_by_user_id(user.id, skip, limit)
else: else:
raise HTTPException( raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED, status_code=status.HTTP_401_UNAUTHORIZED,
@ -92,7 +92,14 @@ async def update_chat_by_id(
user = Users.get_user_by_token(token) user = Users.get_user_by_token(token)
if user: if user:
return Chats.update_chat_by_id_and_user_id(id, user.id, form_data.chat) chat = Chats.get_chat_by_id_and_user_id(id, user.id)
if chat:
return Chats.update_chat_by_id(id, form_data.chat)
else:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
else: else:
raise HTTPException( raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED, status_code=status.HTTP_401_UNAUTHORIZED,