From 00995545e7ed4839bc420b2a2ccecdccfdbb2b47 Mon Sep 17 00:00:00 2001 From: "yunlu.wen" Date: Mon, 22 Jun 2026 11:28:38 +0800 Subject: [PATCH] fix import --- api/controllers/common/wraps.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/api/controllers/common/wraps.py b/api/controllers/common/wraps.py index 57f108679dc..ae2d2a5a29d 100644 --- a/api/controllers/common/wraps.py +++ b/api/controllers/common/wraps.py @@ -20,6 +20,7 @@ Private helpers from collections.abc import Callable from functools import wraps +from typing import TYPE_CHECKING from sqlalchemy import select from werkzeug.exceptions import Forbidden, NotFound @@ -30,9 +31,11 @@ from extensions.ext_database import db from libs.login import current_account_with_tenant from models.dataset import Dataset from models.model import App -from controllers.openapi.auth.data import AuthData from services.enterprise.rbac_service import RBACService +if TYPE_CHECKING: + from controllers.openapi.auth.data import AuthData + __all__ = ["RBACPermission", "RBACResourceScope", "rbac_permission_required"] @@ -51,7 +54,7 @@ def openapi_rbac_permission_required[**P, R]( @wraps(view) def decorated(*args: P.args, **kwargs: P.kwargs) -> R: - auth_data: AuthData | None = kwargs.get("auth_data") + auth_data: "AuthData | None" = kwargs.get("auth_data") if not auth_data: raise Forbidden() # openapi auth pipeline is required if auth_data.caller_kind == "end_user":