JBoltAI_SpringBoot 资源管理：打造一站式 AI 资源管理平台

在 AI 项目开发中，资源管理一直是个让人头疼的问题。各种模型、数据库、嵌入资源等，如果没有一个统一的管理平台，很容易让开发者陷入混乱。今天，就给大家介绍一个超级实用的资源管理功能 ——JBoltAI_SpringBoot 资源管理，让你轻松管理所有资源模型，从此告别资源管理的烦恼。

 一、资源管理的 “痛点”

在 AI 项目中，我们经常会遇到以下问题：

  资源种类繁多 ：有 AI 模型、向量数据库、嵌入模型等，每种资源都有不同的配置和管理方式。

  配置复杂 ：每种资源都有自己的参数和配置，手动管理容易出错。

  缺乏统一管理 ：没有一个统一的平台来管理所有资源，导致资源分散，难以维护。

为了解决这些问题，JBoltAI_SpringBoot提供了一套完善的资源管理功能，让我们可以轻松管理所有资源模型。

二、JBoltAI_SpringBoot 资源管理功能

JBoltAI_SpringBoot 资源管理功能可以分为三个部分：AI 资源管理、VDB 资源管理和 Embedding 资源管理。

（一）AI 资源管理

AI 资源管理主要用于管理各种 AI 模型。我们可以使用以下代码来注册和移除 AI 资源：

注册资源

public Result registerResource(@NotNull AiResource resource) {

    logger.info("AI资源--{}--注册资源", resource.getType());

    try {

        AIResource jboltAIResource = new AIResource(AIModel.get(resource.getType()),

                resource.getId().toString(),

                resource.getApiKey());

        AIResourceCenter.register(jboltAIResource);

    } catch (Exception e) {

        logger.error("AI资源--{}--注册资源失败,异常原因:{}", resource.getType(), e.getMessage());

        return Result.error(e.getMessage());

    }

    return Result.success();

}

这段代码就像是给 AI 模型办理了一个 “身份证”，让系统能够识别和管理它们。我们先创建一个 `AIResource` 对象，然后调用 `AIResourceCenter.register` 方法，把资源注册到系统中。如果注册失败，会记录错误日志并返回错误信息。

移除资源

public void unRegisterResource(AiResource resource) {

    logger.info("AI资源--{}--移除资源", resource.getType());

    try {

        ResourceCenter.unregisterAI(resource.getId().toString());

    } catch (Exception e) {

        logger.error("AI资源--{}--移除资源失败,异常原因:{}", resource.getType(), e.getMessage());

        throw e;

    }

}

这段代码就像是给 AI 模型注销了 “身份证”，让它从系统中消失。我们调用 `ResourceCenter.unregisterAI` 方法，根据资源的 ID 移除资源。如果移除失败，会记录错误日志并抛出异常。

（二）VDB 资源管理

VDB 资源管理主要用于管理向量数据库。我们可以使用以下代码来注册和移除 VDB 资源：

注册资源

public Result registerResource(@NotNull VDBResource newResource) {

    logger.info("VDB资源--{}--注册资源", newResource.getType());

    try {

        Result embeddingResult = embeddingResourceService.getOrCreateModel(newResource.getModelName(), null, null);

        EmbeddingModel embeddingModel = (EmbeddingModel) embeddingResult.getData();



        if (newResource.getType().equals(cn.jbolt.ai.resource.vdb.VDBType.MILVUS.getSn())) {

            MilvusVDBResource milvusVDBResource = new MilvusVDBResource(newResource.getId().toString(), newResource.getUrl(), newResource.getUsername(), newResource.getPassword(), embeddingModel);

            milvusVDBResource.setDatabase(newResource.getDatabaseName());

            ResourceCenter.registerVDB(milvusVDBResource);

        } else if (newResource.getType().equals(cn.jbolt.ai.resource.vdb.VDBType.TENCENT.getSn())) {

            TencentVDBResource tencentVDBResource = new TencentVDBResource(newResource.getId().toString(), newResource.getUrl(), newResource.getUsername(), newResource.getPassword(), embeddingModel);

            tencentVDBResource.setDatabase(newResource.getDatabaseName());

            ResourceCenter.registerVDB(tencentVDBResource);

        }



        try {

            VDBOptResult<Boolean> vdbOptResult = JBoltVDB.use(newResource.getId().toString()).checkDatabaseExist(newResource.getDatabaseName());

            if (vdbOptResult.isSuccess() && vdbOptResult.getResult() == false) {

                JBoltVDB.use(newResource.getId().toString()).createDatabase(newResource.getDatabaseName());

            }

        } catch (Exception e) {

            logger.error("VDB资源--{}--创建数据库失败,异常原因:{}", newResource.getType(), e.getMessage());

        }

    } catch (Exception e) {

        String message = e.getMessage();

        logger.error("VDB资源--{}--注册资源失败,异常原因:{}", newResource.getType(), message);

        return Result.error(message);

    }

    return Result.success();

}

