Package dev.langchain4j.rag
Class DefaultRetrievalAugmentor
java.lang.Object
dev.langchain4j.rag.DefaultRetrievalAugmentor
- All Implemented Interfaces:
RetrievalAugmentor
The default implementation of
It's important to note that while efforts will be made to avoid breaking changes, the default behavior of this class may be updated in the future if it's found that the current behavior does not adequately serve the majority of use cases. Such changes would be made to benefit both current and future users.
This implementation is inspired by this article and this paper. It is recommended to review these resources for a better understanding of the concept.
This implementation orchestrates the flow between the following base components:
For each base component listed above, we offer several ready-to-use implementations, each based on a recognized approach. We intend to introduce more such implementations over time and welcome your contributions.
The flow is as follows:
1. A
2. Each
3. All
4. Lastly, a final list of
By default, each base component (except for
By default, query routing and content retrieval are performed concurrently (for efficiency) using
RetrievalAugmentor
intended to be suitable for the majority of use cases.
It's important to note that while efforts will be made to avoid breaking changes, the default behavior of this class may be updated in the future if it's found that the current behavior does not adequately serve the majority of use cases. Such changes would be made to benefit both current and future users.
This implementation is inspired by this article and this paper. It is recommended to review these resources for a better understanding of the concept.
This implementation orchestrates the flow between the following base components:
-Visual representation of this flow can be found here.QueryTransformer
-QueryRouter
-ContentRetriever
-ContentAggregator
-ContentInjector
For each base component listed above, we offer several ready-to-use implementations, each based on a recognized approach. We intend to introduce more such implementations over time and welcome your contributions.
The flow is as follows:
1. A
Query
(derived from an original UserMessage
) is transformed
using a QueryTransformer
into one or multiple Query
s.
2. Each
Query
is routed to the appropriate ContentRetriever
using a QueryRouter
.
Each ContentRetriever
retrieves one or multiple Content
s using a Query
.
3. All
Content
s retrieved by all ContentRetriever
s using all Query
s are
aggregated (fused/re-ranked/filtered/etc.) into a final list of Content
s using a ContentAggregator
.
4. Lastly, a final list of
Content
s is injected into the original UserMessage
using a ContentInjector
.
By default, each base component (except for
ContentRetriever
, which needs to be provided by you)
is initialized with a sensible default implementation:
-Nonetheless, you are encouraged to use one of the advanced ready-to-use implementations or create a custom one.DefaultQueryTransformer
-DefaultQueryRouter
-DefaultContentAggregator
-DefaultContentInjector
By default, query routing and content retrieval are performed concurrently (for efficiency) using
Executors.newCachedThreadPool()
, but you can provide a custom Executor
.- See Also:
-
Nested Class Summary
Nested ClassesModifier and TypeClassDescriptionstatic class
-
Constructor Summary
ConstructorsConstructorDescriptionDefaultRetrievalAugmentor
(QueryTransformer queryTransformer, QueryRouter queryRouter, ContentAggregator contentAggregator, ContentInjector contentInjector, Executor executor) -
Method Summary
Modifier and TypeMethodDescriptionaugment
(UserMessage userMessage, Metadata metadata) Augments the providedUserMessage
with retrieved content.builder()
-
Constructor Details
-
DefaultRetrievalAugmentor
public DefaultRetrievalAugmentor(QueryTransformer queryTransformer, QueryRouter queryRouter, ContentAggregator contentAggregator, ContentInjector contentInjector, Executor executor)
-
-
Method Details
-
augment
Description copied from interface:RetrievalAugmentor
Augments the providedUserMessage
with retrieved content.- Specified by:
augment
in interfaceRetrievalAugmentor
- Parameters:
userMessage
- TheUserMessage
to be augmented.metadata
- TheMetadata
that may be useful or necessary for retrieval and augmentation.- Returns:
- The augmented
UserMessage
.
-
builder
-