|
|
|
|
|
|
|
|
from controllers.console.workspace.error import AccountNotInitializedError |
|
|
from controllers.console.workspace.error import AccountNotInitializedError |
|
|
from extensions.ext_database import db |
|
|
from extensions.ext_database import db |
|
|
from extensions.ext_redis import redis_client |
|
|
from extensions.ext_redis import redis_client |
|
|
|
|
|
from models.account import AccountStatus |
|
|
from models.dataset import RateLimitLog |
|
|
from models.dataset import RateLimitLog |
|
|
from models.model import DifySetup |
|
|
from models.model import DifySetup |
|
|
from services.feature_service import FeatureService, LicenseStatus |
|
|
from services.feature_service import FeatureService, LicenseStatus |
|
|
|
|
|
|
|
|
# check account initialization |
|
|
# check account initialization |
|
|
account = current_user |
|
|
account = current_user |
|
|
|
|
|
|
|
|
if account.status == "uninitialized": |
|
|
|
|
|
|
|
|
if account.status == AccountStatus.UNINITIALIZED: |
|
|
raise AccountNotInitializedError() |
|
|
raise AccountNotInitializedError() |
|
|
|
|
|
|
|
|
return view(*args, **kwargs) |
|
|
return view(*args, **kwargs) |