diff --git a/backend/src/main/java/com/zl/mjga/controller/AiController.java b/backend/src/main/java/com/zl/mjga/controller/AiController.java index 9095661..3bf8fd3 100644 --- a/backend/src/main/java/com/zl/mjga/controller/AiController.java +++ b/backend/src/main/java/com/zl/mjga/controller/AiController.java @@ -133,4 +133,9 @@ public class AiController { } departmentRepository.deleteByName(name); } + + @PostMapping("/chat/refresh") + void createNewConversation(Principal principal) { + aiChatService.evictChatMemory(principal.getName()); + } } diff --git a/backend/src/main/java/com/zl/mjga/service/AiChatService.java b/backend/src/main/java/com/zl/mjga/service/AiChatService.java index dfa939c..9ea1a5c 100644 --- a/backend/src/main/java/com/zl/mjga/service/AiChatService.java +++ b/backend/src/main/java/com/zl/mjga/service/AiChatService.java @@ -53,4 +53,11 @@ public class AiChatService { AiLlmConfig aiLlmConfig = precedenceLlmBy.orElseThrow(() -> new BusinessException("没有开启的大模型")); return aiLlmConfig.getCode(); } + + public void evictChatMemory(String sessionIdentifier) { + deepSeekChatAssistant.evictChatMemory(sessionIdentifier); + zhiPuChatAssistant.evictChatMemory(sessionIdentifier); + zhiPuToolAssistant.evictChatMemory(sessionIdentifier); + deepSeekToolAssistant.evictChatMemory(sessionIdentifier); + } }