分割文档方案优化

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

@@ -30,6 +30,7 @@ import org.apache.poi.hwpf.usermodel.Paragraph;
 import org.apache.poi.hwpf.usermodel.Range;
 import org.apache.poi.xwpf.usermodel.*;
 import org.apache.xmlbeans.XmlCursor;
+import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTPPrGeneral;
 import org.noear.solon.Solon;
 import org.noear.solon.ai.annotation.ToolMapping;
 import org.noear.solon.ai.chat.ChatModel;
@@ -321,6 +322,120 @@ public class McpServiceImpl implements IMcpService {
         }
     }
 
+    /**
+     * 获取DOCX段落的大纲级别
+     */
+    private int getDocxOutlineLevel(XWPFParagraph paragraph, XWPFDocument document) {
+        int level = 0;
+        String styleId = paragraph.getStyle();
+        if (styleId != null) {
+            XWPFStyle style = document.getStyles().getStyle(styleId);
+            if (style != null) {
+                CTPPrGeneral stylePPr = style.getCTStyle().getPPr();
+                if (stylePPr != null && stylePPr.getOutlineLvl() != null) {
+                    level = stylePPr.getOutlineLvl().getVal().intValue() + 1;
+                }
+            }
+        }
+        return level;
+    }
+
+    /**
+     * 获取DOC段落的大纲级别
+     */
+    private int getDocOutlineLevel(Paragraph paragraph) {
+        int level = 0;
+        short styleIndex = paragraph.getStyleIndex();
+        if (styleIndex >= 1 && styleIndex <= 9) {
+            level = styleIndex;
+        }
+        return level;
+    }
+
+    /**
+     * 检查DOCX段落是否是三级标题
+     */
+    private boolean isDocxLevel3Title(String paraText, XWPFDocument document) {
+        for (XWPFParagraph p : document.getParagraphs()) {
+            if (p.getText().trim().equals(paraText.trim())) {
+                return getDocxOutlineLevel(p, document) == 3;
+            }
+        }
+        return false;
+    }
+
+    /**
+     * 按三级标题分割DOCX内容
+     */
+    private void splitDocxByLevel3(String content, XWPFDocument document, List<TextSegment> segments) {
+        StringBuilder currentLevel3Content = new StringBuilder();
+        String[] paragraphs = content.split("\n");
+        for (String para : paragraphs) {
+            if (para.trim().isEmpty()) continue;
+
+            if (isDocxLevel3Title(para.trim(), document)) {
+                if (currentLevel3Content.length() != 0) {
+                    TextSegment segment = TextSegment.from(currentLevel3Content.toString());
+                    segments.add(segment);
+                    currentLevel3Content = new StringBuilder();
+                }
+                currentLevel3Content.append(para).append("\n");
+            } else {
+                if (currentLevel3Content.length() != 0) {
+                    currentLevel3Content.append(para).append("\n");
+                }
+            }
+        }
+        if (currentLevel3Content.length() != 0) {
+            TextSegment segment = TextSegment.from(currentLevel3Content.toString());
+            segments.add(segment);
+        }
+    }
+
+    /**
+     * 按三级标题分割DOC内容
+     */
+    private void splitDocByLevel3(String content, HWPFDocument document, List<TextSegment> segments) {
+        StringBuilder currentLevel3Content = new StringBuilder();
+        Range level2Range = document.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;
+
+            int pLevel = getDocOutlineLevel(p);
+            if (pLevel == 2 && paraText.equals(content.split("\n")[0].trim())) {
+                foundCurrentLevel2 = true;
+            }
+
+            if (foundCurrentLevel2) {
+                pLevel = getDocOutlineLevel(p);
+
+                if (pLevel == 3) {
+                    if (currentLevel3Content.length() != 0) {
+                        TextSegment segment = TextSegment.from(currentLevel3Content.toString());
+                        segments.add(segment);
+                        currentLevel3Content = new StringBuilder();
+                    }
+                    currentLevel3Content.append(paraText).append("\n");
+                } else if (pLevel == 2 && !paraText.equals(content.split("\n")[0].trim())) {
+                    break;
+                } else {
+                    if (currentLevel3Content.length() != 0 || pLevel == 2) {
+                        currentLevel3Content.append(paraText).append("\n");
+                    }
+                }
+            }
+        }
+
+        if (currentLevel3Content.length() != 0) {
+            TextSegment segment = TextSegment.from(currentLevel3Content.toString());
+            segments.add(segment);
+        }
+    }
+
     /**
      * 获取最低级别子标题列表
      */
