Class BedrockCohereEmbeddingModel

java.lang.Object

dev.langchain4j.model.embedding.DimensionAwareEmbeddingModel

dev.langchain4j.model.bedrock.BedrockCohereEmbeddingModel

All Implemented Interfaces:

EmbeddingModel

public class BedrockCohereEmbeddingModel
extends DimensionAwareEmbeddingModel

Bedrock Cohere embedding model with support for both versions: cohere.embed-english-v3 and cohere.embed-multilingual-v3
See more details here and here.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	BedrockCohereEmbeddingModel.Builder
static enum	BedrockCohereEmbeddingModel.InputType
static enum	BedrockCohereEmbeddingModel.Model
static enum	BedrockCohereEmbeddingModel.Truncate

Field Summary

Fields inherited from class DimensionAwareEmbeddingModel

dimension

Constructor Summary

Constructors

Constructor	Description
BedrockCohereEmbeddingModel(BedrockCohereEmbeddingModel.Builder builder)

Method Summary

Modifier and Type	Method	Description
static BedrockCohereEmbeddingModel.Builder	builder()
Response<List<Embedding>>	embedAll(List<TextSegment> textSegments)	Embeds the text content of a list of TextSegments.
String	modelName()	Returns the name of the underlying embedding model.
ModelProvider	provider()	The ModelProvider of this embedding model (for example ModelProvider.OPEN_AI).

Methods inherited from class DimensionAwareEmbeddingModel

dimension, knownDimension

Methods inherited from class Object

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

Methods inherited from interface EmbeddingModel

addListener, addListeners, defaultRequestParameters, doEmbed, embed, embed, embed, listeners, supportedContentTypes, supportedParameters

Constructor Details

BedrockCohereEmbeddingModel

public BedrockCohereEmbeddingModel(BedrockCohereEmbeddingModel.Builder builder)

Method Details

embedAll

public Response<List<Embedding>> embedAll(List<TextSegment> textSegments)

Description copied from interface: EmbeddingModel

Embeds the text content of a list of TextSegments.

This is a convenience method over EmbeddingModel.embed(EmbeddingRequest), so listeners are notified and the model's default per-call parameters are applied here too. A provider implements its embedding logic by overriding EmbeddingModel.doEmbed(EmbeddingRequest) (preferred); it may still override this method for batch-specific behavior that the request/response API does not carry (for example applying a document title from TextSegment metadata).

Parameters:

textSegments - the text segments to embed.

Returns:

the embeddings.

provider

public ModelProvider provider()

Description copied from interface: EmbeddingModel

The ModelProvider of this embedding model (for example ModelProvider.OPEN_AI). It is made available to listeners through the request, response and error contexts. Defaults to ModelProvider.OTHER.

modelName

public String modelName()

Description copied from interface: EmbeddingModel

Returns the name of the underlying embedding model.

Implementations are encouraged to override this method and provide the actual model name. The default implementation returns "unknown", which indicates that the model name is unknown.

Returns:

the model name or a fallback value if not provided

builder

public static BedrockCohereEmbeddingModel.Builder builder()