Package org.springframework.ai.rag.preretrieval.query.transformation

Class CompressionQueryTransformer

java.lang.Object

org.springframework.ai.rag.preretrieval.query.transformation.CompressionQueryTransformer

All Implemented Interfaces:

Function<Query,Query>, QueryTransformer

public class CompressionQueryTransformer
extends Object
implements QueryTransformer

Uses a large language model to compress a conversation history and a follow-up query into a standalone query that captures the essence of the conversation.

This transformer is useful when the conversation history is long and the follow-up query is related to the conversation context.

Since:

1.0.0

Author:

Thomas Vitale

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static final class	CompressionQueryTransformer.Builder

Constructor Summary

Constructors

Constructor	Description
CompressionQueryTransformer(org.springframework.ai.chat.client.ChatClient.Builder chatClientBuilder, org.springframework.ai.chat.prompt.PromptTemplate promptTemplate)

Method Summary

Modifier and Type	Method	Description
static CompressionQueryTransformer.Builder	builder()
Query	transform(Query query)	Transforms the given query according to the implemented strategy.

Methods inherited from class java.lang.Object

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

Methods inherited from interface java.util.function.Function

andThen, compose

Methods inherited from interface org.springframework.ai.rag.preretrieval.query.transformation.QueryTransformer

apply

Constructor Details

CompressionQueryTransformer

public CompressionQueryTransformer(org.springframework.ai.chat.client.ChatClient.Builder chatClientBuilder,
 @Nullable
 org.springframework.ai.chat.prompt.PromptTemplate promptTemplate)

Method Details

transform

public Query transform(Query query)

Description copied from interface: QueryTransformer

Transforms the given query according to the implemented strategy.

Specified by:

transform in interface QueryTransformer

Parameters:

query - The original query to transform

Returns:

The transformed query

builder

public static CompressionQueryTransformer.Builder builder()