From 53ae0215de46b9bb68d9aa5d5a3e83dde7cb3986 Mon Sep 17 00:00:00 2001
From: suerprisePlus <15810472099@163.com>
Date: 星期五, 27 九月 2024 15:20:11 +0800
Subject: [PATCH] 接口优化

---
 src/main/java/com/yb/controller/agentController.java |   43 ++++++++++++++++++++++++-------------------
 1 files changed, 24 insertions(+), 19 deletions(-)

diff --git a/src/main/java/com/yb/controller/agentController.java b/src/main/java/com/yb/controller/agentController.java
index 47eb4a5..3b81df3 100644
--- a/src/main/java/com/yb/controller/agentController.java
+++ b/src/main/java/com/yb/controller/agentController.java
@@ -78,6 +78,12 @@
             }
 //          鎰忓浘鍑芥暟鍖归厤
             HashMap<String, Object> mapObj = intentionService.getFunctionMatch(orderString, allParams);
+//            hashMap.put("place",place)
+            String args = JSONObject.parseObject(orderString).getString("args");
+            if(args!= null){
+                String place = JSONObject.parseObject(args).getString("place");
+                mapObj.put("place",place);
+            }
             hashMap.put("content", mapObj);
             return R.ok(hashMap);
 
@@ -99,18 +105,9 @@
                 System.out.println("Answer: " + answer1);
                 String jso = JSONObject.parseObject(answer1).getString("answer");
                 String context = JSONObject.parseObject(answer1).getString("context");
-//                JSONArray jsonArray = JSON.parseArray(context);
-//                String metadata = "";
-//                for (int i = 0; i < jsonArray.size(); i++) {
-//                    JSONObject jsonObject = jsonArray.getJSONObject(i);
-//
-//                    metadata = jsonObject.getString("metadata");
-//
-//
-//                }
+
                 if (context != "") {
-//                    String[] fileSource = metadata.split("/");
-                    hashMap.put("fileSrource",  JSON.parseArray(context));
+                    hashMap.put("fileSrource", JSON.parseArray(context));
                 }
                 hashMap.put("content", jso);
                 hashMap.put("type", "Rag");
@@ -130,7 +127,7 @@
         formData.put("token", xzConfig.token);
         formData.put("dbid", xzConfig.dbid);
         String obj = entityHttpUtil.getPostMessage(xzConfig.permissionLayer, formData);
-        System.out.println("Answer: " + 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);
@@ -138,24 +135,32 @@
         for (int i = 0; i < dataArray.size(); i++) {
             JSONObject itmObj = JSONObject.parseObject(dataArray.get(i).toString());
             String item = itmObj.getString("dataformat");
-//            System.out.println("Answer: " + item);
+
             String id = itmObj.getString("id");
             if (item.equals("entity")) {
                 HashMap<String, Object> itmHash = new HashMap<>();
                 String name = itmObj.getString("name");
                 String url = itmObj.getString("lod_url");
                 if (url != null) {
-                    String l1 = JSONObject.parseObject(JSONObject.parseObject(url).getString("3d")).getString("l0");
-                    itmHash.put("url", l1);
-                    itmHash.put("sourceType", "Tileset");
-                    itmHash.put("name", name);
-                    itmHash.put("id", id);
-                    list.add(itmHash);
+                    String url3d = JSONObject.parseObject(url).getString("3d");
+
+                    if (url3d != null) {
+                        String l1 = JSONObject.parseObject(url3d).getString("l0");
+                        itmHash.put("url", l1);
+                        itmHash.put("sourceType", "Tileset");
+                        itmHash.put("name", name);
+                        itmHash.put("id", id);
+                        list.add(itmHash);
+                    }
+
+
                 }
 
             }
 
         }
+
+            System.out.println("layerList: " + list);
         hashMap.put("data", list);
         return R.ok(hashMap);
     }

--
Gitblit v1.9.3