fix: 初始化 valueMemory 函数

src/main/java/cn/com/poc/agent_application/rest/impl/AgentApplicationInfoRestImpl.java

@@ -16,11 +16,13 @@ import cn.com.poc.support.security.oauth.entity.UserBaseEntity;
 import cn.com.poc.thirdparty.resource.demand.ai.entity.dialogue.Tool;
 import cn.com.poc.thirdparty.resource.demand.ai.function.LargeModelFunctionEnum;
 import cn.com.poc.thirdparty.resource.demand.ai.function.long_memory.SetLongMemoryConstants;
+import cn.com.poc.thirdparty.resource.demand.ai.function.value_memory.GetValueMemory;
 import cn.com.yict.framemax.core.exception.BusinessException;
 import cn.com.yict.framemax.data.model.PagingInfo;
 import cn.hutool.core.collection.ListUtil;
+import com.alibaba.fastjson.JSONObject;
 import org.apache.commons.collections4.CollectionUtils;
-import org.apache.commons.lang3.ArrayUtils;
+import org.apache.commons.collections4.MapUtils;
 import org.apache.commons.lang3.StringUtils;
 import org.springframework.stereotype.Component;
 import org.springframework.util.Assert;
@@ -29,7 +31,10 @@ import javax.annotation.Resource;
 import javax.servlet.http.HttpServletResponse;
 import java.io.IOException;
 import java.io.PrintWriter;
-import java.util.*;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+import java.util.Set;
 import java.util.stream.Collectors;
 
 @Component
