diff --git a/spring-ai-rag/src/main/java/org/springframework/ai/rag/preretrieval/query/expansion/MultiQueryExpander.java b/spring-ai-rag/src/main/java/org/springframework/ai/rag/preretrieval/query/expansion/MultiQueryExpander.java index 314a455ba25..101750a5b27 100644 --- a/spring-ai-rag/src/main/java/org/springframework/ai/rag/preretrieval/query/expansion/MultiQueryExpander.java +++ b/spring-ai-rag/src/main/java/org/springframework/ai/rag/preretrieval/query/expansion/MultiQueryExpander.java @@ -111,9 +111,11 @@ public List expand(Query query) { return List.of(query); } - var queryVariants = Arrays.asList(response.split("\n")); + var queryVariants = Arrays.stream(response.split("\n")) + .filter(StringUtils::hasText) + .toList(); - if (CollectionUtils.isEmpty(queryVariants) || this.numberOfQueries != queryVariants.size()) { + if (CollectionUtils.isEmpty(queryVariants) || this.numberOfQueries > queryVariants.size()) { logger.warn( "Query expansion result does not contain the requested {} variants. Returning the input query unchanged.", this.numberOfQueries);