|
@@ -1,6 +1,6 @@
|
|
from fastapi import Depends, Request, HTTPException, status
|
|
from fastapi import Depends, Request, HTTPException, status
|
|
from datetime import datetime, timedelta
|
|
from datetime import datetime, timedelta
|
|
-from typing import List, Union, Optional
|
|
|
|
|
|
+from typing import Union, Optional
|
|
from utils.utils import get_verified_user, get_admin_user
|
|
from utils.utils import get_verified_user, get_admin_user
|
|
from fastapi import APIRouter
|
|
from fastapi import APIRouter
|
|
from pydantic import BaseModel
|
|
from pydantic import BaseModel
|
|
@@ -40,8 +40,8 @@ router = APIRouter()
|
|
############################
|
|
############################
|
|
|
|
|
|
|
|
|
|
-@router.get("/", response_model=List[ChatTitleIdResponse])
|
|
|
|
-@router.get("/list", response_model=List[ChatTitleIdResponse])
|
|
|
|
|
|
+@router.get("/", response_model=list[ChatTitleIdResponse])
|
|
|
|
+@router.get("/list", response_model=list[ChatTitleIdResponse])
|
|
async def get_session_user_chat_list(
|
|
async def get_session_user_chat_list(
|
|
user=Depends(get_verified_user), page: Optional[int] = None
|
|
user=Depends(get_verified_user), page: Optional[int] = None
|
|
):
|
|
):
|
|
@@ -80,7 +80,7 @@ async def delete_all_user_chats(request: Request, user=Depends(get_verified_user
|
|
############################
|
|
############################
|
|
|
|
|
|
|
|
|
|
-@router.get("/list/user/{user_id}", response_model=List[ChatTitleIdResponse])
|
|
|
|
|
|
+@router.get("/list/user/{user_id}", response_model=list[ChatTitleIdResponse])
|
|
async def get_user_chat_list_by_user_id(
|
|
async def get_user_chat_list_by_user_id(
|
|
user_id: str,
|
|
user_id: str,
|
|
user=Depends(get_admin_user),
|
|
user=Depends(get_admin_user),
|
|
@@ -119,7 +119,7 @@ async def create_new_chat(form_data: ChatForm, user=Depends(get_verified_user)):
|
|
############################
|
|
############################
|
|
|
|
|
|
|
|
|
|
-@router.get("/all", response_model=List[ChatResponse])
|
|
|
|
|
|
+@router.get("/all", response_model=list[ChatResponse])
|
|
async def get_user_chats(user=Depends(get_verified_user)):
|
|
async def get_user_chats(user=Depends(get_verified_user)):
|
|
return [
|
|
return [
|
|
ChatResponse(**{**chat.model_dump(), "chat": json.loads(chat.chat)})
|
|
ChatResponse(**{**chat.model_dump(), "chat": json.loads(chat.chat)})
|
|
@@ -132,7 +132,7 @@ async def get_user_chats(user=Depends(get_verified_user)):
|
|
############################
|
|
############################
|
|
|
|
|
|
|
|
|
|
-@router.get("/all/archived", response_model=List[ChatResponse])
|
|
|
|
|
|
+@router.get("/all/archived", response_model=list[ChatResponse])
|
|
async def get_user_archived_chats(user=Depends(get_verified_user)):
|
|
async def get_user_archived_chats(user=Depends(get_verified_user)):
|
|
return [
|
|
return [
|
|
ChatResponse(**{**chat.model_dump(), "chat": json.loads(chat.chat)})
|
|
ChatResponse(**{**chat.model_dump(), "chat": json.loads(chat.chat)})
|
|
@@ -145,7 +145,7 @@ async def get_user_archived_chats(user=Depends(get_verified_user)):
|
|
############################
|
|
############################
|
|
|
|
|
|
|
|
|
|
-@router.get("/all/db", response_model=List[ChatResponse])
|
|
|
|
|
|
+@router.get("/all/db", response_model=list[ChatResponse])
|
|
async def get_all_user_chats_in_db(user=Depends(get_admin_user)):
|
|
async def get_all_user_chats_in_db(user=Depends(get_admin_user)):
|
|
if not ENABLE_ADMIN_EXPORT:
|
|
if not ENABLE_ADMIN_EXPORT:
|
|
raise HTTPException(
|
|
raise HTTPException(
|
|
@@ -163,7 +163,7 @@ async def get_all_user_chats_in_db(user=Depends(get_admin_user)):
|
|
############################
|
|
############################
|
|
|
|
|
|
|
|
|
|
-@router.get("/archived", response_model=List[ChatTitleIdResponse])
|
|
|
|
|
|
+@router.get("/archived", response_model=list[ChatTitleIdResponse])
|
|
async def get_archived_session_user_chat_list(
|
|
async def get_archived_session_user_chat_list(
|
|
user=Depends(get_verified_user), skip: int = 0, limit: int = 50
|
|
user=Depends(get_verified_user), skip: int = 0, limit: int = 50
|
|
):
|
|
):
|
|
@@ -216,7 +216,7 @@ class TagNameForm(BaseModel):
|
|
limit: Optional[int] = 50
|
|
limit: Optional[int] = 50
|
|
|
|
|
|
|
|
|
|
-@router.post("/tags", response_model=List[ChatTitleIdResponse])
|
|
|
|
|
|
+@router.post("/tags", response_model=list[ChatTitleIdResponse])
|
|
async def get_user_chat_list_by_tag_name(
|
|
async def get_user_chat_list_by_tag_name(
|
|
form_data: TagNameForm, user=Depends(get_verified_user)
|
|
form_data: TagNameForm, user=Depends(get_verified_user)
|
|
):
|
|
):
|
|
@@ -241,7 +241,7 @@ async def get_user_chat_list_by_tag_name(
|
|
############################
|
|
############################
|
|
|
|
|
|
|
|
|
|
-@router.get("/tags/all", response_model=List[TagModel])
|
|
|
|
|
|
+@router.get("/tags/all", response_model=list[TagModel])
|
|
async def get_all_tags(user=Depends(get_verified_user)):
|
|
async def get_all_tags(user=Depends(get_verified_user)):
|
|
try:
|
|
try:
|
|
tags = Tags.get_tags_by_user_id(user.id)
|
|
tags = Tags.get_tags_by_user_id(user.id)
|
|
@@ -417,7 +417,7 @@ async def delete_shared_chat_by_id(id: str, user=Depends(get_verified_user)):
|
|
############################
|
|
############################
|
|
|
|
|
|
|
|
|
|
-@router.get("/{id}/tags", response_model=List[TagModel])
|
|
|
|
|
|
+@router.get("/{id}/tags", response_model=list[TagModel])
|
|
async def get_chat_tags_by_id(id: str, user=Depends(get_verified_user)):
|
|
async def get_chat_tags_by_id(id: str, user=Depends(get_verified_user)):
|
|
tags = Tags.get_tags_by_chat_id_and_user_id(id, user.id)
|
|
tags = Tags.get_tags_by_chat_id_and_user_id(id, user.id)
|
|
|
|
|