FastEmbed 本地检索嵌入
将以下提示词粘贴到你的 AI 对话框中:
请根据 https://skillhub.cn/install/skillhub.md,安装 @user_3c6cb52e/prepare-local-retrieval-embeddings-with-fastembed
技能介绍
解决什么问题
在构建 Agent 记忆或 RAG 工作流时,嵌入生成往往依赖外部 API,带来延迟、成本与数据出境的顾虑。当检索栈使用 Qdrant 等向量数据库时,你需要一种能在本地快速运行、且同时支持多种嵌入类型的方案,而不是为不同任务拼接多个互不兼容的库。
核心能力与关键步骤
该技能基于 FastEmbed,在写入向量之前于本地完成嵌入计算。它覆盖四类嵌入:稠密、稀疏、图像与重排序。FastEmbed 是一个轻量、快速的 Python 库,支持流行文本模型,包括 MTEB 榜单上的 Flag Embedding。该模型区分 query 与 passage 前缀,能在检索时提供更匹配的编码语义。
工作流程大致分为三步:
- 准备环境:Python 环境下安装 FastEmbed,按需选择 CPU 或 GPU 版本;
- 生成嵌入:调用库对文本、图像或查询进行编码,必要时附加前缀并选择重排序模型;
- 写入检索栈:将生成的向量写入 Qdrant 或其他向量数据库,供 Agent 记忆与 RAG 检索使用。
适用边界与注意点
该方案面向本地部署场景。FastEmbed 支持的模型以官方文档为准,若你需要特定领域模型,需确认其是否在支持列表中。对于 GPU 用户,上游文档指出需关注 CUDA 12.x 兼容性,并提供了常见问题排查指引。此外,qdrant-client 与 Qdrant 向量数据库是可选组件,你可以根据检索栈的实际选型决定是否引入。
使用场景
- 在离线环境为 RAG 管线批量生成文本嵌入并写入 Qdrant,避免调用外部 API 带来的数据出境风险
- 构建 Agent 长期记忆时,用 query 和 passage 两种前缀分别编码检索请求与候选文档,提升召回匹配度
- 为图文混合知识库在本地生成图像嵌入,统一存入向量数据库供后续相似度检索使用
- 检索召回候选文档后,用重排序模型在本地对结果重新打分,改善回答引用的准确性
适合人员
- 正在搭建 RAG 系统的工程师,需要在不依赖外部嵌入 API 的前提下完成文本向量化与入库
- 为 Agent 设计记忆模块的开发者,想要在本地批量生成并管理稠密与稀疏嵌入向量
- 处理敏感数据的合规工程师,要求嵌入计算全程在本地完成并可写入私有 Qdrant 实例
- 构建图文检索应用的开发者,需要在本地同时生成图像嵌入并支持重排序来优化召回质量