Spring/spring-ai
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix Keyword extractor test as chatgpt's response changed

Browse Source
This commit is contained in:
Mark Pollack
2024-10-07 16:25:19 +02:00
parent 38082de0f8
commit 22a61548ac
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
models/spring-ai-openai/src/test/java/org/springframework/ai/openai/transformer/MetadataTransformerIT.java
View File

@@ -84,7 +84,7 @@ public class MetadataTransformerIT {
assertThat(keywords2).containsKeys("excerpt_keywords");
assertThat((String) keywords1.get("excerpt_keywords")).contains("Andes", "Aymara");
assertThat((String) keywords2.get("excerpt_keywords")).contains("Spring Framework", "dependency injection");
assertThat((String) keywords2.get("excerpt_keywords")).contains("Spring MVC", "Dependency Injection");
}
@Test
@@ -169,13 +169,13 @@ public class MetadataTransformerIT {
}
@Bean
public KeywordMetadataEnricher keywordMetadata(OpenAiChatModel aiClient) {
return new KeywordMetadataEnricher(aiClient, 5);
public KeywordMetadataEnricher keywordMetadata(OpenAiChatModel chatModel) {
return new KeywordMetadataEnricher(chatModel, 5);
}
@Bean
public SummaryMetadataEnricher summaryMetadata(OpenAiChatModel aiClient) {
return new SummaryMetadataEnricher(aiClient,
public SummaryMetadataEnricher summaryMetadata(OpenAiChatModel chatModel) {
return new SummaryMetadataEnricher(chatModel,
List.of(SummaryType.PREVIOUS, SummaryType.CURRENT, SummaryType.NEXT));
}