Skip to content

Commit a07906a

Browse files
author
ogbozoyan
committed
feat: added missed builder methods; deprecated all constructors except one with all fields
1 parent 928ec41 commit a07906a

File tree

1 file changed

+32
-32
lines changed

1 file changed

+32
-32
lines changed

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

Lines changed: 32 additions & 32 deletions
Original file line numberDiff line numberDiff line change
@@ -78,18 +78,18 @@ public VectorStoreChatMemoryAdvisor(VectorStore vectorStore, String systemTextAd
7878

7979
@Deprecated
8080
public VectorStoreChatMemoryAdvisor(VectorStore vectorStore, String defaultConversationId,
81-
int chatHistoryWindowSize) {
81+
int chatHistoryWindowSize) {
8282
this(vectorStore, defaultConversationId, chatHistoryWindowSize, DEFAULT_SYSTEM_TEXT_ADVISE);
8383
}
8484

8585
@Deprecated
8686
public VectorStoreChatMemoryAdvisor(VectorStore vectorStore, String defaultConversationId,
87-
int chatHistoryWindowSize, int order) {
87+
int chatHistoryWindowSize, int order) {
8888
this(vectorStore, defaultConversationId, chatHistoryWindowSize, DEFAULT_SYSTEM_TEXT_ADVISE, order);
8989
}
9090

9191
public VectorStoreChatMemoryAdvisor(VectorStore vectorStore, String defaultConversationId,
92-
int chatHistoryWindowSize, String systemTextAdvise) {
92+
int chatHistoryWindowSize, String systemTextAdvise) {
9393
this(vectorStore, defaultConversationId, chatHistoryWindowSize, systemTextAdvise,
9494
Advisor.DEFAULT_CHAT_MEMORY_PRECEDENCE_ORDER);
9595
}
@@ -105,7 +105,7 @@ public VectorStoreChatMemoryAdvisor(VectorStore vectorStore, String defaultConve
105105
* @param order the order of precedence for this advisor in the chain.
106106
*/
107107
public VectorStoreChatMemoryAdvisor(VectorStore vectorStore, String defaultConversationId,
108-
int chatHistoryWindowSize, String systemTextAdvise, int order) {
108+
int chatHistoryWindowSize, String systemTextAdvise, int order) {
109109
super(vectorStore, defaultConversationId, chatHistoryWindowSize, true, order);
110110
this.systemTextAdvise = systemTextAdvise;
111111
}
@@ -148,60 +148,60 @@ private AdvisedRequest before(AdvisedRequest request) {
148148
}
149149

150150
var searchRequest = SearchRequest.query(request.userText())
151-
.withTopK(this.doGetChatMemoryRetrieveSize(request.adviseContext()))
152-
.withFilterExpression(DOCUMENT_METADATA_CONVERSATION_ID + "=='"
153-
+ this.doGetConversationId(request.adviseContext()) + "'");
151+
.withTopK(this.doGetChatMemoryRetrieveSize(request.adviseContext()))
152+
.withFilterExpression(DOCUMENT_METADATA_CONVERSATION_ID + "=='"
153+
+ this.doGetConversationId(request.adviseContext()) + "'");
154154

155155
List<Document> documents = this.getChatMemoryStore().similaritySearch(searchRequest);
156156

157157
String longTermMemory = documents.stream()
158-
.map(Content::getContent)
159-
.collect(Collectors.joining(System.lineSeparator()));
158+
.map(Content::getContent)
159+
.collect(Collectors.joining(System.lineSeparator()));
160160

161161
Map<String, Object> advisedSystemParams = new HashMap<>(request.systemParams());
162162
advisedSystemParams.put("long_term_memory", longTermMemory);
163163

164164
AdvisedRequest advisedRequest = AdvisedRequest.from(request)
165-
.withSystemText(advisedSystemText)
166-
.withSystemParams(advisedSystemParams)
167-
.build();
165+
.withSystemText(advisedSystemText)
166+
.withSystemParams(advisedSystemParams)
167+
.build();
168168

169169
UserMessage userMessage = new UserMessage(request.userText(), request.media());
170170
this.getChatMemoryStore()
171-
.write(toDocuments(List.of(userMessage), this.doGetConversationId(request.adviseContext())));
171+
.write(toDocuments(List.of(userMessage), this.doGetConversationId(request.adviseContext())));
172172

173173
return advisedRequest;
174174
}
175175

176176
private void observeAfter(AdvisedResponse advisedResponse) {
177177

178178
List<Message> assistantMessages = advisedResponse.response()
179-
.getResults()
180-
.stream()
181-
.map(g -> (Message) g.getOutput())
182-
.toList();
179+
.getResults()
180+
.stream()
181+
.map(g -> (Message) g.getOutput())
182+
.toList();
183183

184184
this.getChatMemoryStore()
185-
.write(toDocuments(assistantMessages, this.doGetConversationId(advisedResponse.adviseContext())));
185+
.write(toDocuments(assistantMessages, this.doGetConversationId(advisedResponse.adviseContext())));
186186
}
187187

188188
private List<Document> toDocuments(List<Message> messages, String conversationId) {
189189

190190
List<Document> docs = messages.stream()
191-
.filter(m -> m.getMessageType() == MessageType.USER || m.getMessageType() == MessageType.ASSISTANT)
192-
.map(message -> {
193-
var metadata = new HashMap<>(message.getMetadata() != null ? message.getMetadata() : new HashMap<>());
194-
metadata.put(DOCUMENT_METADATA_CONVERSATION_ID, conversationId);
195-
metadata.put(DOCUMENT_METADATA_MESSAGE_TYPE, message.getMessageType().name());
196-
if (message instanceof UserMessage userMessage) {
197-
return new Document(userMessage.getContent(), userMessage.getMedia(), metadata);
198-
}
199-
else if (message instanceof AssistantMessage assistantMessage) {
200-
return new Document(assistantMessage.getContent(), metadata);
201-
}
202-
throw new RuntimeException("Unknown message type: " + message.getMessageType());
203-
})
204-
.toList();
191+
.filter(m -> m.getMessageType() == MessageType.USER || m.getMessageType() == MessageType.ASSISTANT)
192+
.map(message -> {
193+
var metadata = new HashMap<>(message.getMetadata() != null ? message.getMetadata() : new HashMap<>());
194+
metadata.put(DOCUMENT_METADATA_CONVERSATION_ID, conversationId);
195+
metadata.put(DOCUMENT_METADATA_MESSAGE_TYPE, message.getMessageType().name());
196+
if (message instanceof UserMessage userMessage) {
197+
return new Document(userMessage.getContent(), userMessage.getMedia(), metadata);
198+
}
199+
else if (message instanceof AssistantMessage assistantMessage) {
200+
return new Document(assistantMessage.getContent(), metadata);
201+
}
202+
throw new RuntimeException("Unknown message type: " + message.getMessageType());
203+
})
204+
.toList();
205205

206206
return docs;
207207
}

0 commit comments

Comments
 (0)