dev.langchain4j.model.embedding.onnx.bgesmallen

Class BgeSmallEnEmbeddingModel

java.lang.Object

dev.langchain4j.model.embedding.DimensionAwareEmbeddingModel

dev.langchain4j.model.embedding.onnx.AbstractInProcessEmbeddingModel

dev.langchain4j.model.embedding.onnx.bgesmallen.BgeSmallEnEmbeddingModel

All Implemented Interfaces:

dev.langchain4j.model.embedding.EmbeddingModel, dev.langchain4j.model.embedding.TokenCountEstimator

public class BgeSmallEnEmbeddingModel
extends AbstractInProcessEmbeddingModel

BAAI bge-small-en embedding model that runs within your Java application's process.

Maximum length of text (in tokens) that can be embedded at once: unlimited. However, while you can embed very long texts, the quality of the embedding degrades as the text lengthens. It is recommended to embed segments of no more than 512 tokens long.

Embedding dimensions: 384

It is recommended to add "Represent this sentence for searching relevant passages:" prefix to a query.

Uses an Executor to parallelize the embedding process. By default, uses a cached thread pool with the number of threads equal to the number of available processors. Threads are cached for 1 second.

More details here

Field Summary

Fields inherited from class dev.langchain4j.model.embedding.DimensionAwareEmbeddingModel

dimension

Constructor Summary

Constructors

Constructor and Description
BgeSmallEnEmbeddingModel() Creates an instance of an BgeSmallEnEmbeddingModel.
BgeSmallEnEmbeddingModel(Executor executor) Creates an instance of an BgeSmallEnEmbeddingModel.

Method Summary

Modifier and Type	Method and Description
protected Integer	knownDimension()
protected OnnxBertBiEncoder	model()

Methods inherited from class dev.langchain4j.model.embedding.onnx.AbstractInProcessEmbeddingModel

embedAll, estimateTokenCount, loadFromJar

Methods inherited from class dev.langchain4j.model.embedding.DimensionAwareEmbeddingModel

dimension

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface dev.langchain4j.model.embedding.TokenCountEstimator

estimateTokenCount, estimateTokenCount

Methods inherited from interface dev.langchain4j.model.embedding.EmbeddingModel

embed, embed

Constructor Detail

BgeSmallEnEmbeddingModel

public BgeSmallEnEmbeddingModel()

Creates an instance of an BgeSmallEnEmbeddingModel. Uses a cached thread pool with the number of threads equal to the number of available processors. Threads are cached for 1 second.

BgeSmallEnEmbeddingModel

public BgeSmallEnEmbeddingModel(Executor executor)

Creates an instance of an BgeSmallEnEmbeddingModel.

Parameters:

executor - The executor to use to parallelize the embedding process.

Method Detail

model

protected OnnxBertBiEncoder model()

Specified by:

model in class AbstractInProcessEmbeddingModel

knownDimension

protected Integer knownDimension()

Overrides:

knownDimension in class dev.langchain4j.model.embedding.DimensionAwareEmbeddingModel