feat(ai): 完善模型管理和智能体导入功能

云

9 小时以前 95ae0c83d2aa58d86b4039e3f7f16ee93afb8f00

 yudao-module-system/src/main/java/cn/iocoder/yudao/module/system/framework/agent/SystemAgentBridgeGuard.java

@@ -2,7 +2,6 @@

import cn.iocoder.yudao.framework.security.core.service.SecurityFrameworkService;
import cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils;
import cn.iocoder.yudao.framework.tenant.core.context.TenantContextHolder;
import jakarta.annotation.Resource;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.HttpStatus;
@@ -10,6 +9,7 @@

import java.nio.charset.StandardCharsets;
import java.security.MessageDigest;
import java.util.List;

/** 桥接统一鉴权：服务令牌 + 当前登录用户 + 业务权限，缺一不可。 */
@Component
@@ -17,11 +17,22 @@

    public static final String PERMISSION_ITEM_IMPORT = "mdm:item:import";

    /** 各业务模块的导入权限；会话创建与附件读取只需命中其中任意一项。 */
    public static final List<String> IMPORT_PERMISSIONS = List.of(
            PERMISSION_ITEM_IMPORT,
            "srm:supplier:create",
            "mes:md-item:create",
            "mes:qc-template:create",
            "mes:pro-mps:create",
            "mes:pro-route:create");

    @Value("${yudao.agent.bridge-service-token:}")
    private String bridgeServiceToken;

    @Resource
    private SecurityFrameworkService securityFrameworkService;
    @Resource
    private SystemAgentDelegationService delegationService;

    public void verifyServiceToken(String provided) {
        if (bridgeServiceToken == null || bridgeServiceToken.isBlank() || provided == null
@@ -32,6 +43,16 @@
        }
    }

    /** 只确认服务身份与当前登录用户；具体能力由调用方逐项收紧。 */
    public AgentPrincipal requireAuthenticated(String providedServiceToken) {
        verifyServiceToken(providedServiceToken);
        Long userId = SecurityFrameworkUtils.getLoginUserId();
        if (userId == null) {
            throw new AgentBridgeException(HttpStatus.UNAUTHORIZED, "UNAUTHORIZED", "未登录或登录已过期");
        }
        return principalOf(userId);
    }

    public AgentPrincipal require(String providedServiceToken, String permission) {
        verifyServiceToken(providedServiceToken);
        Long userId = SecurityFrameworkUtils.getLoginUserId();
@@ -40,9 +61,56 @@
        }
        if (!securityFrameworkService.hasPermission(permission)) {
            throw new AgentBridgeException(HttpStatus.FORBIDDEN, "FORBIDDEN",
                    "当前用户缺少物料导入权限（" + permission + "）");
                    "当前用户缺少数据导入权限（" + permission + "）");
        }
        Long tenantId = TenantContextHolder.getTenantId();
        return new AgentPrincipal(userId, tenantId == null ? 0L : tenantId, "live");
        return principalOf(userId);
    }

    public AgentPrincipal require(String providedServiceToken, String delegationToken, String taskId,
                                  String operation, String permission, String moduleId, Long attachmentId) {
        verifyServiceToken(providedServiceToken);
        if (delegationToken != null && !delegationToken.isBlank()) {
            return delegationService.require(delegationToken, taskId, operation, permission, moduleId, attachmentId);
        }
        return require(providedServiceToken, permission);
    }

    /** 会话与附件入口无法预知模块，只要求具备任意一项导入权限。 */
    public AgentPrincipal requireAny(String providedServiceToken) {
        verifyServiceToken(providedServiceToken);
        Long userId = SecurityFrameworkUtils.getLoginUserId();
        if (userId == null) {
            throw new AgentBridgeException(HttpStatus.UNAUTHORIZED, "UNAUTHORIZED", "未登录或登录已过期");
        }
        if (!securityFrameworkService.hasAnyPermissions(IMPORT_PERMISSIONS.toArray(String[]::new))) {
            throw new AgentBridgeException(HttpStatus.FORBIDDEN, "FORBIDDEN", "当前用户缺少数据导入权限");
        }
        return principalOf(userId);
    }


    public AgentPrincipal requireAny(String providedServiceToken, String delegationToken, String taskId,
                                     String operation, String moduleId, Long attachmentId) {
        if (delegationToken == null || delegationToken.isBlank()) {
            return requireAny(providedServiceToken);
        }
        verifyServiceToken(providedServiceToken);
        AgentBridgeException last = null;
        for (String permission : IMPORT_PERMISSIONS) {
            try {
                return delegationService.require(delegationToken, taskId, operation, permission, moduleId, attachmentId);
            } catch (AgentBridgeException ex) {
                last = ex;
                if (!"DELEGATION_SCOPE_MISMATCH".equals(ex.getErrorCode())
                        && !"DELEGATION_PERMISSION_REVOKED".equals(ex.getErrorCode())) {
                    throw ex;
                }
            }
        }
        throw last == null ? new AgentBridgeException(HttpStatus.FORBIDDEN, "FORBIDDEN", "委托票据权限不足") : last;
    }

    private AgentPrincipal principalOf(Long userId) {
        return new AgentPrincipal(userId, "live");
    }
}