open-webui/backend/apps/web/routers/chats.py

287 lines
7.5 KiB
Python
Raw Normal View History

from fastapi import Depends, Request, HTTPException, status
2023-12-26 06:44:28 +01:00
from datetime import datetime, timedelta
from typing import List, Union, Optional
from utils.utils import get_current_user, get_admin_user
2023-12-26 06:44:28 +01:00
from fastapi import APIRouter
from pydantic import BaseModel
2023-12-26 10:27:43 +01:00
import json
2023-12-26 06:44:28 +01:00
from apps.web.models.users import Users
from apps.web.models.chats import (
ChatModel,
2023-12-26 10:27:43 +01:00
ChatResponse,
2023-12-26 19:41:55 +01:00
ChatTitleForm,
2023-12-26 06:44:28 +01:00
ChatForm,
ChatTitleIdResponse,
Chats,
)
2024-01-18 09:58:45 +01:00
from apps.web.models.tags import (
TagModel,
2024-01-18 11:10:07 +01:00
ChatIdTagModel,
2024-01-18 09:58:45 +01:00
ChatIdTagForm,
ChatTagsResponse,
Tags,
)
2023-12-26 06:44:28 +01:00
from constants import ERROR_MESSAGES
router = APIRouter()
############################
# GetChats
############################
@router.get("/", response_model=List[ChatTitleIdResponse])
async def get_user_chats(
2024-01-17 23:47:56 +01:00
user=Depends(get_current_user), skip: int = 0, limit: int = 50
):
return Chats.get_chat_lists_by_user_id(user.id, skip, limit)
2023-12-26 06:44:28 +01:00
2023-12-27 07:10:22 +01:00
############################
# GetAllChats
############################
@router.get("/all", response_model=List[ChatResponse])
async def get_all_user_chats(user=Depends(get_current_user)):
return [
2024-01-17 23:47:56 +01:00
ChatResponse(**{**chat.model_dump(), "chat": json.loads(chat.chat)})
for chat in Chats.get_all_chats_by_user_id(user.id)
]
2023-12-27 07:10:22 +01:00
############################
# GetAllChatsInDB
############################
@router.get("/all/db", response_model=List[ChatResponse])
async def get_all_user_chats_in_db(user=Depends(get_admin_user)):
return [
ChatResponse(**{**chat.model_dump(), "chat": json.loads(chat.chat)})
for chat in Chats.get_all_chats()
]
2023-12-26 06:44:28 +01:00
############################
# CreateNewChat
############################
2023-12-26 10:27:43 +01:00
@router.post("/new", response_model=Optional[ChatResponse])
async def create_new_chat(form_data: ChatForm, user=Depends(get_current_user)):
2024-01-17 23:47:56 +01:00
try:
chat = Chats.insert_new_chat(user.id, form_data)
return ChatResponse(**{**chat.model_dump(), "chat": json.loads(chat.chat)})
except Exception as e:
print(e)
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST, detail=ERROR_MESSAGES.DEFAULT()
)
2023-12-26 06:44:28 +01:00
2024-01-18 11:55:25 +01:00
############################
# GetAllTags
############################
@router.get("/tags/all", response_model=List[TagModel])
async def get_all_tags(user=Depends(get_current_user)):
try:
tags = Tags.get_tags_by_user_id(user.id)
return tags
except Exception as e:
print(e)
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST, detail=ERROR_MESSAGES.DEFAULT()
)
############################
# GetChatsByTags
############################
@router.get("/tags/tag/{tag_name}", response_model=List[ChatTitleIdResponse])
async def get_user_chats_by_tag_name(
tag_name: str, user=Depends(get_current_user), skip: int = 0, limit: int = 50
):
chat_ids = [
chat_id_tag.chat_id
for chat_id_tag in Tags.get_chat_ids_by_tag_name_and_user_id(tag_name, user.id)
]
2024-02-25 22:03:26 +01:00
chats = Chats.get_chat_lists_by_chat_ids(chat_ids, skip, limit)
2024-01-18 11:55:25 +01:00
2024-02-25 22:03:26 +01:00
if len(chats) == 0:
Tags.delete_tag_by_tag_name_and_user_id(tag_name, user.id)
return chats
2024-01-18 11:55:25 +01:00
2023-12-26 06:44:28 +01:00
############################
# GetChatById
############################
2023-12-26 10:27:43 +01:00
@router.get("/{id}", response_model=Optional[ChatResponse])
async def get_chat_by_id(id: str, user=Depends(get_current_user)):
chat = Chats.get_chat_by_id_and_user_id(id, user.id)
2023-12-26 06:44:28 +01:00
if chat:
2024-01-17 23:47:56 +01:00
return ChatResponse(**{**chat.model_dump(), "chat": json.loads(chat.chat)})
2023-12-26 06:44:28 +01:00
else:
2024-01-17 23:47:56 +01:00
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED, detail=ERROR_MESSAGES.NOT_FOUND
)
2023-12-26 06:44:28 +01:00
############################
# UpdateChatById
############################
2023-12-26 10:27:43 +01:00
@router.post("/{id}", response_model=Optional[ChatResponse])
2024-01-17 23:47:56 +01:00
async def update_chat_by_id(
id: str, form_data: ChatForm, user=Depends(get_current_user)
):
chat = Chats.get_chat_by_id_and_user_id(id, user.id)
if chat:
updated_chat = {**json.loads(chat.chat), **form_data.chat}
2023-12-26 19:41:55 +01:00
chat = Chats.update_chat_by_id(id, updated_chat)
2024-01-17 23:47:56 +01:00
return ChatResponse(**{**chat.model_dump(), "chat": json.loads(chat.chat)})
else:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
2023-12-26 10:27:43 +01:00
############################
# DeleteChatById
############################
@router.delete("/{id}", response_model=bool)
2024-02-14 10:17:43 +01:00
async def delete_chat_by_id(request: Request, id: str, user=Depends(get_current_user)):
if (
user.role == "user"
and not request.app.state.USER_PERMISSIONS["chat"]["deletion"]
):
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
result = Chats.delete_chat_by_id_and_user_id(id, user.id)
return result
2023-12-30 09:15:37 +01:00
2024-01-18 09:58:45 +01:00
############################
# GetChatTagsById
############################
@router.get("/{id}/tags", response_model=List[TagModel])
async def get_chat_tags_by_id(id: str, user=Depends(get_current_user)):
tags = Tags.get_tags_by_chat_id_and_user_id(id, user.id)
2024-01-18 11:10:07 +01:00
if tags != None:
2024-01-18 09:58:45 +01:00
return tags
else:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED, detail=ERROR_MESSAGES.NOT_FOUND
)
############################
# AddChatTagById
############################
2024-01-18 11:10:07 +01:00
@router.post("/{id}/tags", response_model=Optional[ChatIdTagModel])
2024-01-18 09:58:45 +01:00
async def add_chat_tag_by_id(
id: str, form_data: ChatIdTagForm, user=Depends(get_current_user)
):
2024-01-18 11:10:07 +01:00
tags = Tags.get_tags_by_chat_id_and_user_id(id, user.id)
if form_data.tag_name not in tags:
tag = Tags.add_tag_to_chat(user.id, form_data)
2024-01-18 09:58:45 +01:00
2024-01-18 11:10:07 +01:00
if tag:
return tag
else:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
detail=ERROR_MESSAGES.NOT_FOUND,
)
2024-01-18 09:58:45 +01:00
else:
raise HTTPException(
2024-01-18 11:10:07 +01:00
status_code=status.HTTP_401_UNAUTHORIZED, detail=ERROR_MESSAGES.DEFAULT()
2024-01-18 09:58:45 +01:00
)
############################
# DeleteChatTagById
############################
@router.delete("/{id}/tags", response_model=Optional[bool])
2024-01-18 10:04:24 +01:00
async def delete_chat_tag_by_id(
2024-01-18 09:58:45 +01:00
id: str, form_data: ChatIdTagForm, user=Depends(get_current_user)
):
2024-01-18 10:04:24 +01:00
result = Tags.delete_tag_by_tag_name_and_chat_id_and_user_id(
2024-01-18 09:58:45 +01:00
form_data.tag_name, id, user.id
)
2024-01-18 10:04:24 +01:00
if result:
return result
else:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED, detail=ERROR_MESSAGES.NOT_FOUND
)
############################
# DeleteAllChatTagsById
############################
@router.delete("/{id}/tags/all", response_model=Optional[bool])
async def delete_all_chat_tags_by_id(id: str, user=Depends(get_current_user)):
result = Tags.delete_tags_by_chat_id_and_user_id(id, user.id)
if result:
return result
2024-01-18 09:58:45 +01:00
else:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED, detail=ERROR_MESSAGES.NOT_FOUND
)
2023-12-30 09:15:37 +01:00
############################
# DeleteAllChats
############################
@router.delete("/", response_model=bool)
2024-03-02 09:07:50 +01:00
async def delete_all_user_chats(request: Request, user=Depends(get_current_user)):
if (
user.role == "user"
and not request.app.state.USER_PERMISSIONS["chat"]["deletion"]
):
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
detail=ERROR_MESSAGES.ACCESS_PROHIBITED,
)
result = Chats.delete_chats_by_user_id(user.id)
return result