From 11b24ae16afd91cacbb22855f04f5f262bf2a5cb Mon Sep 17 00:00:00 2001 From: magicgone Date: Fri, 28 Feb 2025 18:12:44 +0800 Subject: [PATCH] Add Context Support to Query in RetrievalAugmentationAdvisor Signed-off-by: magicgone --- .../ai/chat/client/advisor/RetrievalAugmentationAdvisor.java | 1 + 1 file changed, 1 insertion(+) diff --git a/spring-ai-core/src/main/java/org/springframework/ai/chat/client/advisor/RetrievalAugmentationAdvisor.java b/spring-ai-core/src/main/java/org/springframework/ai/chat/client/advisor/RetrievalAugmentationAdvisor.java index 686e7727fc4..f2dfbde24aa 100644 --- a/spring-ai-core/src/main/java/org/springframework/ai/chat/client/advisor/RetrievalAugmentationAdvisor.java +++ b/spring-ai-core/src/main/java/org/springframework/ai/chat/client/advisor/RetrievalAugmentationAdvisor.java @@ -106,6 +106,7 @@ public AdvisedRequest before(AdvisedRequest request) { Query originalQuery = Query.builder() .text(new PromptTemplate(request.userText(), request.userParams()).render()) .history(request.messages()) + .context(context) .build(); // 1. Transform original user query based on a chain of query transformers.