Class ContextualQueryAugmenter

java.lang.Object

org.springframework.ai.rag.generation.augmentation.ContextualQueryAugmenter

All Implemented Interfaces:

BiFunction<Query, List<Document>, Query>, QueryAugmenter

public final class ContextualQueryAugmenter
extends Object
implements QueryAugmenter

Augments the user query with contextual data from the content of the provided documents.

Example usage:

QueryAugmenter augmenter = ContextualQueryAugmenter.builder()
   .allowEmptyContext(false)
   .build();
Query augmentedQuery = augmenter.augment(query, documents);

Since:

1.0.0

Author:

Thomas Vitale

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static final class	ContextualQueryAugmenter.Builder

Constructor Summary

Constructors

Constructor	Description
ContextualQueryAugmenter(PromptTemplate promptTemplate, PromptTemplate emptyContextPromptTemplate, Boolean allowEmptyContext, Function<List<Document>, String> documentFormatter)

Method Summary

Modifier and Type	Method	Description
Query	augment(Query query, List<Document> documents)	Augments the user query with contextual data.
static ContextualQueryAugmenter.Builder	builder()

Methods inherited from class Object

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

Methods inherited from interface BiFunction

andThen

Methods inherited from interface QueryAugmenter

apply

Constructor Details

ContextualQueryAugmenter

public ContextualQueryAugmenter(@Nullable
 PromptTemplate promptTemplate,
 @Nullable
 PromptTemplate emptyContextPromptTemplate,
 @Nullable
 Boolean allowEmptyContext,
 @Nullable
 Function<List<Document>, String> documentFormatter)

Method Details

augment

public Query augment(Query query,
 List<Document> documents)

Description copied from interface: QueryAugmenter

Augments the user query with contextual data.

Specified by:

augment in interface QueryAugmenter

Parameters:

query - The user query to augment

documents - The contextual data to use for augmentation

Returns:

The augmented query

builder

public static ContextualQueryAugmenter.Builder builder()