From c64f2eb0f6b7268152b2ea41f4e9f27770c0f933 Mon Sep 17 00:00:00 2001 From: Chuck1sn Date: Sat, 14 Jun 2025 14:04:58 +0800 Subject: [PATCH] add clear api --- .../src/main/java/com/zl/mjga/controller/AiController.java | 5 +++++ .../src/main/java/com/zl/mjga/service/AiChatService.java | 7 +++++++ 2 files changed, 12 insertions(+) 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); + } }