Merge pull request #982 from SuanmoSuanyangTechnology/fix/wxy_031

fix(quota_manager): retrieve workspace_id from api_key_auth context
This commit is contained in:
山程漫悟
2026-04-23 00:17:04 +08:00
committed by GitHub

View File

@@ -37,6 +37,11 @@ def _get_workspace_id_from_kwargs(kwargs: dict):
if workspace_id: if workspace_id:
return workspace_id return workspace_id
# 从 api_key_auth.workspace_id 获取API Key 认证场景)
api_key_auth = kwargs.get("api_key_auth")
if api_key_auth and hasattr(api_key_auth, 'workspace_id'):
return api_key_auth.workspace_id
# 从 user.current_workspace_id 获取 # 从 user.current_workspace_id 获取
user = _get_user_from_kwargs(kwargs) user = _get_user_from_kwargs(kwargs)
if user: if user: