Skip to content

Commit cc9c2cd

Browse files
committed
Add test for OpenAiModerationProperties
See GH-4708 Signed-off-by: Yanming Zhou <zhouyanming@gmail.com>
1 parent f6ff20a commit cc9c2cd

File tree

2 files changed

+20
-1
lines changed
  • auto-configurations/models

2 files changed

+20
-1
lines changed

auto-configurations/models/spring-ai-autoconfigure-model-mistral-ai/src/test/java/org/springframework/ai/model/mistralai/autoconfigure/MistralAiPropertiesTests.java

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -140,11 +140,14 @@ public void embeddingOptionsTest() {
140140
@Test
141141
public void moderationOptionsTest() {
142142
new ApplicationContextRunner()
143-
.withPropertyValues("spring.ai.mistralai.base-url=TEST_BASE_URL", "spring.ai.mistralai.api-key=abc123",
143+
.withPropertyValues("spring.ai.mistralai.moderation.base-url=TEST_BASE_URL",
144+
"spring.ai.mistralai.moderation.api-key=abc123",
144145
"spring.ai.mistralai.moderation.options.model=MODERATION_MODEL")
145146
.withConfiguration(BaseMistralAiIT.mistralAiModerationAutoConfig())
146147
.run(context -> {
147148
var moderationProperties = context.getBean(MistralAiModerationProperties.class);
149+
assertThat(moderationProperties.getBaseUrl()).isEqualTo("TEST_BASE_URL");
150+
assertThat(moderationProperties.getApiKey()).isEqualTo("abc123");
148151
assertThat(moderationProperties.getOptions().getModel()).isEqualTo("MODERATION_MODEL");
149152
});
150153
}

auto-configurations/models/spring-ai-autoconfigure-model-openai/src/test/java/org/springframework/ai/model/openai/autoconfigure/OpenAiPropertiesTests.java

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@
4646
* @author Thomas Vitale
4747
* @author Jonghoon Park
4848
* @author Issam El-atif
49+
* @author Yanming Zhou
4950
* @since 0.8.0
5051
*/
5152
public class OpenAiPropertiesTests {
@@ -689,4 +690,19 @@ void audioTranscriptionActivation() {
689690

690691
}
691692

693+
@Test
694+
public void moderationOptionsTest() {
695+
this.contextRunner
696+
.withPropertyValues("spring.ai.openai.moderation.base-url=TEST_BASE_URL",
697+
"spring.ai.openai.moderation.api-key=abc123",
698+
"spring.ai.openai.moderation.options.model=MODERATION_MODEL")
699+
.withConfiguration(AutoConfigurations.of(OpenAiModerationAutoConfiguration.class))
700+
.run(context -> {
701+
var moderationProperties = context.getBean(OpenAiModerationProperties.class);
702+
assertThat(moderationProperties.getBaseUrl()).isEqualTo("TEST_BASE_URL");
703+
assertThat(moderationProperties.getApiKey()).isEqualTo("abc123");
704+
assertThat(moderationProperties.getOptions().getModel()).isEqualTo("MODERATION_MODEL");
705+
});
706+
}
707+
692708
}

0 commit comments

Comments
 (0)