@@ -337,27 +452,15 @@ public class McpServiceImpl implements IMcpService {
 
             for (XWPFParagraph  paragraph : document.getParagraphs()) {
                 String text = paragraph.getText();
-                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;
-
+                int level = getDocxOutlineLevel(paragraph, document);
+                if (level > 0) {
                     // 维护当前路径
                     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;
+                        int lastLevel = getDocxOutlineLevel(lastPara, document);
+                        if (lastLevel <= 0) {
+                            break;
+                        }
 
                         if (lastLevel >= level) {
                             currentPath.remove(currentPath.size() - 1);
@@ -397,15 +500,8 @@ public class McpServiceImpl implements IMcpService {
 
         // 检查后续段落
         XWPFParagraph nextP = doc.getParagraphs().get(index + 1);
-        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;
+        int nextLevel = getDocxOutlineLevel(nextP, doc);
+        if (nextLevel > 0) {
             // 遇到同级或更高级别标题，是叶子节点
             return nextLevel <= level; // 存在更低级别的标题，不是叶子节点
         }
@@ -421,14 +517,10 @@ public class McpServiceImpl implements IMcpService {
         try (XWPFDocument document = new XWPFDocument(fileInputStream)) {
             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 (styleName.equals("heading 3") || styleName.equals("heading 4")) {
-                        subTitles.append(text).append("\n");
-                    }
+                int level = getDocxOutlineLevel(paragraph, document);
+                // 判断主标题（大纲级别3或4，即原来的heading 3或heading 4）
+                if (level == 3 || level == 4) {
+                    subTitles.append(text).append("\n");
                 }
             }
         }
@@ -477,11 +569,9 @@ public class McpServiceImpl implements IMcpService {
      * 检索知识库
      */
     private List<TextSegment> splitDocument(File transferFile) throws IOException {
-        // 加载文档
         InputStream fileInputStream = new FileInputStream(transferFile);
         String filename = transferFile.getName().toLowerCase();
         if (filename.endsWith(".docx")) {
-            // 使用XWPFDocument处理DOCX文件，按二级标题分割（二级标题样式为2）
             List<TextSegment> segments = new ArrayList<>();
             try (XWPFDocument xwpfDocument = new XWPFDocument(fileInputStream)) {
                 StringBuilder currentLevel2Content = new StringBuilder();
@@ -491,145 +581,41 @@ public class McpServiceImpl implements IMcpService {
                     String text = paragraph.getText().trim();
                     if (text.isEmpty()) continue;
 
-                    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) {
-                                // 检查当前二级标题内容的字节数
-                                int level2Length = currentLevel2Content.toString().getBytes().length;
-                                if (level2Length <= 65535) {
-                                    // 字数不大于65535，按二级标题分割
-                                    TextSegment segment = TextSegment.from(currentLevel2Content.toString());
-                                    segments.add(segment);
-                                } else {
-                                    // 字数大于65535，按三级标题分割
-                                    // 将当前二级标题内容按三级标题重新分割
-                                    StringBuilder currentLevel3Content = new StringBuilder();
-                                    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) {
-                                                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) {
-                                                TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                                                segments.add(segment);
-                                                currentLevel3Content = new StringBuilder();
-                                            }
-                                            // 开始新的三级标题内容
-                                            currentLevel3Content.append(para).append("\n");
-                                        } else {
-                                            // 普通内容，添加到当前三级标题
-                                            if (currentLevel3Content.length() != 0) {
-                                                currentLevel3Content.append(para).append("\n");
-                                            }
-                                        }
-                                    }
-                                    // 保存最后一个三级标题内容
-                                    if (currentLevel3Content.length() != 0) {
-                                        TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                                        segments.add(segment);
-                                    }
-                                }
-                                currentLevel2Content = new StringBuilder();
-                            }
-                            // 开始新的二级标题内容
-                            currentLevel2Content.append(text).append("\n");
-                            inLevel2Section = true;
-                        }
-                        // 其他标题样式（包括三级标题）
-                        else {
-                            if (inLevel2Section) {
-                                currentLevel2Content.append(text).append("\n");
+                    int paraLevel = getDocxOutlineLevel(paragraph, xwpfDocument);
+
+                    if (paraLevel == 2) {
+                        if (currentLevel2Content.length() != 0) {
+                            int level2Length = currentLevel2Content.toString().getBytes().length;
+                            if (level2Length <= 65535) {
+                                TextSegment segment = TextSegment.from(currentLevel2Content.toString());
+                                segments.add(segment);
+                            } else {
+                                splitDocxByLevel3(currentLevel2Content.toString(), xwpfDocument, segments);
                             }
+                            currentLevel2Content = new StringBuilder();
                         }
-                    }
-                    else {
-                        // 普通内容段落
+                        currentLevel2Content.append(text).append("\n");
+                        inLevel2Section = true;
+                    } else {
                         if (inLevel2Section) {
                             currentLevel2Content.append(text).append("\n");
                         }
                     }
                 }
 
-                // 保存最后一个二级标题内容
                 if (currentLevel2Content.length() != 0) {
-                    // 检查字节数
                     int level2Length = currentLevel2Content.toString().getBytes().length;
                     if (level2Length <= 65535) {
-                        // 字数不大于65535，按二级标题分割
                         TextSegment segment = TextSegment.from(currentLevel2Content.toString());
                         segments.add(segment);
                     } else {
-                        // 字数大于65535，按三级标题分割
-                        StringBuilder currentLevel3Content = new StringBuilder();
-                        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) {
-                                    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) {
-                                    TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                                    segments.add(segment);
-                                    currentLevel3Content = new StringBuilder();
-                                }
-                                // 开始新的三级标题内容
-                                currentLevel3Content.append(para).append("\n");
-                            } else {
-                                // 普通内容，添加到当前三级标题
-                                if (currentLevel3Content.length() != 0) {
-                                    currentLevel3Content.append(para).append("\n");
-                                }
-                            }
-                        }
-                        // 保存最后一个三级标题内容
-                        if (currentLevel3Content.length() != 0) {
-                            TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                            segments.add(segment);
-                        }
+                        splitDocxByLevel3(currentLevel2Content.toString(), xwpfDocument, segments);
                     }
                 }
             }
             return segments;
         }
         else if (filename.endsWith(".doc")) {
-            // 使用HWPFDocument处理DOC文件
             List<TextSegment> segments = new ArrayList<>();
             try (HWPFDocument hwpfDocument = new HWPFDocument(fileInputStream)) {
                 StringBuilder currentLevel2Content = new StringBuilder();
@@ -641,134 +627,35 @@ public class McpServiceImpl implements IMcpService {
                     String text = paragraph.text().trim();
                     if (text.isEmpty()) continue;
 
-                    // 获取段落样式索引
-                    short styleIndex = paragraph.getStyleIndex();
+                    int paraLevel = getDocOutlineLevel(paragraph);
 
-                    if (styleIndex == 2) {
-                        // 二级标题，开始新的分段
-                        // 保存当前二级标题下的内容
+                    if (paraLevel == 2) {
                         if (currentLevel2Content.length() != 0) {
-                            // 检查当前二级标题内容的字节数
                             int level2Length = currentLevel2Content.toString().getBytes().length;
                             if (level2Length <= 65535) {
-                                // 字数不大于65535，按二级标题分割
                                 TextSegment segment = TextSegment.from(currentLevel2Content.toString());
                                 segments.add(segment);
                             } else {
-                                // 字数大于65535，按三级标题分割
-                                // 将当前二级标题内容按三级标题重新分割
-                                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) {
-                                            // 三级标题
-                                            // 保存当前三级标题内容
-                                            if (currentLevel3Content.length() != 0) {
-                                                TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                                                segments.add(segment);
-                                                currentLevel3Content = new StringBuilder();
-                                            }
-                                            // 开始新的三级标题内容
-                                            currentLevel3Content.append(paraText).append("\n");
-                                        } else if (paraStyleIndex == 2 && !paraText.equals(currentLevel2Content.toString().split("\n")[0].trim())) {
-                                            // 下一个二级标题，结束当前二级标题的处理
-                                            break;
-                                        } else {
-                                            // 普通内容或当前二级标题
-                                            if (currentLevel3Content.length() != 0 || paraStyleIndex == 2) {
-                                                currentLevel3Content.append(paraText).append("\n");
-                                            }
-                                        }
-                                    }
-                                }
-
-                                // 保存最后一个三级标题内容
-                                if (currentLevel3Content.length() != 0) {
-                                    TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                                    segments.add(segment);
-                                }
+                                splitDocByLevel3(currentLevel2Content.toString(), hwpfDocument, segments);
                             }
                             currentLevel2Content = new StringBuilder();
                         }
-                        // 开始新的二级标题内容
                         currentLevel2Content.append(text).append("\n");
                         inLevel2Section = true;
                     } else {
-                        // 非二级标题或普通内容段落
                         if (inLevel2Section) {
                             currentLevel2Content.append(text).append("\n");
                         }
                     }
                 }
 
-                // 保存最后一个二级标题内容
                 if (currentLevel2Content.length() != 0) {
-                    // 检查字节数
                     int level2Length = currentLevel2Content.toString().getBytes().length;
                     if (level2Length <= 65535) {
-                        // 字数不大于65535，按二级标题分割
                         TextSegment segment = TextSegment.from(currentLevel2Content.toString());
                         segments.add(segment);
                     } else {
-                        // 字数大于65535，按三级标题分割
-                        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) {
-                                    // 三级标题
-                                    // 保存当前三级标题内容
-                                    if (currentLevel3Content.length() != 0) {
-                                        TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                                        segments.add(segment);
-                                        currentLevel3Content = new StringBuilder();
-                                    }
-                                    // 开始新的三级标题内容
-                                    currentLevel3Content.append(paraText).append("\n");
-                                } else if (paraStyleIndex == 2 && !paraText.equals(currentLevel2Content.toString().split("\n")[0].trim())) {
-                                    // 下一个二级标题，结束当前二级标题的处理
-                                    break;
-                                } else {
-                                    // 普通内容或当前二级标题
-                                    if (currentLevel3Content.length() != 0 || paraStyleIndex == 2) {
-                                        currentLevel3Content.append(paraText).append("\n");
-                                    }
-                                }
-                            }
-                        }
-
-                        // 保存最后一个三级标题内容
-                        if (currentLevel3Content.length() != 0) {
-                            TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                            segments.add(segment);
-                        }
+                        splitDocByLevel3(currentLevel2Content.toString(), hwpfDocument, segments);
                     }
                 }
             }

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

@@ -14,7 +14,7 @@ import java.util.List;
 
 /**
  * cmc知识库Controller
- * 
+ *
  * @author cmc
  * @date 2025-04-08
  */
@@ -142,4 +142,14 @@ public class KnowLedgeController extends BaseController
         return success(contents);
     }
 
+    /**
+     * 根据文件名获取标题列表
+     */
+    @GetMapping("/listTilesByFile")
+    public AjaxResult listTilesByFile(String collectionName, String fileName)
+    {
+        List<String> titles = milvusService.listTilesByFile(collectionName, fileName);
+        return success(titles);
+    }
+
 }

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

@@ -55,4 +55,9 @@ public interface IMilvusService {
      * 根据title名查询相关content
      */
     public JSONArray listByTitle(String collectionName, String title);
+
+    /**
+     * 根据文件名获取标题列表
+     */
+    public List<String> listTilesByFile(String collectionName, String fileName);
 }

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

@@ -44,6 +44,7 @@ import org.noear.solon.ai.chat.ChatSession;
 import org.noear.solon.ai.chat.message.ChatMessage;
 import org.noear.solon.ai.chat.prompt.Prompt;
 import org.noear.solon.ai.chat.session.InMemoryChatSession;
+import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTPPrGeneral;
 import org.reactivestreams.Publisher;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
@@ -422,26 +423,18 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
         try (XWPFDocument document = new XWPFDocument(fileInputStream)) {
             for (XWPFParagraph paragraph : document.getParagraphs()) {
                 String text = paragraph.getText().trim();
-                if (paragraph.getStyleID() != null) {
-                    String styleId = paragraph.getStyleID();
-                    XWPFStyle style = document.getStyles().getStyle(styleId);
-                    String styleName = style.getName();
-                    // 判断主标题
-                    if (styleName.equals("heading 3") &&
-                            text.contains(question)) {
-                        inTargetSection = true;
-                        continue;
-                    }
+                int level = getDocxOutlineLevel(paragraph, document);
+                if (level == 3 && text.contains(question)) {
+                    inTargetSection = true;
+                    continue;
+                }
 
-                    // 如果已经在目标节中，收集标题3级别的子标题
-                    if (inTargetSection) {
-                        if (styleName.equals("heading 4")) {
-                            subTitles.add(text);
-                        }
-                        // 遇到下一个Heading2则退出
-                        else if (styleName.equals("heading 3")) {
-                            break;
-                        }
+                if (inTargetSection) {
+                    if (level == 4) {
+                        subTitles.add(text);
+                    }
+                    else if (level == 3) {
+                        break;
                     }
                 }
             }
@@ -501,14 +494,9 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
                 for (XWPFParagraph paragraph : document.getParagraphs()) {
                     String text = paragraph.getText().trim();
                     if (text.isEmpty()) continue;
-                    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");
-                        }
+                    int outlineLevel = getDocxOutlineLevel(paragraph, document);
+                    if (outlineLevel == 2 || outlineLevel == 3) {
+                        titlesBuilder.append(text).append("\n");
                     }
                 }
             }
@@ -519,8 +507,8 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
                     Paragraph paragraph = range.getParagraph(i);
                     String text = paragraph.text().trim();
                     if (text.isEmpty()) continue;
-                    short styleIndex = paragraph.getStyleIndex();
-                    if (styleIndex == 2 || styleIndex == 3) {
+                    int outlineLevel = getDocOutlineLevel(paragraph);
+                    if (outlineLevel == 2 || outlineLevel == 3) {
                         titlesBuilder.append(text).append("\n");
                     }
                 }
@@ -529,6 +517,132 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
         return titlesBuilder.toString();
     }
 
