[feature][chat]Support creation of chat memory via REST API.#1603

tencentmusic · Oct 12, 2024 · dd93934 · dd93934
1 parent f31b877
commit dd93934
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/chat/api/src/main/java/com/tencent/supersonic/chat/api/pojo/request/ChatMemoryCreateReq.java b/chat/api/src/main/java/com/tencent/supersonic/chat/api/pojo/request/ChatMemoryCreateReq.java
@@ -1,11 +1,10 @@
 package com.tencent.supersonic.chat.api.pojo.request;
 
 import com.tencent.supersonic.chat.api.pojo.enums.MemoryStatus;
-import com.tencent.supersonic.common.pojo.RecordInfo;
 import lombok.Data;
 
 @Data
-public class ChatMemoryCreateReq extends RecordInfo {
+public class ChatMemoryCreateReq {
 
     private Integer agentId;
 

diff --git a/chat/api/src/main/java/com/tencent/supersonic/chat/api/pojo/request/ChatMemoryUpdateReq.java b/chat/api/src/main/java/com/tencent/supersonic/chat/api/pojo/request/ChatMemoryUpdateReq.java
@@ -4,11 +4,10 @@
 
 import com.tencent.supersonic.chat.api.pojo.enums.MemoryReviewResult;
 import com.tencent.supersonic.chat.api.pojo.enums.MemoryStatus;
-import com.tencent.supersonic.common.pojo.RecordInfo;
 import lombok.Data;
 
 @Data
-public class ChatMemoryUpdateReq extends RecordInfo {
+public class ChatMemoryUpdateReq {
 
     @NotNull(message = "id不可为空")
     private Long id;

diff --git a/...rver/src/main/java/com/tencent/supersonic/chat/server/service/impl/MemoryServiceImpl.java b/...rver/src/main/java/com/tencent/supersonic/chat/server/service/impl/MemoryServiceImpl.java
@@ -20,6 +20,7 @@
 import org.springframework.stereotype.Service;
 import org.springframework.util.CollectionUtils;
 
+import java.util.Date;
 import java.util.List;
 
 @Service
@@ -46,10 +47,11 @@ public void createMemory(ChatMemoryDO memory) {
 
     @Override
     public void updateMemory(ChatMemoryUpdateReq chatMemoryUpdateReq, User user) {
-        chatMemoryUpdateReq.updatedBy(user.getName());
         ChatMemoryDO chatMemoryDO = chatMemoryRepository.getMemory(chatMemoryUpdateReq.getId());
-        boolean hadEnabled = MemoryStatus.ENABLED.equals(chatMemoryDO.getStatus());
+        chatMemoryDO.setUpdatedBy(user.getName());
+        chatMemoryDO.setUpdatedAt(new Date());
         BeanMapper.mapper(chatMemoryUpdateReq, chatMemoryDO);
+        boolean hadEnabled = MemoryStatus.ENABLED.equals(chatMemoryDO.getStatus());
         if (MemoryStatus.ENABLED.equals(chatMemoryUpdateReq.getStatus()) && !hadEnabled) {
             enableMemory(chatMemoryDO);
         } else if (MemoryStatus.DISABLED.equals(chatMemoryUpdateReq.getStatus()) && hadEnabled) {