From 7e98fb78b77dc14a2e2ee2018230b9ab2e1ce5a6 Mon Sep 17 00:00:00 2001 From: Christian Tzolov Date: Mon, 15 Jul 2024 12:38:56 +0200 Subject: [PATCH] Fix incorect VertexAi Embedding model atuo-config condition Resolves #1028 --- .../vertexai/embedding/VertexAiEmbeddingAutoConfiguration.java | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/vertexai/embedding/VertexAiEmbeddingAutoConfiguration.java b/spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/vertexai/embedding/VertexAiEmbeddingAutoConfiguration.java index 70aa7ec58..16bde0c6c 100644 --- a/spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/vertexai/embedding/VertexAiEmbeddingAutoConfiguration.java +++ b/spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/vertexai/embedding/VertexAiEmbeddingAutoConfiguration.java @@ -20,7 +20,6 @@ import java.io.IOException; import org.springframework.ai.vertexai.embedding.VertexAiEmbeddigConnectionDetails; import org.springframework.ai.vertexai.embedding.multimodal.VertexAiMultimodalEmbeddingModel; import org.springframework.ai.vertexai.embedding.text.VertexAiTextEmbeddingModel; -import org.springframework.ai.vertexai.gemini.VertexAiGeminiChatModel; import org.springframework.boot.autoconfigure.condition.ConditionalOnClass; import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean; import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; @@ -37,7 +36,7 @@ import com.google.cloud.vertexai.VertexAI; * @author Christian Tzolov * @since 1.0.0 */ -@ConditionalOnClass({ VertexAI.class, VertexAiGeminiChatModel.class }) +@ConditionalOnClass({ VertexAI.class, VertexAiTextEmbeddingModel.class }) @EnableConfigurationProperties({ VertexAiEmbeddingConnectionProperties.class, VertexAiTextEmbeddingProperties.class, VertexAiMultimodalEmbeddingProperties.class, }) public class VertexAiEmbeddingAutoConfiguration {