+    /**
+     * 获取DOCX段落的大纲级别
+     */
+    private int getDocxOutlineLevel(XWPFParagraph paragraph, XWPFDocument document) {
+        int level = 0;
+        String styleId = paragraph.getStyleID();
+        if (styleId != null) {
+            XWPFStyle style = document.getStyles().getStyle(styleId);
+            if (style != null) {
+                CTPPrGeneral stylePPr = style.getCTStyle().getPPr();
+                if (stylePPr != null && stylePPr.getOutlineLvl() != null) {
+                    level = stylePPr.getOutlineLvl().getVal().intValue() + 1;
+                }
+            }
+        }
+        return level;
+    }
+
+    /**
+     * 获取DOC段落的大纲级别
+     */
+    private int getDocOutlineLevel(Paragraph paragraph) {
+        int level = 0;
+        short styleIndex = paragraph.getStyleIndex();
+        if (styleIndex >= 1 && styleIndex <= 9) {
+            level = styleIndex;
+        }
+        return level;
+    }
+
+    /**
+     * 检查DOCX段落是否是三级标题
+     */
+    private boolean isDocxLevel3Title(String paraText, XWPFDocument document) {
+        for (XWPFParagraph p : document.getParagraphs()) {
+            if (p.getText().trim().equals(paraText.trim())) {
+                return getDocxOutlineLevel(p, document) == 3;
+            }
+        }
+        return false;
+    }
+
+    /**
+     * 按三级标题分割DOCX内容
+     */
+    private void splitDocxByLevel3(String content, XWPFDocument document, List<TextSegment> segments) {
+        StringBuilder currentLevel3Content = new StringBuilder();
+        String[] paragraphs = content.split("\n");
+        for (String para : paragraphs) {
+            if (para.trim().isEmpty()) continue;
+
+            if (isDocxLevel3Title(para.trim(), document)) {
+                // 保存当前三级标题内容
+                if (currentLevel3Content.length() != 0) {
+                    TextSegment segment = TextSegment.from(currentLevel3Content.toString());
+                    segments.add(segment);
+                    currentLevel3Content = new StringBuilder();
+                }
+                // 开始新的三级标题内容
+                currentLevel3Content.append(para).append("\n");
+            } else {
+                // 普通内容，添加到当前三级标题
+                if (currentLevel3Content.length() != 0) {
+                    currentLevel3Content.append(para).append("\n");
+                }
+            }
+        }
+        // 保存最后一个三级标题内容
+        if (currentLevel3Content.length() != 0) {
+            TextSegment segment = TextSegment.from(currentLevel3Content.toString());
+            segments.add(segment);
+        }
+    }
+
+    /**
+     * 按三级标题分割DOC内容
+     */
+    private void splitDocByLevel3(String content, HWPFDocument document, List<TextSegment> segments) {
+        StringBuilder currentLevel3Content = new StringBuilder();
+        Range level2Range = document.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;
+
+            // 找到当前二级标题的开始
+            int pLevel = getDocOutlineLevel(p);
+            if (pLevel == 2 && paraText.equals(content.split("\n")[0].trim())) {
+                foundCurrentLevel2 = true;
+            }
+
+            if (foundCurrentLevel2) {
+                // 重新获取当前段落的大纲级别
+                pLevel = getDocOutlineLevel(p);
+
+                if (pLevel == 3) {
+                    // 三级标题
+                    // 保存当前三级标题内容
+                    if (currentLevel3Content.length() != 0) {
+                        TextSegment segment = TextSegment.from(currentLevel3Content.toString());
+                        segments.add(segment);
+                        currentLevel3Content = new StringBuilder();
+                    }
+                    // 开始新的三级标题内容
+                    currentLevel3Content.append(paraText).append("\n");
+                } else if (pLevel == 2 && !paraText.equals(content.split("\n")[0].trim())) {
+                    // 下一个二级标题，结束当前二级标题的处理
+                    break;
+                } else {
+                    // 普通内容或当前二级标题
+                    if (currentLevel3Content.length() != 0 || pLevel == 2) {
+                        currentLevel3Content.append(paraText).append("\n");
+                    }
+                }
+            }
+        }
+
+        // 保存最后一个三级标题内容
+        if (currentLevel3Content.length() != 0) {
+            TextSegment segment = TextSegment.from(currentLevel3Content.toString());
+            segments.add(segment);
+        }
+    }
+
     /**
      * 检索知识库
      */
@@ -538,7 +652,7 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
         InputStream fileInputStream = new FileInputStream(transferFile);
         String filename = transferFile.getName().toLowerCase();
         if (filename.endsWith(".docx")) {
-            // 使用XWPFDocument处理DOCX文件，按二级标题分割（二级标题样式为2）
+            // 使用XWPFDocument处理DOCX文件，按段落层级分割
             List<TextSegment> segments = new ArrayList<>();
             try (XWPFDocument xwpfDocument = new XWPFDocument(fileInputStream)) {
                 StringBuilder currentLevel2Content = new StringBuilder();
@@ -548,82 +662,30 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
                     String text = paragraph.getText().trim();
                     if (text.isEmpty()) continue;
 
-                    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) {
-                                // 检查当前二级标题内容的字节数
-                                int level2Length = currentLevel2Content.toString().getBytes().length;
-                                if (level2Length <= 65535) {
-                                    // 字数不大于65535，按二级标题分割
-                                    TextSegment segment = TextSegment.from(currentLevel2Content.toString());
-                                    segments.add(segment);
-                                } else {
-                                    // 字数大于65535，按三级标题分割
-                                    // 将当前二级标题内容按三级标题重新分割
-                                    StringBuilder currentLevel3Content = new StringBuilder();
-                                    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) {
-                                                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) {
-                                                TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                                                segments.add(segment);
-                                                currentLevel3Content = new StringBuilder();
-                                            }
-                                            // 开始新的三级标题内容
-                                            currentLevel3Content.append(para).append("\n");
-                                        } else {
-                                            // 普通内容，添加到当前三级标题
-                                            if (currentLevel3Content.length() != 0) {
-                                                currentLevel3Content.append(para).append("\n");
-                                            }
-                                        }
-                                    }
-                                    // 保存最后一个三级标题内容
-                                    if (currentLevel3Content.length() != 0) {
-                                        TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                                        segments.add(segment);
-                                    }
-                                }
-                                currentLevel2Content = new StringBuilder();
-                            }
-                            // 开始新的二级标题内容
-                            currentLevel2Content.append(text).append("\n");
-                            inLevel2Section = true;
-                        }
-                        // 其他标题样式（包括三级标题）
-                        else {
-                            if (inLevel2Section) {
-                                currentLevel2Content.append(text).append("\n");
+                    int paraLevel = getDocxOutlineLevel(paragraph, xwpfDocument);
+
+                    // 二级标题（大纲级别2）开始新的分段
+                    if (paraLevel == 2) {
+                        // 保存当前二级标题下的内容
+                        if (currentLevel2Content.length() != 0) {
+                            // 检查当前二级标题内容的字节数
+                            int level2Length = currentLevel2Content.toString().getBytes().length;
+                            if (level2Length <= 65535) {
+                                // 字数不大于65535，按二级标题分割
+                                TextSegment segment = TextSegment.from(currentLevel2Content.toString());
+                                segments.add(segment);
+                            } else {
+                                // 字数大于65535，按三级标题分割
+                                splitDocxByLevel3(currentLevel2Content.toString(), xwpfDocument, segments);
                             }
+                            currentLevel2Content = new StringBuilder();
                         }
+                        // 开始新的二级标题内容
+                        currentLevel2Content.append(text).append("\n");
+                        inLevel2Section = true;
                     }
+                    // 其他层级（包括三级标题）
                     else {
-                        // 普通内容段落
                         if (inLevel2Section) {
                             currentLevel2Content.append(text).append("\n");
                         }
@@ -640,53 +702,14 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
                         segments.add(segment);
                     } else {
                         // 字数大于65535，按三级标题分割
-                        StringBuilder currentLevel3Content = new StringBuilder();
-                        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) {
-                                    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) {
-                                    TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                                    segments.add(segment);
-                                    currentLevel3Content = new StringBuilder();
-                                }
-                                // 开始新的三级标题内容
-                                currentLevel3Content.append(para).append("\n");
-                            } else {
-                                // 普通内容，添加到当前三级标题
-                                if (currentLevel3Content.length() != 0) {
-                                    currentLevel3Content.append(para).append("\n");
-                                }
-                            }
-                        }
-                        // 保存最后一个三级标题内容
-                        if (currentLevel3Content.length() != 0) {
-                            TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                            segments.add(segment);
-                        }
+                        splitDocxByLevel3(currentLevel2Content.toString(), xwpfDocument, segments);
                     }
                 }
             }
             return segments;
         }
         else if (filename.endsWith(".doc")) {
-            // 使用HWPFDocument处理DOC文件
+            // 使用HWPFDocument处理DOC文件，按段落层级分割
             List<TextSegment> segments = new ArrayList<>();
             try (HWPFDocument hwpfDocument = new HWPFDocument(fileInputStream)) {
                 StringBuilder currentLevel2Content = new StringBuilder();
@@ -698,10 +721,9 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
                     String text = paragraph.text().trim();
                     if (text.isEmpty()) continue;
 
-                    // 获取段落样式索引
-                    short styleIndex = paragraph.getStyleIndex();
+                    int paraLevel = getDocOutlineLevel(paragraph);
 
-                    if (styleIndex == 2) {
+                    if (paraLevel == 2) {
                         // 二级标题，开始新的分段
                         // 保存当前二级标题下的内容
                         if (currentLevel2Content.length() != 0) {
@@ -713,51 +735,7 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
                                 segments.add(segment);
                             } else {
                                 // 字数大于65535，按三级标题分割
-                                // 将当前二级标题内容按三级标题重新分割
-                                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) {
-                                            // 三级标题
-                                            // 保存当前三级标题内容
-                                            if (currentLevel3Content.length() != 0) {
-                                                TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                                                segments.add(segment);
-                                                currentLevel3Content = new StringBuilder();
-                                            }
-                                            // 开始新的三级标题内容
-                                            currentLevel3Content.append(paraText).append("\n");
-                                        } else if (paraStyleIndex == 2 && !paraText.equals(currentLevel2Content.toString().split("\n")[0].trim())) {
-                                            // 下一个二级标题，结束当前二级标题的处理
-                                            break;
-                                        } else {
-                                            // 普通内容或当前二级标题
-                                            if (currentLevel3Content.length() != 0 || paraStyleIndex == 2) {
-                                                currentLevel3Content.append(paraText).append("\n");
-                                            }
-                                        }
-                                    }
-                                }
-
-                                // 保存最后一个三级标题内容
-                                if (currentLevel3Content.length() != 0) {
-                                    TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                                    segments.add(segment);
-                                }
+                                splitDocByLevel3(currentLevel2Content.toString(), hwpfDocument, segments);
                             }
                             currentLevel2Content = new StringBuilder();
                         }
