查询向量库列表详细信息

llm-back/ruoyi-agent/src/main/java/com/ruoyi/agent/controller/KnowLedgeController.java

@@ -1,6 +1,6 @@
 package com.ruoyi.agent.controller;
 
-import com.alibaba.fastjson2.JSONObject;
+import com.alibaba.fastjson2.JSONArray;
 import com.ruoyi.agent.service.LangChainMilvusService;
 import com.ruoyi.common.config.RuoYiConfig;
 import com.ruoyi.common.core.controller.BaseController;
@@ -13,7 +13,6 @@ import org.springframework.web.multipart.MultipartFile;
 
 import java.io.File;
 import java.io.IOException;
-import java.util.List;
 
 /**
  * cmc知识库Controller
@@ -34,7 +33,7 @@ public class KnowLedgeController extends BaseController
     @GetMapping("/list")
     public AjaxResult listKnowLedgeCollection()
     {
-        JSONObject collectionNames = milvusService.getCollectionNames();
+        JSONArray collectionNames = milvusService.getCollectionNames();
         return success(collectionNames);
     }
 

llm-back/ruoyi-agent/src/main/java/com/ruoyi/agent/service/MilvusService.java

@@ -1,13 +1,15 @@
 package com.ruoyi.agent.service;
 
+import com.alibaba.fastjson2.JSONArray;
 import com.alibaba.fastjson2.JSONObject;
 import io.milvus.client.MilvusServiceClient;
-import io.milvus.grpc.DataType;
+import io.milvus.grpc.*;
 import io.milvus.param.ConnectParam;
 import io.milvus.param.IndexType;
 import io.milvus.param.MetricType;
 import io.milvus.param.collection.*;
 import io.milvus.param.highlevel.collection.ListCollectionsParam;
+import io.milvus.param.highlevel.collection.response.ListCollectionsResponse;
 import io.milvus.param.index.CreateIndexParam;
 
 import java.text.SimpleDateFormat;
@@ -84,21 +86,27 @@ public class MilvusService {
     /**
      * 查询知识库Collection
      */
-    public JSONObject getCollectionNames() {
-        JSONObject jsonObject = new JSONObject();
+    public JSONArray getCollectionNames() {
+        JSONArray jsonArray = new JSONArray();
         ListCollectionsParam listParam = ListCollectionsParam.newBuilder().build();
-        List<String> collectionNames = milvusClient.listCollections(listParam).getData().collectionNames;
-        ShowCollectionsParam showParam = ShowCollectionsParam.newBuilder().withCollectionNames(collectionNames).build();
-        int size = milvusClient.showCollections(showParam).getData().getCollectionIdsCount();
-        for (int i = 0; i < size; i++) {
-            jsonObject.put("collectionId", milvusClient.showCollections(showParam).getData().getCollectionIds(i));
-            jsonObject.put("collectionName", milvusClient.showCollections(showParam).getData().getCollectionNames(i));
-            SimpleDateFormat beijingFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
-            beijingFormat.setTimeZone(TimeZone.getTimeZone("Asia/Shanghai"));
-            String beijingTime = beijingFormat.format(milvusClient.showCollections(showParam).getData().getCreatedUtcTimestamps(i));
-            jsonObject.put("createdTime", beijingTime);
+        ListCollectionsResponse listResponse = milvusClient.listCollections(listParam).getData();
+        if (listResponse != null) {
+            List<String> collectionNames = milvusClient.listCollections(listParam).getData().collectionNames;
+            for (int i = 0; i < collectionNames.size(); i++) {
+                JSONObject jsonObject = new JSONObject();
+                DescribeCollectionParam describeParam = DescribeCollectionParam.newBuilder().withCollectionName(collectionNames.get(i)).build();
+                DescribeCollectionResponse describeResponse = milvusClient.describeCollection(describeParam).getData();
+                jsonObject.put("collectionId", describeResponse.getCollectionID());
+                jsonObject.put("collectionName", collectionNames.get(i));
+                SimpleDateFormat beijingFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
+                beijingFormat.setTimeZone(TimeZone.getTimeZone("Asia/Shanghai"));
+                String beijingTime = beijingFormat.format(describeResponse.getCreatedUtcTimestamp());
+                jsonObject.put("createdTime", beijingTime);
+                jsonObject.put("description", describeResponse.getSchema().getDescription());
+                jsonArray.add(jsonObject);
+            }
         }
-        return jsonObject;
+        return jsonArray;
     }
 
     /**