Package org.springframework.ai.vertexai.embedding.text

Class VertexAiTextEmbeddingModel

java.lang.Object

org.springframework.ai.embedding.AbstractEmbeddingModel

org.springframework.ai.vertexai.embedding.text.VertexAiTextEmbeddingModel

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 VertexAiTextEmbeddingModel
extends org.springframework.ai.embedding.AbstractEmbeddingModel

A class representing a Vertex AI Text Embedding Model.

Since:

1.0.0

Author:

Christian Tzolov, Mark Pollack, Rodrigo Malara, Soby Chacko

Field Summary

Fields

Modifier and Type	Field	Description
final VertexAiTextEmbeddingOptions	defaultOptions

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

embeddingDimensions

Constructor Summary

Constructors

Constructor	Description
VertexAiTextEmbeddingModel(VertexAiEmbeddingConnectionDetails connectionDetails, VertexAiTextEmbeddingOptions defaultEmbeddingOptions)
VertexAiTextEmbeddingModel(VertexAiEmbeddingConnectionDetails connectionDetails, VertexAiTextEmbeddingOptions defaultEmbeddingOptions, org.springframework.retry.support.RetryTemplate retryTemplate)
VertexAiTextEmbeddingModel(VertexAiEmbeddingConnectionDetails connectionDetails, VertexAiTextEmbeddingOptions defaultEmbeddingOptions, org.springframework.retry.support.RetryTemplate retryTemplate, io.micrometer.observation.ObservationRegistry observationRegistry)

Method Summary

Modifier and Type	Method	Description
org.springframework.ai.embedding.EmbeddingResponse	call(org.springframework.ai.embedding.EmbeddingRequest request)
int	dimensions()
float[]	embed(org.springframework.ai.document.Document document)
protected com.google.cloud.aiplatform.v1.PredictRequest.Builder	getPredictRequestBuilder(org.springframework.ai.embedding.EmbeddingRequest request, com.google.cloud.aiplatform.v1.EndpointName endpointName, VertexAiTextEmbeddingOptions finalOptions)
void	setObservationConvention(org.springframework.ai.embedding.observation.EmbeddingModelObservationConvention observationConvention)	Use the provided convention for reporting observation data

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

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

Field Details

defaultOptions

public final VertexAiTextEmbeddingOptions defaultOptions

Constructor Details

VertexAiTextEmbeddingModel

public VertexAiTextEmbeddingModel(VertexAiEmbeddingConnectionDetails connectionDetails,
 VertexAiTextEmbeddingOptions defaultEmbeddingOptions)

VertexAiTextEmbeddingModel

public VertexAiTextEmbeddingModel(VertexAiEmbeddingConnectionDetails connectionDetails,
 VertexAiTextEmbeddingOptions defaultEmbeddingOptions,
 org.springframework.retry.support.RetryTemplate retryTemplate)

VertexAiTextEmbeddingModel

public VertexAiTextEmbeddingModel(VertexAiEmbeddingConnectionDetails connectionDetails,
 VertexAiTextEmbeddingOptions defaultEmbeddingOptions,
 org.springframework.retry.support.RetryTemplate retryTemplate,
 io.micrometer.observation.ObservationRegistry observationRegistry)

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)

getPredictRequestBuilder

protected com.google.cloud.aiplatform.v1.PredictRequest.Builder getPredictRequestBuilder(org.springframework.ai.embedding.EmbeddingRequest request,
 com.google.cloud.aiplatform.v1.EndpointName endpointName,
 VertexAiTextEmbeddingOptions finalOptions)

dimensions

public int dimensions()

Specified by:

dimensions in interface org.springframework.ai.embedding.EmbeddingModel

Overrides:

dimensions in class org.springframework.ai.embedding.AbstractEmbeddingModel

setObservationConvention

public void setObservationConvention(org.springframework.ai.embedding.observation.EmbeddingModelObservationConvention observationConvention)

Use the provided convention for reporting observation data

Parameters:

observationConvention - The provided convention