@@ -782,50 +760,7 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
                         segments.add(segment);
                     } else {
                         // 字数大于65535，按三级标题分割
-                        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) {
-                                    // 三级标题
-                                    // 保存当前三级标题内容
-                                    if (currentLevel3Content.length() != 0) {
-                                        TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                                        segments.add(segment);
-                                        currentLevel3Content = new StringBuilder();
-                                    }
-                                    // 开始新的三级标题内容
-                                    currentLevel3Content.append(paraText).append("\n");
-                                } else if (paraStyleIndex == 2 && !paraText.equals(currentLevel2Content.toString().split("\n")[0].trim())) {
-                                    // 下一个二级标题，结束当前二级标题的处理
-                                    break;
-                                } else {
-                                    // 普通内容或当前二级标题
-                                    if (currentLevel3Content.length() != 0 || paraStyleIndex == 2) {
-                                        currentLevel3Content.append(paraText).append("\n");
-                                    }
-                                }
-                            }
-                        }
-
-                        // 保存最后一个三级标题内容
-                        if (currentLevel3Content.length() != 0) {
-                            TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                            segments.add(segment);
-                        }
+                        splitDocByLevel3(currentLevel2Content.toString(), hwpfDocument, segments);
                     }
                 }
             }

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

@@ -304,6 +304,39 @@ public class MilvusServiceImpl implements IMilvusService {
         return titleList.stream().distinct().collect(Collectors.toList());
     }
 
+    /**
+     * 根据文件名获取标题列表
+     */
+    @Override
+    public List<String> listTilesByFile(String collectionName, String fileName) {
+        List<String> titleList = new ArrayList<>();
+        LoadCollectionReq loadCollectionReq = LoadCollectionReq.builder()
+                .collectionName(collectionName)
+                .build();
+        milvusClient.loadCollection(loadCollectionReq);
+        QueryReq queryParam = QueryReq.builder()
+                .collectionName(collectionName)
+                .filter(String.format("id > 0 && title != \"\" && file_name == \"%s\"", fileName))
+                .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
      */

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

@@ -42,6 +42,10 @@ import org.noear.solon.ai.chat.ChatResponse;
 import org.noear.solon.ai.chat.ChatSession;
 import org.noear.solon.ai.chat.message.ChatMessage;
 import org.noear.solon.ai.chat.session.InMemoryChatSession;
+import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTDecimalNumber;
+import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTP;
+import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTPPr;
+import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTPPrGeneral;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
@@ -205,9 +209,24 @@ public class CmcAgentServiceImpl implements ICmcAgentService
             String question = "工作大纲/工作范围/招标范围/服务范围";
             String chapters = generateAnswerWithDocument(profilePath + "/" + file.getOriginalFilename(),
                     RuoYiConfig.getProfile() + outputFilename, question);
+
+            // 生成详细目录结构（包含二三级标题）
+            String detailedDirectory = generateDetailedDirectory(profilePath + "/" + file.getOriginalFilename());
+
+            // 分析项目概况
+            String projectOverview = analyzeProjectOverview(profilePath + "/" + file.getOriginalFilename());
+
+            // 分析评分要求
+            String scoringRequirements = analyzeScoringRequirements(profilePath + "/" + file.getOriginalFilename());
+
             message = "好的，我已经收到您上传的招标文件。以下为根据招标文件生成的章节大纲：\n\n"+ chapters + "\n\n" +
                     "若您对章节标题有异议，请打开" + "【<a href='/profile" + outputFilename + "'> 技术文件 " + "</a>】" + "进行修改，后续将根据修改后的章节标题，帮您生成对应章节内容。\n\n" +
                     "思考时间可能较长，请耐心等待！\n";
+
+            // 将分析结果添加到返回对象中
+            jsonObject.put("detailedDirectory", detailedDirectory);
+            jsonObject.put("projectOverview", projectOverview);
+            jsonObject.put("scoringRequirements", scoringRequirements);
         }
         else if (agentName.contains("检查")) {
             message = generateAnswerWithDocumentContent(profilePath + "/" + file.getOriginalFilename());
@@ -599,6 +618,139 @@ public class CmcAgentServiceImpl implements ICmcAgentService
         return writeChapters(sb.toString(), templatePath);
     }
 
