添加向量库

qianbao
1 parent 9cc28448
Showing 3 changed files with 136 additions and 41 deletions
src/main/java/com/xly/service/XlyErpService.java
src/main/java/com/xly/tts/service/PythonTtsProxyService.java
src/main/java/com/xly/web/TTSStreamController.java
@@ -37,6 +37,7 @@ import dev.langchain4j.service.MemoryId;
 import dev.langchain4j.service.V;
 import io.milvus.v2.common.DataType;
 import io.milvus.v2.service.collection.request.CreateCollectionReq;
+import jnr.ffi.annotations.In;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.time.DateFormatUtils;
@@ -227,6 +228,25 @@ public class XlyErpService {
         }
     }
+    /****
+     * @Author 钱豹
+     * @Date 8:54 2026/4/7
+     * @Param [userInput, userId, sUserName, sBrandsId, sSubsidiaryId, sUserType, authorization]
+     * @return com.xly.entity.AiResponseDTO
+     * @Description 换一换
+     **/
+    public AiResponseDTO change(String  userId ,
+                                      String sUserName ,
+                                      String sBrandsId ,
+                                      String sSubsidiaryId,
+                                      String sUserType,
+                                      String authorization,
+                                      Integer iPage) {
+        // 1. 初始化用户场景会话（权限内场景）
+        UserSceneSession session = userSceneSessionService.getUserSceneSession(userId,sUserName,sBrandsId,sSubsidiaryId,sUserType,authorization);
+        return getChange(session, iPage);
+    }
+
     /***
      * @Author 钱豹
      * @Date 19:18 2026/1/27
@@ -270,7 +290,7 @@ public class XlyErpService {
             // 3. 未选场景：先展示场景选择界面，处理用户序号选择
             if (!session.isSceneSelected() && ValiDataUtil.me().isPureNumber(input)){
                 // 3.1 尝试处理场景选择（输入序号则匹配，否则展示选择提示）
-                return handleSceneSelect(userId, input, session);
+                return handleSceneSelect(userId, input, session,1);
             }
             // 4. 构建Agent，执行业务交互,如果返回为null,说明大模型没有判段出场景，必判断出后才能继续
             ErpAiAgent aiAgent = createErpAiAgent(userId, input, session);
@@ -942,7 +962,7 @@ public class XlyErpService {
     /**
      * 处理用户场景选择：输入序号→匹配场景→更新会话状态
      */
