From 22a61548ac09715ab3fcada677a24048e62b132a Mon Sep 17 00:00:00 2001 From: Mark Pollack Date: Mon, 7 Oct 2024 16:25:19 +0200 Subject: [PATCH] Fix Keyword extractor test as chatgpt's response changed --- .../ai/openai/transformer/MetadataTransformerIT.java | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/models/spring-ai-openai/src/test/java/org/springframework/ai/openai/transformer/MetadataTransformerIT.java b/models/spring-ai-openai/src/test/java/org/springframework/ai/openai/transformer/MetadataTransformerIT.java index d984cb73d..7c8dba5fe 100644 --- a/models/spring-ai-openai/src/test/java/org/springframework/ai/openai/transformer/MetadataTransformerIT.java +++ b/models/spring-ai-openai/src/test/java/org/springframework/ai/openai/transformer/MetadataTransformerIT.java @@ -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)); }