dev.langchain4j.rag.query.router

Class LanguageModelQueryRouter

java.lang.Object

dev.langchain4j.rag.query.router.LanguageModelQueryRouter

All Implemented Interfaces:

QueryRouter

public class LanguageModelQueryRouter
extends Object
implements QueryRouter

A QueryRouter that utilizes a ChatLanguageModel to make a routing decision.
Each ContentRetriever provided in the constructor is accompanied by a description which is provided to the LLM to aid in making a decision on where a Query should be routed.
Refer to DEFAULT_PROMPT_TEMPLATE and implementation for more details.

Configurable parameters (optional):
- promptTemplate: The prompt template used to ask the LLM for routing decisions.

See Also:

DefaultQueryRouter

Field Summary

Fields

Modifier and Type	Field and Description
static PromptTemplate	DEFAULT_PROMPT_TEMPLATE

Constructor Summary

Constructors

Constructor and Description
LanguageModelQueryRouter(ChatLanguageModel chatLanguageModel, Map<ContentRetriever,String> retrieverToDescription)
LanguageModelQueryRouter(ChatLanguageModel chatLanguageModel, Map<ContentRetriever,String> retrieverToDescription, PromptTemplate promptTemplate)

Method Summary

Modifier and Type	Method and Description
protected Prompt	createPrompt(Query query)
protected Collection<ContentRetriever>	parse(String choices)
Collection<ContentRetriever>	route(Query query) Routes the given Query to one or multiple ContentRetrievers.

Methods inherited from class java.lang.Object

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

Field Detail

DEFAULT_PROMPT_TEMPLATE

public static final PromptTemplate DEFAULT_PROMPT_TEMPLATE

Constructor Detail

LanguageModelQueryRouter

public LanguageModelQueryRouter(ChatLanguageModel chatLanguageModel,
                                Map<ContentRetriever,String> retrieverToDescription)

LanguageModelQueryRouter

public LanguageModelQueryRouter(ChatLanguageModel chatLanguageModel,
                                Map<ContentRetriever,String> retrieverToDescription,
                                PromptTemplate promptTemplate)

Method Detail

route

public Collection<ContentRetriever> route(Query query)

Description copied from interface: QueryRouter

Routes the given Query to one or multiple ContentRetrievers.

Specified by:

route in interface QueryRouter

Parameters:

query - The Query to be routed.

Returns:

A collection of one or more ContentRetrievers to which the Query should be routed.

createPrompt

protected Prompt createPrompt(Query query)

parse

protected Collection<ContentRetriever> parse(String choices)