Package dev.langchain4j.rag.query.transformer

Class ExpandingQueryTransformer

java.lang.Object

dev.langchain4j.rag.query.transformer.ExpandingQueryTransformer

All Implemented Interfaces:

QueryTransformer

public class ExpandingQueryTransformer
extends Object
implements QueryTransformer

A QueryTransformer that utilizes a ChatLanguageModel to expand a given Query.
Refer to DEFAULT_PROMPT_TEMPLATE and implementation for more details.

Configurable parameters (optional):
- promptTemplate: The prompt template used to instruct the LLM to expand the provided Query.
- n: The number of Querys to generate. Default value is 3.

See Also:

• DefaultQueryTransformer
• CompressingQueryTransformer

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	ExpandingQueryTransformer.ExpandingQueryTransformerBuilder

Field Summary

Fields

Modifier and Type	Field	Description
protected final ChatLanguageModel	chatLanguageModel
static final int	DEFAULT_N
static final PromptTemplate	DEFAULT_PROMPT_TEMPLATE
protected final int	n
protected final PromptTemplate	promptTemplate

Constructor Summary

Constructors

Constructor	Description
ExpandingQueryTransformer(ChatLanguageModel chatLanguageModel)
ExpandingQueryTransformer(ChatLanguageModel chatLanguageModel, int n)
ExpandingQueryTransformer(ChatLanguageModel chatLanguageModel, PromptTemplate promptTemplate)
ExpandingQueryTransformer(ChatLanguageModel chatLanguageModel, PromptTemplate promptTemplate, Integer n)

Method Summary

Modifier and Type	Method	Description
static ExpandingQueryTransformer.ExpandingQueryTransformerBuilder	builder()
protected Prompt	createPrompt(Query query)
protected List<String>	parse(String queries)
Collection<Query>	transform(Query query)	Transforms the given Query into one or multiple Querys.

Methods inherited from class java.lang.Object

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

Field Details

DEFAULT_PROMPT_TEMPLATE

public static final PromptTemplate DEFAULT_PROMPT_TEMPLATE

DEFAULT_N

public static final int DEFAULT_N

See Also:

• Constant Field Values

chatLanguageModel

protected final ChatLanguageModel chatLanguageModel

promptTemplate

protected final PromptTemplate promptTemplate

n

protected final int n

Constructor Details

ExpandingQueryTransformer

public ExpandingQueryTransformer(ChatLanguageModel chatLanguageModel)

ExpandingQueryTransformer

public ExpandingQueryTransformer(ChatLanguageModel chatLanguageModel,
 int n)

ExpandingQueryTransformer

public ExpandingQueryTransformer(ChatLanguageModel chatLanguageModel,
 PromptTemplate promptTemplate)

ExpandingQueryTransformer

public ExpandingQueryTransformer(ChatLanguageModel chatLanguageModel,
 PromptTemplate promptTemplate,
 Integer n)

Method Details

builder

public static ExpandingQueryTransformer.ExpandingQueryTransformerBuilder builder()

transform

public Collection<Query> transform(Query query)

Description copied from interface: QueryTransformer

Transforms the given Query into one or multiple Querys.

Specified by:

transform in interface QueryTransformer

Parameters:

query - The Query to be transformed.

Returns:

A collection of one or more Querys derived from the original Query.

createPrompt

protected Prompt createPrompt(Query query)

parse

protected List<String> parse(String queries)