@langchain/community/dist/embeddings/hf.cjs

Third-party integrations for LangChain.js

Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
const require_runtime = require("../_virtual/_rolldown/runtime.cjs");
let _langchain_core_utils_env = require("@langchain/core/utils/env");
let _langchain_core_embeddings = require("@langchain/core/embeddings");
let _huggingface_inference = require("@huggingface/inference");
//#region src/embeddings/hf.ts
var hf_exports = /* @__PURE__ */ require_runtime.__exportAll({ HuggingFaceInferenceEmbeddings: () => HuggingFaceInferenceEmbeddings });
/**
* Class that extends the Embeddings class and provides methods for
* generating embeddings using Hugging Face models through the
* HuggingFaceInference API.
*/
var HuggingFaceInferenceEmbeddings = class extends _langchain_core_embeddings.Embeddings {
	apiKey;
	model;
	endpointUrl;
	provider;
	client;
	constructor(fields) {
		super(fields ?? {});
		if (fields?.model) this.model = fields.model;
		else {
			console.warn("[HuggingFaceInferenceEmbeddings] No \"model\" provided. Using default: \"BAAI/bge-base-en-v1.5\".");
			this.model = "BAAI/bge-base-en-v1.5";
		}
		this.apiKey = fields?.apiKey ?? (0, _langchain_core_utils_env.getEnvironmentVariable)("HUGGINGFACEHUB_API_KEY");
		this.endpointUrl = fields?.endpointUrl;
		this.provider = fields?.provider;
		this.client = this.endpointUrl ? new _huggingface_inference.InferenceClient(this.apiKey).endpoint(this.endpointUrl) : new _huggingface_inference.InferenceClient(this.apiKey);
	}
	async _embed(texts) {
		const clean = texts.map((text) => text.replace(/\n/g, " "));
		return this.caller.call(() => this.client.featureExtraction({
			model: this.model,
			inputs: clean,
			provider: this.provider
		}));
	}
	/**
	* Method that takes a document as input and returns a promise that
	* resolves to an embedding for the document. It calls the _embed method
	* with the document as the input and returns the first embedding in the
	* resulting array.
	* @param document Document to generate an embedding for.
	* @returns Promise that resolves to an embedding for the document.
	*/
	embedQuery(document) {
		return this._embed([document]).then((embeddings) => embeddings[0]);
	}
	/**
	* Method that takes an array of documents as input and returns a promise
	* that resolves to a 2D array of embeddings for each document. It calls
	* the _embed method with the documents as the input.
	* @param documents Array of documents to generate embeddings for.
	* @returns Promise that resolves to a 2D array of embeddings for each document.
	*/
	embedDocuments(documents) {
		return this._embed(documents);
	}
};
//#endregion
exports.HuggingFaceInferenceEmbeddings = HuggingFaceInferenceEmbeddings;
Object.defineProperty(exports, "hf_exports", {
	enumerable: true,
	get: function() {
		return hf_exports;
	}
});

//# sourceMappingURL=hf.cjs.map