-    private AiResponseDTO handleSceneSelect(String userId, String userInput, UserSceneSession session) {
+    private AiResponseDTO handleSceneSelect(String userId, String userInput, UserSceneSession session,Integer page) {
         // 1. 尝试根据序号匹配场景
         boolean selectSuccess = session.selectSceneByInput(userInput);
         String sceneName = StrUtil.EMPTY;
@@ -954,20 +974,34 @@ public class XlyErpService {
             UserSceneSessionService.ERP_AGENT_CACHE.remove(userId);
             //清除记忆缓存
             operableChatMemoryProvider.clearSpecifiedMemory(userId);
-            StringBuffer aiText = new StringBuffer().append(" <div style='line-height:1.8;width:100%;'>")
-                    .append("<div style=\"color: #333;\">")
-                    .append("智能体选择成功！ 现在可以问她相关问题(如" + String.join("、", session.getCurrentScene().getSSceneContext()) + "）")
-                    .append("</div>");
-            //插入用户常用问题
-            aiText.append(getSelectAgent(session,1));
-            sceneName = ObjectUtil.isNotEmpty(session.getCurrentScene())?session.getCurrentScene().getSSceneName():StrUtil.EMPTY;
-            methodName = ObjectUtil.isNotEmpty(session.getCurrentTool())?session.getCurrentTool().getSControlName():StrUtil.EMPTY;
-            return AiResponseDTO.builder().sSceneName(sceneName).sMethodName(methodName).aiText(aiText.toString()).sSceneName(session.getCurrentScene().getSSceneName()).build();
+            return getChange( session, page);
         } else {
             // 3. 选择失败：重新展示场景选择提示
             return AiResponseDTO.builder().sSceneName(sceneName).sMethodName(methodName).aiText(session.buildSceneSelectHint()).build();
         }
     }
+
+    /***
+     * @Author 钱豹
+     * @Date 9:00 2026/4/7
+     * @Param [session, page]
+     * @return com.xly.entity.AiResponseDTO
+     * @Description 用户输入数字换一换
+     **/
+    private AiResponseDTO getChange(UserSceneSession session,Integer page){
+        StringBuffer aiText = new StringBuffer().append(" <div style='line-height:1.8;width:100%;'>")
+                .append("<div style=\"color: #333;\">")
+                .append("智能体选择成功！ 现在可以问她相关问题(如" + String.join("、", session.getCurrentScene().getSSceneContext()) + "）")
+                .append("</div>");
+        //插入用户常用问题
+        aiText.append(getSelectAgent(session,page));
+        String sceneName = ObjectUtil.isNotEmpty(session.getCurrentScene())?session.getCurrentScene().getSSceneName():StrUtil.EMPTY;
+        String methodName = ObjectUtil.isNotEmpty(session.getCurrentTool())?session.getCurrentTool().getSControlName():StrUtil.EMPTY;
+        return AiResponseDTO.builder().sSceneName(sceneName).sMethodName(methodName).aiText(aiText.toString()).sSceneName(session.getCurrentScene().getSSceneName()).build();
+    }
+
+
+
     /***
      * @Author 钱豹
      * @Date 11:45 2026/3/13
@@ -978,18 +1012,24 @@ public class XlyErpService {
     private String getSelectAgent(UserSceneSession session,Integer page){
         List<ToolMeta> toolMetaAll = session.getAuthTool();
         String sSceneId = session.getCurrentScene().getSId();
+        String sToolId = session.getCurrentTool().getSId();
         toolMetaAll = toolMetaAll.stream().filter(to-> to.getSSceneId().equals(session.getCurrentScene().getSId())).collect(Collectors.toUnmodifiableList());
         StringBuffer sb = new StringBuffer();
-        List<List<ToolMeta>> firstFive = ListUtil.split(toolMetaAll,5);
-        List<ToolMeta> showList;
-        if(ObjectUtil.isNotEmpty(firstFive.get(page-1))){
-            showList = firstFive.get(page-1);
-            page = page + 1;
-        }else{
-            showList = firstFive.get(0);
-            page = 1;
+        //获取用户最近五次问题
+        List<Map<String,Object>> data = getAiUserAgentQuestion(session.getUserId(),sSceneId, sToolId, page,3);
+        Integer iPageCount = 0;
+        if(page==1){
+            iPageCount =getAiUserAgentQuestionCount(session.getUserId(),  sSceneId, sToolId);
+        }
+        List<ToolMeta> showListAll = new ArrayList<>();
+        if(ObjectUtil.isNotEmpty(data)){
+            data.forEach(one->{
+                ToolMeta tm = new ToolMeta();
+                tm.setSMethodName(one.get("sUserInput").toString());
+            });
         }
-        showList.forEach(one->{
+        showListAll.addAll(toolMetaAll);
+        showListAll.forEach(one->{
             sb.append("<div style=\"color: #4096ff; margin-top: 5px;display:flex;align-items:center;font-size:12px;\" data-action=\"reset\" data-text=\"")
                     .append(one.getSMethodName()).append("\" onclick=\"reset(").append(one.getSMethodName()).append("\">")
                     .append("<span style=\"display:inline-block; width:5px; height:5px; background-color:#ccc; border-radius:50%; margin-right:5px;\"></span>")
@@ -997,11 +1037,53 @@ public class XlyErpService {
                     .append(" </div>");
         });
         sb.append("</div>");
-        sb.append(" <div style=\"color: #4096ff;  margin-top: 5px;width:100%;text-align:right;margin-right:10px;font-size:12px;\" data-action=\"resetTag\" data-text=\"").append(sSceneId).append(",").append(page).append("\" onclick=\"reset(换一换)\">");
+        sb.append(" <div style=\"color: #4096ff;  margin-top: 5px;width:100%;text-align:right;margin-right:10px;font-size:12px;\" data-action=\"resetTag\" data-text=\"").append(sSceneId).append(",").append(sToolId).append(",").append(page+1).append(",").append(iPageCount).append("\" onclick=\"reset(换一换)\">");
         sb.append(" 换一换").append("</div>");
         return sb.toString();
     }
+    /***
+     * @Author 钱豹
+     * @Date 10:02 2026/4/7
+     * @Param [sLoginId, sSceneId, sMethodId, iPageNum, iPageSize]
+     * @return java.util.List<java.util.Map<java.lang.String,java.lang.Object>>
+     * @Description 换一换获取最近数据
+     **/
+    private List<Map<String,Object>> getAiUserAgentQuestion(String sLoginId, String sSceneId, String sMethodId, Integer iPageNum,Integer iPageSize){
+        StringBuffer sb = new StringBuffer().append("SELECT sId,iUpdate,sUserInput FROM ai_user_agent_question ")
+                .append("WHERE sLoginId = #{sLoginId} ")
+                .append("AND sSceneId = #{sSceneId} ")
+                .append("AND sMethodId = #{sMethodId}  ")
+                .append("AND IFNULL(sUserInput,'') <> '' ")
+                .append("ORDER BY iUpdate DESC,tUpdateDate DESC  ")
+                .append("LIMIT #{iPageNum},#{iPageSize} ");
+        Map<String,Object> serMap = new HashMap<>();
+        serMap.put("sLoginId",sLoginId);
+        serMap.put("sSceneId",sSceneId);
+        serMap.put("sMethodId",sMethodId);
+        serMap.put("iPageNum",iPageNum);
+        serMap.put("iPageSize",iPageSize);
+        return dynamicExeDbService.findSql(serMap,sb.toString());
+    }
+
+    private Integer getAiUserAgentQuestionCount(String sLoginId, String sSceneId, String sMethodId){
+        StringBuffer sb = new StringBuffer().append("SELECT COUNT(1) AS iCount FROM ai_user_agent_question ")
+                .append("WHERE sLoginId = #{sLoginId} ")
+                .append("AND sSceneId = #{sSceneId} ")
+                .append("AND sMethodId = #{sMethodId}  ")
+                .append("AND IFNULL(sUserInput,'') <> '' ");
+        Map<String,Object> serMap = new HashMap<>();
+        serMap.put("sLoginId",sLoginId);
+        serMap.put("sSceneId",sSceneId);
+        serMap.put("sMethodId",sMethodId);
+        List<Map<String,Object>>  data = dynamicExeDbService.findSql(serMap,sb.toString());
+        if(ObjectUtil.isNotEmpty(data)){
+            return (Integer) data.get(0).get("iCount");
+        }
+        return 0;
+    }
+
+
     /**
      * @Author 钱豹
      * @Date 13:32 2026/2/6
@@ -1056,25 +1138,5 @@ public class XlyErpService {
         String sChatMessage = chatiAgent.chat(session.getUserId(), input);
         return AiResponseDTO.builder().sSceneName(sceneName).sMethodName(methodName).aiText(sChatMessage).systemText(StrUtil.EMPTY).sReturnType(ReturnTypeCode.HTML.getCode()).build();
     }
-    /***
-     * @Author 钱豹
-     * @Date 12:14 2026/3/8
-     * @Param [chatMessage, memoryId]
-     * @return void
-     * @Description 随便聊聊移除最后一个AI 返回信息 跟系统询问的随便聊聊
-     **/
-    private void removeMssageSbll(List<ChatMessage> chatMessage,String memoryId){
-        if(chatMessage!=null){
-//            operableChatMemoryProvider.deleteSingleMessage(memoryId,chatMessage.get(chatMessage.size()-1));
-            for(int i=chatMessage.size()-1;i>0;i--){
-                ChatMessage data = chatMessage.get(i);
-                ChatMessageType sType  = data.type();
-                if(ChatMessageType.SYSTEM.equals(sType)){
-                    operableChatMemoryProvider.deleteSingleMessage(memoryId,data);
-                    return;
-                }
-            }
-        }
-    }
 }
 \ No newline at end of file