+    /**
+     * 生成详细目录结构（包含二三级标题）
+     */
+    public String generateDetailedDirectory(String uploadFilePath) throws IOException {
+        processValue = "生成详细目录结构中: 0%";
+        StringBuilder sb = new StringBuilder("招标文件内容：\n\n");
+        File profilePath = new File(uploadFilePath);
+        InMemoryEmbeddingStore<TextSegment> embeddingStore = new InMemoryEmbeddingStore<>();
+        List<TextSegment> segments = splitDocument(profilePath, 300, 50);
+        List<Embedding> embeddings = embeddingModel.embedAll(segments).content();
+        embeddingStore.addAll(embeddings, segments);
+
+        // 搜索与目录相关的内容
+        String directoryQuery = "目录结构、章节划分、文件结构";
+        Embedding queryEmbedding = embeddingModel.embed(directoryQuery).content();
+        EmbeddingSearchRequest embeddingSearchRequest = EmbeddingSearchRequest.builder()
+                .queryEmbedding(queryEmbedding)
+                .minScore(0.7)
+                .build();
+        List<EmbeddingMatch<TextSegment>> results = embeddingStore.search(embeddingSearchRequest).matches();
+        results.sort(Comparator.comparingDouble(EmbeddingMatch<TextSegment>::score).reversed());
+
+        for (EmbeddingMatch<TextSegment> embeddingMatch : results) {
+            String content = embeddingMatch.embedded().toString();
+            sb.append(content).append("\n\n");
+        }
+
+        // 生成详细目录结构
+        sb.append("请基于上述招标文件内容，生成详细的目录结构，包含一级、二级和三级标题，严格按以下格式输出：\n")
+                .append("1. 一级标题\n")
+                .append("  1.1 二级标题\n")
+                .append("    1.1.1 三级标题\n")
+                .append("    1.1.2 三级标题\n")
+                .append("  1.2 二级标题\n")
+                .append("2. 一级标题\n")
+                .append("  2.1 二级标题\n")
+                .append("    2.1.1 三级标题\n")
+                .append("......\n");
+
+        processValue = "生成详细目录结构中: 50%";
+        String directory = generateAnswer(sb.toString());
+        processValue = "生成详细目录结构中: 100%";
+        return directory;
+    }
+
+    /**
+     * 分析项目概况
+     */
+    public String analyzeProjectOverview(String uploadFilePath) throws IOException {
+        processValue = "分析项目概况中: 0%";
+        StringBuilder sb = new StringBuilder("招标文件内容：\n\n");
+        File profilePath = new File(uploadFilePath);
+        InMemoryEmbeddingStore<TextSegment> embeddingStore = new InMemoryEmbeddingStore<>();
+        List<TextSegment> segments = splitDocument(profilePath, 300, 50);
+        List<Embedding> embeddings = embeddingModel.embedAll(segments).content();
+        embeddingStore.addAll(embeddings, segments);
+
+        // 搜索与项目概况相关的内容
+        String overviewQuery = "项目概况、项目基本信息、项目背景、项目目标、项目范围";
+        Embedding queryEmbedding = embeddingModel.embed(overviewQuery).content();
+        EmbeddingSearchRequest embeddingSearchRequest = EmbeddingSearchRequest.builder()
+                .queryEmbedding(queryEmbedding)
+                .minScore(0.7)
+                .build();
+        List<EmbeddingMatch<TextSegment>> results = embeddingStore.search(embeddingSearchRequest).matches();
+        results.sort(Comparator.comparingDouble(EmbeddingMatch<TextSegment>::score).reversed());
+
+        for (EmbeddingMatch<TextSegment> embeddingMatch : results) {
+            String content = embeddingMatch.embedded().toString();
+            sb.append(content).append("\n\n");
+        }
+
+        // 分析项目概况
+        sb.append("请基于上述招标文件内容，分析项目概况，包括但不限于以下内容：\n")
+                .append("1. 项目名称\n")
+                .append("2. 项目类型\n")
+                .append("3. 项目预算\n")
+                .append("4. 项目周期\n")
+                .append("5. 项目地点\n")
+                .append("6. 招标人\n")
+                .append("7. 项目背景\n")
+                .append("8. 项目目标\n")
+                .append("9. 项目范围\n")
+                .append("请以清晰的结构输出分析结果。\n");
+
+        processValue = "分析项目概况中: 50%";
+        String overview = generateAnswer(sb.toString());
+        processValue = "分析项目概况中: 100%";
+        return overview;
+    }
+
+    /**
+     * 分析评分要求
+     */
+    public String analyzeScoringRequirements(String uploadFilePath) throws IOException {
+        processValue = "分析评分要求中: 0%";
+        StringBuilder sb = new StringBuilder("招标文件内容：\n\n");
+        File profilePath = new File(uploadFilePath);
+        InMemoryEmbeddingStore<TextSegment> embeddingStore = new InMemoryEmbeddingStore<>();
+        List<TextSegment> segments = splitDocument(profilePath, 300, 50);
+        List<Embedding> embeddings = embeddingModel.embedAll(segments).content();
+        embeddingStore.addAll(embeddings, segments);
+
+        // 搜索与评分要求相关的内容
+        String scoringQuery = "评分要求、评分标准、评标办法、打分规则";
+        Embedding queryEmbedding = embeddingModel.embed(scoringQuery).content();
+        EmbeddingSearchRequest embeddingSearchRequest = EmbeddingSearchRequest.builder()
+                .queryEmbedding(queryEmbedding)
+                .minScore(0.7)
+                .build();
+        List<EmbeddingMatch<TextSegment>> results = embeddingStore.search(embeddingSearchRequest).matches();
+        results.sort(Comparator.comparingDouble(EmbeddingMatch<TextSegment>::score).reversed());
+
+        for (EmbeddingMatch<TextSegment> embeddingMatch : results) {
+            String content = embeddingMatch.embedded().toString();
+            sb.append(content).append("\n\n");
+        }
+
+        // 分析评分要求
+        sb.append("请基于上述招标文件内容，分析评分要求，包括但不限于以下内容：\n")
+                .append("1. 评分项及权重\n")
+                .append("2. 技术方案评分标准\n")
+                .append("3. 商务方案评分标准\n")
+                .append("4. 服务方案评分标准\n")
+                .append("5. 其他评分项\n")
+                .append("请以清晰的结构输出分析结果，包括各项的具体分值和评分细则。\n");
+
+        processValue = "分析评分要求中: 50%";
+        String scoring = generateAnswer(sb.toString());
+        processValue = "分析评分要求中: 100%";
+        return scoring;
+    }
+
     /**
      * 编排章节
      * @return
@@ -837,6 +989,24 @@ public class CmcAgentServiceImpl implements ICmcAgentService
         }
     }
 
+    /**
+     * 获取DOCX段落的大纲级别
+     */
+    private int getDocxOutlineLevel(XWPFParagraph paragraph, XWPFDocument document) {
+        int level = 0;
+        String styleId = paragraph.getStyle();
+        if (styleId != null) {
+            XWPFStyle style = document.getStyles().getStyle(styleId);
+            if (style != null) {
+                CTPPrGeneral stylePPr = style.getCTStyle().getPPr();
+                if (stylePPr != null && stylePPr.getOutlineLvl() != null) {
+                    level = stylePPr.getOutlineLvl().getVal().intValue() + 1;
+                }
+            }
+        }
+        return level;
+    }
+
     /**
      * 获取最低级别子标题列表
      */
