diff --git a/api/controllers/console/app/wraps.py b/api/controllers/console/app/wraps.py index 2f0e16ac79..e687d980fa 100644 --- a/api/controllers/console/app/wraps.py +++ b/api/controllers/console/app/wraps.py @@ -24,7 +24,6 @@ def _load_app_model(app_id: str) -> App | None: def _load_app_model_with_trial(app_id: str) -> App | None: - assert isinstance(current_user, Account) app_model = db.session.query(App).where(App.id == app_id, App.status == "normal").first() return app_model diff --git a/api/controllers/console/explore/wraps.py b/api/controllers/console/explore/wraps.py index 475a617aae..0f2f8d6dc9 100644 --- a/api/controllers/console/explore/wraps.py +++ b/api/controllers/console/explore/wraps.py @@ -76,6 +76,8 @@ def trial_app_required(view: Callable[Concatenate[App, P], R] | None = None): def decorator(view: Callable[Concatenate[App, P], R]): @wraps(view) def decorated(app_id: str, *args: P.args, **kwargs: P.kwargs): + current_user, _ = current_account_with_tenant() + trial_app = db.session.query(TrialApp).where(TrialApp.app_id == str(app_id)).first() if trial_app is None: @@ -85,8 +87,6 @@ def trial_app_required(view: Callable[Concatenate[App, P], R] | None = None): if app is None: raise TrialAppNotAllowed() - assert isinstance(current_user, Account) - account_trial_app_record = ( db.session.query(AccountTrialAppRecord) .where(AccountTrialAppRecord.account_id == current_user.id, AccountTrialAppRecord.app_id == app_id)