面向 AI 应用的开源数据基础设施,用于存储文档与向量并支持语义、混合及全文检索。
FlagOpen/FlagEmbedding
北京智源推出的检索工具包,提供 BGE 向量模型与重排序模型,并包含微调、评测和教程,用于搜索与 RAG。
- Star
- 12,259
- Fork
- 1,105
- 未关闭 Issue
- 924
- 最近提交
- 2个月前
这是什么?
FlagEmbedding(BGE,BAAI General Embedding)是一个面向文本检索和检索增强 LLM 应用的工具包。它提供一系列将文本转为向量的嵌入模型用于语义搜索,以及对嵌入结果前 k 条文档重新排序的交叉编码器(reranker)模型。
项目覆盖完整流程:可通过 pip 安装 Python 包进行推理,提供嵌入模型和重排序模型的微调脚本、评测工具和数据集。模型发布在 Hugging Face 上,包括英文、中文和多语言版本。其中 BGE-M3 支持稠密、稀疏和多向量检索,输入长度可达 8192 个 token。
除核心模型外,仓库还包含 LLM-Embedder(面向 LLM 检索增强的统一嵌入模型)、Activation-Beacon(扩展 LLM 上下文长度)以及中文嵌入评测基准 C-MTEB 等研究项目。
核心功能
- 01提供可通过 pip 安装的 Python 包,用于运行 BGE 嵌入模型和重排序模型。
- 02多语言 BGE-M3 模型支持稠密、稀疏和多向量检索。
- 03提供嵌入模型与重排序模型的微调脚本。
- 04包含检索评测工具和数据集。
- 05模型托管在 Hugging Face,覆盖英文、中文和多语言场景。
适用场景
- 为文档库构建语义搜索。
- 在 RAG 流程中加入检索与重排序环节,对前 k 条结果重新排序。
- 基于领域内的查询与段落对微调嵌入模型。
- 评测中文或多语言检索效果。
快速上手
- 1
使用 pip 安装该包
pip install -U FlagEmbedding - 2
在 Python 中加载 BGE 嵌入模型
- 3
对句子编码并计算相似度
使用前须知
- 微调功能需要单独安装额外的 finetune 依赖。
标签
常见问题
- FlagEmbedding 是什么?
- 北京智源推出的检索工具包,提供 BGE 向量模型与重排序模型,并包含微调、评测和教程,用于搜索与 RAG。 FlagEmbedding(BGE,BAAI General Embedding)是一个面向文本检索和检索增强 LLM 应用的工具包。它提供一系列将文本转为向量的嵌入模型用于语义搜索,以及对嵌入结果前 k 条文档重新排序的交叉编码器(reranker)模型。
- FlagEmbedding 可以免费使用吗?用的什么许可证?
- FlagEmbedding 以 MIT 许可证开源。用于商业产品前,请先确认许可证条款。
- FlagEmbedding 还在维护吗?
- 半年内有更新。FlagEmbedding 最近一次提交是在 2026年8月24日。
- FlagEmbedding 适合用来做什么?
- 为文档库构建语义搜索。 在 RAG 流程中加入检索与重排序环节,对前 k 条结果重新排序。 基于领域内的查询与段落对微调嵌入模型。 评测中文或多语言检索效果。
- FlagEmbedding 怎么安装和运行?
- 使用 pip 安装该包: pip install -U FlagEmbedding 在 Python 中加载 BGE 嵌入模型 对句子编码并计算相似度
- 在哪里获取 FlagEmbedding?
- 源代码和安装说明在 GitHub 上:https://github.com/FlagOpen/FlagEmbedding
概要由 AI 根据项目 README 生成;GitHub 数据每日更新。
相关项目
3万2571RustApache-2.0更新于 20小时前
用于构建生产级 LLM 应用的开源 Python 框架,提供模块化管道与智能体工作流,适用于 RAG、语义搜索等场景。
2.7万3307PythonApache-2.0更新于 11小时前
自托管推理服务器,通过兼容 OpenAI 的 API 在一个集群中提供嵌入、重排序、OCR、抽取和大模型等多种模型。
3414323PythonApache-2.0更新于 昨天
开源 RAG 引擎,将检索增强生成与智能体能力结合,为大模型应用提供上下文层。
9.3万1.1万GoApache-2.0更新于 11小时前
本地优先的 AI 记忆系统,逐字保存对话历史,并通过语义搜索检索,提供 MCP 服务与命令行工具。
6万7569PythonMIT更新于 12小时前
轻量级知识图谱 RAG 框架与服务,将文档构建为实体关系图谱,用于增强 LLM 问答。
4万5709PythonMIT更新于 5天前