查看知识库内容

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

@@ -122,4 +122,24 @@ public class KnowLedgeController extends BaseController
         return success();
     }
 
+    /**
+     * 列出所有的title
+     */
+    @GetMapping("/listTiles")
+    public AjaxResult listTiles(String collectionName)
+    {
+        List<String> titles = milvusService.listTiles(collectionName);
+        return success(titles);
+    }
+
+    /**
+     * 根据title名查询相关content
+     */
+    @GetMapping("/listByTitle")
+    public AjaxResult listByTitle(String collectionName, String title)
+    {
+        JSONArray contents = milvusService.listByTitle(collectionName, title);
+        return success(contents);
+    }
+
 }

oa-back/ruoyi-llm/src/main/java/com/ruoyi/web/llm/service/IMilvusService.java

@@ -54,5 +54,5 @@ public interface IMilvusService {
     /**
      * 根据title名查询相关content
      */
-    public List<String> listByTitle(String collectionName, String title);
+    public JSONArray listByTitle(String collectionName, String title);
 }

oa-back/ruoyi-llm/src/main/java/com/ruoyi/web/llm/service/impl/LangChainMilvusServiceImpl.java

@@ -158,7 +158,7 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
                     fastjsonObj.put("file_name", file.getOriginalFilename());
                     String[] fileName = file.getOriginalFilename().split("\\.");
                     fastjsonObj.put("file_type", fileName[fileName.length - 1]);
-                    fastjsonObj.put("title", text.split("\n")[0]);
+                    fastjsonObj.put("title", text.split("\n")[0].split(" ")[1]);
                     fastjsonObj.put("content", text);
                     fastjsonObj.put("embedding", embeddingModel.embed(text).content().vectorAsList());
                     String jsonString = fastjsonObj.toJSONString();

oa-back/ruoyi-llm/src/main/java/com/ruoyi/web/llm/service/impl/MilvusServiceImpl.java

