feat(hf-transformer): dynamically load library

This commit is contained in:
ItzCrazyKns
2025-10-19 21:06:52 +05:30
parent eef6ebb924
commit d9ebf611ff

View File

@@ -1,6 +1,5 @@
import { Embeddings, type EmbeddingsParams } from '@langchain/core/embeddings';
import { chunkArray } from '@langchain/core/utils/chunk_array';
import { pipeline } from '@huggingface/transformers';
export interface HuggingFaceTransformersEmbeddingsParams
extends EmbeddingsParams {
@@ -68,6 +67,7 @@ export class HuggingFaceTransformersEmbeddings
}
private async runEmbedding(texts: string[]) {
const { pipeline } = await import('@huggingface/transformers');
const pipe = await pipeline('feature-extraction', this.model);
return this.caller.call(async () => {