上传知识库文件，修改判断二三级标题

oa-back/ruoyi-agent/src/main/java/com/ruoyi/agent/service/impl/McpServiceImpl.java

@@ -79,7 +79,7 @@ public class McpServiceImpl implements IMcpService {
 //                ConnectConfig.builder()
 //                        .uri(milvusServiceUrl)
 //                        .build());
-    }    
+    }
 
     @PreDestroy
     public void destroyMilvusClient() {
@@ -104,16 +104,16 @@ public class McpServiceImpl implements IMcpService {
                                            @Param(description = "章节名称") String title,
                                            @Param(description = "技术文件地址") String templatePath) throws IOException
     {
-            try {
-                if (templatePath.startsWith("/dev-api"))
-                    templatePath = templatePath.replace("/dev-api/profile", profile);
-                templatePath =profile + templatePath;
-                List<String> subTitles = extractSubTitles(templatePath, title);
-                List<JSONObject> contexts = retrieveFromMilvus(collectionName, title, 10);
-                return generateAnswerWithDocumentAndCollection(agentName, templatePath, subTitles, contexts);
-            } catch (IOException e) {
-                throw new RuntimeException(e);
-            }
+        try {
+            if (templatePath.startsWith("/dev-api"))
+                templatePath = templatePath.replace("/dev-api/profile", profile);
+            templatePath =profile + templatePath;
+            List<String> subTitles = extractSubTitles(templatePath, title);
+            List<JSONObject> contexts = retrieveFromMilvus(collectionName, title, 10);
+            return generateAnswerWithDocumentAndCollection(agentName, templatePath, subTitles, contexts);
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
     }
 
     /**
@@ -122,27 +122,27 @@ public class McpServiceImpl implements IMcpService {
     @ToolMapping(description = "执行SQL查询")
     public JSONArray SQLQuery(@Param(description = "sql语句") String sqlString)
     {
-            sqlString = sqlString.replace("[","").replace("]","");
-            return executeQuery(sqlString);
+        sqlString = sqlString.replace("[","").replace("]","");
+        return executeQuery(sqlString);
     }
 
     /**
-      * 获取所有表名列表
-      */
-     @ToolMapping(description = "获取数据库中所有表名列表")
-     public JSONArray GetAllTableNames()
-     {
-             return getAllTableNames();
-     }
-
-     /**
-      * 获取指定表的结构信息
-      */
-     @ToolMapping(description = "获取指定表的结构信息")
-     public JSONObject GetTableStructure(@Param(description = "表名") String tableName)
-     {
-             return getTableStructure(tableName);
-     }
+     * 获取所有表名列表
+     */
+    @ToolMapping(description = "获取数据库中所有表名列表")
+    public JSONArray GetAllTableNames()
+    {
+        return getAllTableNames();
+    }
+
+    /**
+     * 获取指定表的结构信息
+     */
+    @ToolMapping(description = "获取指定表的结构信息")
+    public JSONObject GetTableStructure(@Param(description = "表名") String tableName)
+    {
+        return getTableStructure(tableName);
+    }
 
     public JSONArray executeQuery(String sql) {
         JSONArray result = new JSONArray();
@@ -337,25 +337,48 @@ public class McpServiceImpl implements IMcpService {
 
             for (XWPFParagraph  paragraph : document.getParagraphs()) {
                 String text = paragraph.getText();
-                int level = Integer.parseInt(paragraph.getStyle());
-
-                // 维护当前路径
-                while (!currentPath.isEmpty() && Integer.parseInt(currentPath.get(currentPath.size() - 1).getStyle()) >= level) {
-                    currentPath.remove(currentPath.size() - 1);
-                }
-                currentPath.add(paragraph);
-                if (foundParent) {
-                    if (level > parentLevel) { // 是子标题
-                        if (isLeafNode(document, paragraph, level)) {
-                            subTitles.add(text);
+                if (paragraph.getStyle() != null) {
+                    String styleId = paragraph.getStyle();
+                    XWPFStyle style = document.getStyles().getStyle(styleId);
+                    String styleName = style.getName();
+                    int level = 0;
+                    if (styleName.equals("heading 2")) level = 2;
+                    else if (styleName.equals("heading 3")) level = 3;
+                    else if (styleName.equals("heading 4")) level = 4;
+                    else continue;
+
+                    // 维护当前路径
+                    while (!currentPath.isEmpty()) {
+                        XWPFParagraph lastPara = currentPath.get(currentPath.size() - 1);
+                        String lastStyleId = lastPara.getStyle();
+                        XWPFStyle lastStyle = document.getStyles().getStyle(lastStyleId);
+                        String lastStyleName = lastStyle.getName();
+                        int lastLevel = 0;
+                        if (lastStyleName.equals("heading 2")) lastLevel = 2;
+                        else if (lastStyleName.equals("heading 3")) lastLevel = 3;
+                        else if (lastStyleName.equals("heading 4")) lastLevel = 4;
+                        else break;
+
+                        if (lastLevel >= level) {
+                            currentPath.remove(currentPath.size() - 1);
+                        } else {
+                            break;
                         }
-                    } else {
-                        // 遇到同级或更高级别的标题，停止搜索
-                        break;
                     }
-                } else if (text.equals(question)) {
-                    foundParent = true;
-                    parentLevel = level;
+                    currentPath.add(paragraph);
+                    if (foundParent) {
+                        if (level > parentLevel) { // 是子标题
+                            if (isLeafNode(document, paragraph, level)) {
+                                subTitles.add(text);
+                            }
+                        } else {
+                            // 遇到同级或更高级别的标题，停止搜索
+                            break;
+                        }
+                    } else if (text.equals(question)) {
+                        foundParent = true;
+                        parentLevel = level;
+                    }
                 }
             }
 
@@ -374,10 +397,19 @@ public class McpServiceImpl implements IMcpService {
 
         // 检查后续段落
         XWPFParagraph nextP = doc.getParagraphs().get(index + 1);
-        int nextLevel = Integer.parseInt(nextP.getStyle());
-        // 遇到同级或更高级别标题，是叶子节点
-        return nextLevel <= level; // 存在更低级别的标题，不是叶子节点
-
+        if (nextP.getStyle() != null) {
+            String styleId = nextP.getStyle();
+            XWPFStyle style = doc.getStyles().getStyle(styleId);
+            String styleName = style.getName();
+            int nextLevel = 0;
+            if (styleName.equals("heading 2")) nextLevel = 2;
+            else if (styleName.equals("heading 3")) nextLevel = 3;
+            else if (styleName.equals("heading 4")) nextLevel = 4;
+            else return true;
+            // 遇到同级或更高级别标题，是叶子节点
+            return nextLevel <= level; // 存在更低级别的标题，不是叶子节点
+        }
+        return true;
     }
 
     /**
@@ -390,8 +422,11 @@ public class McpServiceImpl implements IMcpService {
             for (XWPFParagraph paragraph : document.getParagraphs()) {
                 String text = paragraph.getText().trim();
                 if (paragraph.getStyle() != null) {
+                    String styleId = paragraph.getStyle();
+                    XWPFStyle style = document.getStyles().getStyle(styleId);
+                    String styleName = style.getName();
                     // 判断主标题
-                    if (paragraph.getStyle().equals("3") || paragraph.getStyle().equals("4") ) {
+                    if (styleName.equals("heading 3") || styleName.equals("heading 4")) {
                         subTitles.append(text).append("\n");
                     }
                 }
@@ -456,9 +491,12 @@ public class McpServiceImpl implements IMcpService {
                     String text = paragraph.getText().trim();
                     if (text.isEmpty()) continue;
 
-                    if (paragraph.getStyle() != null) {
-                        // 二级标题（样式2）开始新的分段
-                        if (paragraph.getStyle().equals("2")) {
+                    if (paragraph.getStyleID() != null) {
+                        String styleId = paragraph.getStyleID();
+                        XWPFStyle style = xwpfDocument.getStyles().getStyle(styleId);
+                        String styleName = style.getName();
+                        // 二级标题开始新的分段
+                        if (styleName.equals("heading 2")) {
                             // 保存当前二级标题下的内容
                             if (currentLevel2Content.length() != 0) {
                                 // 检查当前二级标题内容的字节数
@@ -474,16 +512,24 @@ public class McpServiceImpl implements IMcpService {
                                     String[] paragraphs = currentLevel2Content.toString().split("\n");
                                     for (String para : paragraphs) {
                                         if (para.trim().isEmpty()) continue;
-                                        
-                                        // 检查是否是三级标题
+
+                                        // 检查是否是三级标题（简单判断：如果是三级标题，样式应该是3，但这里需要重新解析）
+                                        // 由于已经将内容转为字符串，这里采用另一种方式：假设三级标题以数字+点+空格开头（如"1.1."）
+                                        // 或者可以重新遍历段落，但为了效率，这里采用简单的判断方式
                                         boolean isLevel3Title = false;
                                         for (XWPFParagraph p : xwpfDocument.getParagraphs()) {
-                                            if (p.getText().trim().equals(para.trim()) && p.getStyle() != null && p.getStyle().equals("3")) {
-                                                isLevel3Title = true;
-                                                break;
+                                            if (p.getText().trim().equals(para.trim()) && p.getStyle() != null) {
+                                                styleId = p.getStyle();
+                                                style = xwpfDocument.getStyles().getStyle(styleId);
+                                                styleName = style.getName();
+                                                // 二级标题（样式2）开始新的分段
+                                                if (styleName.equals("heading 3")) {
+                                                    isLevel3Title = true;
+                                                    break;
+                                                }
                                             }
                                         }
-                                        
+
                                         if (isLevel3Title) {
                                             // 保存当前三级标题内容
                                             if (currentLevel3Content.length() != 0) {
@@ -541,15 +587,21 @@ public class McpServiceImpl implements IMcpService {
                         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")) {
-                                    isLevel3Title = true;
-                                    break;
+                                if (p.getText().trim().equals(para.trim()) && p.getStyle() != null) {
+                                    String styleId = p.getStyle();
+                                    XWPFStyle style = xwpfDocument.getStyles().getStyle(styleId);
+                                    String styleName = style.getName();
+                                    // 二级标题（样式2）开始新的分段
+                                    if (styleName.equals("heading 3")) {
+                                        isLevel3Title = true;
+                                        break;
+                                    }
                                 }
                             }
-                            
+
                             if (isLevel3Title) {
                                 // 保存当前三级标题内容
                                 if (currentLevel3Content.length() != 0) {
@@ -608,20 +660,20 @@ public class McpServiceImpl implements IMcpService {
                                 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) {
                                             // 三级标题
                                             // 保存当前三级标题内容
@@ -643,7 +695,7 @@ public class McpServiceImpl implements IMcpService {
                                         }
                                     }
                                 }
-                                
+
                                 // 保存最后一个三级标题内容
                                 if (currentLevel3Content.length() != 0) {
                                     TextSegment segment = TextSegment.from(currentLevel3Content.toString());
@@ -676,20 +728,20 @@ public class McpServiceImpl implements IMcpService {
                         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) {
                                     // 三级标题
                                     // 保存当前三级标题内容
@@ -711,7 +763,7 @@ public class McpServiceImpl implements IMcpService {
                                 }
                             }
                         }
-                        
+
                         // 保存最后一个三级标题内容
                         if (currentLevel3Content.length() != 0) {
                             TextSegment segment = TextSegment.from(currentLevel3Content.toString());
@@ -743,7 +795,7 @@ public class McpServiceImpl implements IMcpService {
      */
     private JSONObject getTableStructure(String tableName) {
         JSONObject result = new JSONObject();
-        
+
         try {
             Class.forName("com.mysql.cj.jdbc.Driver");
         } catch (ClassNotFoundException e) {
@@ -751,12 +803,12 @@ public class McpServiceImpl implements IMcpService {
         }
 
         try (Connection conn = DriverManager.getConnection(Solon.cfg().getProperty("cmc.mysqlService.jdbcUrl"),Solon.cfg().getProperty("cmc.mysqlService.username"), Solon.cfg().getProperty("cmc.mysqlService.password"))) {
-            
+
             DatabaseMetaData metaData = conn.getMetaData();
-            
+
             ResultSet columns = metaData.getColumns(null, null, tableName, null);
             JSONArray columnsInfo = new JSONArray();
-            
+
             while (columns.next()) {
                 JSONObject column = new JSONObject();
                 column.put("columnName", columns.getString("COLUMN_NAME"));
@@ -767,23 +819,23 @@ public class McpServiceImpl implements IMcpService {
                 columnsInfo.add(column);
             }
             columns.close();
-            
+
             ResultSet primaryKeys = metaData.getPrimaryKeys(null, null, tableName);
             JSONArray primaryKeyColumns = new JSONArray();
-            
+
             while (primaryKeys.next()) {
                 primaryKeyColumns.add(primaryKeys.getString("COLUMN_NAME"));
             }
             primaryKeys.close();
-            
+
             result.put("tableName", tableName);
             result.put("columns", columnsInfo);
             result.put("primaryKey", primaryKeyColumns);
-            
+
         } catch (SQLException e) {
             throw new RuntimeException("获取表结构失败: " + tableName, e);
         }
-        
+
         return result;
     }
 
@@ -792,7 +844,7 @@ public class McpServiceImpl implements IMcpService {
      */
     private JSONArray getAllTableNames() {
         JSONArray tableNames = new JSONArray();
-        
+
         try {
             Class.forName("com.mysql.cj.jdbc.Driver");
         } catch (ClassNotFoundException e) {
@@ -800,10 +852,10 @@ public class McpServiceImpl implements IMcpService {
         }
 
         try (Connection conn = DriverManager.getConnection(Solon.cfg().getProperty("cmc.mysqlService.jdbcUrl"),Solon.cfg().getProperty("cmc.mysqlService.username"), Solon.cfg().getProperty("cmc.mysqlService.password"))) {
-            
+
             DatabaseMetaData metaData = conn.getMetaData();
             ResultSet tables = metaData.getTables(null, null, null, new String[]{"TABLE"});
-            
+
             while (tables.next()) {
                 String tableName = tables.getString("TABLE_NAME");
                 String tableComment = tables.getString("REMARKS");
@@ -816,11 +868,11 @@ public class McpServiceImpl implements IMcpService {
                 }
             }
             tables.close();
-            
+
         } catch (SQLException e) {
             throw new RuntimeException("获取表名列表失败", e);
         }
-        
+
         return tableNames;
     }
 

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

@@ -37,6 +37,7 @@ import org.apache.poi.hwpf.usermodel.Paragraph;
 import org.apache.poi.hwpf.usermodel.Range;
 import org.apache.poi.xwpf.usermodel.XWPFDocument;
 import org.apache.poi.xwpf.usermodel.XWPFParagraph;
+import org.apache.poi.xwpf.usermodel.XWPFStyle;
 import org.noear.solon.ai.chat.ChatModel;
 import org.noear.solon.ai.chat.ChatResponse;
 import org.noear.solon.ai.chat.ChatSession;
@@ -421,9 +422,12 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
         try (XWPFDocument document = new XWPFDocument(fileInputStream)) {
             for (XWPFParagraph paragraph : document.getParagraphs()) {
                 String text = paragraph.getText().trim();
-                if (paragraph.getStyle() != null) {
+                if (paragraph.getStyleID() != null) {
+                    String styleId = paragraph.getStyleID();
+                    XWPFStyle style = document.getStyles().getStyle(styleId);
+                    String styleName = style.getName();
                     // 判断主标题
-                    if (paragraph.getStyle().equals("3") &&
+                    if (styleName.equals("heading 3") &&
                             text.contains(question)) {
                         inTargetSection = true;
                         continue;
@@ -431,11 +435,11 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
 
                     // 如果已经在目标节中，收集标题3级别的子标题
                     if (inTargetSection) {
-                        if (paragraph.getStyle().equals("4")) {
+                        if (styleName.equals("heading 4")) {
                             subTitles.add(text);
                         }
-                        // 遇到下一个Heading1则退出
-                        else if (paragraph.getStyle().equals("3")) {
+                        // 遇到下一个Heading2则退出
+                        else if (styleName.equals("heading 3")) {
                             break;
                         }
                     }
@@ -497,8 +501,12 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
                 for (XWPFParagraph paragraph : document.getParagraphs()) {
                     String text = paragraph.getText().trim();
                     if (text.isEmpty()) continue;
-                    if (paragraph.getStyle() != null) {
-                        if (paragraph.getStyle().equals("2") || paragraph.getStyle().equals("3")) {
+                    if (paragraph.getStyleID() != null) {
+                        String styleId = paragraph.getStyleID();
+                        XWPFStyle style = document.getStyles().getStyle(styleId);
+                        String styleName = style.getName();
+                        // 二级标题（样式2）开始新的分段
+                        if (styleName.equals("heading 2") || styleName.equals("heading 3")) {
                             titlesBuilder.append(text).append("\n");
                         }
                     }
@@ -540,9 +548,12 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
                     String text = paragraph.getText().trim();
                     if (text.isEmpty()) continue;
 
-                    if (paragraph.getStyle() != null) {
-                        // 二级标题（样式2）开始新的分段
-                        if (paragraph.getStyle().equals("2")) {
+                    if (paragraph.getStyleID() != null) {
+                        String styleId = paragraph.getStyleID();
+                        XWPFStyle style = xwpfDocument.getStyles().getStyle(styleId);
+                        String styleName = style.getName();
+                        // 二级标题开始新的分段
+                        if (styleName.equals("heading 2")) {
                             // 保存当前二级标题下的内容
                             if (currentLevel2Content.length() != 0) {
                                 // 检查当前二级标题内容的字节数
@@ -564,9 +575,15 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
                                         // 或者可以重新遍历段落，但为了效率，这里采用简单的判断方式
                                         boolean isLevel3Title = false;
                                         for (XWPFParagraph p : xwpfDocument.getParagraphs()) {
-                                            if (p.getText().trim().equals(para.trim()) && p.getStyle() != null && p.getStyle().equals("3")) {
-                                                isLevel3Title = true;
-                                                break;
+                                            if (p.getText().trim().equals(para.trim()) && p.getStyle() != null) {
+                                                styleId = p.getStyle();
+                                                style = xwpfDocument.getStyles().getStyle(styleId);
+                                                styleName = style.getName();
+                                                // 二级标题（样式2）开始新的分段
+                                                if (styleName.equals("heading 3")) {
+                                                    isLevel3Title = true;
+                                                    break;
+                                                }
                                             }
                                         }
 
@@ -630,9 +647,15 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
 
                             boolean isLevel3Title = false;
                             for (XWPFParagraph p : xwpfDocument.getParagraphs()) {
-                                if (p.getText().trim().equals(para.trim()) && p.getStyle() != null && p.getStyle().equals("3")) {
-                                    isLevel3Title = true;
-                                    break;
+                                if (p.getText().trim().equals(para.trim()) && p.getStyle() != null) {
+                                    String styleId = p.getStyle();
+                                    XWPFStyle style = xwpfDocument.getStyles().getStyle(styleId);
+                                    String styleName = style.getName();
+                                    // 二级标题（样式2）开始新的分段
+                                    if (styleName.equals("heading 3")) {
+                                        isLevel3Title = true;
+                                        break;
+                                    }
                                 }
                             }
 

oa-back/ruoyi-system/src/main/java/com/ruoyi/llm/service/impl/CmcAgentServiceImpl.java

@@ -96,7 +96,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService
 //                        .uri(milvusServiceUrl)
 //                        .build());
     }
-    
+
     @PreDestroy
     public void destroyMilvusClient() {
         if (milvusClient != null) {
@@ -436,7 +436,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService
         }
         message.append("\n技术文件已更新，请查看：\n");
         message.append("【<a href='/profile").append(templatePath.replace("\\", "/").replace(RuoYiConfig.getProfile(), ""))
-               .append("'>点击查看技术文件</a>】\n\n");
+                .append("'>点击查看技术文件</a>】\n\n");
         message.append("如需修改特定章节，请输入章节标题名称。\n").append("若提示未检测到工具，麻烦重新发送，谢谢");
 
         return message.toString();
@@ -591,11 +591,11 @@ public class CmcAgentServiceImpl implements ICmcAgentService
                 .append(question)
                 .append("，先列出二级章节标题，严格按以下格式，仅输出标题列表：\n")
                 .append("6.1 XX\n" +
-                "6.2 XX\n" +
-                "6.3 XX\n" +
-                "......\n" +
-                "6.n-1 XX\n" +
-                "6.n XX\n");
+                        "6.2 XX\n" +
+                        "6.3 XX\n" +
+                        "......\n" +
+                        "6.n-1 XX\n" +
+                        "6.n XX\n");
         return writeChapters(sb.toString(), templatePath);
     }
 
@@ -711,7 +711,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService
                 .metricType(IndexParam.MetricType.COSINE)
                 .searchParams(searchParams)
                 .build();
-        
+
         if (filter != null && !filter.isEmpty()) {
             searchReq = SearchReq.builder()
                     .collectionName(collectionName)
@@ -853,25 +853,48 @@ public class CmcAgentServiceImpl implements ICmcAgentService
 
             for (XWPFParagraph  paragraph : document.getParagraphs()) {
                 String text = paragraph.getText();
-                int level = Integer.parseInt(paragraph.getStyle());
-
-                // 维护当前路径
-                while (!currentPath.isEmpty() && Integer.parseInt(currentPath.get(currentPath.size() - 1).getStyle()) >= level) {
-                    currentPath.remove(currentPath.size() - 1);
-                }
-                currentPath.add(paragraph);
-                if (foundParent) {
-                    if (level > parentLevel) { // 是子标题
-                        if (isLeafNode(document, paragraph, level)) {
-                            subTitles.add(text.replace("\n", ""));
+                if (paragraph.getStyle() != null) {
+                    String styleId = paragraph.getStyle();
+                    XWPFStyle style = document.getStyles().getStyle(styleId);
+                    String styleName = style.getName();
+                    int level = 0;
+                    if (styleName.equals("heading 2")) level = 2;
+                    else if (styleName.equals("heading 3")) level = 3;
+                    else if (styleName.equals("heading 4")) level = 4;
+                    else continue;
+
+                    // 维护当前路径
+                    while (!currentPath.isEmpty()) {
+                        XWPFParagraph lastPara = currentPath.get(currentPath.size() - 1);
+                        String lastStyleId = lastPara.getStyle();
+                        XWPFStyle lastStyle = document.getStyles().getStyle(lastStyleId);
+                        String lastStyleName = lastStyle.getName();
+                        int lastLevel = 0;
+                        if (lastStyleName.equals("heading 2")) lastLevel = 2;
+                        else if (lastStyleName.equals("heading 3")) lastLevel = 3;
+                        else if (lastStyleName.equals("heading 4")) lastLevel = 4;
+                        else break;
+
+                        if (lastLevel >= level) {
+                            currentPath.remove(currentPath.size() - 1);
+                        } else {
+                            break;
                         }
-                    } else {
-                        // 遇到同级或更高级别的标题，停止搜索
-                        break;
                     }
-                } else if (text.equals(question)) {
-                    foundParent = true;
-                    parentLevel = level;
+                    currentPath.add(paragraph);
+                    if (foundParent) {
+                        if (level > parentLevel) { // 是子标题
+                            if (isLeafNode(document, paragraph, level)) {
+                                subTitles.add(text.replace("\n", ""));
+                            }
+                        } else {
+                            // 遇到同级或更高级别的标题，停止搜索
+                            break;
+                        }
+                    } else if (text.equals(question)) {
+                        foundParent = true;
+                        parentLevel = level;
+                    }
                 }
             }
 
@@ -892,10 +915,19 @@ public class CmcAgentServiceImpl implements ICmcAgentService
 
         // 检查后续段落
         XWPFParagraph nextP = doc.getParagraphs().get(index + 1);
-        int nextLevel = Integer.parseInt(nextP.getStyle());
-        // 遇到同级或更高级别标题，是叶子节点
-        return nextLevel <= level; // 存在更低级别的标题，不是叶子节点
-
+        if (nextP.getStyle() != null) {
+            String styleId = nextP.getStyle();
+            XWPFStyle style = doc.getStyles().getStyle(styleId);
+            String styleName = style.getName();
+            int nextLevel = 0;
+            if (styleName.equals("heading 2")) nextLevel = 2;
+            else if (styleName.equals("heading 3")) nextLevel = 3;
+            else if (styleName.equals("heading 4")) nextLevel = 4;
+            else return true;
+            // 遇到同级或更高级别标题，是叶子节点
+            return nextLevel <= level; // 存在更低级别的标题，不是叶子节点
+        }
+        return true;
     }
 
     /**
@@ -908,8 +940,11 @@ public class CmcAgentServiceImpl implements ICmcAgentService
             for (XWPFParagraph paragraph : document.getParagraphs()) {
                 String text = paragraph.getText().trim();
                 if (paragraph.getStyle() != null) {
+                    String styleId = paragraph.getStyle();
+                    XWPFStyle style = document.getStyles().getStyle(styleId);
+                    String styleName = style.getName();
                     // 判断主标题
-                    if (paragraph.getStyle().equals("3") || paragraph.getStyle().equals("4") ) {
+                    if (styleName.equals("heading 3") || styleName.equals("heading 4")) {
                         subTitles.append(text).append("\n");
                     }
                 }
@@ -936,9 +971,12 @@ public class CmcAgentServiceImpl implements ICmcAgentService
                     String text = paragraph.getText().trim();
                     if (text.isEmpty()) continue;
 
-                    if (paragraph.getStyle() != null) {
-                        // 二级标题（样式2）开始新的分段
-                        if (paragraph.getStyle().equals("3")) {
+                    if (paragraph.getStyleID() != null) {
+                        String styleId = paragraph.getStyleID();
+                        XWPFStyle style = xwpfDocument.getStyles().getStyle(styleId);
+                        String styleName = style.getName();
+                        // 二级标题开始新的分段
+                        if (styleName.equals("heading 2")) {
                             // 保存当前二级标题下的内容
                             if (currentLevel2Content.length() != 0) {
                                 // 检查当前二级标题内容的字节数
@@ -954,16 +992,24 @@ public class CmcAgentServiceImpl implements ICmcAgentService
                                     String[] paragraphs = currentLevel2Content.toString().split("\n");
                                     for (String para : paragraphs) {
                                         if (para.trim().isEmpty()) continue;
-                                        
-                                        // 检查是否是三级标题
+
+                                        // 检查是否是三级标题（简单判断：如果是三级标题，样式应该是3，但这里需要重新解析）
+                                        // 由于已经将内容转为字符串，这里采用另一种方式：假设三级标题以数字+点+空格开头（如"1.1."）
+                                        // 或者可以重新遍历段落，但为了效率，这里采用简单的判断方式
                                         boolean isLevel3Title = false;
                                         for (XWPFParagraph p : xwpfDocument.getParagraphs()) {
-                                            if (p.getText().trim().equals(para.trim()) && p.getStyle() != null && p.getStyle().equals("4")) {
-                                                isLevel3Title = true;
-                                                break;
+                                            if (p.getText().trim().equals(para.trim()) && p.getStyle() != null) {
+                                                styleId = p.getStyle();
+                                                style = xwpfDocument.getStyles().getStyle(styleId);
+                                                styleName = style.getName();
+                                                // 二级标题（样式2）开始新的分段
+                                                if (styleName.equals("heading 3")) {
+                                                    isLevel3Title = true;
+                                                    break;
+                                                }
                                             }
                                         }
-                                        
+
                                         if (isLevel3Title) {
                                             // 保存当前三级标题内容
                                             if (currentLevel3Content.length() != 0) {
@@ -1021,15 +1067,21 @@ public class CmcAgentServiceImpl implements ICmcAgentService
                         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("4")) {
-                                    isLevel3Title = true;
-                                    break;
+                                if (p.getText().trim().equals(para.trim()) && p.getStyle() != null) {
+                                    String styleId = p.getStyle();
+                                    XWPFStyle style = xwpfDocument.getStyles().getStyle(styleId);
+                                    String styleName = style.getName();
+                                    // 二级标题（样式2）开始新的分段
+                                    if (styleName.equals("heading 3")) {
+                                        isLevel3Title = true;
+                                        break;
+                                    }
                                 }
                             }
-                            
+
                             if (isLevel3Title) {
                                 // 保存当前三级标题内容
                                 if (currentLevel3Content.length() != 0) {
@@ -1088,20 +1140,20 @@ public class CmcAgentServiceImpl implements ICmcAgentService
                                 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) {
                                             // 三级标题
                                             // 保存当前三级标题内容
@@ -1123,7 +1175,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService
                                         }
                                     }
                                 }
-                                
+
                                 // 保存最后一个三级标题内容
                                 if (currentLevel3Content.length() != 0) {
                                     TextSegment segment = TextSegment.from(currentLevel3Content.toString());
@@ -1156,20 +1208,20 @@ public class CmcAgentServiceImpl implements ICmcAgentService
                         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) {
                                     // 三级标题
                                     // 保存当前三级标题内容
@@ -1191,7 +1243,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService
                                 }
                             }
                         }
-                        
+
                         // 保存最后一个三级标题内容
                         if (currentLevel3Content.length() != 0) {
                             TextSegment segment = TextSegment.from(currentLevel3Content.toString());
@@ -1367,7 +1419,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService
                         blocks.add(new Block("para", line));
                     }
                 }
-                
+
                 // 更新上一行记录
                 previousLine = line;
             }
@@ -1439,13 +1491,13 @@ public class CmcAgentServiceImpl implements ICmcAgentService
                     case "boldOnly": {
                         // 处理格式：**文本** 或 **文本**： 转换为 Word 中的部分加粗
                         XWPFParagraph p = document.insertNewParagraph(cursor);
-                        
+
                         // 加粗文本部分
                         XWPFRun run1 = p.createRun();
                         run1.setText(b.boldText);
                         run1.setFontSize(12);
                         run1.setBold(true);
-                        
+
                         // 后缀部分（不加粗，如冒号及后续内容，如果有的话）
                         if (b.suffixText != null && !b.suffixText.isEmpty()) {
                             XWPFRun run2 = p.createRun();

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

@@ -71,31 +71,31 @@
       <!-- 右侧面板 -->
       <div class="right-panel">
         <div class="panel-header">
-          <h3>{{ isChatMode ? '知识库对话' : (isTitleMode ? '标题列表' : '文件列表') }}</h3>
+          <h3>{{ isChatMode ? '知识库对话' : (isScanMode ? '内容列表' : '文件列表') }}</h3>
           <div class="header-actions">
-            <el-button v-if="!isChatMode && !isTitleMode" type="primary" icon="el-icon-chat-round" @click="handleChat(selectedKnowledge)"
+            <el-button v-if="!isChatMode && !isScanMode" type="primary" icon="el-icon-chat-round" @click="handleChat(selectedKnowledge)"
               :disabled="!selectedKnowledge" v-hasPermi="['llm:knowledge:chat']">
               开始对话
             </el-button>
-            <el-button v-if="!isChatMode && !isTitleMode" type="primary" icon="el-icon-upload" @click="handleUpload(selectedKnowledge)"
+            <el-button v-if="!isChatMode && !isScanMode" 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"
+            <el-button v-if="!isChatMode && !isScanMode" 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 v-if="isScanMode" type="default" icon="el-icon-document" @click="switchToFileMode">
               返回文件列表
             </el-button>
           </div>
         </div>
 
         <!-- 文件列表模式 -->
-        <div v-if="!isChatMode && !isTitleMode" class="file-content" v-loading="fileLoading">
+        <div v-if="!isChatMode && !isScanMode" class="file-content" v-loading="fileLoading">
           <div v-if="selectedKnowledge" class="selected-knowledge">
             <i class="el-icon-folder folder-icon">
             </i>
@@ -140,13 +140,13 @@
             @pagination="handleFilePagination" :autoScroll="false" />
         </div>
 
-        <!-- 标题列表模式 -->
-        <div v-if="!isChatMode && isTitleMode" class="title-content" v-loading="titleLoading">
+        <!-- 内容列表模式 -->
+        <div v-if="!isChatMode && isScanMode" 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>
+            <span class="title-mode-badge">浏览模式</span>
           </div>
 
           <div v-if="!selectedKnowledge" class="empty-state">
@@ -158,7 +158,7 @@
           <div v-else-if="titleList.length === 0" class="empty-state">
             <i class="el-icon-notebook-2 empty-icon">
             </i>
-            <p>该知识库暂无标题</p>
+            <p>该知识库暂无内容</p>
           </div>
 
           <div v-else class="title-content-container">
@@ -518,7 +518,7 @@ export default {
       timer: null,
 
       // 标题和内容相关状态
-      isTitleMode: false, // 是否显示标题模式
+      isScanMode: false, // 是否显示浏览模式
       titleList: [], // 标题列表
       contentList: [], // 内容列表
       selectedTitle: null, // 选中的标题
@@ -527,9 +527,9 @@ export default {
     }
   },
   mounted() {
-    // 重置状态，确保页面刷新后不会停留在对话模式或标题模式
+    // 重置状态，确保页面刷新后不会停留在对话模式或浏览模式
     this.isChatMode = false;
-    this.isTitleMode = false;
+    this.isScanMode = false;
     this.chatMessages = [];
     this.chatInput = '';
     this.streamingStarted = false;
@@ -663,12 +663,12 @@ export default {
       }
 
       this.isChatMode = false;
-      this.isTitleMode = false;
+      this.isScanMode = false;
     },
 
-    /** 切换到标题模式 */
+    /** 切换到浏览模式 */
     switchToTitleMode() {
-      this.isTitleMode = true;
+      this.isScanMode = true;
       this.isChatMode = false;
       this.titleList = [];
       this.contentList = [];