这段代码就像是给向量数据库搭建了一个 “小家”，让它能够安稳地待在系统中。我们先根据传入的参数创建一个 `MilvusVDBResource` 或 `TencentVDBResource` 对象，然后调用 `ResourceCenter.registerVDB` 方法，把资源注册到系统中。如果注册失败，会记录错误日志并返回错误信息。同时，我们还会检查数据库是否存在，如果不存在就创建一个。

移除资源

public void unRegisterResource(VDBResource resource) {

    logger.info("VDB资源--{}--移除资源", resource.getType());

    try {

        ResourceCenter.unregisterVDB(resource.getId().toString());

    } catch (Exception e) {

        logger.error("VDB资源--{}--移除资源失败,异常原因:{}", resource.getType(), e.getMessage());

        throw e;

    }

}

这段代码就像是给向量数据库拆掉了 “小家”，让它从系统中消失。我们调用 `ResourceCenter.unregisterVDB` 方法，根据资源的 ID 移除资源。如果移除失败，会记录错误日志并抛出异常。

（三）Embedding 资源管理

Embedding 资源管理主要用于管理嵌入模型。我们可以使用以下代码来注册 Embedding 资源：

public Result registerResource(@NotNull EmbeddingResource embeddingResource) {

    logger.info("Embedding资源-id:{}-{}--注册资源", embeddingResource.getId());

    Result modelResult = getOrCreateModel(embeddingResource.getType(), embeddingResource.getDefaultModel(), embeddingResource.getDimension());

    com.jboltai.resource.embedding.EmbeddingResource jboltEmbeddingResource = new com.jboltai.resource.embedding.EmbeddingResource((EmbeddingModel) modelResult.getData());

    try {

        EmbeddingResourceCenter.register(jboltEmbeddingResource);

    } catch (Exception e) {

        logger.error("Embedding资源--注册资源失败,异常原因:{}", e.getMessage());

        return Result.error(e.getMessage());

    }

    return Result.success();

}

这段代码就像是给嵌入模型穿上了一件 “新衣”，让它能够更好地融入系统。我们先调用 `getOrCreateModel` 方法创建一个模型，然后创建一个 `EmbeddingResource` 对象，最后调用 `EmbeddingResourceCenter.register` 方法，把资源注册到系统中。如果注册失败，会记录错误日志并返回错误信息。

移除资源

    /**
     * 移除资源
     *
     * @param resource 资源
     */
    public void unRegisterResource(EmbeddingResource resource) {
        logger.info("Embedding资源--{}-{}---移除资源", resource.getType(), resource.getDefaultModel());

        //从资源池
        try {
            com.jboltai.resource.embedding.EmbeddingResource jboltEmbeddingResource = EmbeddingResourceCenter.getResourceById(resource.getId().toString());
            EmbeddingResourceCenter.unregister(jboltEmbeddingResource);
        } catch (Exception e) {
            e.printStackTrace();
            logger.error("Embedding资源--{}-{}---移除资源失败,异常原因:{}", resource.getType(), resource.getDefaultModel(), e.getMessage());
            throw e;
        }
    }

三、资源管理的优势

JBoltAI_SpringBoot 资源管理功能具有以下优势：

 统一管理：可以将所有资源模型统一管理，方便维护和更新。就像把所有的玩具都放在一个箱子里，想找哪个就找哪个，再也不用到处乱找了。

简单易用：提供简洁的 API，让开发者可以轻松地注册和移除资源。就像用积木搭建房子一样，只需要几块积木就能搭建出一个漂亮的房子。

灵活配置：支持多种资源类型和配置方式，满足不同项目的需求。就像一个百宝箱，里面有各种各样的宝贝，总有一款适合你。

 高效稳定：经过优化的代码和算法，保证了资源管理的高效性和稳定性。就像一辆高性能的赛车，不仅速度快，而且稳定性强，让你在开发过程中一路飞驰。

四、资源管理的实际应用

JBoltAI_SpringBoot资源管理功能在实际项目中有着广泛的应用。比如，在一个大型的 AI 项目中，我们需要管理多个 AI 模型、向量数据库和嵌入模型。通过使用 JBoltAI_SpringBoot资源管理功能，我们可以轻松地将这些资源注册到系统中，并且可以随时移除不需要的资源。同时，我们还可以对资源进行灵活的配置，满足项目的需求。通过统一管理各种资源模型，我们可以更加高效地开发和维护 AI 项目。无论是 AI 模型、向量数据库还是嵌入模型，都可以在这个平台上轻松管理。

资源管理