rag-retriever（RAG retriever 搜索）

🦞 rag-retriever - RAG 2.0 检索系统

版本: 2.0.0 创建时间: 2026-03-17 更新时间: 2026-03-20 状态: ✅ RAG 2.0 完成

📋 简介

RAG 2.0 (Retrieval-Augmented Generation) 检索系统，为 OpenClaw 提供：

• 📄 文档分块 - 智能文本分割策略（可配置 chunk size + overlap）
• 🔢 向量嵌入 - OpenAI Embedding 或本地简单嵌入
• 🔍 混合搜索 - 向量搜索 + BM25 关键词搜索（RRF 融合）
• 🔄 加权重排 - 可配置向量/BM25 权重
• 📖 来源引用 - 自动引用编号和来源追溯
• 🤖 上下文增强 - 自动生成 RAG 增强提示词
• 🇨🇳 中文分词 - 支持中英文混合搜索

🚀 快速开始

安装

cd skills/rag-retriever
npm install

基本使用

1. 初始化

node src/rag-skill.js init my_docs

2. 添加文档

node src/rag-skill.js add ./readme.json '{"source":"github"}'

3. 检索

node src/rag-skill.js search "MCP protocol" 5

4. RAG 查询

node src/rag-skill.js rag "什么是 RAG" 3

✅ 已完成功能

• [x] 文档分块策略 (ChunkingStrategy)
• [x] 简单文本嵌入 (SimpleTextEmbedding) - 词频 TF 归一化
• [x] LanceDB 向量存储
• [x] 相似检索（向量相似度）
• [x] RAG 格式化输出
• [x] CLI 工具
• [x] 基础测试
• [x] 中文分词支持 (@node-rs/jieba) 🆕

📅 开发计划

Week 1 (03-17~03-23)

• [x] Day 1: 原型验证 ✅
• [x] Day 1: 测试文档完成 ✅
• [x] Day 1: 检索功能测试 ✅ (2026-03-17 15:21)
• [ ] Day 2: 中文分词支持
• [ ] Day 3-7: 优化 + 文档

Week 2 (03-24~03-30)

• [ ] 集成真正嵌入模型 (OpenAI/Cohere)
• [ ] 混合检索 (关键词 + 向量)
• [ ] Reranker 重排序
• [ ] 多集合管理

🔧 API 使用

JavaScript API

import { RAGRetriever } from './src/retriever.js';

const rag = new RAGRetriever({
  dbPath: './data/lancedb',
  dimensions: 384
});

// 初始化
await rag.initialize('my_docs');

// 添加文档
const result = await rag.addDocument(text, {
  source: 'github',
  timestamp: new Date().toISOString()
});

// 检索
const results = await rag.retrieve('MCP protocol', {
  limit: 5
});

// RAG 格式化
const ragResult = await rag.retrieveForRAG('什么是 MCP', {
  limit: 3
});
console.log(ragResult.context); // 格式化的上下文

// 统计
const stats = await rag.getStats();

🎯 核心组件

1. ChunkingStrategy (分块策略)

const chunking = new ChunkingStrategy({
  chunkSize: 500,    // 每块字符数
  overlap: 50,       // 重叠字符数
  separator: '\
'    // 分隔符
});

const chunks = chunking.chunk(text, metadata);

2. SimpleTextEmbedding (简单嵌入)

const embedding = new SimpleTextEmbedding(384);

// 构建词汇表
embedding.buildVocabulary(texts);

// 生成向量
const vector = embedding.embed('人工智能');

3. RAGRetriever (检索器)

const retriever = new RAGRetriever({
  dbPath: './data/lancedb',
  dimensions: 384
});

await retriever.initialize();
await retriever.addDocument(text, metadata);
const results = await retriever.retrieve(query);

📚 配置选项

RAGRetriever 选项

{
  dbPath: './data/lancedb',     // LanceDB 路径
  dimensions: 384,              // 向量维度
  chunking: {
    chunkSize: 500,             // 分块大小
    overlap: 50,                // 重叠大小
    separator: '\
'             // 分隔符
  }
}

🧪 测试

运行测试

node test/retriever.test.js

测试结果

✅ 分块策略
✅ 简单嵌入
✅ RAG 初始化
✅ 添加文档
⚠️ 检索功能 (需优化)
✅ 统计信息

🔗 参考资料

• LanceDB 文档
• RAG 最佳实践
• 嵌入模型对比 2026

📊 性能指标

⚠️ 已知限制

1. 简单嵌入 - 当前使用词频 TF 嵌入，效果有限

- 改进: 集成 OpenAI/Cohere embeddings

1. 检索精度 - 简单词频嵌入精度有限

- 改进: 使用真正的语义嵌入模型

1. 中文分词 - 使用 jieba 基础分词，未优化专业术语

- 改进: 添加用户词典支持

*最后更新：2026-03-17 17:00*