@@ -283,7 +283,7 @@ public class MilvusServiceImpl implements IMilvusService {
         milvusClient.loadCollection(loadCollectionReq);
         QueryReq queryParam = QueryReq.builder()
                 .collectionName(collectionName)
-                .filter("id > 0")
+                .filter("id > 0 && title != \"\"")
                 .outputFields(Arrays.asList("title"))
                 .build();
         QueryResp queryResp = milvusClient.query(queryParam);
@@ -308,8 +308,8 @@ public class MilvusServiceImpl implements IMilvusService {
      * 根据title名查询相关content
      */
     @Override
-    public List<String> listByTitle(String collectionName, String title) {
-        List<String> contentList = new ArrayList<>();
+    public JSONArray listByTitle(String collectionName, String title) {
+        JSONArray resultArray = new JSONArray();
         LoadCollectionReq loadCollectionReq = LoadCollectionReq.builder()
                 .collectionName(collectionName)
                 .build();
@@ -317,24 +317,30 @@ public class MilvusServiceImpl implements IMilvusService {
         QueryReq queryParam = QueryReq.builder()
                 .collectionName(collectionName)
                 .filter(String.format("title == \"%s\"", title))
-                .outputFields(Arrays.asList("content"))
+                .outputFields(Arrays.asList("content", "file_name"))
                 .build();
         QueryResp queryResp = milvusClient.query(queryParam);
         List<QueryResp.QueryResult> rowRecordList;
         if (queryResp != null) {
             rowRecordList = queryResp.getQueryResults();
             for (QueryResp.QueryResult rowRecord : rowRecordList) {
+                JSONObject item = new JSONObject();
                 Object content = rowRecord.getEntity().get("content");
+                Object fileName = rowRecord.getEntity().get("file_name");
                 if (content != null) {
-                    contentList.add(content.toString());
+                    item.put("content", content.toString());
+                }
+                if (fileName != null) {
+                    item.put("file_name", fileName.toString());
                 }
+                resultArray.add(item);
             }
         }
         ReleaseCollectionReq releaseCollectionReq = ReleaseCollectionReq.builder()
                 .collectionName(collectionName)
                 .build();
         milvusClient.releaseCollection(releaseCollectionReq);
-        return contentList;
+        return resultArray;
     }
 
 }

oa-ui-app/pages/form/borrow/borrow.vue

@@ -650,8 +650,8 @@ export default {
         let formData = new FormData();
         let message = "您有一条新的借款申请： \n>" + 
         "申请人：<font color='info'>" + this.getUserName(this.form.applier) + "</font> \n>" + 
-        "借款金额：<font color='warning'>" + this.form.managerAmount ? this.form.managerAmount : this.form.applyAmount + "</font> 元 \n>" + 
-        "借款说明：" + (this.form.applyReason ? this.form.applyReason : this.form.remark) + " \n>" + 
+        "借款金额：<font color='warning'>" + (this.form.managerAmount ? this.form.managerAmount : this.form.applyAmount) + "</font> 元 \n>" + 
+        "借款说明：" + (this.form.applyReason ? this.form.applyReason : this.form.remark) + " \n>" + 
         "\n>" + 
 		"已办流程：<font color='comment'>" + this.taskName + "</font> \n>";
         formData.append('message', message);

oa-ui/src/views/flowable/form/finance/borrowForm.vue

@@ -856,7 +856,7 @@ export default {
               }
               message = "您有一条新的借款申请： \n>" +
                 "申请人：<font color='info'>" + this.getUserName(this.form.applier) + "</font> \n>" +
-                "借款金额：<font color='warning'>" + this.form.managerAmount ? this.form.managerAmount : this.form.applyAmount + "</font> 元 \n>" +
+                "借款金额：<font color='warning'>" + (this.form.managerAmount ? this.form.managerAmount : this.form.applyAmount) + "</font> 元 \n>" +
                 "借款说明：" + (this.form.applyReason ? this.form.applyReason : this.form.remark) + " \n>" +
                 "\n>" +
                 "已办流程：<font color='comment'>" + this.taskName + "</font> \n>";
@@ -888,7 +888,7 @@ export default {
               }
               message = "您有一条新的借款申请： \n>" +
                 "申请人：<font color='info'>" + this.getUserName(this.form.applier) + "</font> \n>" +
-                "借款金额：<font color='warning'>" + this.form.managerAmount ? this.form.managerAmount : this.form.applyAmount + "</font> 元 \n>" +
+                "借款金额：<font color='warning'>" + (this.form.managerAmount ? this.form.managerAmount : this.form.applyAmount) + "</font> 元 \n>" +
                 "借款说明：" + (this.form.applyReason ? this.form.applyReason : this.form.remark) + " \n>" +
                 "\n>" +
                 "已办流程：<font color='comment'>" + this.taskName + "</font> \n>";

oa-ui/src/views/llm/knowledge/index.vue

@@ -186,9 +186,10 @@
                 <div v-for="(content, index) in contentList" :key="index" class="content-item">
                   <div class="content-header">
                     <span class="content-index">{{ index + 1 }}</span>
+                    <span v-if="content.file_name" class="content-file-name">{{ content.file_name }}</span>
                   </div>
                   <div class="content-body">
-                    {{ content }}
+                    {{ content.content }}
                   </div>
                 </div>
               </div>
@@ -197,8 +198,8 @@
         </div>
 
         <!-- 聊天模式 -->
-        <div v-else class="chat-content">
-          <div v-if="selectedKnowledge" class="selected-knowledge">
+        <div v-if="selectedKnowledge && isChatMode" class="chat-content">
+          <div class="selected-knowledge">
             <i class="el-icon-folder folder-icon">
             </i>
             <span class="knowledge-name">{{ selectedKnowledge.collectionName }}</span>
@@ -1498,6 +1499,7 @@ export default {
   .content-header {
     display: flex;
     align-items: center;
+    gap: 12px;
     margin-bottom: 12px;
 
     .content-index {
@@ -1512,6 +1514,18 @@ export default {
       font-size: 12px;
       font-weight: 600;
     }
+
+    .content-file-name {
+      font-size: 12px;
+      color: #909399;
+      background: #f0f2f5;
+      padding: 4px 8px;
+      border-radius: 4px;
+      flex: 1;
+      overflow: hidden;
+      text-overflow: ellipsis;
+      white-space: nowrap;
+    }
   }
 
   .content-body {