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.