xuyonghao
2024-12-05 f7f4430d30ba5f41da69c99082eabe5727198dca
app/api/auth.py
@@ -1,6 +1,5 @@
import json
from Tools.scripts.mailerdaemon import emparse_list_from
from fastapi import APIRouter, Depends
from sqlalchemy.orm import Session
from sqlalchemy.ext.asyncio import AsyncSession
@@ -99,10 +98,11 @@
        return Response(code=400, msg="token not found")
    return Response(code=200, msg="success", data={
        "ragflow_token": token.ragflow_token,
        "bisheng_token": token.bisheng_token,
    })
@router.post("/login_test", response_model=Response)
@router.post("/v2/login", response_model=Response)
async def login_test(login_data: LoginData, db: Session = Depends(get_db), pdb: AsyncSession = Depends(get_pdb)):
    user = authenticate_user(db, login_data.username, login_data.password)
    if not user:
@@ -153,7 +153,7 @@
@router.post("/register_test", response_model=Response)
@router.post("/v2/register", response_model=Response)
async def register_test(user: UserCreate, db=Depends(get_db)):
    db_user = db.query(UserModel).filter(UserModel.username == user.username).first()
    if db_user: