admin/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

chore: update the error message (#29325)

Browse Source
This commit is contained in:
wangxiaolei
2025-12-09 15:29:04 +08:00
committed by GitHub
parent 022cfbd186
commit c24835ca87
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
api/services/datasource_provider_service.py
View File

@@ -29,8 +29,14 @@ def get_current_user():
from models.account import Account from models.account import Account
from models.model import EndUser from models.model import EndUser
if not isinstance(current_user._get_current_object(), (Account, EndUser)): # type: ignore try:
raise TypeError(f"current_user must be Account or EndUser, got {type(current_user).__name__}") user_object = current_user._get_current_object()
except AttributeError:
# Handle case where current_user might not be a LocalProxy in test environments
user_object = current_user
if not isinstance(user_object, (Account, EndUser)):
raise TypeError(f"current_user must be Account or EndUser, got {type(user_object).__name__}")
return current_user return current_user