Skip to content

Commit 958dbd3

Browse files
committed
Fix TransformersEmbeddingModelAutoConfigurationIT
- Update the right settings to enable the configuration properties
1 parent c11cc73 commit 958dbd3

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

auto-configurations/models/spring-ai-autoconfigure-model-transformers/src/test/java/org/springframework/ai/model/transformers/autoconfigure/TransformersEmbeddingModelAutoConfigurationIT.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -92,12 +92,12 @@ public void remoteOnnxModel() {
9292

9393
@Test
9494
void embeddingActivation() {
95-
this.contextRunner.withPropertyValues("spring.ai.embedding.transformer.enabled=false").run(context -> {
96-
assertThat(context.getBeansOfType(TransformersEmbeddingModelProperties.class)).isNotEmpty();
95+
this.contextRunner.withPropertyValues("spring.ai.model.embedding=none").run(context -> {
96+
assertThat(context.getBeansOfType(TransformersEmbeddingModelProperties.class)).isEmpty();
9797
assertThat(context.getBeansOfType(TransformersEmbeddingModel.class)).isEmpty();
9898
});
9999

100-
this.contextRunner.withPropertyValues("spring.ai.embedding.transformer.enabled=true").run(context -> {
100+
this.contextRunner.withPropertyValues("spring.ai.model.embedding=transformers").run(context -> {
101101
assertThat(context.getBeansOfType(TransformersEmbeddingModelProperties.class)).isNotEmpty();
102102
assertThat(context.getBeansOfType(TransformersEmbeddingModel.class)).isNotEmpty();
103103
});

0 commit comments

Comments
 (0)