Package org.springframework.ai.rag.retrieval.join

Class ConcatenationDocumentJoiner

java.lang.Object

org.springframework.ai.rag.retrieval.join.ConcatenationDocumentJoiner

All Implemented Interfaces:

Function<Map<Query,List<List<Document>>>,List<Document>>, DocumentJoiner

public class ConcatenationDocumentJoiner
extends Object
implements DocumentJoiner

Combines documents retrieved based on multiple queries and from multiple data sources by concatenating them into a single collection of documents. In case of duplicate documents, the first occurrence is kept. The score of each document is kept as is.

Since:

1.0.0

Author:

Thomas Vitale

Constructor Summary

Constructors

Constructor	Description
ConcatenationDocumentJoiner()

Method Summary

Modifier and Type	Method	Description
List<Document>	join(Map<Query,List<List<Document>>> documentsForQuery)	Joins documents retrieved across multiple queries and daa sources.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.springframework.ai.rag.retrieval.join.DocumentJoiner

apply

Methods inherited from interface java.util.function.Function

andThen, compose

Constructor Details

ConcatenationDocumentJoiner

public ConcatenationDocumentJoiner()

Method Details

join

public List<Document> join(Map<Query,List<List<Document>>> documentsForQuery)

Description copied from interface: DocumentJoiner

Joins documents retrieved across multiple queries and daa sources.

Specified by:

join in interface DocumentJoiner

Parameters:

documentsForQuery - a map of queries and the corresponding list of documents retrieved

Returns:

a single collection of documents