Fix OllamaEmbeddingAutoConfigurationTests

- Use OllamaEmbeddingAutoConfiguration for the test

Signed-off-by: Ilayaperumal Gopinathan <ilayaperumal.gopinathan@broadcom.com>
This commit is contained in:
Ilayaperumal Gopinathan
2025-06-12 18:12:22 +01:00
parent dc9ea17fc6
commit 330986998a

View File

@@ -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);