From 389d592e2e5340ff3ece8868836e375d245a0df1 Mon Sep 17 00:00:00 2001 From: "ext-ensky-jeremy.jj.he" Date: Wed, 13 Aug 2025 08:52:10 +0800 Subject: [PATCH] new-feature:add a new feature in two utils --- backend/app/api/routes/utils.py | 5 ++--- backend/app/utils.py | 9 +++++++++ 2 files changed, 11 insertions(+), 3 deletions(-) diff --git a/backend/app/api/routes/utils.py b/backend/app/api/routes/utils.py index dd2302b..5be4db2 100644 --- a/backend/app/api/routes/utils.py +++ b/backend/app/api/routes/utils.py @@ -5,7 +5,7 @@ from pydantic import BaseModel from app.api.deps import get_current_active_superuser from app.models import Message -from app.utils import generate_test_email, send_email +from app.utils import generate_test_email, send_email, fast_json_response router = APIRouter(prefix="/utils", tags=["utils"]) @@ -43,5 +43,4 @@ async def bentoml_classifiy(data: ClassifyRequest): json={"text": f"{data.text}"} ) result = response.json() - - return result + return fast_json_response(data=result) diff --git a/backend/app/utils.py b/backend/app/utils.py index ac029f6..ddd15d0 100644 --- a/backend/app/utils.py +++ b/backend/app/utils.py @@ -121,3 +121,12 @@ def verify_password_reset_token(token: str) -> str | None: return str(decoded_token["sub"]) except InvalidTokenError: return None + +from fastapi.responses import JSONResponse + +def fast_json_response(data=None, code=200, msg="success"): + return JSONResponse({ + "code": code, + "data": data or {}, + "msg": msg + })