topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-10 11:07:06 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

(fix)MemoryReviewTask run null #1312 (#1314)

Browse Source
This commit is contained in:
zhaodongsheng
2024-07-01 17:40:35 +08:00
committed by GitHub
parent d26eeb3367
commit 717a2d26a9
1 changed files with 21 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
chat/server/src/main/java/com/tencent/supersonic/chat/server/memory/MemoryReviewTask.java
View File

@@ -16,6 +16,7 @@ import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
import java.util.Collections; import java.util.Collections;
import java.util.Objects;
import java.util.regex.Matcher; import java.util.regex.Matcher;
import java.util.regex.Pattern; import java.util.regex.Pattern;
@@ -50,19 +51,28 @@ public class MemoryReviewTask {
memoryService.getMemoriesForLlmReview().stream() memoryService.getMemoriesForLlmReview().stream()
.forEach(m -> { .forEach(m -> {
Agent chatAgent = agentService.getAgent(m.getAgentId()); Agent chatAgent = agentService.getAgent(m.getAgentId());
String promptStr = String.format(INSTRUCTION, m.getQuestion(), m.getDbSchema(), m.getS2sql()); if (Objects.nonNull(chatAgent)) {
Prompt prompt = PromptTemplate.from(promptStr).apply(Collections.EMPTY_MAP); String promptStr = String.format(INSTRUCTION, m.getQuestion(), m.getDbSchema(), m.getS2sql());
Prompt prompt = PromptTemplate.from(promptStr).apply(Collections.EMPTY_MAP);
keyPipelineLog.info("MemoryReviewTask reqPrompt:{}", promptStr); keyPipelineLog.info("MemoryReviewTask reqPrompt:{}", promptStr);
ChatLanguageModel chatLanguageModel = ChatLanguageModelProvider.provide(chatAgent.getLlmConfig()); ChatLanguageModel chatLanguageModel = ChatLanguageModelProvider.provide(
String response = chatLanguageModel.generate(prompt.toUserMessage()).content().text(); chatAgent.getLlmConfig());
keyPipelineLog.info("MemoryReviewTask modelResp:{}", response); if (Objects.nonNull(chatLanguageModel)) {
String response = chatLanguageModel.generate(prompt.toUserMessage()).content().text();
keyPipelineLog.info("MemoryReviewTask modelResp:{}", response);
Matcher matcher = OUTPUT_PATTERN.matcher(response); Matcher matcher = OUTPUT_PATTERN.matcher(response);
if (matcher.find()) { if (matcher.find()) {
m.setLlmReviewRet(MemoryReviewResult.valueOf(matcher.group(1))); m.setLlmReviewRet(MemoryReviewResult.valueOf(matcher.group(1)));
m.setLlmReviewCmt(matcher.group(2)); m.setLlmReviewCmt(matcher.group(2));
memoryService.updateMemory(m); memoryService.updateMemory(m);
}
} else {
log.debug("ChatLanguageModel not found for agent:{}", chatAgent.getId());
}
} else {
log.debug("Agent not found for memory:{}", m.getAgentId());
} }
}); });
} }