@@ -853,27 +1023,15 @@ public class CmcAgentServiceImpl implements ICmcAgentService
 
             for (XWPFParagraph  paragraph : document.getParagraphs()) {
                 String text = paragraph.getText();
-                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;
-
+                int level = getDocxOutlineLevel(paragraph, document);
+                if (level > 0) {
                     // 维护当前路径
                     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;
+                        int lastLevel = getDocxOutlineLevel(lastPara, document);
+                        if (lastLevel <= 0) {
+                            break;
+                        }
 
                         if (lastLevel >= level) {
                             currentPath.remove(currentPath.size() - 1);
@@ -915,15 +1073,8 @@ public class CmcAgentServiceImpl implements ICmcAgentService
 
         // 检查后续段落
         XWPFParagraph nextP = doc.getParagraphs().get(index + 1);
-        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;
+        int nextLevel = getDocxOutlineLevel(nextP, doc);
+        if (nextLevel > 0) {
             // 遇到同级或更高级别标题，是叶子节点
             return nextLevel <= level; // 存在更低级别的标题，不是叶子节点
         }
@@ -939,29 +1090,119 @@ public class CmcAgentServiceImpl implements ICmcAgentService
         try (XWPFDocument document = new XWPFDocument(fileInputStream)) {
             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 (styleName.equals("heading 3") || styleName.equals("heading 4")) {
-                        subTitles.append(text).append("\n");
-                    }
+                int level = getDocxOutlineLevel(paragraph, document);
+                // 判断主标题（大纲级别3或4，即原来的heading 3或heading 4）
+                if (level == 3 || level == 4) {
+                    subTitles.append(text).append("\n");
                 }
             }
         }
         return subTitles.toString();
     }
 
+    /**
+     * 获取DOC段落的大纲级别
+     */
+    private int getDocOutlineLevel(Paragraph paragraph) {
+        int level = 0;
+        short styleIndex = paragraph.getStyleIndex();
+        if (styleIndex >= 1 && styleIndex <= 9) {
+            level = styleIndex;
+        }
+        return level;
+    }
+
+    /**
+     * 检查DOCX段落是否是三级标题
+     */
+    private boolean isDocxLevel3Title(String paraText, XWPFDocument document) {
+        for (XWPFParagraph p : document.getParagraphs()) {
+            if (p.getText().trim().equals(paraText.trim())) {
+                return getDocxOutlineLevel(p, document) == 3;
+            }
+        }
+        return false;
+    }
+
+    /**
+     * 按三级标题分割DOCX内容
+     */
+    private void splitDocxByLevel3(String content, XWPFDocument document, List<TextSegment> segments) {
+        StringBuilder currentLevel3Content = new StringBuilder();
+        String[] paragraphs = content.split("\n");
+        for (String para : paragraphs) {
+            if (para.trim().isEmpty()) continue;
+
+            if (isDocxLevel3Title(para.trim(), document)) {
+                if (currentLevel3Content.length() != 0) {
+                    TextSegment segment = TextSegment.from(currentLevel3Content.toString());
+                    segments.add(segment);
+                    currentLevel3Content = new StringBuilder();
+                }
+                currentLevel3Content.append(para).append("\n");
+            } else {
+                if (currentLevel3Content.length() != 0) {
+                    currentLevel3Content.append(para).append("\n");
+                }
+            }
+        }
+        if (currentLevel3Content.length() != 0) {
+            TextSegment segment = TextSegment.from(currentLevel3Content.toString());
+            segments.add(segment);
+        }
+    }
+
+    /**
+     * 按三级标题分割DOC内容
+     */
+    private void splitDocByLevel3(String content, HWPFDocument document, List<TextSegment> segments) {
+        StringBuilder currentLevel3Content = new StringBuilder();
+        Range level2Range = document.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;
+
+            int pLevel = getDocOutlineLevel(p);
+            if (pLevel == 2 && paraText.equals(content.split("\n")[0].trim())) {
+                foundCurrentLevel2 = true;
+            }
+
+            if (foundCurrentLevel2) {
+                pLevel = getDocOutlineLevel(p);
+
+                if (pLevel == 3) {
+                    if (currentLevel3Content.length() != 0) {
+                        TextSegment segment = TextSegment.from(currentLevel3Content.toString());
+                        segments.add(segment);
+                        currentLevel3Content = new StringBuilder();
+                    }
+                    currentLevel3Content.append(paraText).append("\n");
+                } else if (pLevel == 2 && !paraText.equals(content.split("\n")[0].trim())) {
+                    break;
+                } else {
+                    if (currentLevel3Content.length() != 0 || pLevel == 2) {
+                        currentLevel3Content.append(paraText).append("\n");
+                    }
+                }
+            }
+        }
+
+        if (currentLevel3Content.length() != 0) {
+            TextSegment segment = TextSegment.from(currentLevel3Content.toString());
+            segments.add(segment);
+        }
+    }
+
     /**
      * 分割文档
      */
     private List<TextSegment> splitDocument(File transferFile, int maxSegmentSizeInChars, int maxOverlapSizeInChars) throws IOException {
-        // 加载文档
         InputStream fileInputStream = new FileInputStream(transferFile);
         String filename = transferFile.getName().toLowerCase();
         if (filename.endsWith(".docx")) {
-            // 使用XWPFDocument处理DOCX文件，按二级标题分割（二级标题样式为2）
             List<TextSegment> segments = new ArrayList<>();
             try (XWPFDocument xwpfDocument = new XWPFDocument(fileInputStream)) {
                 StringBuilder currentLevel2Content = new StringBuilder();
@@ -971,145 +1212,41 @@ public class CmcAgentServiceImpl implements ICmcAgentService
                     String text = paragraph.getText().trim();
                     if (text.isEmpty()) continue;
 
-                    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) {
-                                // 检查当前二级标题内容的字节数
-                                int level2Length = currentLevel2Content.toString().getBytes().length;
-                                if (level2Length <= 65535) {
-                                    // 字数不大于65535，按二级标题分割
-                                    TextSegment segment = TextSegment.from(currentLevel2Content.toString());
-                                    segments.add(segment);
-                                } else {
-                                    // 字数大于65535，按三级标题分割
-                                    // 将当前二级标题内容按三级标题重新分割
-                                    StringBuilder currentLevel3Content = new StringBuilder();
-                                    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) {
-                                                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) {
-                                                TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                                                segments.add(segment);
-                                                currentLevel3Content = new StringBuilder();
-                                            }
-                                            // 开始新的三级标题内容
-                                            currentLevel3Content.append(para).append("\n");
-                                        } else {
-                                            // 普通内容，添加到当前三级标题
-                                            if (currentLevel3Content.length() != 0) {
-                                                currentLevel3Content.append(para).append("\n");
-                                            }
-                                        }
-                                    }
-                                    // 保存最后一个三级标题内容
-                                    if (currentLevel3Content.length() != 0) {
-                                        TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                                        segments.add(segment);
-                                    }
-                                }
-                                currentLevel2Content = new StringBuilder();
-                            }
-                            // 开始新的二级标题内容
-                            currentLevel2Content.append(text).append("\n");
-                            inLevel2Section = true;
-                        }
-                        // 其他标题样式（包括三级标题）
-                        else {
-                            if (inLevel2Section) {
-                                currentLevel2Content.append(text).append("\n");
+                    int paraLevel = getDocxOutlineLevel(paragraph, xwpfDocument);
+
+                    if (paraLevel == 2) {
+                        if (currentLevel2Content.length() != 0) {
+                            int level2Length = currentLevel2Content.toString().getBytes().length;
+                            if (level2Length <= 65535) {
+                                TextSegment segment = TextSegment.from(currentLevel2Content.toString());
+                                segments.add(segment);
+                            } else {
+                                splitDocxByLevel3(currentLevel2Content.toString(), xwpfDocument, segments);
                             }
+                            currentLevel2Content = new StringBuilder();
                         }
-                    }
-                    else {
-                        // 普通内容段落
+                        currentLevel2Content.append(text).append("\n");
+                        inLevel2Section = true;
+                    } else {
                         if (inLevel2Section) {
                             currentLevel2Content.append(text).append("\n");
                         }
                     }
                 }
 
-                // 保存最后一个二级标题内容
                 if (currentLevel2Content.length() != 0) {
-                    // 检查字节数
                     int level2Length = currentLevel2Content.toString().getBytes().length;
                     if (level2Length <= 65535) {
-                        // 字数不大于65535，按二级标题分割
                         TextSegment segment = TextSegment.from(currentLevel2Content.toString());
                         segments.add(segment);
                     } else {
-                        // 字数大于65535，按三级标题分割
-                        StringBuilder currentLevel3Content = new StringBuilder();
-                        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) {
-                                    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) {
-                                    TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                                    segments.add(segment);
-                                    currentLevel3Content = new StringBuilder();
-                                }
-                                // 开始新的三级标题内容
-                                currentLevel3Content.append(para).append("\n");
-                            } else {
-                                // 普通内容，添加到当前三级标题
-                                if (currentLevel3Content.length() != 0) {
-                                    currentLevel3Content.append(para).append("\n");
-                                }
-                            }
-                        }
-                        // 保存最后一个三级标题内容
-                        if (currentLevel3Content.length() != 0) {
-                            TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                            segments.add(segment);
-                        }
+                        splitDocxByLevel3(currentLevel2Content.toString(), xwpfDocument, segments);
                     }
                 }
             }
             return segments;
         }
         else if (filename.endsWith(".doc")) {
-            // 使用HWPFDocument处理DOC文件
             List<TextSegment> segments = new ArrayList<>();
             try (HWPFDocument hwpfDocument = new HWPFDocument(fileInputStream)) {
                 StringBuilder currentLevel2Content = new StringBuilder();
@@ -1121,134 +1258,35 @@ public class CmcAgentServiceImpl implements ICmcAgentService
                     String text = paragraph.text().trim();
                     if (text.isEmpty()) continue;
 
-                    // 获取段落样式索引
-                    short styleIndex = paragraph.getStyleIndex();
+                    int paraLevel = getDocOutlineLevel(paragraph);
 
-                    if (styleIndex == 2) {
-                        // 二级标题，开始新的分段
-                        // 保存当前二级标题下的内容
+                    if (paraLevel == 2) {
                         if (currentLevel2Content.length() != 0) {
-                            // 检查当前二级标题内容的字节数
                             int level2Length = currentLevel2Content.toString().getBytes().length;
                             if (level2Length <= 65535) {
-                                // 字数不大于65535，按二级标题分割
                                 TextSegment segment = TextSegment.from(currentLevel2Content.toString());
                                 segments.add(segment);
                             } else {
-                                // 字数大于65535，按三级标题分割
-                                // 将当前二级标题内容按三级标题重新分割
-                                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) {
-                                            // 三级标题
-                                            // 保存当前三级标题内容
-                                            if (currentLevel3Content.length() != 0) {
-                                                TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                                                segments.add(segment);
-                                                currentLevel3Content = new StringBuilder();
-                                            }
-                                            // 开始新的三级标题内容
-                                            currentLevel3Content.append(paraText).append("\n");
-                                        } else if (paraStyleIndex == 2 && !paraText.equals(currentLevel2Content.toString().split("\n")[0].trim())) {
-                                            // 下一个二级标题，结束当前二级标题的处理
-                                            break;
-                                        } else {
-                                            // 普通内容或当前二级标题
-                                            if (currentLevel3Content.length() != 0 || paraStyleIndex == 2) {
-                                                currentLevel3Content.append(paraText).append("\n");
-                                            }
-                                        }
-                                    }
-                                }
-
-                                // 保存最后一个三级标题内容
-                                if (currentLevel3Content.length() != 0) {
-                                    TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                                    segments.add(segment);
-                                }
+                                splitDocByLevel3(currentLevel2Content.toString(), hwpfDocument, segments);
                             }
                             currentLevel2Content = new StringBuilder();
                         }
-                        // 开始新的二级标题内容
                         currentLevel2Content.append(text).append("\n");
                         inLevel2Section = true;
                     } else {
-                        // 非二级标题或普通内容段落
                         if (inLevel2Section) {
                             currentLevel2Content.append(text).append("\n");
                         }
                     }
                 }
 
-                // 保存最后一个二级标题内容
                 if (currentLevel2Content.length() != 0) {
-                    // 检查字节数
                     int level2Length = currentLevel2Content.toString().getBytes().length;
                     if (level2Length <= 65535) {
-                        // 字数不大于65535，按二级标题分割
                         TextSegment segment = TextSegment.from(currentLevel2Content.toString());
                         segments.add(segment);
                     } else {
-                        // 字数大于65535，按三级标题分割
-                        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) {
-                                    // 三级标题
-                                    // 保存当前三级标题内容
-                                    if (currentLevel3Content.length() != 0) {
-                                        TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                                        segments.add(segment);
-                                        currentLevel3Content = new StringBuilder();
-                                    }
-                                    // 开始新的三级标题内容
-                                    currentLevel3Content.append(paraText).append("\n");
-                                } else if (paraStyleIndex == 2 && !paraText.equals(currentLevel2Content.toString().split("\n")[0].trim())) {
-                                    // 下一个二级标题，结束当前二级标题的处理
-                                    break;
-                                } else {
-                                    // 普通内容或当前二级标题
-                                    if (currentLevel3Content.length() != 0 || paraStyleIndex == 2) {
-                                        currentLevel3Content.append(paraText).append("\n");
-                                    }
-                                }
-                            }
-                        }
-
-                        // 保存最后一个三级标题内容
-                        if (currentLevel3Content.length() != 0) {
-                            TextSegment segment = TextSegment.from(currentLevel3Content.toString());
-                            segments.add(segment);
-                        }
+                        splitDocByLevel3(currentLevel2Content.toString(), hwpfDocument, segments);
                     }
                 }
             }

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

@@ -2,7 +2,7 @@
  * @Author: ysh
  * @Date: 2025-06-30 09:56:10
  * @LastEditors: wrh
- * @LastEditTime: 2026-04-17 09:54:52
+ * @LastEditTime: 2026-04-21 13:42:00
  */
 import request from '@/utils/request'
 
@@ -110,3 +110,12 @@ export function listByTitle(collectionName, title) {
     params: { collectionName, title }
   })
 }
+
+// 根据文件名获取标题列表
+export function listTilesByFile(collectionName, fileName) {
+  return request({
+    url: '/llm/knowledge/listTilesByFile',
+    method: 'get',
+    params: { collectionName, fileName }
+  })
+}

oa-ui/src/views/llm/agent/AgentDetail.vue

