From 1efa47bc58fe0673a231233f644d3a5f8277e42c Mon Sep 17 00:00:00 2001 From: suerprisePlus <15810472099@163.com> Date: 星期五, 18 十月 2024 14:46:22 +0800 Subject: [PATCH] 接口优化修改 --- src/main/java/com/yb/controller/agentController.java | 20 ++++++++++++-------- 1 files changed, 12 insertions(+), 8 deletions(-) diff --git a/src/main/java/com/yb/controller/agentController.java b/src/main/java/com/yb/controller/agentController.java index 3b81df3..0c152c4 100644 --- a/src/main/java/com/yb/controller/agentController.java +++ b/src/main/java/com/yb/controller/agentController.java @@ -50,10 +50,12 @@ public R getIntention(@RequestParam Map<String, String> allParams) throws Exception { System.out.println("rec getMessage:" + allParams); String msg = allParams.get("message"); + String sessionid = allParams.get("session_id"); + String libid = allParams.get("lib_id"); // 鍒ゆ柇msg鏄惁涓虹┖ if (StringUtils.isEmpty(msg)) { - String info = agentService.getAgentIntention(xzConfig.agentUrl, xzConfig.agentIntent, "''"); + String info = agentService.getAgentIntention(xzConfig.agentUrl, xzConfig.agentIntent, "''",sessionid,libid); JSONObject obj = JSONObject.parseObject(info); String cont = obj.getString("content"); System.out.println("rec isEmpty:" + cont); @@ -65,7 +67,8 @@ HashMap<String, Object> hashMap = new HashMap<String, Object>(); if (isMap) { hashMap.put("type", "Map"); - String funcInfo = agentService.getAgentIntention(xzConfig.agentUrl, xzConfig.agentIntent, msg); + hashMap.put("session_id", allParams.get("session_id")); + String funcInfo = agentService.getAgentIntention(xzConfig.agentUrl, xzConfig.agentIntent, msg,sessionid,libid); JSONObject jsonObject = JSON.parseObject(funcInfo); String orderString = jsonObject.getString("order"); if (orderString == null) { @@ -76,13 +79,14 @@ return R.error(erroContent); } + String report =jsonObject.getString("report"); // 鎰忓浘鍑芥暟鍖归厤 - HashMap<String, Object> mapObj = intentionService.getFunctionMatch(orderString, allParams); + HashMap<String, Object> mapObj = intentionService.getFunctionMatch(orderString, allParams,report); // hashMap.put("place",place) String args = JSONObject.parseObject(orderString).getString("args"); - if(args!= null){ + if (args != null) { String place = JSONObject.parseObject(args).getString("place"); - mapObj.put("place",place); + mapObj.put("place", place); } hashMap.put("content", mapObj); return R.ok(hashMap); @@ -92,7 +96,7 @@ String ragInfo1 = agentService.getAgentRag(xzConfig.agentUrl, xzConfig.agentRag, allParams.get("session_id"), allParams.get("lib_id"), msg); System.out.println("rec rag:" + ragInfo1); if (ragInfo1.equals("")) { - String ragInfo = agentService.getAgentIntention(xzConfig.agentUrl, xzConfig.agentChat, msg); + String ragInfo = agentService.getAgentIntention(xzConfig.agentUrl, xzConfig.agentChat, msg,sessionid,libid); JSONObject jsonObject = JSON.parseObject(ragInfo); String answer = jsonObject.getString("content"); System.out.println("Answer: " + answer); @@ -127,7 +131,7 @@ formData.put("token", xzConfig.token); formData.put("dbid", xzConfig.dbid); String obj = entityHttpUtil.getPostMessage(xzConfig.permissionLayer, formData); - System.out.println("layerList: " + obj); +// System.out.println("layerList: " + obj); HashMap<String, Object> hashMap = new HashMap<String, Object>(); String layers = JSON.parseObject(JSONObject.parseObject(obj).getString("data")).getString("layers"); JSONArray dataArray = JSONArray.parseArray(layers); @@ -160,7 +164,7 @@ } - System.out.println("layerList: " + list); +// System.out.println("layerList: " + list); hashMap.put("data", list); return R.ok(hashMap); } -- Gitblit v1.9.3