@@ -205,11 +210,23 @@ public class AgentApplicationInfoRestImpl implements AgentApplicationInfoRest {
             List<Tool> tools = new ArrayList<>();
             //开启对话变量
             if (CollectionUtils.isNotEmpty(infoEntity.getVariableStructure())) {
-                //todo 修正
                 String functionName = LargeModelFunctionEnum.set_value_memory.name();
-//                String llmConfig = LargeModelFunctionEnum.valueOf(functionName).getFunction().getVariableStructureLLMConfig(infoEntity.getVariableStructure()).get(0);
-//                Tool tool = JsonUtils.deSerialize(llmConfig, Tool.class);
-//                tools.add(tool);
+                String llmConfig = LargeModelFunctionEnum.valueOf(functionName).getFunction().getVariableStructureLLMConfig(infoEntity.getVariableStructure()).get(0);
+                Tool tool = JsonUtils.deSerialize(llmConfig, Tool.class);
+                tools.add(tool);
+
+                //初始化变量函数
+                Map<Object, Object> map = GetValueMemory.get(agentId);
+                if (MapUtils.isEmpty(map)) {
+                    List<Variable> variableStructure = infoEntity.getVariableStructure();
+                    for (Variable variable : variableStructure) {
+                        String key = variable.getKey();
+                        String variableDefault = variable.getVariableDefault();
+                        JSONObject jsonObject = new JSONObject();
+                        jsonObject.put(key, variableDefault);
+                        LargeModelFunctionEnum.valueOf(functionName).getFunction().doFunction(jsonObject.toJSONString(), agentId);
+                    }
+                }
             }
             //开启长期记忆
             if (CommonConstant.YOrN.Y.equals(infoEntity.getIsLongMemory())) {

src/main/java/cn/com/poc/expose/aggregate/impl/AgentApplicationServiceImpl.java

@@ -6,6 +6,7 @@ import cn.com.poc.agent_application.constant.AgentApplicationGCConfigConstants;
 import cn.com.poc.agent_application.entity.BizAgentApplicationDialoguesRecordEntity;
 import cn.com.poc.agent_application.entity.BizAgentApplicationGcConfigEntity;
 import cn.com.poc.agent_application.entity.BizAgentApplicationPublishEntity;
+import cn.com.poc.agent_application.entity.Variable;
 import cn.com.poc.agent_application.query.MemberCollectQueryCondition;
 import cn.com.poc.agent_application.query.MemberCollectQueryItem;
 import cn.com.poc.agent_application.service.BizAgentApplicationDialoguesRecordService;
@@ -26,11 +27,14 @@ import cn.com.poc.thirdparty.resource.demand.ai.constants.LLMRoleEnum;
 import cn.com.poc.thirdparty.resource.demand.ai.entity.largemodel.LargeModelDemandResult;
 import cn.com.poc.thirdparty.resource.demand.ai.entity.largemodel.LargeModelResponse;
 import cn.com.poc.thirdparty.resource.demand.ai.function.LargeModelFunctionEnum;
+import cn.com.poc.thirdparty.resource.demand.ai.function.value_memory.GetValueMemory;
 import cn.com.poc.thirdparty.service.LLMService;
 import cn.com.yict.framemax.core.exception.BusinessException;
 import cn.com.yict.framemax.data.model.PagingInfo;
+import com.alibaba.fastjson.JSONObject;
 import com.fasterxml.jackson.core.type.TypeReference;
 import org.apache.commons.collections4.CollectionUtils;
+import org.apache.commons.collections4.MapUtils;
 import org.apache.commons.lang3.ArrayUtils;
 import org.apache.commons.lang3.StringUtils;
 import org.slf4j.Logger;
@@ -40,10 +44,7 @@ import org.springframework.stereotype.Service;
 import javax.annotation.Resource;
 import javax.servlet.http.HttpServletResponse;
 import java.security.SecureRandom;
-import java.util.ArrayList;
-import java.util.HashSet;
-import java.util.List;
-import java.util.Set;
+import java.util.*;
 
 @Service
 public class AgentApplicationServiceImpl implements AgentApplicationService {
@@ -77,7 +78,6 @@ public class AgentApplicationServiceImpl implements AgentApplicationService {
     @Resource
     private RedisService redisService;
 
-    //todo 优化记录保存，对话保持链接/重新链接推送消息
     @Override
     public void callAgentApplication(String agentId, String dialogsId, String input, HttpServletResponse httpServletResponse) {
 
@@ -106,7 +106,7 @@ public class AgentApplicationServiceImpl implements AgentApplicationService {
             List<Message> messages = buildMessages(dialogsId, agentId, userBaseEntity.getUserId(), input);
 
             //配置对话function
-            List<Tool> tools = buildMemoryConfig(infoEntity);
+            List<Tool> tools = buildMemoryConfig(infoEntity, dialogsId);
 
             // 保存用户输入记录
             Long inputTimestamp = System.currentTimeMillis();
@@ -307,14 +307,27 @@ public class AgentApplicationServiceImpl implements AgentApplicationService {
         bizAgentApplicationDialoguesRecordService.save(outputRecord);
     }
 
-    private List<Tool> buildMemoryConfig(BizAgentApplicationPublishEntity infoEntity) {
+    private List<Tool> buildMemoryConfig(BizAgentApplicationPublishEntity infoEntity, String identifier) {
         List<Tool> tools = new ArrayList<>();
         //开启对话变量
         if (CollectionUtils.isNotEmpty(infoEntity.getVariableStructure())) {
-//            String functionName = LargeModelFunctionEnum.set_value_memory.name();
-//            String llmConfig = LargeModelFunctionEnum.valueOf(functionName).getFunction().getVariableStructureLLMConfig(infoEntity.getVariableStructure()).get(0);
-//            Tool tool = JsonUtils.deSerialize(llmConfig, Tool.class);
-//            tools.add(tool);
+            String functionName = LargeModelFunctionEnum.set_value_memory.name();
+            String llmConfig = LargeModelFunctionEnum.valueOf(functionName).getFunction().getVariableStructureLLMConfig(infoEntity.getVariableStructure()).get(0);
+            Tool tool = JsonUtils.deSerialize(llmConfig, Tool.class);
+            tools.add(tool);
+
+            //初始化变量函数
+            Map<Object, Object> map = GetValueMemory.get(identifier);
+            if (MapUtils.isEmpty(map)) {
+                List<Variable> variableStructure = infoEntity.getVariableStructure();
+                for (Variable variable : variableStructure) {
+                    String key = variable.getKey();
+                    String variableDefault = variable.getVariableDefault();
+                    JSONObject jsonObject = new JSONObject();
+                    jsonObject.put(key, variableDefault);
+                    LargeModelFunctionEnum.valueOf(functionName).getFunction().doFunction(jsonObject.toJSONString(), identifier);
+                }
+            }
         }
         //开启长期记忆
         if (CommonConstant.YOrN.Y.equals(infoEntity.getIsLongMemory())) {