diff --git a/api/controllers/console/app/wraps.py b/api/controllers/console/app/wraps.py index 520955c247..a3437d541d 100644 --- a/api/controllers/console/app/wraps.py +++ b/api/controllers/console/app/wraps.py @@ -21,6 +21,7 @@ def _load_app_model(app_id: str) -> App | None: ) return app_model + def _load_app_model_with_trial(app_id: str) -> App | None: assert isinstance(current_user, Account) app_model = ( @@ -30,6 +31,7 @@ def _load_app_model_with_trial(app_id: str) -> App | None: ) return app_model + def get_app_model(view: Callable[P, R] | None = None, *, mode: Union[AppMode, list[AppMode], None] = None): def decorator(view_func: Callable[P, R]): @wraps(view_func) diff --git a/api/controllers/console/explore/wraps.py b/api/controllers/console/explore/wraps.py index 61d7a39e14..56a5c1070b 100644 --- a/api/controllers/console/explore/wraps.py +++ b/api/controllers/console/explore/wraps.py @@ -12,10 +12,10 @@ from controllers.console.wraps import account_initialization_required from extensions.ext_database import db from libs.login import login_required from models import AccountTrialAppRecord, App, InstalledApp, TrialApp +from models.account import Account from services.app_service import AppService from services.enterprise.enterprise_service import EnterpriseService from services.feature_service import FeatureService -from models.account import Account P = ParamSpec("P") R = TypeVar("R")