@@ -2,13 +2,13 @@
  * @Author: wrh
  * @Date: 2025-01-01 00:00:00
  * @LastEditors: wrh
- * @LastEditTime: 2026-02-07 15:49:42
+ * @LastEditTime: 2026-04-21 19:34:14
 -->
 <template>
   <div class="agent-detail-container" v-loading="loading">
     <div v-if="agentInfo" class="detail-content">
       <!-- 智能体头部信息 -->
-      <!-- <div class="agent-header">
+      <div class="agent-header">
         <div class="agent-basic-info">
           <h2 class="agent-title">{{ agentInfo.agentName }}</h2>
           <p class="agent-description">{{ agentInfo.description || '暂无描述' }}</p>
@@ -21,10 +21,18 @@
             </span>
           </div>
         </div>
-      </div> -->
+        
+        <!-- 模式切换 -->
+        <div class="mode-switcher">
+          <el-radio-group v-model="currentMode" @change="handleModeChange">
+            <el-radio-button label="conversation">对话模式</el-radio-button>
+            <el-radio-button label="task">任务模式</el-radio-button>
+          </el-radio-group>
+        </div>
+      </div>
 
-      <!-- 对话区域 -->
-      <div class="chat-section">
+      <!-- 对话模式 -->
+      <div v-if="currentMode === 'conversation'" class="chat-section">
         <!-- 话题列表 -->
         <div v-if="topicList.length > 0" class="topic-list">
           <div class="topic-header">
@@ -183,6 +191,133 @@
           </div>
         </div>
       </div>
+
+      <!-- 任务模式 -->
+      <div v-else-if="currentMode === 'task'" class="task-section">
+        <el-tabs v-model="activeTaskTab">
+          <!-- 文档上传 -->
+          <el-tab-pane label="文档上传" name="upload">
+            <div class="task-content">
+              <div class="upload-section">
+                <h4>上传招标文件</h4>
+                <el-upload class="upload-area" :action="uploadAction" :multiple="false" :auto-upload="false"
+                  :file-list="taskFileList" :on-change="handleTaskFileChange" :before-upload="beforeUpload"
+                  :show-file-list="true" :limit="1">
+                  <el-button type="primary" size="large" icon="el-icon-upload">
+                    选择文件
+                  </el-button>
+                  <div class="upload-tip">支持 PDF、DOC、DOCX、TXT 格式，单个文件不超过 10MB</div>
+                </el-upload>
+                <div v-if="taskFileList.length > 0" class="task-upload-actions">
+                  <el-button size="medium" type="success" @click="submitTaskUpload" icon="el-icon-check">
+                    确认上传
+                  </el-button>
+                </div>
+                <div v-if="taskPercentageDisplay" class="upload-progress">
+                  <p class="upload-tip"></p>
+                  <p class="upload-tip">{{ taskStep }}</p>
+                  <el-progress :percentage="taskPercentage" :stroke-width="25" :text-inside="true">
+                    <el-button text>{{ taskProgress }}</el-button>
+                  </el-progress>
+                </div>
+              </div>
+            </div>
+          </el-tab-pane>
+
+          <!-- 文档分析 -->
+          <el-tab-pane label="文档分析" name="analysis">
+            <div class="task-content">
+              <el-tabs v-model="activeAnalysisTab">
+                <el-tab-pane label="评分要求" name="scoring">
+                  <div class="analysis-content">
+                    <h4>评分要求分析</h4>
+                    <div v-if="analysisResults.scoring" class="analysis-result">
+                      <div v-html="analysisResults.scoring"></div>
+                    </div>
+                    <div v-else class="empty-state">
+                      <i class="el-icon-document"></i>
+                      <p>请先上传文件进行分析</p>
+                    </div>
+                  </div>
+                </el-tab-pane>
+                <el-tab-pane label="项目概述" name="overview">
+                  <div class="analysis-content">
+                    <h4>项目概述分析</h4>
+                    <div v-if="analysisResults.overview" class="analysis-result">
+                      <div v-html="analysisResults.overview"></div>
+                    </div>
+                    <div v-else class="empty-state">
+                      <i class="el-icon-document"></i>
+                      <p>请先上传文件进行分析</p>
+                    </div>
+                  </div>
+                </el-tab-pane>
+                <el-tab-pane label="目录管理" name="directory">
+                  <div class="analysis-content">
+                    <h4>目录管理</h4>
+                    <div v-if="analysisResults.directory" class="analysis-result">
+                      <div v-html="analysisResults.directory"></div>
+                    </div>
+                    <div v-else class="empty-state">
+                      <i class="el-icon-document"></i>
+                      <p>请先上传文件进行分析</p>
+                    </div>
+                  </div>
+                </el-tab-pane>
+              </el-tabs>
+            </div>
+          </el-tab-pane>
+
+          <!-- 标书内容 -->
+          <el-tab-pane label="标书内容" name="bid">
+            <div class="task-content">
+              <el-tabs v-model="activeBidTab">
+                <el-tab-pane label="生成标书" name="generate">
+                  <div class="bid-content">
+                    <h4>生成标书</h4>
+                    <el-form :model="bidForm" label-width="120px">
+                      <el-form-item label="标书标题">
+                        <el-input v-model="bidForm.title" placeholder="请输入标书标题" />
+                      </el-form-item>
+                      <el-form-item label="项目名称">
+                        <el-input v-model="bidForm.projectName" placeholder="请输入项目名称" />
+                      </el-form-item>
+                      <el-form-item label="生成选项">
+                        <el-checkbox-group v-model="bidForm.options">
+                          <el-checkbox label="技术方案">技术方案</el-checkbox>
+                          <el-checkbox label="商务方案">商务方案</el-checkbox>
+                          <el-checkbox label="服务方案">服务方案</el-checkbox>
+                        </el-checkbox-group>
+                      </el-form-item>
+                      <el-form-item>
+                        <el-button type="primary" @click="generateBid">生成标书</el-button>
+                      </el-form-item>
+                    </el-form>
+                    <div v-if="bidResult" class="bid-result">
+                      <h5>生成结果</h5>
+                      <div v-html="bidResult"></div>
+                    </div>
+                  </div>
+                </el-tab-pane>
+                <el-tab-pane label="导出Word" name="export">
+                  <div class="bid-content">
+                    <h4>导出Word</h4>
+                    <div v-if="bidResult" class="export-section">
+                      <el-button type="primary" icon="el-icon-download" @click="exportWord">
+                        导出Word文档
+                      </el-button>
+                    </div>
+                    <div v-else class="empty-state">
+                      <i class="el-icon-document"></i>
+                      <p>请先生成标书内容</p>
+                    </div>
+                  </div>
+                </el-tab-pane>
+              </el-tabs>
+            </div>
+          </el-tab-pane>
+        </el-tabs>
+      </div>
     </div>
 
     <!-- 空状态 -->
