HuggingFace Transformers
TransformerEmbeddings 类使用 Transformers.js 包为给定的文本生成嵌入。
它在本地运行，甚至可以直接在浏览器中使用，允许您创建具有内置嵌入的 Web 应用程序。
安装
您需要安装 @xenova/transformers 包作为对等依赖项
- npm
- Yarn
- pnpm
npm install @xenova/transformers
yarn add @xenova/transformers
pnpm add @xenova/transformers
提示
- npm
- Yarn
- pnpm
npm install @langchain/community @langchain/core
yarn add @langchain/community @langchain/core
pnpm add @langchain/community @langchain/core
示例
请注意，如果您在浏览器环境中使用，您可能希望将所有与推理相关的代码放在 Web 工作器中，以避免阻塞主线程。
查看 本指南以及 Transformers.js 文档中的其他资源，了解如何设置您的项目。
import { HuggingFaceTransformersEmbeddings } from "@langchain/community/embeddings/hf_transformers";
const model = new HuggingFaceTransformersEmbeddings({
model: "Xenova/all-MiniLM-L6-v2",
});
/* Embed queries */
const res = await model.embedQuery(
"What would be a good company name for a company that makes colorful socks?"
);
console.log({ res });
/* Embed documents */
const documentRes = await model.embedDocuments(["Hello world", "Bye bye"]);
console.log({ documentRes });
API 参考
- HuggingFaceTransformersEmbeddings 来自
@langchain/community/embeddings/hf_transformers