rag-lifecycle-management

RAG生命周期管理重构方案

问题背景

当前RAG管理系统存在以下核心问题：

1. 私有RAG使用困难：用户创建的RAG添加新文档后，需要重新发布并审核才能在Agent中使用最新数据
2. 版本管理不灵活：每个版本需要单独安装，无法像工具市场一样在已安装RAG中切换版本
3. 数据来源不统一：Agent需要从多个数据源获取RAG，增加了复杂度

解决方案

核心设计思路

引入动态引用机制，区分两种安装类型：

• REFERENCE类型：引用原始数据集，支持实时更新（用于私有RAG）
• SNAPSHOT类型：使用版本快照，内容固定（用于公开RAG）

版本管理模式

对标工具市场的版本管理：

1. 安装RAG本身（而不是特定版本）
2. 支持在已安装RAG中切换版本
3. 统一从user_rags表获取可用RAG

技术实施方案

1. 数据库表结构调整

user_rags表重新设计

ALTER TABLE user_rags ADD COLUMN original_rag_id VARCHAR(64) COMMENT '原始RAG数据集ID';
ALTER TABLE user_rags ADD COLUMN install_type VARCHAR(20) DEFAULT 'SNAPSHOT' COMMENT '安装类型：REFERENCE/SNAPSHOT';
ALTER TABLE user_rags ADD COLUMN current_version_id VARCHAR(64) COMMENT '当前选择的版本ID';

-- 为向后兼容，保留 rag_version_id 字段
-- 新增复合索引
CREATE INDEX idx_user_original_rag ON user_rags(user_id, original_rag_id);
CREATE INDEX idx_user_install_type ON user_rags(user_id, install_type);

安装类型枚举

• REFERENCE：动态引用原始数据集，数据实时更新
• SNAPSHOT：使用版本快照数据，内容固定不变

2. 领域模型重构

新增InstallType枚举

public enum InstallType {
    REFERENCE("REFERENCE", "引用类型"),
    SNAPSHOT("SNAPSHOT", "快照类型");
}

UserRagEntity实体扩展

新增字段：

• originalRagId：原始RAG数据集ID
• installType：安装类型
• currentVersionId：当前选择的版本ID

3. 核心业务逻辑重构

创建RAG时的自动安装

// RagQaDatasetAppService.createDataset()
1. 创建原始数据集
2. 创建0.0.1私有版本（用于版本管理）
3. 自动为创建者安装REFERENCE类型记录
   - originalRagId: 数据集ID
   - installType: REFERENCE  
   - currentVersionId: 0.0.1版本ID

RAG安装逻辑重构

// UserRagDomainService.installRag()
1. 检查是否已安装同一个RAG（按originalRagId检查）
2. 如果已安装，则更新版本；如果未安装，则创建新记录
3. 自己的RAG：安装REFERENCE类型
4. 他人的RAG：安装SNAPSHOT类型

版本切换功能

// UserRagDomainService.switchRagVersion()
1. 验证用户是否已安装该RAG
2. 验证目标版本是否可用
3. 更新currentVersionId
4. 如果切换到私有版本，更新installType为REFERENCE

4. Agent集成机制

统一RAG数据获取

// RagDataService.getRagData()
根据installType决定数据来源：
- REFERENCE类型：从原始数据集表获取最新数据
- SNAPSHOT类型：从版本快照表获取固定数据

检索服务适配

// 检索时的数据路由
if (userRag.getInstallType() == REFERENCE) {
    // 从 rag_qa_dataset, rag_files, rag_documents 表查询
    return getRealTimeRagData(userRag.getOriginalRagId());
} else {
    // 从 rag_version_files, rag_version_documents 表查询  
    return getSnapshotRagData(userRag.getCurrentVersionId());
}