Bladeren bron

optimize account status field hard coded (#18771)

Co-authored-by: crazywoola <427733928@qq.com>
tags/1.3.1
gsmini 6 maanden geleden
bovenliggende
commit
ec82534a1e
No account linked to committer's email address
1 gewijzigde bestanden met toevoegingen van 2 en 1 verwijderingen
  1. 2
    1
      api/controllers/console/wraps.py

+ 2
- 1
api/controllers/console/wraps.py Bestand weergeven

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)

Laden…
Annuleren
Opslaan