@@ -68,6 +68,26 @@ public class PythonTtsProxyService {
     }
     /**
+     * @Author 钱豹
+     * @Date 8:54 2026/4/7
+     * @Param [request]
+     * @return org.springframework.http.ResponseEntity<com.xly.tts.bean.TTSResponseDTO>
+     * @Description 换一换
+     **/
+    public ResponseEntity<TTSResponseDTO> change(TTSRequestDTO request) {
+        String userInput = request.getText();
+        String sUserId = request.getUserid();
+        String sUserName = request.getUsername();
+        String sBrandsId = request.getBrandsid();
+        String sSubsidiaryId = request.getSubsidiaryid();
+        String sUserType = request.getUsertype();
+        String authorization = request.getAuthorization();
+        Integer iPage = Integer.valueOf(userInput);
+        AiResponseDTO voiceText = xlyErpService.change(sUserId, sUserName , sBrandsId , sSubsidiaryId, sUserType, authorization,iPage);
+        return synthesizeStreamAi(request, voiceText);
+    }
+
+    /**
      * 【保持原有返回类型】AI对话 + 流式TTS
      */
     public ResponseEntity<TTSResponseDTO> synthesizeStreamAi(TTSRequestDTO request) {
@@ -77,6 +77,19 @@ public class TTSStreamController {
         pythonTtsProxyService.shutdown();
     }
+    /***
+     * @Author 钱豹
+     * @Date 8:53 2026/4/7
+     * @Param [request]
+     * @return org.springframework.http.ResponseEntity<com.xly.tts.bean.TTSResponseDTO>
+     * @Description 换一换
+     **/
+    @PostMapping(value = "/stream/change", consumes = {MediaType.APPLICATION_JSON_VALUE, MediaType.ALL_VALUE})
+    public ResponseEntity<TTSResponseDTO> change(@RequestBody TTSRequestDTO request) {
+        return pythonTtsProxyService.change(request);
+    }
+
+
     /**
      * 流式合成语音（代理到Python服务）
      */