Agent Skills
返回列表
💻

本地文本嵌入引擎 v2.0

开发编程 更新于 2026.08.30

将以下提示词粘贴到你的 AI 对话框中:

请根据 https://skillhub.cn/install/skillhub.md,将 @user_287b0fec/local-embedding-engine 安装到当前 AI 助手中。

技能介绍

解决什么问题

做本地 RAG、语义搜索或文风对齐时,embedding 往往是最先卡住的一环:调云厂商 API 要 Key、按量计费,文本还要离开内网;不少项目因此只能退回关键词匹配,召回质量明显打折。这个技能把 embedding 模型拉到本地跑,不依赖任何 OpenAI 风格的 API,适合离线、内网或隐私敏感的部署。

核心架构与工作方式

双路径设计是它的基础:

  • mode="nn":基于 sentence-transformers/all-MiniLM-L6-v2,输出 768 维稠密向量,短文本约 55 句/秒(AMD 5500U,纯 CPU),官方基准下质量约为 ada-002 的 85–90%
  • mode="stat":TF-IDF 关键词重叠 + N-gram Jaccard + 编辑距离归一化,纯标准库实现,吞吐量可达万句/秒
  • mode="auto":优先 nn,环境缺依赖时自动降级到 stat,零配置即可运行

长文本处理走滑动窗口:超过 1024 字符自动分段,chunk_size=512、overlap=128,各块独立编码后平均池化成最终向量。5000 字左右的文本若不分段,nn 路径会 OOM。

v2.0 的四项新增能力:

  • SQLite 持久化向量库:向量与元数据落盘,重启不丢;按 hash 做增量更新,rebuild_index() 支持全量重建
  • 多文档索引:index_file / index_directory 按 ##/### 标题自动分段,IDF 基于整个语料库做全局归一化
  • BM25 混合搜索:语义向量与关键词双通道,α=0.7, β=0.3 权重可调,可返回 high/medium/low 置信度
  • 自进化反馈:learn_feedback() 记录人工纠正,同一 label 累计 ≥3 条后给出权重调整建议,持久化到 feedback.jsonl

适用边界

  • nn 路径需要 sentence-transformers(torch CPU 约 200MB + 模型 22MB);stat 路径语义能力较弱,只适合兜底或关键词偏重的场景
  • 85–90% 是相对 ada-002 的估计值,高精度检索(如法律、医疗)仍需自行评测,或考虑更大的模型
  • SQLite 支持多进程读,但不支持并发写,多写场景要自行串行化或加锁
  • 内网访问 Hugging Face 受限时需配置镜像(已内置 HF_ENDPOINT)

使用场景

  • 在内网离线环境搭建 RAG:把产品说明书按标题切片存入本地 SQLite 向量库,并做语义检索召回相关章节
  • 给一批 AI 生成稿做文风对齐:计算每篇与参考范文的语义相似度并排序,标记偏离风格的稿件
  • 把整个技术文档目录批量索引进向量库,回答提问时同时跑 BM25 关键词与语义向量双通道并调权重
  • 客户反馈工单涉及隐私不能出内网,需在本地计算文本相似度,做相似工单去重与关联推荐

适合人员

  • 后端工程师:要在不能连外网、无 GPU 的服务器上部署语义检索,需要零 API 依赖的本地 embedding 方案
  • RAG 算法工程师:正在调召回质量,需要对比向量检索与 BM25 混合检索的效果并逐步调权重
  • 内容审核或技术编辑:负责检查 AI 生成稿是否贴合范文风格,需要批量计算文风相似度
  • 注重数据合规的产品负责人:用户文本不允许调用第三方 API,需要在本地完成向量化与检索