@@ -230,6 +365,28 @@ export default {
       selectDocument: '',
       selectDocumentTip: '',
       chatFileList: [], // 聊天内的文件列表
+      
+      // 任务模式相关数据
+      currentMode: 'conversation', // conversation 或 task
+      activeTaskTab: 'upload', // upload, analysis, bid
+      activeAnalysisTab: 'scoring', // scoring, overview, directory
+      activeBidTab: 'generate', // generate, export
+      taskFileList: [], // 任务模式文件列表
+      taskPercentage: 0,
+      taskProgress: "",
+      taskStep: "",
+      taskPercentageDisplay: false,
+      analysisResults: {
+        scoring: '',
+        overview: '',
+        directory: ''
+      },
+      bidForm: {
+        title: '',
+        projectName: '',
+        options: []
+      },
+      bidResult: ''
     }
   },
   computed: {
@@ -790,6 +947,103 @@ export default {
       formattedContent = formattedContent.replace(/\n/g, '<br>')
 
       return formattedContent
+    },
+
+    // 模式切换处理
+    handleModeChange(mode) {
+      this.currentMode = mode
+      if (mode === 'task') {
+        // 切换到任务模式时的初始化
+        this.activeTaskTab = 'upload'
+        this.taskFileList = []
+        this.taskPercentageDisplay = false
+      }
+    },
+
+    // 任务模式文件上传相关方法
+    handleTaskFileChange(file, fileList) {
+      // 只保留最新选择的文件（限制为单文件）
+      this.taskFileList = fileList.slice(-1)
+    },
+
+    // 任务模式文件上传提交
+    async submitTaskUpload() {
+      if (this.taskFileList.length === 0) {
+        this.$message.warning('请先选择文件')
+        return
+      }
+      try {
+        this.taskPercentageDisplay = true
+        var timer
+        clearInterval(timer)
+        const getProcess = () => {
+          timer = setInterval(() => { //隔2000毫秒获取进度
+            getProcessValue().then(res => {
+              if (res.code == 200 & res.msg.includes(":")) {
+                this.taskProgress = res.msg
+                this.taskStep = res.msg.split(":")[0]
+                this.taskPercentage = Number(res.msg.split(":")[1].replace("%", ""))
+              }
+            })
+          }, 2000)
+        }
+        getProcess()
+        const file = this.taskFileList[0].raw // 只取第一个文件
+        const fileName = file.name
+        try {
+          const response = await uploadFile(file, this.agentInfo.agentName)
+          const chatId = response.data.chatId //获取保存后的chatId
+          // 解析返回的数据
+          if (response.data && response.data.assistantMessage) {
+            this.taskPercentageDisplay = false
+            // 格式化链接：在href前加上基础API地址
+            let assistantContent = this.formatContentLinks(response.data.assistantMessage)
+            clearInterval(timer)
+            
+            // 模拟分析结果（实际项目中应该从API获取）
+            this.analysisResults = {
+              scoring: `<h4>评分要求分析结果</h4><p>根据上传的文件，系统分析出以下评分要求：</p><ul><li>技术方案：40分</li><li>商务方案：30分</li><li>服务方案：30分</li></ul>`,
+              overview: `<h4>项目概述分析结果</h4><p>项目名称：${fileName}</p><p>项目类型：技术服务</p><p>项目预算：100万元</p><p>项目周期：6个月</p><p>项目地点：北京市</p><p>招标人：某某公司</p>`,
+              directory: `<h4>目录管理</h4><ul><li>1. 技术方案</li><ul><li>1.1 系统架构</li><ul><li>1.1.1 系统设计</li><li>1.1.2 技术选型</li></ul><li>1.2 功能模块</li><ul><li>1.2.1 核心功能</li><li>1.2.2 辅助功能</li></ul></ul><li>2. 商务方案</li><ul><li>2.1 报价方案</li><li>2.2 交付计划</li></ul><li>3. 服务方案</li><ul><li>3.1 售后服务</li><li>3.2 培训计划</li></ul><li>4. 附件</li><ul><li>4.1 资质证明</li><li>4.2 案例介绍</li></ul></ul>`
+            }
+            
+            this.$message.success('文件上传成功并已分析')
+            
+            // 切换到分析标签
+            this.activeTaskTab = 'analysis'
+          }
+        }
+        catch (error) {
+          clearInterval(timer)
+          console.error('文件上传失败:', error)
+        }
+
+      } catch (error) {
+        console.error('文件上传失败:', error)
+        this.$message.error('文件上传失败')
+      }
+    },
+
+    // 生成标书
+    generateBid() {
+      if (!this.bidForm.title || !this.bidForm.projectName) {
+        this.$message.warning('请填写标书标题和项目名称')
+        return
+      }
+      if (this.bidForm.options.length === 0) {
+        this.$message.warning('请至少选择一个生成选项')
+        return
+      }
+      
+      // 模拟生成标书（实际项目中应该调用API）
+      this.bidResult = `<h4>${this.bidForm.title}</h4><p><strong>项目名称：</strong>${this.bidForm.projectName}</p><p><strong>生成选项：</strong>${this.bidForm.options.join('、')}</p><p><strong>生成时间：</strong>${new Date().toLocaleString()}</p><p>标书内容已生成，包含技术方案、商务方案和服务方案等内容。</p>`
+      this.$message.success('标书生成成功')
+    },
+
+    // 导出Word
+    exportWord() {
+      // 模拟导出Word（实际项目中应该调用API）
+      this.$message.success('Word文档导出成功')
     }
   }
 }
@@ -810,12 +1064,12 @@ export default {
 }
 
 .agent-header {
-  padding: 15px;
+  padding: 15px 24px;
   border-bottom: 1px solid #e4e4e4;
   background: white;
   display: flex;
   justify-content: space-between;
-  align-items: flex-start;
+  align-items: center;
 
   .agent-basic-info {
     flex: 1;
@@ -856,6 +1110,10 @@ export default {
     display: flex;
     gap: 12px;
   }
+  
+  .mode-switcher {
+    margin-left: 20px;
+  }
 }
 
 .chat-section {
@@ -1311,4 +1569,139 @@ export default {
     font-size: 14px;
   }
 }
+
+/* 任务模式样式 */
+.task-section {
+  background: white;
+  margin: 16px 24px;
+  border-radius: 8px;
+  border: 1px solid #e4e4e4;
+  overflow: hidden;
+  height: calc(100vh - 220px);
+  
+  :deep(.el-tabs__content) {
+    height: calc(100% - 40px);
+    overflow: auto;
+  }
+}
+
+.task-content {
+  padding: 20px;
+  height: 100%;
+}
+
+.upload-section {
+  text-align: center;
+  padding: 40px 20px;
+  background: #f8f9fa;
+  border-radius: 8px;
+  
+  h4 {
+    margin: 0 0 20px 0;
+    font-size: 16px;
+    color: #333;
+  }
+  
+  .upload-area {
+    margin: 20px 0;
+    
+    :deep(.el-upload-dragger) {
+      width: 100%;
+      max-width: 500px;
+      margin: 0 auto;
+    }
+  }
+  
+  .upload-tip {
+    margin-top: 12px;
+    font-size: 14px;
+    color: #666;
+  }
+  
+  .task-upload-actions {
+    margin-top: 20px;
+  }
+  
+  .upload-progress {
+    margin-top: 30px;
+    max-width: 500px;
+    margin-left: auto;
+    margin-right: auto;
+  }
+}
+
+.analysis-content {
+  height: 100%;
+  
+  h4 {
+    margin: 0 0 20px 0;
+    font-size: 16px;
+    color: #333;
+  }
+  
+  .analysis-result {
+    background: #f8f9fa;
+    padding: 20px;
+    border-radius: 8px;
+    height: calc(100% - 60px);
+    overflow-y: auto;
+    
+    h4 {
+      margin: 0 0 15px 0;
+      font-size: 14px;
+      color: #333;
+    }
+    
+    p {
+      margin: 8px 0;
+      line-height: 1.5;
+    }
+    
+    ul {
+      margin: 10px 0;
+      padding-left: 20px;
+    }
+    
+    li {
+      margin: 5px 0;
+    }
+  }
+}
+
+.bid-content {
+  height: 100%;
+  
+  h4 {
+    margin: 0 0 20px 0;
+    font-size: 16px;
+    color: #333;
+  }
+  
+  .el-form {
+    margin-bottom: 30px;
+  }
+  
+  .bid-result {
+    background: #f8f9fa;
+    padding: 20px;
+    border-radius: 8px;
+    margin-top: 20px;
+    
+    h5 {
+      margin: 0 0 15px 0;
+      font-size: 14px;
+      color: #333;
+    }
+    
+    p {
+      margin: 8px 0;
+      line-height: 1.5;
+    }
+  }
+  
+  .export-section {
+    margin-top: 40px;
+    text-align: center;
+  }
+}
 </style>

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

@@ -116,7 +116,7 @@
           </div>
 
           <div v-else class="file-list">
-            <div v-for="(file, index) in pagedFileList" :key="index" class="file-item">
+            <div v-for="(file, index) in pagedFileList" :key="index" class="file-item" @click="handleFileClick(file)">
               <div class="file-info">
                 <i class="el-icon-document file-icon">
                 </i>
@@ -128,7 +128,10 @@
                 </div>
               </div>
               <div class="file-actions">
-                <el-button type="danger" size="small" icon="el-icon-delete" @click="handleDeleteFile(file)">
+                <el-button type="primary" size="small" icon="el-icon-view" @click.stop="handleFileClick(file)">
+                  浏览
+                </el-button>
+                <el-button type="danger" size="small" icon="el-icon-delete" @click.stop="handleDeleteFile(file)">
                   删除
                 </el-button>
               </div>
@@ -146,6 +149,7 @@
             <i class="el-icon-folder folder-icon">
             </i>
             <span class="knowledge-name">{{ selectedKnowledge.collectionName }}</span>
+            <span v-if="currentFileName" class="file-name-badge">{{ currentFileName }}</span>
             <span class="title-mode-badge">浏览模式</span>
           </div>
 
@@ -366,7 +370,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, listTiles, listByTitle } from "@/api/llm/knowLedge";
+import { listKnowledge, listKnowLedgeByCollectionName, addKnowledge, updateKnowledge, delKnowledge, insertKnowledgeFile, listKnowledgeDocument, deleteKnowledgeFile, getProcessValue, listTiles, listByTitle, listTilesByFile } from "@/api/llm/knowLedge";
 import { getAnswer, getAnswerStream, getContextFile } from '@/api/llm/rag';
 import { marked } from 'marked';
 
@@ -522,6 +526,7 @@ export default {
       titleList: [], // 标题列表
       contentList: [], // 内容列表
       selectedTitle: null, // 选中的标题
+      currentFileName: null, // 当前浏览的文件名
       titleLoading: false, // 标题加载状态
       contentLoading: false // 内容加载状态
     }
@@ -664,29 +669,52 @@ export default {
 
       this.isChatMode = false;
       this.isScanMode = false;
+      this.currentFileName = null;
     },
 
     /** 切换到浏览模式 */
-    switchToTitleMode() {
+    switchToTitleMode(fileName) {
       this.isScanMode = true;
       this.isChatMode = false;
       this.titleList = [];
       this.contentList = [];
       this.selectedTitle = null;
+      this.currentFileName = fileName;
       
       // 获取标题列表
       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);
-      });
+      if (fileName) {
+        // 使用新接口根据文件名获取标题列表
+        listTilesByFile(this.selectedKnowledge.collectionName, fileName).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);
+        });
+      } else {
+        // 获取所有标题
+        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);
+        });
+      }
+    },
+    
+    /** 点击文件进入浏览模式 */
+    handleFileClick(file) {
+      this.switchToTitleMode(file);
     },
 
     /** 选择标题 */