gmakstutis/ragflow
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Continue...

Browse source
This commit is contained in:
Kevin Hu 2025-11-17 16:23:56 +08:00
parent 3f652088e8
commit df9d90307d
1 changed files with 12 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
api/utils/api_utils.py
View file

@ -211,8 +211,7 @@ def construct_json_result(code: RetCode = RetCode.SUCCESS, message="success", da
def token_required(func): def token_required(func):
@wraps(func) def get_tenant_id(**kwargs):
async def decorated_function(*args, **kwargs):
if os.environ.get("DISABLE_SDK"): if os.environ.get("DISABLE_SDK"):
return get_json_result(data=False, message="`Authorization` can't be empty") return get_json_result(data=False, message="`Authorization` can't be empty")
authorization_str = request.headers.get("Authorization") authorization_str = request.headers.get("Authorization")
@ -226,11 +225,20 @@ def token_required(func):
if not objs: if not objs:
return get_json_result(data=False, message="Authentication error: API key is invalid!", code=RetCode.AUTHENTICATION_ERROR) return get_json_result(data=False, message="Authentication error: API key is invalid!", code=RetCode.AUTHENTICATION_ERROR)
kwargs["tenant_id"] = objs[0].tenant_id kwargs["tenant_id"] = objs[0].tenant_id
return kwargs
if inspect.iscoroutinefunction(func): @wraps(func)
return await func(*args, **kwargs) def decorated_function(*args, **kwargs):
kwargs = get_tenant_id(**kwargs)
return func(*args, **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 return decorated_function