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

[improvement][chat] Optimize the MemoryReviewTask task by adding individual exception handling (#1788)

Browse Source
This commit is contained in:
lexluo09
2024-10-11 23:09:13 +08:00
committed by GitHub
parent 0a86a63937
commit 72d01bb419
4 changed files with 13 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -49,11 +49,14 @@ public class MemoryReviewTask {
@Scheduled(fixedDelay = 60 * 1000) @Scheduled(fixedDelay = 60 * 1000)
public void review() { public void review() {
memoryService.getMemoriesForLlmReview().stream().forEach(memory -> {
try { try {
memoryService.getMemoriesForLlmReview().stream().forEach(this::processMemory); processMemory(memory);
} catch (Exception e) { } catch (Exception e) {
log.error("Exception occurred during memory review task", e); log.error("Exception occurred while processing memory with id {}: {}",
memory.getId(), e.getMessage(), e);
} }
});
} }
private void processMemory(ChatMemoryDO m) { private void processMemory(ChatMemoryDO m) {

3
chat/server/src/main/java/com/tencent/supersonic/chat/server/service/impl/AgentServiceImpl.java
View File

@@ -105,8 +105,7 @@ public class AgentServiceImpl extends ServiceImpl<AgentDOMapper, AgentDO> implem
} }
private synchronized void doExecuteAgentExamples(Agent agent) { private synchronized void doExecuteAgentExamples(Agent agent) {
if (!agent.containsDatasetTool() if (!agent.containsDatasetTool() || !agent.enableMemoryReview()
|| !agent.enableMemoryReview()
|| !ModelConfigHelper.testConnection( || !ModelConfigHelper.testConnection(
ModelConfigHelper.getChatModelConfig(agent, ChatModelType.TEXT_TO_SQL)) ModelConfigHelper.getChatModelConfig(agent, ChatModelType.TEXT_TO_SQL))
|| CollectionUtils.isEmpty(agent.getExamples())) { || CollectionUtils.isEmpty(agent.getExamples())) {

3
headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/ModelSchema.java
View File

@@ -1,8 +1,9 @@
package com.tencent.supersonic.headless.api.pojo; package com.tencent.supersonic.headless.api.pojo;
import java.util.List;
import lombok.Data; import lombok.Data;
import java.util.List;
@Data @Data
public class ModelSchema { public class ModelSchema {

3
headless/chat/src/main/java/com/tencent/supersonic/headless/chat/corrector/LLMSqlCorrector.java
View File

@@ -56,7 +56,8 @@ public class LLMSqlCorrector extends BaseSemanticCorrector {
return; return;
} }
ChatLanguageModel chatLanguageModel = ModelProvider.getChatModel(chatQueryContext.getModelConfig()); ChatLanguageModel chatLanguageModel =
ModelProvider.getChatModel(chatQueryContext.getModelConfig());
SemanticSqlExtractor extractor = SemanticSqlExtractor extractor =
AiServices.create(SemanticSqlExtractor.class, chatLanguageModel); AiServices.create(SemanticSqlExtractor.class, chatLanguageModel);
Prompt prompt = generatePrompt(chatQueryContext.getQueryText(), semanticParseInfo); Prompt prompt = generatePrompt(chatQueryContext.getQueryText(), semanticParseInfo);