From 330986998a31b5221bdd2de16a3bcdcb21f16a31 Mon Sep 17 00:00:00 2001 From: Ilayaperumal Gopinathan Date: Thu, 12 Jun 2025 18:12:22 +0100 Subject: [PATCH] Fix OllamaEmbeddingAutoConfigurationTests - Use OllamaEmbeddingAutoConfiguration for the test Signed-off-by: Ilayaperumal Gopinathan --- .../autoconfigure/OllamaEmbeddingAutoConfigurationTests.java | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/test/java/org/springframework/ai/model/ollama/autoconfigure/OllamaEmbeddingAutoConfigurationTests.java b/auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/test/java/org/springframework/ai/model/ollama/autoconfigure/OllamaEmbeddingAutoConfigurationTests.java index bc6f2a6d3..2490e5258 100644 --- a/auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/test/java/org/springframework/ai/model/ollama/autoconfigure/OllamaEmbeddingAutoConfigurationTests.java +++ b/auto-configurations/models/spring-ai-autoconfigure-model-ollama/src/test/java/org/springframework/ai/model/ollama/autoconfigure/OllamaEmbeddingAutoConfigurationTests.java @@ -45,7 +45,7 @@ public class OllamaEmbeddingAutoConfigurationTests { ) .withConfiguration(AutoConfigurations.of(SpringAiRetryAutoConfiguration.class, - RestClientAutoConfiguration.class, OllamaChatAutoConfiguration.class)) + RestClientAutoConfiguration.class, OllamaEmbeddingAutoConfiguration.class)) .run(context -> { var embeddingProperties = context.getBean(OllamaEmbeddingProperties.class); var connectionProperties = context.getBean(OllamaConnectionProperties.class);