diff --git a/app/api/user/routes.py b/app/api/user/routes.py index 240c045..f2906b7 100644 --- a/app/api/user/routes.py +++ b/app/api/user/routes.py @@ -71,7 +71,7 @@ async def register_email_verify(req: Request, user_email: UserResetEmailRequest) @users_router.put("/update", deprecated=False) -async def user_modification(updated_user: UpdateUserRequest, current_user: User = Depends(get_current_user)): +async def user_modification(updated_user: UpdateUserRequest, current_user: Tuple[User, Dict] = Depends(get_current_user)): """ :param updated_user: Pydantic 模型验证修改内容(根据JSON内容修改对应字段) @@ -169,7 +169,7 @@ async def forget_password(request: Request, user_request: UserResetPhoneRequest) @users_router.post("/auth/varify_code", deprecated=True) async def varify_code(data: VerifyPhoneCodeRequest, request: Request): redis = request.app.state.redis - if not await service.varify_code(redis=redis, phone=data.phone, input_code=data.code): + if not await service.verify_code(redis=redis, phone=data.phone, input_code=data.code): raise HTTPException(status_code=400, detail="验证码错误或已过期") return {"message": "验证成功,可以重置密码"} diff --git a/app/models/base.py b/app/models/base.py index ab88ba2..5a97af5 100644 --- a/app/models/base.py +++ b/app/models/base.py @@ -30,6 +30,7 @@ class User(Model): encrypted_phone = fields.CharField(max_length=11, description="用户手机号", null=True) language = fields.ForeignKeyField("models.Language", related_name="users", on_delete=fields.CASCADE) is_admin = fields.BooleanField(default=False, description="管理员权限") + token_usage = fields.IntField(default=0, description="AI答疑使用量") class Meta: table = "users" diff --git a/main.py b/main.py index e135de4..094156b 100644 --- a/main.py +++ b/main.py @@ -7,6 +7,7 @@ from tortoise.contrib.fastapi import register_tortoise import app.models.signals from app.api.admin.router import admin_router +from app.api.ai_assist.routes import ai_router from app.api.redis_test import redis_test_router from app.api.search import dict_search from app.api.translator import translator_router @@ -52,5 +53,7 @@ app.include_router(redis_test_router, tags=["Redis Test-Only API"]) app.include_router(translator_router, tags=["Translation API"]) +app.include_router(ai_router, tags=["AI Assist API"], prefix="/ai_assist") + if __name__ == "__main__": uvicorn.run("main:app", host="127.0.0.1", port=8000, reload=True)