Top Fields Constructors Methods

org.jabref.logic.ai.embedding

public Class DeepJavaEmbeddingModel

Class Inheritance

extends Object
implements dev.langchain4j.model.embedding.EmbeddingModel, AutoCloseable

java.lang.Object
org.jabref.logic.ai.embedding.DeepJavaEmbeddingModel

All Implemented Interfaces

java.lang.AutoCloseable

Imports

java.io.IOException, java.util.ArrayList, .List, .OptionalInt, org.jabref.logic.util.ProgressCounter, org.jabref.logic.util.strings.StringUtil, ai.djl.MalformedModelException, ai.djl.huggingface.translator.TextEmbeddingTranslatorFactory, ai.djl.inference.Predictor, ai.djl.repository.zoo.Criteria, ai.djl.repository.zoo.ModelNotFoundException, ai.djl.repository.zoo.ZooModel, ai.djl.translate.TranslateException, dev.langchain4j.data.embedding.Embedding, dev.langchain4j.data.segment.TextSegment, dev.langchain4j.model.embedding.EmbeddingModel, dev.langchain4j.model.output.Response, org.slf4j.Logger, .LoggerFactory

Field Summary

Modifier and Type	Field and Description
public static final String	DJL_EMBEDDING_MODEL_URL_PREFIX
private static final Logger	LOGGER
private final EmbeddingModelMetadataService	metadataService
private final ai.djl.repository.zoo.ZooModel<String, float[]>	model
private final String	modelName
private final ai.djl.inference.Predictor<String, float[]>	predictor

Constructor Summary

Access	Constructor and Description
public	DeepJavaEmbeddingModel(String modelName, ProgressCounter progressCounter, EmbeddingModelMetadataService metadataService)

Method Summary

Modifier and Type	Method and Description
public void	close() Implements java.lang.AutoCloseable.close. Closes this resource, relinquishing any underlying resources.
public dev.langchain4j.model.output.Response<List<dev.langchain4j.data.embedding.Embedding>>	embedAll(List<dev.langchain4j.data.segment.TextSegment> list)
public OptionalInt	getMaxSnippetTokens() Returns the maximum snippet size (sequence length in tokens) supported by this embedding model.
public String	getModelName()
public static boolean	isDownloaded(String modelName)
private static ai.djl.repository.zoo.Criteria<String, float[]>	makeCriteria(String modelUrl)
private static ai.djl.repository.zoo.Criteria.Builder<String, float[]>	makeCriteriaBuilder()

Inherited from java.lang.Object:: clone equals finalize getClass hashCode notify notifyAll toString wait wait wait

Field Detail

DJL_EMBEDDING_MODEL_URL_PREFIX	back to summary
public static final String DJL_EMBEDDING_MODEL_URL_PREFIX

LOGGER	back to summary
private static final Logger LOGGER

metadataService	back to summary
private final EmbeddingModelMetadataService metadataService

model	back to summary
private final ai.djl.repository.zoo.ZooModel<String, float[]> model

modelName	back to summary
private final String modelName

predictor	back to summary
private final ai.djl.inference.Predictor<String, float[]> predictor

Constructor Detail

DeepJavaEmbeddingModel	back to summary
public DeepJavaEmbeddingModel(String modelName, ProgressCounter progressCounter, EmbeddingModelMetadataService metadataService) throws ai.djl.repository.zoo.ModelNotFoundException, ai.djl.MalformedModelException, IOException

Method Detail

close back to summary

public void close()

Implements java.lang.AutoCloseable.close.

Doc from java.lang.AutoCloseable.close.

Closes this resource, relinquishing any underlying resources. This method is invoked automatically on objects managed by the try-with-resources statement.

Annotations

@Override

embedAll	back to summary
public dev.langchain4j.model.output.Response<List<dev.langchain4j.data.embedding.Embedding>> embedAll(List<dev.langchain4j.data.segment.TextSegment> list) Annotations @Override

getMaxSnippetTokens	back to summary
public OptionalInt getMaxSnippetTokens() Returns the maximum snippet size (sequence length in tokens) supported by this embedding model.

getModelName	back to summary
public String getModelName()

isDownloaded	back to summary
public static boolean isDownloaded(String modelName)

makeCriteria	back to summary
private static ai.djl.repository.zoo.Criteria<String, float[]> makeCriteria(String modelUrl)

makeCriteriaBuilder	back to summary
private static ai.djl.repository.zoo.Criteria.Builder<String, float[]> makeCriteriaBuilder()