Package com.redis.om.spring.vectorize

Class DefaultEmbedder

java.lang.Object

com.redis.om.spring.vectorize.DefaultEmbedder

All Implemented Interfaces:

Embedder

public class DefaultEmbedder
extends Object
implements Embedder

Field Summary

Fields

Modifier and Type	Field	Description
final ai.djl.translate.Pipeline	imagePipeline
final ai.djl.huggingface.tokenizers.HuggingFaceTokenizer	sentenceTokenizer

Constructor Summary

Constructors

Constructor

Description

DefaultEmbedder(org.springframework.context.ApplicationContext applicationContext, ai.djl.repository.zoo.ZooModel<ai.djl.modality.cv.Image,byte[]> imageEmbeddingModel, ai.djl.repository.zoo.ZooModel<ai.djl.modality.cv.Image,float[]> faceEmbeddingModel, ai.djl.modality.cv.ImageFactory imageFactory, ai.djl.translate.Pipeline imagePipeline, ai.djl.huggingface.tokenizers.HuggingFaceTokenizer sentenceTokenizer, org.springframework.ai.openai.OpenAiEmbeddingModel openAITextVectorizer, com.azure.ai.openai.OpenAIClient azureOpenAIClient, org.springframework.ai.vertexai.palm2.VertexAiPaLm2EmbeddingModel vertexAiPaLm2EmbeddingModel, org.springframework.ai.bedrock.cohere.BedrockCohereEmbeddingModel bedrockCohereEmbeddingModel, org.springframework.ai.bedrock.titan.BedrockTitanEmbeddingModel bedrockTitanEmbeddingModel, RedisOMProperties properties)

Method Summary

Modifier and Type	Method	Description
List<byte[]>	getTextEmbeddingsAsBytes(List<String> texts, MetamodelField<?,?> metamodelField)
List<byte[]>	getTextEmbeddingsAsBytes(List<String> texts, Field field)
List<float[]>	getTextEmbeddingsAsFloats(List<String> texts, MetamodelField<?,?> metamodelField)
List<float[]>	getTextEmbeddingsAsFloats(List<String> texts, Field field)
boolean	isReady()
void	processEntity(Object item)

Methods inherited from class java.lang.Object

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

Field Details

imagePipeline

public final ai.djl.translate.Pipeline imagePipeline

sentenceTokenizer

public final ai.djl.huggingface.tokenizers.HuggingFaceTokenizer sentenceTokenizer

Constructor Details

DefaultEmbedder

public DefaultEmbedder(org.springframework.context.ApplicationContext applicationContext,
 ai.djl.repository.zoo.ZooModel<ai.djl.modality.cv.Image,byte[]> imageEmbeddingModel,
 ai.djl.repository.zoo.ZooModel<ai.djl.modality.cv.Image,float[]> faceEmbeddingModel,
 ai.djl.modality.cv.ImageFactory imageFactory,
 ai.djl.translate.Pipeline imagePipeline,
 ai.djl.huggingface.tokenizers.HuggingFaceTokenizer sentenceTokenizer,
 org.springframework.ai.openai.OpenAiEmbeddingModel openAITextVectorizer,
 com.azure.ai.openai.OpenAIClient azureOpenAIClient,
 org.springframework.ai.vertexai.palm2.VertexAiPaLm2EmbeddingModel vertexAiPaLm2EmbeddingModel,
 org.springframework.ai.bedrock.cohere.BedrockCohereEmbeddingModel bedrockCohereEmbeddingModel,
 org.springframework.ai.bedrock.titan.BedrockTitanEmbeddingModel bedrockTitanEmbeddingModel,
 RedisOMProperties properties)

Method Details

processEntity

public void processEntity(Object item)

Specified by:

processEntity in interface Embedder

isReady

public boolean isReady()

Specified by:

isReady in interface Embedder

getTextEmbeddingsAsBytes

public List<byte[]> getTextEmbeddingsAsBytes(List<String> texts,
 Field field)

Specified by:

getTextEmbeddingsAsBytes in interface Embedder

getTextEmbeddingsAsFloats

public List<float[]> getTextEmbeddingsAsFloats(List<String> texts,
 Field field)

Specified by:

getTextEmbeddingsAsFloats in interface Embedder

getTextEmbeddingsAsBytes

public List<byte[]> getTextEmbeddingsAsBytes(List<String> texts,
 MetamodelField<?,?> metamodelField)

Specified by:

getTextEmbeddingsAsBytes in interface Embedder

getTextEmbeddingsAsFloats

public List<float[]> getTextEmbeddingsAsFloats(List<String> texts,
 MetamodelField<?,?> metamodelField)

Specified by:

getTextEmbeddingsAsFloats in interface Embedder