From df9d90307d62f3e45cf7c8da751f508f34768129 Mon Sep 17 00:00:00 2001
From: Kevin Hu <kevinhu.sh@gmail.com>
Date: Mon, 17 Nov 2025 16:23:56 +0800
Subject: [PATCH] Continue...

---
 api/utils/api_utils.py | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/api/utils/api_utils.py b/api/utils/api_utils.py
index 20dc88086..508dae6e6 100644
--- a/api/utils/api_utils.py
+++ b/api/utils/api_utils.py
@@ -211,8 +211,7 @@ def construct_json_result(code: RetCode = RetCode.SUCCESS, message="success", da
 
 
 def token_required(func):
-    @wraps(func)
-    async def decorated_function(*args, **kwargs):
+    def get_tenant_id(**kwargs):
         if os.environ.get("DISABLE_SDK"):
             return get_json_result(data=False, message="`Authorization` can't be empty")
         authorization_str = request.headers.get("Authorization")
@@ -226,11 +225,20 @@ def token_required(func):
         if not objs:
             return get_json_result(data=False, message="Authentication error: API key is invalid!", code=RetCode.AUTHENTICATION_ERROR)
         kwargs["tenant_id"] = objs[0].tenant_id
+        return kwargs
 
-        if inspect.iscoroutinefunction(func):
-            return await func(*args, **kwargs)
+    @wraps(func)
+    def decorated_function(*args, **kwargs):
+        kwargs = get_tenant_id(**kwargs)
         return func(*args, **kwargs)
 
+    @wraps(func)
+    async def adecorated_function(*args, **kwargs):
+        kwargs = get_tenant_id(**kwargs)
+        return await func(*args, **kwargs)
+
+    if inspect.iscoroutinefunction(func):
+        return adecorated_function
     return decorated_function