基本更新

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": "验证成功，可以重置密码"}
 

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"

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)