opensource
/
dify
mirror of https://github.com/langgenius/dify.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

chore: update the error message (#29325)

This commit is contained in:
wangxiaolei 2025-12-09 15:29:04 +08:00 committed by GitHub
parent 022cfbd186
commit c24835ca87
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 2 deletions
Show Stats 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.model import EndUser
if not isinstance(current_user._get_current_object(), (Account, EndUser)): # type: ignore
raise TypeError(f"current_user must be Account or EndUser, got {type(current_user).__name__}")
try:
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