Skip to content

Commit 13fc058

Browse files
fix memory data issue
Signed-off-by: Jast <[email protected]>
1 parent 3756e16 commit 13fc058

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

spring-ai-client-chat/src/main/java/org/springframework/ai/chat/client/advisor/MessageChatMemoryAdvisor.java

+8-1
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,8 @@
1919
import java.util.ArrayList;
2020
import java.util.List;
2121

22+
import org.springframework.ai.chat.prompt.PromptTemplate;
23+
import org.springframework.util.CollectionUtils;
2224
import reactor.core.publisher.Flux;
2325

2426
import org.springframework.ai.chat.client.advisor.api.AdvisedRequest;
@@ -94,7 +96,12 @@ private AdvisedRequest before(AdvisedRequest request) {
9496
AdvisedRequest advisedRequest = AdvisedRequest.from(request).messages(advisedMessages).build();
9597

9698
// 4. Add the new user input to the conversation memory.
97-
UserMessage userMessage = new UserMessage(request.userText(), request.media());
99+
String processedUserText = request.userText();
100+
if (!CollectionUtils.isEmpty(request.userParams())) {
101+
processedUserText = new PromptTemplate(processedUserText, request.userParams()).render();
102+
}
103+
UserMessage userMessage = new UserMessage(processedUserText, request.media());
104+
98105
this.getChatMemoryStore().add(this.doGetConversationId(request.adviseContext()), userMessage);
99106

100107
return advisedRequest;

0 commit comments

Comments
 (0)