HuggingFace Inference

This Embeddings integration uses the HuggingFace Inference API to generate embeddings for a given text using by default the sentence-transformers/distilbert-base-nli-mean-tokens model. You can pass a different model name to the constructor to use a different model.

Setup

You'll first need to install the @langchain/community package and the required peer dep:

tip

See this section for general instructions on installing integration packages.

npm

npm install @langchain/community @huggingface/inference@2

Yarn

yarn add @langchain/community @huggingface/inference@2

pnpm

pnpm add @langchain/community @huggingface/inference@2

Usage

import { HuggingFaceInferenceEmbeddings } from "@langchain/community/embeddings/hf";

const embeddings = new HuggingFaceInferenceEmbeddings({
  apiKey: "YOUR-API-KEY", // In Node.js defaults to process.env.HUGGINGFACEHUB_API_KEY
});