Package org.springframework.ai.bedrock.cohere

Class BedrockCohereEmbeddingModel

java.lang.Object

org.springframework.ai.embedding.AbstractEmbeddingModel

org.springframework.ai.bedrock.cohere.BedrockCohereEmbeddingModel

All Implemented Interfaces:

org.springframework.ai.embedding.EmbeddingModel, org.springframework.ai.model.Model<org.springframework.ai.embedding.EmbeddingRequest,org.springframework.ai.embedding.EmbeddingResponse>

public class BedrockCohereEmbeddingModel
extends org.springframework.ai.embedding.AbstractEmbeddingModel

EmbeddingModel implementation that uses the Bedrock Cohere Embedding API. Note: The invocation metrics are not exposed by AWS for this API. If this change in the future we will add it as metadata.

Since:

0.8.0

Author:

Christian Tzolov, Soby Chacko

Field Summary

Fields inherited from class org.springframework.ai.embedding.AbstractEmbeddingModel

embeddingDimensions

Constructor Summary

Constructors

Constructor	Description
BedrockCohereEmbeddingModel(CohereEmbeddingBedrockApi cohereEmbeddingBedrockApi)
BedrockCohereEmbeddingModel(CohereEmbeddingBedrockApi cohereEmbeddingBedrockApi, BedrockCohereEmbeddingOptions options)

Method Summary

Modifier and Type	Method	Description
org.springframework.ai.embedding.EmbeddingResponse	call(org.springframework.ai.embedding.EmbeddingRequest request)
float[]	embed(org.springframework.ai.document.Document document)

Methods inherited from class org.springframework.ai.embedding.AbstractEmbeddingModel

dimensions, dimensions

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.springframework.ai.embedding.EmbeddingModel

embed, embed, embed, embedForResponse

Constructor Details

BedrockCohereEmbeddingModel

public BedrockCohereEmbeddingModel(CohereEmbeddingBedrockApi cohereEmbeddingBedrockApi)

BedrockCohereEmbeddingModel

public BedrockCohereEmbeddingModel(CohereEmbeddingBedrockApi cohereEmbeddingBedrockApi,
 BedrockCohereEmbeddingOptions options)

Method Details

embed

public float[] embed(org.springframework.ai.document.Document document)

call

public org.springframework.ai.embedding.EmbeddingResponse call(org.springframework.ai.embedding.EmbeddingRequest request)