知识库更新

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

@@ -45,4 +45,14 @@ public interface IMilvusService {
      * 删除知识库所有文件
      */
     public void removeAllDocument(String collectionName);
+
+    /**
+     * 列出所有的title
+     */
+    public List<String> listTiles(String collectionName);
+
+    /**
+     * 根据title名查询相关content
+     */
+    public List<String> listByTitle(String collectionName, String title);
 }

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

@@ -75,7 +75,7 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
     private String milvusServiceUrl;
 
     private MilvusClientV2 milvusClient;
-    
+
     // 复用 ChatModel 实例
     private ChatModel chatModel;
 
@@ -84,12 +84,12 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
         if (milvusServiceUrl == null || milvusServiceUrl.isEmpty()) {
             throw new IllegalArgumentException("milvusServiceUrl 配置不能为空");
         }
-//        milvusClient = new MilvusClientV2(
-//                ConnectConfig.builder()
-//                        .uri(milvusServiceUrl)
-//                        .build());
+        milvusClient = new MilvusClientV2(
+                ConnectConfig.builder()
+                        .uri(milvusServiceUrl)
+                        .build());
     }
-    
+
     @PostConstruct
     public void initChatModel() {
         // 初始化 ChatModel
@@ -97,7 +97,7 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
                 .model("Qwen")
                 .build();
     }
