minRAG RAG 系统

联合创作 · 2025-02-18 07:49

RAG 从未如此简单

minRAG是从零开始的RAG系统,追求极致的简单和强大,不超过1万行代码,无需安装,双击启动.支持OpenAI、Gitee AI、百度千帆、腾讯云LKE、阿里云百炼、字节火山引擎等AI平台.

使用FTS5实现BM25全文检索,使用Vec实现向量检索,实现了 MarkdownConverter、DocumentSplitter、OpenAIDocumentEmbedder、SQLiteVecDocumentStore、OpenAITextEmbedder、VecEmbeddingRetriever、FtsKeywordRetriever、DocumentChunkReranker、PromptBuilder、OpenAIChatMemory、OpenAIChatGenerator、ChatMessageLogStore、Pipeline等组件,支持流水线设置和扩展.

支持的AI平台

因为 reranker 没有统一标准,组件参数中base_url要填写完整的路径

OpenAI

minRAG实现了OpenAI的标准规范,所有兼容OpenAI的平台都可以使用.

Gitee AI(默认)

AI平台默认是 Gitee AI,Gitee AI每天100次免费调用

  • 注册或设置页面的AI平台base_url 填写 https://ai.gitee.com/v1
  • 注册或设置页面的AI平台api_key 填写 免费或者付费的token
  • OpenAITextEmbedder 默认使用 bge-m3 模型
  • GiteeDocumentChunkReranker 组件参数 {"base_url":"https://ai.gitee.com/api/serverless/bge-reranker-v2-m3/rerank","model":"bge-reranker-v2-m3"}
  • OpenAIChatGenerator 建议使用 DeepSeek-V3 模型

腾讯云LKE知识引擎

  • 注册或设置页面的AI平台base_url 填写 SecretId ,或在组件参数配置{"SecretId":"xxx"}
  • 注册或设置页面的AI平台api_key 填写 SecretKey ,或在组件参数配置{"SecretKey":"xxx"}
  • LKETextEmbedder和LKEDocumentEmbedder 默认使用 lke-text-embedding-v1 模型
  • LKEDocumentChunkReranker 默认使用 lke-reranker-base 模型
  • OpenAIChatGenerator 使用OpenAI SDK方式接入,组件参数配置 {"base_url":"https://api.lkeap.cloud.tencent.com/v1","api_key":"xxx","model":"deepseek-v3"}
  • 记得修改流水线中的组件

百度千帆

  • 注册或设置页面的AI平台base_url 填写 https://qianfan.baidubce.com/v2
  • 注册或设置页面的AI平台api_key 填写 永久有效API Key
  • OpenAITextEmbedderOpenAIDocumentEmbedder 默认使用 bge-large-zh 模型,1024维度
  • DocumentChunkReranker 组件参数配置 {"base_url":"https://qianfan.baidubce.com/v2/rerankers","model":"bce-reranker-base","top_n":5,"score":0.1}
  • OpenAIChatGenerator 建议使用 deepseek-v3 模型
  • 记得修改流水线中的组件

阿里云百炼

  • 注册或设置页面的AI平台base_url 填写 https://dashscope.aliyuncs.com/compatible-mode/v1
  • 注册或设置页面的AI平台api_key 填写 申请的API KEY
  • OpenAITextEmbedderOpenAIDocumentEmbedder 默认使用 text-embedding-v3 模型,1024维度
  • BaiLianDocumentChunkReranker 组件参数配置 {"base_url":"https://dashscope.aliyuncs.com/api/v1/services/rerank/text-rerank/text-rerank","model":"gte-rerank","top_n":5,"score":0.1}
  • OpenAIChatGenerator 建议使用 deepseek-v3 模型
  • 记得修改流水线中的组件

字节火山引擎

  • 注册或设置页面的AI平台base_url 填写 https://ark.cn-beijing.volces.com/api/v3
  • 注册或设置页面的AI平台api_key 填写 申请的API KEY
  • OpenAITextEmbedderOpenAIDocumentEmbedder 建议使用doubao-embedding模型,兼容1024维度
  • DocumentChunkReranker 火山引擎暂时没有Reranker模型,建议使用其他平台的Reranker模型或者去掉
  • OpenAIChatGenerator 建议使用 deepseek-v3模型
  • 记得修改流水线中的组件

界面预览

 

浏览 2
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报