feat:隔离用户空间

xservice-service/src/main/java/com/xiang/xservice/ai/agent/BaseAgent.java

@@ -3,5 +3,12 @@ package com.xiang.xservice.ai.agent;
 import com.xiang.xservice.ai.core.enums.ModelTypeEnum;
 
 public interface BaseAgent {
-    void chat(ModelTypeEnum modelType, Long memoryId, String message);
+    /**
+     * chat
+     * @param modelType 模型枚举
+     * @param memoryId
+     * @param userId
+     * @param message
+     */
+    void chat(ModelTypeEnum modelType, Long memoryId, Long userId, String message);
 }

xservice-service/src/main/java/com/xiang/xservice/ai/agent/SimpleChatAgent.java

@@ -32,16 +32,17 @@ public class SimpleChatAgent implements BaseAgent {
 
 
     @Override
-    public void chat(ModelTypeEnum modelType, Long id, String message) {
+    public void chat(ModelTypeEnum modelType, Long memoryId, Long userId, String message) {
 
         DbPersistentStore store = new DbPersistentStore(aiSimpleChatMessageManage);
 
-        ChatMemoryProvider chatMemoryProvider = memoryId -> MessageWindowChatMemory.builder()
+        ChatMemoryProvider chatMemoryProvider = args -> MessageWindowChatMemory.builder()
                 .id(memoryId)
                 .maxMessages(10)
                 .chatMemoryStore(store)
                 .build();
 
+
         OpenAIConfig openAIConfig = openAIBaseConfig.getConfigs().get("bai-lian");
         ModelConfig modelConfig = ModelConfig.builder()
                 .baseUrl(openAIConfig.getBaseUrl())
@@ -60,6 +61,7 @@ public class SimpleChatAgent implements BaseAgent {
                 .build();
         UserMessage userMessage = new UserMessage(message);
         SystemMessage systemMessage = new SystemMessage("你是一个能够与人聊天的AI智能助手，名字叫龙虾");
+        String id = userId + "-" + memoryId;
         assistant.chat(id, Lists.newArrayList(systemMessage, userMessage))
                 .onPartialResponse(System.out::print)
                 .onCompleteResponse(res -> System.out.println("\n完成"))

xservice-service/src/main/java/com/xiang/xservice/ai/core/storage/DbPersistentStore.java

@@ -22,8 +22,12 @@ public class DbPersistentStore implements ChatMemoryStore {
     private final IAiSimpleChatMessageManage aiSimpleChatMessageManage;
 
     @Override
-    public List<ChatMessage> getMessages(Object memoryId) {
-        List<AiSimpleChatMessageDO> msg = aiSimpleChatMessageManage.getMsgByMemoryId((Long) memoryId);
+    public List<ChatMessage> getMessages(Object args) {
+        String argsStr = String.valueOf(args);
+        String[] split = argsStr.split("-");
+        Long userId = Long.valueOf(split[0]);
+        Long memoryId = Long.valueOf(split[1]);
+        List<AiSimpleChatMessageDO> msg = aiSimpleChatMessageManage.getMsgByMemoryId(userId, memoryId);
         if (CollectionUtils.isEmpty(msg)) {
             return Lists.newArrayList();
         }
@@ -52,7 +56,11 @@ public class DbPersistentStore implements ChatMemoryStore {
     }
 
     @Override
-    public void updateMessages(Object memoryId, List<ChatMessage> list) {
+    public void updateMessages(Object args, List<ChatMessage> list) {
+        String argsStr = String.valueOf(args);
+        String[] split = argsStr.split("-");
+        Long userId = Long.valueOf(split[0]);
+        Long memoryId = Long.valueOf(split[1]);
         List<AiSimpleChatMessageDO> result = Lists.newArrayList();
         for (ChatMessage chatMessage : list) {
             StringBuilder str = new StringBuilder();
@@ -77,8 +85,8 @@ public class DbPersistentStore implements ChatMemoryStore {
             }
 
             AiSimpleChatMessageDO message = AiSimpleChatMessageDO.builder()
-                    .userId(1L)
-                    .memoryId((Long) memoryId)
+                    .userId(userId)
+                    .memoryId(memoryId)
                     .message(str.toString())
                     .createTime(LocalDateTime.now())
                     .delFlag(0)
@@ -91,7 +99,11 @@ public class DbPersistentStore implements ChatMemoryStore {
     }
 
     @Override
-    public void deleteMessages(Object memoryId) {
-        aiSimpleChatMessageManage.deleteByMemoryId((Long) memoryId);
+    public void deleteMessages(Object args) {
+        String argsStr = String.valueOf(args);
+        String[] split = argsStr.split("-");
+        Long userId = Long.valueOf(split[0]);
+        Long memoryId = Long.valueOf(split[1]);
+        aiSimpleChatMessageManage.deleteByMemoryId(userId, memoryId);
     }
 }

xservice-service/src/main/java/com/xiang/xservice/ai/repository/manage/AiSimpleChatMessageManageImpl.java

@@ -12,15 +12,16 @@ import java.util.List;
 @Service
 public class AiSimpleChatMessageManageImpl extends ServiceImpl<IAiSimpleChatMessageMapper, AiSimpleChatMessageDO> implements IAiSimpleChatMessageManage {
     @Override
-    public List<AiSimpleChatMessageDO> getMsgByMemoryId(Long memoryId) {
+    public List<AiSimpleChatMessageDO> getMsgByMemoryId(Long userId, Long memoryId) {
         LambdaQueryWrapper<AiSimpleChatMessageDO> lambdaQueryWrapper = Wrappers.lambdaQuery();
+        lambdaQueryWrapper.eq(AiSimpleChatMessageDO::getUserId, userId);
         lambdaQueryWrapper.eq(AiSimpleChatMessageDO::getMemoryId, memoryId);
         lambdaQueryWrapper.eq(AiSimpleChatMessageDO::getDelFlag, 0);
         return baseMapper.selectList(lambdaQueryWrapper);
     }
 
     @Override
-    public boolean deleteByMemoryId(Long memoryId) {
-        return baseMapper.deleteByMemoryId(memoryId) > 0;
+    public boolean deleteByMemoryId(Long userId, Long memoryId) {
+        return baseMapper.deleteByMemoryId(userId, memoryId) > 0;
     }
 }

xservice-service/src/main/java/com/xiang/xservice/ai/repository/manage/IAiSimpleChatMessageManage.java

@@ -6,6 +6,6 @@ import com.xiang.xservice.ai.pojo.entity.AiSimpleChatMessageDO;
 import java.util.List;
 
 public interface IAiSimpleChatMessageManage extends IService<AiSimpleChatMessageDO> {
-    List<AiSimpleChatMessageDO> getMsgByMemoryId(Long memoryId);
-    boolean deleteByMemoryId(Long memoryId);
+    List<AiSimpleChatMessageDO> getMsgByMemoryId(Long userId, Long memoryId);
+    boolean deleteByMemoryId(Long userId, Long memoryId);
 }

xservice-service/src/main/java/com/xiang/xservice/ai/repository/mapper/IAiSimpleChatMessageMapper.java

@@ -9,6 +9,6 @@ import org.springframework.stereotype.Repository;
 @Mapper
 @Repository
 public interface IAiSimpleChatMessageMapper extends BaseMapper<AiSimpleChatMessageDO> {
-    int deleteByMemoryId(@Param("id") Long memoryId);
+    int deleteByMemoryId(@Param("userId") Long userId, @Param("memoryId") Long memoryId);
 
 }

xservice-service/src/main/resources/mapper/AiSimpleChatMessageMapper.xml

@@ -24,7 +24,7 @@
     </sql>
 
     <update id="deleteByMemoryId">
-        update ai_simple_chat_message set del_flag = 1 where memory_id = #{id}
+        update ai_simple_chat_message set del_flag = 1 where memory_id = #{memoryId} and user_id = #{userId}
     </update>
 
 </mapper>