-    
+
     @PreDestroy
     public void destroyMilvusClient() {
         if (milvusClient != null) {
@@ -137,7 +137,7 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
                 // 构建上传目录路径
                 String uploadDir = RuoYiConfig.getProfile() + "/upload/rag/knowledge/" + collectionName;
                 File profilePath = new File(uploadDir);
-                
+
                 // 确保目录存在，创建目录并设置权限
                 if (!profilePath.exists())
                     profilePath.mkdirs();
@@ -491,7 +491,7 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
     private String extractAllTitles(File transferFile) throws Exception {
         StringBuilder titlesBuilder = new StringBuilder();
         String filename = transferFile.getName().toLowerCase();
-        
+
         if (filename.endsWith(".docx")) {
             try (XWPFDocument document = new XWPFDocument(new FileInputStream(transferFile))) {
                 for (XWPFParagraph paragraph : document.getParagraphs()) {
@@ -558,7 +558,7 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
                                     String[] paragraphs = currentLevel2Content.toString().split("\n");
                                     for (String para : paragraphs) {
                                         if (para.trim().isEmpty()) continue;
-                                        
+
                                         // 检查是否是三级标题（简单判断：如果是三级标题，样式应该是3，但这里需要重新解析）
                                         // 由于已经将内容转为字符串，这里采用另一种方式：假设三级标题以数字+点+空格开头（如"1.1."）
                                         // 或者可以重新遍历段落，但为了效率，这里采用简单的判断方式
@@ -569,7 +569,7 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
                                                 break;
                                             }
                                         }
-                                        
+
                                         if (isLevel3Title) {
                                             // 保存当前三级标题内容
                                             if (currentLevel3Content.length() != 0) {
@@ -627,7 +627,7 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
                         String[] paragraphs = currentLevel2Content.toString().split("\n");
                         for (String para : paragraphs) {
                             if (para.trim().isEmpty()) continue;
-                            
+
                             boolean isLevel3Title = false;
                             for (XWPFParagraph p : xwpfDocument.getParagraphs()) {
                                 if (p.getText().trim().equals(para.trim()) && p.getStyle() != null && p.getStyle().equals("3")) {
@@ -635,7 +635,7 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
                                     break;
                                 }
                             }
-                            
+
                             if (isLevel3Title) {
                                 // 保存当前三级标题内容
                                 if (currentLevel3Content.length() != 0) {
@@ -694,20 +694,20 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
                                 StringBuilder currentLevel3Content = new StringBuilder();
                                 Range level2Range = hwpfDocument.getRange();
                                 boolean foundCurrentLevel2 = false;
-                                
+
                                 for (int j = 0; j < level2Range.numParagraphs(); j++) {
                                     Paragraph p = level2Range.getParagraph(j);
                                     String paraText = p.text().trim();
                                     if (paraText.isEmpty()) continue;
-                                    
+
                                     // 找到当前二级标题的开始
                                     if (p.getStyleIndex() == 2 && paraText.equals(currentLevel2Content.toString().split("\n")[0].trim())) {
                                         foundCurrentLevel2 = true;
                                     }
-                                    
+
                                     if (foundCurrentLevel2) {
                                         short paraStyleIndex = p.getStyleIndex();
-                                        
+
                                         if (paraStyleIndex == 3) {
                                             // 三级标题
                                             // 保存当前三级标题内容
@@ -729,7 +729,7 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
                                         }
                                     }
                                 }
-                                
+
                                 // 保存最后一个三级标题内容
                                 if (currentLevel3Content.length() != 0) {
                                     TextSegment segment = TextSegment.from(currentLevel3Content.toString());
@@ -762,20 +762,20 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
                         StringBuilder currentLevel3Content = new StringBuilder();
                         Range level2Range = hwpfDocument.getRange();
                         boolean foundCurrentLevel2 = false;
-                        
+
                         for (int j = 0; j < level2Range.numParagraphs(); j++) {
                             Paragraph p = level2Range.getParagraph(j);
                             String paraText = p.text().trim();
                             if (paraText.isEmpty()) continue;
-                            
+
                             // 找到当前二级标题的开始
                             if (p.getStyleIndex() == 2 && paraText.equals(currentLevel2Content.toString().split("\n")[0].trim())) {
                                 foundCurrentLevel2 = true;
                             }
-                            
+
                             if (foundCurrentLevel2) {
                                 short paraStyleIndex = p.getStyleIndex();
-                                
+
                                 if (paraStyleIndex == 3) {
                                     // 三级标题
                                     // 保存当前三级标题内容
@@ -797,7 +797,7 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
                                 }
                             }
                         }
-                        
+
                         // 保存最后一个三级标题内容
                         if (currentLevel3Content.length() != 0) {
                             TextSegment segment = TextSegment.from(currentLevel3Content.toString());

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

@@ -35,12 +35,12 @@ public class MilvusServiceImpl implements IMilvusService {
         if (milvusServiceUrl == null || milvusServiceUrl.isEmpty()) {
             throw new IllegalArgumentException("milvusServiceUrl 配置不能为空");
         }
-//        milvusClient = new MilvusClientV2(
-//                ConnectConfig.builder()
-//                        .uri(milvusServiceUrl)
-//                        .build());
+        milvusClient = new MilvusClientV2(
+                ConnectConfig.builder()
+                        .uri(milvusServiceUrl)
+                        .build());
     }
-    
+
     @PreDestroy
     public void destroyMilvusClient() {
         if (milvusClient != null) {
@@ -76,15 +76,15 @@ public class MilvusServiceImpl implements IMilvusService {
                 .build());
 
         schema.addField(AddFieldReq.builder()
-                .fieldName("file_type")
+                .fieldName("title")
                 .dataType(DataType.VarChar)
-                .maxLength(10)
+                .maxLength(256)
                 .build());
 
         schema.addField(AddFieldReq.builder()
-                .fieldName("title")
+                .fieldName("file_type")
                 .dataType(DataType.VarChar)
-                .maxLength(256)
+                .maxLength(10)
                 .build());
 
         schema.addField(AddFieldReq.builder()
@@ -220,10 +220,10 @@ public class MilvusServiceImpl implements IMilvusService {
                 .build();
         if (fileType != null && !fileType.equals(""))
             queryParam = QueryReq.builder()
-                .collectionName(collectionName)
-                .filter(String.format("file_type == \"%s\"", fileType))
-                .outputFields(Arrays.asList("file_type", "file_name"))
-                .build();
+                    .collectionName(collectionName)
+                    .filter(String.format("file_type == \"%s\"", fileType))
+                    .outputFields(Arrays.asList("file_type", "file_name"))
+                    .build();
         QueryResp queryResp = milvusClient.query(queryParam);
         List<QueryResp.QueryResult> rowRecordList;
         if (queryResp != null) {
@@ -271,4 +271,70 @@ public class MilvusServiceImpl implements IMilvusService {
         milvusClient.delete(deleteReq);
     }
 
+    /**
+     * 列出所有的title
+     */
+    @Override
+    public List<String> listTiles(String collectionName) {
+        List<String> titleList = new ArrayList<>();
+        LoadCollectionReq loadCollectionReq = LoadCollectionReq.builder()
+                .collectionName(collectionName)
+                .build();
+        milvusClient.loadCollection(loadCollectionReq);
+        QueryReq queryParam = QueryReq.builder()
+                .collectionName(collectionName)
+                .filter("id > 0")
+                .outputFields(Arrays.asList("title"))
+                .build();
+        QueryResp queryResp = milvusClient.query(queryParam);
+        List<QueryResp.QueryResult> rowRecordList;
+        if (queryResp != null) {
+            rowRecordList = queryResp.getQueryResults();
+            for (QueryResp.QueryResult rowRecord : rowRecordList) {
+                Object title = rowRecord.getEntity().get("title");
+                if (title != null) {
+                    titleList.add(title.toString());
+                }
+            }
+        }
+        ReleaseCollectionReq releaseCollectionReq = ReleaseCollectionReq.builder()
+                .collectionName(collectionName)
+                .build();
+        milvusClient.releaseCollection(releaseCollectionReq);
+        return titleList.stream().distinct().collect(Collectors.toList());
+    }
+
+    /**
+     * 根据title名查询相关content
+     */
+    @Override
+    public List<String> listByTitle(String collectionName, String title) {
+        List<String> contentList = new ArrayList<>();
+        LoadCollectionReq loadCollectionReq = LoadCollectionReq.builder()
+                .collectionName(collectionName)
+                .build();
+        milvusClient.loadCollection(loadCollectionReq);
+        QueryReq queryParam = QueryReq.builder()
+                .collectionName(collectionName)
+                .filter(String.format("title == \"%s\"", title))
+                .outputFields(Arrays.asList("content"))
+                .build();
+        QueryResp queryResp = milvusClient.query(queryParam);
+        List<QueryResp.QueryResult> rowRecordList;
+        if (queryResp != null) {
+            rowRecordList = queryResp.getQueryResults();
+            for (QueryResp.QueryResult rowRecord : rowRecordList) {
+                Object content = rowRecord.getEntity().get("content");
+                if (content != null) {
+                    contentList.add(content.toString());
+                }
+            }
+        }
+        ReleaseCollectionReq releaseCollectionReq = ReleaseCollectionReq.builder()
+                .collectionName(collectionName)
+                .build();
+        milvusClient.releaseCollection(releaseCollectionReq);
+        return contentList;
+    }
+
 }

oa-ui/src/api/llm/knowLedge.js

@@ -2,7 +2,7 @@
  * @Author: ysh
  * @Date: 2025-06-30 09:56:10
  * @LastEditors: wrh
- * @LastEditTime: 2025-07-18 15:40:00
+ * @LastEditTime: 2026-04-17 09:54:52
  */
 import request from '@/utils/request'
 
@@ -92,3 +92,21 @@ export function deleteKnowledgeFile(fileName,collectionName) {
     params: { fileName, collectionName }
   })
 }
+
+// 列出所有的title
+export function listTiles(collectionName) {
+  return request({
+    url: '/llm/knowledge/listTiles',
+    method: 'get',
+    params: { collectionName }
+  })
+}
+
+// 根据title名查询相关content
+export function listByTitle(collectionName, title) {
+  return request({
+    url: '/llm/knowledge/listByTitle',
+    method: 'get',
+    params: { collectionName, title }
+  })
+}

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

@@ -71,24 +71,31 @@
       <!-- 右侧面板 -->
       <div class="right-panel">
         <div class="panel-header">
-          <h3>{{ isChatMode ? '知识库对话' : '文件列表' }}</h3>
+          <h3>{{ isChatMode ? '知识库对话' : (isTitleMode ? '标题列表' : '文件列表') }}</h3>
           <div class="header-actions">
-            <el-button v-if="!isChatMode" type="primary" icon="el-icon-chat-round" @click="handleChat(selectedKnowledge)"
+            <el-button v-if="!isChatMode && !isTitleMode" type="primary" icon="el-icon-chat-round" @click="handleChat(selectedKnowledge)"
               :disabled="!selectedKnowledge" v-hasPermi="['llm:knowledge:chat']">
               开始对话
             </el-button>
-            <el-button v-if="!isChatMode" type="primary" icon="el-icon-upload" @click="handleUpload(selectedKnowledge)"
+            <el-button v-if="!isChatMode && !isTitleMode" type="primary" icon="el-icon-upload" @click="handleUpload(selectedKnowledge)"
               :disabled="!selectedKnowledge" v-hasPermi="['llm:knowledge:upload']">
               上传文件
             </el-button>
+            <el-button v-if="!isChatMode && !isTitleMode" type="default" icon="el-icon-notebook-2" @click="switchToTitleMode"
+              :disabled="!selectedKnowledge">
+              查看内容
+            </el-button>
             <el-button v-if="isChatMode" type="default" icon="el-icon-document" @click="switchToFileMode">
               返回文件列表
             </el-button>
+            <el-button v-if="isTitleMode" type="default" icon="el-icon-document" @click="switchToFileMode">
+              返回文件列表
+            </el-button>
           </div>
         </div>
 
         <!-- 文件列表模式 -->
-        <div v-if="!isChatMode" class="file-content" v-loading="fileLoading">
+        <div v-if="!isChatMode && !isTitleMode" class="file-content" v-loading="fileLoading">
           <div v-if="selectedKnowledge" class="selected-knowledge">
             <i class="el-icon-folder folder-icon">
             </i>
@@ -133,6 +140,62 @@
             @pagination="handleFilePagination" :autoScroll="false" />
         </div>
 
+        <!-- 标题列表模式 -->
+        <div v-if="!isChatMode && isTitleMode" class="title-content" v-loading="titleLoading">
+          <div v-if="selectedKnowledge" class="selected-knowledge">
+            <i class="el-icon-folder folder-icon">
+            </i>
+            <span class="knowledge-name">{{ selectedKnowledge.collectionName }}</span>
+            <span class="title-mode-badge">标题模式</span>
+          </div>
+
+          <div v-if="!selectedKnowledge" class="empty-state">
+            <i class="el-icon-notebook-2 empty-icon">
+            </i>
+            <p>请选择一个知识库查看内容</p>
+          </div>
+
+          <div v-else-if="titleList.length === 0" class="empty-state">
+            <i class="el-icon-notebook-2 empty-icon">
+            </i>
+            <p>该知识库暂无标题</p>
+          </div>
+
+          <div v-else class="title-content-container">
+            <!-- 左侧标题列表 -->
+            <div class="title-list">
+              <div v-for="(title, index) in titleList" :key="index" class="title-item"
+                :class="{ 'active': selectedTitle === title }"
+                @click="selectTitle(title)">
+                <i class="el-icon-notebook-2 title-icon"></i>
+                <span class="title-text">{{ title }}</span>
+              </div>
+            </div>
+
+            <!-- 右侧内容列表 -->
+            <div class="content-list" v-loading="contentLoading">
+              <div v-if="!selectedTitle" class="empty-state">
+                <i class="el-icon-document empty-icon"></i>
+                <p>请选择一个标题查看内容</p>
+              </div>
+              <div v-else-if="contentList.length === 0" class="empty-state">
+                <i class="el-icon-document empty-icon"></i>
+                <p>该标题暂无内容</p>
+              </div>
+              <div v-else class="content-items">
+                <div v-for="(content, index) in contentList" :key="index" class="content-item">
+                  <div class="content-header">
+                    <span class="content-index">{{ index + 1 }}</span>
+                  </div>
+                  <div class="content-body">
+                    {{ content }}
+                  </div>
+                </div>
+              </div>
+            </div>
+          </div>
+        </div>
+
         <!-- 聊天模式 -->
         <div v-else class="chat-content">
           <div v-if="selectedKnowledge" class="selected-knowledge">
@@ -302,7 +365,7 @@
 import { parseTime } from "@/utils/ruoyi";
 import { Message } from 'element-ui'
 import { getToken } from "@/utils/auth";
-import { listKnowledge, listKnowLedgeByCollectionName, addKnowledge, updateKnowledge, delKnowledge, insertKnowledgeFile, listKnowledgeDocument, deleteKnowledgeFile, getProcessValue } from "@/api/llm/knowLedge";
+import { listKnowledge, listKnowLedgeByCollectionName, addKnowledge, updateKnowledge, delKnowledge, insertKnowledgeFile, listKnowledgeDocument, deleteKnowledgeFile, getProcessValue, listTiles, listByTitle } from "@/api/llm/knowLedge";
 import { getAnswer, getAnswerStream, getContextFile } from '@/api/llm/rag';
 import { marked } from 'marked';
 
@@ -451,10 +514,28 @@ export default {
       },
 
       // 定时器
-      timer: null
+      timer: null,
+
+      // 标题和内容相关状态
+      isTitleMode: false, // 是否显示标题模式
+      titleList: [], // 标题列表
+      contentList: [], // 内容列表
+      selectedTitle: null, // 选中的标题
+      titleLoading: false, // 标题加载状态
+      contentLoading: false // 内容加载状态
     }
   },
   mounted() {
+    // 重置状态，确保页面刷新后不会停留在对话模式或标题模式
+    this.isChatMode = false;
+    this.isTitleMode = false;
+    this.chatMessages = [];
+    this.chatInput = '';
+    this.streamingStarted = false;
+    this.titleList = [];
+    this.contentList = [];
+    this.selectedTitle = null;
+    
     this.getList();
   },
 
@@ -581,6 +662,50 @@ export default {
       }
 
       this.isChatMode = false;
+      this.isTitleMode = false;
+    },
+
+    /** 切换到标题模式 */
+    switchToTitleMode() {
+      this.isTitleMode = true;
+      this.isChatMode = false;
+      this.titleList = [];
+      this.contentList = [];
+      this.selectedTitle = null;
+      
+      // 获取标题列表
+      this.titleLoading = true;
+      listTiles(this.selectedKnowledge.collectionName).then(response => {
+        if (Array.isArray(response.data)) {
+          this.titleList = response.data;
+        } else {
+          this.titleList = [];
+        }
+        this.titleLoading = false;
+      }).catch(error => {
+        this.titleLoading = false;
+        this.$modal.msgError("获取标题列表失败：" + error.message);
+      });
+    },
+
+    /** 选择标题 */
+    selectTitle(title) {
+      this.selectedTitle = title;
+      this.contentList = [];
+      
+      // 获取内容列表
+      this.contentLoading = true;
+      listByTitle(this.selectedKnowledge.collectionName, title).then(response => {
+        if (Array.isArray(response.data)) {
+          this.contentList = response.data;
+        } else {
+          this.contentList = [];
+        }
+        this.contentLoading = false;
+      }).catch(error => {
+        this.contentLoading = false;
+        this.$modal.msgError("获取内容列表失败：" + error.message);
+      });
     },
 
     /** 发送消息 */
@@ -1282,6 +1407,131 @@ export default {
   overflow-y: auto;
 }
 
+.title-content {
+  flex: 1;
+  padding: 20px;
+  overflow: hidden;
+}
+
+.title-content-container {
+  display: flex;
+  height: calc(100vh - 240px);
+  gap: 20px;
+  overflow: hidden;
+}
+
+.title-list {
+  width: 300px;
+  background: #fafbfc;
+  border-radius: 8px;
+  border: 1px solid #e4e7ed;
+  padding: 12px;
+  overflow-y: auto;
+}
+
+.title-item {
+  display: flex;
+  align-items: center;
+  gap: 8px;
+  padding: 10px 12px;
+  margin-bottom: 8px;
+  border-radius: 6px;
+  cursor: pointer;
+  transition: all 0.3s ease;
+  background: white;
+  border: 1px solid #e4e7ed;
+
+  &:hover {
+    border-color: #409eff;
+    background: #f0f9ff;
+  }
+
+  &.active {
+    border-color: #409eff;
+    background: #e6f7ff;
+    box-shadow: 0 2px 8px rgba(64, 158, 255, 0.15);
+  }
+
+  .title-icon {
+    color: #409eff;
+    font-size: 16px;
+  }
+
+  .title-text {
+    flex: 1;
+    font-size: 14px;
+    color: #303133;
+    line-height: 1.4;
+    overflow: hidden;
+    text-overflow: ellipsis;
+    white-space: nowrap;
+  }
+}
+
+.content-list {
+  flex: 1;
+  background: #fafbfc;
+  border-radius: 8px;
+  border: 1px solid #e4e7ed;
+  padding: 20px;
+  overflow-y: auto;
+}
+
+.content-items {
+  display: flex;
+  flex-direction: column;
+  gap: 16px;
+}
+
+.content-item {
+  background: white;
+  border: 1px solid #e4e7ed;
+  border-radius: 8px;
+  padding: 16px;
+  transition: all 0.3s ease;
+
+  &:hover {
+    border-color: #409eff;
+    box-shadow: 0 2px 8px rgba(64, 158, 255, 0.15);
+  }
+
+  .content-header {
+    display: flex;
+    align-items: center;
+    margin-bottom: 12px;
+
+    .content-index {
+      background: #409eff;
+      color: white;
+      width: 24px;
+      height: 24px;
+      border-radius: 50%;
+      display: flex;
+      align-items: center;
+      justify-content: center;
+      font-size: 12px;
+      font-weight: 600;
+    }
+  }
+
+  .content-body {
+    font-size: 14px;
+    color: #303133;
+    line-height: 1.6;
+    white-space: pre-wrap;
+    word-break: break-word;
+  }
+}
+
+.title-mode-badge {
+  background: #67c23a;
+  color: white;
+  padding: 2px 8px;
+  border-radius: 12px;
+  font-size: 12px;
+  margin-left: auto;
+}
+
 // 聊天模式样式
 .chat-content {
   flex: 1;