技术标书智能体

oa-back/ruoyi-llm/pom.xml

@@ -76,6 +76,12 @@
             <version>3.9.5</version>
         </dependency>
 
+        <!-- Spring Boot Web -->
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-web</artifactId>
+        </dependency>
+
     </dependencies>
 
 </project>

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

@@ -3,9 +3,12 @@ package com.ruoyi.web.llm.controller;
 import java.io.IOException;
 import java.util.Date;
 import java.util.List;
+import java.util.concurrent.ExecutionException;
+
 import javax.servlet.http.HttpServletResponse;
 
 import org.noear.solon.ai.chat.message.ChatMessage;
+import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PostMapping;
@@ -17,6 +20,7 @@ import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
 
 import com.alibaba.fastjson2.JSONObject;
+import com.ruoyi.common.annotation.Anonymous;
 import com.ruoyi.common.annotation.Log;
 import com.ruoyi.common.core.controller.BaseController;
 import com.ruoyi.common.core.domain.AjaxResult;
@@ -110,9 +114,19 @@ public class CmcAgentController extends BaseController
      * @return
      */
     @PostMapping("/modifyFile")
-    public AjaxResult uploadModifyFile(String topicId, MultipartFile file, String agentName) throws IOException
+    public AjaxResult uploadModifyFile(String topicId, MultipartFile file, String agentName, String selectedNode) throws IOException, InterruptedException, ExecutionException
+    {
+        return success(cmcAgentService.uploadModifyFile(topicId, file, agentName, selectedNode));
+    }
+
+    /**
+     * 流式生成章节内容（SSE方式）
+     */
+    @Anonymous
+    @PostMapping(value = "/streamGenerate", produces = "text/event-stream;charset=UTF-8")
+    public SseEmitter streamGenerateChapters(String topicId, MultipartFile file, String agentName, String selectedNode) throws IOException
     {
-        return success(cmcAgentService.uploadModifyFile(topicId, file, agentName));
+        return cmcAgentService.streamGenerateChapters(topicId, file, agentName, selectedNode);
     }
 
     /**

oa-back/ruoyi-system/pom.xml

@@ -63,6 +63,12 @@
             <version>3.9.5</version>
         </dependency>
 
+        <!-- Spring Boot Web -->
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-web</artifactId>
+        </dependency>
+
         <dependency>
             <groupId>io.milvus</groupId>
             <artifactId>milvus-sdk-java</artifactId>

oa-back/ruoyi-system/src/main/java/com/ruoyi/llm/domain/ChapterStreamResponse.java

@@ -0,0 +1,158 @@
+package com.ruoyi.llm.domain;
+
+/**
+ * 章节流式响应DTO
+ */
+public class ChapterStreamResponse {
+
+    /**
+     * 章节标题
+     */
+    private String title;
+
+    /**
+     * 当前章节内容片段
+     */
+    private String content;
+
+    /**
+     * 是否是当前章节的最后一块内容
+     */
+    private boolean isLast;
+
+    /**
+     * 当前章节索引（从1开始）
+     */
+    private int chapterIndex;
+
+    /**
+     * 总章节数
+     */
+    private int totalChapters;
+
+    /**
+     * 是否发生错误
+     */
+    private boolean error;
+
+    /**
+     * 错误信息
+     */
+    private String errorMessage;
+
+    /**
+     * 是否全部完成
+     */
+    private boolean completed;
+
+    public ChapterStreamResponse() {
+    }
+
+    public static ChapterStreamResponse content(String title, String content, boolean isLast, int chapterIndex, int totalChapters) {
+        ChapterStreamResponse response = new ChapterStreamResponse();
+        response.title = title;
+        response.content = content;
+        response.isLast = isLast;
+        response.chapterIndex = chapterIndex;
+        response.totalChapters = totalChapters;
+        response.error = false;
+        response.completed = false;
+        return response;
+    }
+
+    public static ChapterStreamResponse error(String title, String errorMessage) {
+        ChapterStreamResponse response = new ChapterStreamResponse();
+        response.title = title;
+        response.content = "该章节内容生成失败，请手动填写。";
+        response.isLast = true;
+        response.error = true;
+        response.errorMessage = errorMessage;
+        response.completed = false;
+        return response;
+    }
+
+    public static ChapterStreamResponse completed() {
+        ChapterStreamResponse response = new ChapterStreamResponse();
+        response.completed = true;
+        response.isLast = true;
+        return response;
+    }
+
+    // Getters and Setters
+    public String getTitle() {
+        return title;
+    }
+
+    public void setTitle(String title) {
+        this.title = title;
+    }
+
+    public String getContent() {
+        return content;
+    }
+
+    public void setContent(String content) {
+        this.content = content;
+    }
+
+    public boolean isLast() {
+        return isLast;
+    }
+
+    public void setLast(boolean last) {
+        isLast = last;
+    }
+
+    public int getChapterIndex() {
+        return chapterIndex;
+    }
+
+    public void setChapterIndex(int chapterIndex) {
+        this.chapterIndex = chapterIndex;
+    }
+
+    public int getTotalChapters() {
+        return totalChapters;
+    }
+
+    public void setTotalChapters(int totalChapters) {
+        this.totalChapters = totalChapters;
+    }
+
+    public boolean isError() {
+        return error;
+    }
+
+    public void setError(boolean error) {
+        this.error = error;
+    }
+
+    public String getErrorMessage() {
+        return errorMessage;
+    }
+
+    public void setErrorMessage(String errorMessage) {
+        this.errorMessage = errorMessage;
+    }
+
+    public boolean isCompleted() {
+        return completed;
+    }
+
+    public void setCompleted(boolean completed) {
+        this.completed = completed;
+    }
+
+    @Override
+    public String toString() {
+        return "ChapterStreamResponse{" +
+                "title='" + title + '\'' +
+                ", content='" + content + '\'' +
+                ", isLast=" + isLast +
+                ", chapterIndex=" + chapterIndex +
+                ", totalChapters=" + totalChapters +
+                ", error=" + error +
+                ", completed=" + completed +
+                '}';
+    }
+}

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

@@ -2,6 +2,8 @@ package com.ruoyi.llm.service;
 
 import com.alibaba.fastjson2.JSONObject;
 import com.ruoyi.llm.domain.CmcAgent;
+
+import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
 import org.springframework.web.multipart.MultipartFile;
 
 import java.io.IOException;
@@ -50,10 +52,13 @@ public interface ICmcAgentService
     /**
      * 上传修改文件
      *
+     * @param topicId 话题ID
      * @param file 文件
+     * @param agentName 智能体名称
+     * @param selectedNode 选中的节点标题（可选，为空时生成全部章节）
      * @return 结果
      */
-    public JSONObject uploadModifyFile(String topicId, MultipartFile file, String agentName) throws IOException;
+    public JSONObject uploadModifyFile(String topicId, MultipartFile file, String agentName, String selectedNode) throws IOException, InterruptedException, java.util.concurrent.ExecutionException;
 
     /**
      * 上传多文件
@@ -109,4 +114,16 @@ public interface ICmcAgentService
      * @return 结果
      */
     public JSONObject writeTitles(JSONObject data) throws IOException;
+
+    /**
+     * 流式生成章节内容（SSE方式）
+     *
+     * @param topicId 主题ID
+     * @param file 招标文件
+     * @param agentName 智能体名称
+     * @param selectedNode 选中的节点标题（可选，为空时生成全部章节）
+     * @return SseEmitter
+     */
+    public SseEmitter streamGenerateChapters(String topicId, MultipartFile file, String agentName, String selectedNode) throws IOException;
+
 }

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

@@ -0,0 +1,112 @@
+package com.ruoyi.llm.service;
+
+import io.milvus.v2.client.ConnectConfig;
+import io.milvus.v2.client.MilvusClientV2;
+import io.milvus.v2.service.collection.request.LoadCollectionReq;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.stereotype.Component;
+
+import javax.annotation.PostConstruct;
+import javax.annotation.PreDestroy;
+import java.util.*;
+import java.util.concurrent.BlockingQueue;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.LinkedBlockingQueue;
+import java.util.concurrent.TimeUnit;
+import java.util.concurrent.locks.ReadWriteLock;
+import java.util.concurrent.locks.ReentrantReadWriteLock;
+
+@Component
+public class MilvusConnectionPool {
+
+    @Value("${cmc.milvusService.url}")
+    private String milvusServiceUrl;
+
+    private static final int POOL_SIZE = 5;
+    private final List<MilvusClientV2> clientPool = new ArrayList<>();
+    private final BlockingQueue<MilvusClientV2> availableClients = new LinkedBlockingQueue<>();
+    private final Set<String> loadedCollections = ConcurrentHashMap.newKeySet();
+    private final ReadWriteLock collectionLock = new ReentrantReadWriteLock();
+    private volatile boolean initialized = false;
+
+    @PostConstruct
+    public void init() {
+        if (milvusServiceUrl == null || milvusServiceUrl.isEmpty()) {
+            throw new IllegalArgumentException("milvusServiceUrl 配置不能为空");
+        }
+
+        for (int i = 0; i < POOL_SIZE; i++) {
+            MilvusClientV2 client = new MilvusClientV2(
+                    ConnectConfig.builder()
+                            .uri(milvusServiceUrl)
+                            .build());
+            clientPool.add(client);
+            availableClients.offer(client);
+        }
+        initialized = true;
+        System.out.println("Milvus连接池初始化完成，连接数: " + POOL_SIZE);
+    }
+
+    @PreDestroy
+    public void destroy() {
+        for (MilvusClientV2 client : clientPool) {
+            try {
+                if (client != null) {
+                    client.close();
+                }
+            } catch (Exception e) {
+                System.err.println("关闭Milvus连接时出错: " + e.getMessage());
+            }
+        }
+        clientPool.clear();
+        availableClients.clear();
+        System.out.println("Milvus连接池已关闭");
+    }
+
+    public MilvusClientV2 getClient() throws InterruptedException {
+        MilvusClientV2 client = availableClients.poll(5, TimeUnit.SECONDS);
+        if (client == null) {
+            throw new RuntimeException("获取Milvus连接超时");
+        }
+        return client;
+    }
+
+    public void returnClient(MilvusClientV2 client) {
+        if (client != null) {
+            availableClients.offer(client);
+        }
+    }
+
+    public void ensureCollectionLoaded(MilvusClientV2 client, String collectionName) {
+        collectionLock.readLock().lock();
+        try {
+            if (loadedCollections.contains(collectionName)) {
+                return;
+            }
+        } finally {
+            collectionLock.readLock().unlock();
+        }
+
+        collectionLock.writeLock().lock();
+        try {
+            if (!loadedCollections.contains(collectionName)) {
+                LoadCollectionReq loadCollectionReq = LoadCollectionReq.builder()
+                        .collectionName(collectionName)
+                        .build();
+                client.loadCollection(loadCollectionReq);
+                loadedCollections.add(collectionName);
+                System.out.println("集合已加载: " + collectionName);
+            }
+        } finally {
+            collectionLock.writeLock().unlock();
+        }
+    }
+
+    public boolean isInitialized() {
+        return initialized;
+    }
+
+    public Set<String> getLoadedCollections() {
+        return new HashSet<>(loadedCollections);
+    }
+}

oa-back/ruoyi-system/src/main/resources/mapper/system/SysUserMapper.xml

@@ -145,7 +145,7 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
 		</if>
 		<!-- 数据范围过滤 -->
 		${params.dataScope}
-		order by u.dept_id, u.user_id
+		order by u.dept_id, ps.post_level desc, ps.salary_level desc, u.user_id
 	</select>
 
     <select id="selectUserServingList" parameterType="SysUser" resultMap="SysUserResult">
@@ -192,7 +192,7 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
 		</if>
 		<!-- 数据范围过滤 -->
 		${params.dataScope}
-		order by u.dept_id, u.user_id
+		order by u.dept_id, ps.post_level desc, ps.salary_level desc, u.user_id
 	</select>
 
 	<select id="selectAllocatedList" parameterType="SysUser" resultMap="SysUserResult">

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

@@ -2,9 +2,10 @@
  * @Author: wrh
  * @Date: 2025-07-17 18:06:24
  * @LastEditors: wrh
- * @LastEditTime: 2026-04-24 13:52:03
+ * @LastEditTime: 2026-05-14 19:52:03
  */
 import request from '@/utils/request'
+import { getToken } from '@/utils/auth'
 
 // 查询智能体列表
 export function listAgent(query) {
@@ -64,14 +65,14 @@ export function uploadFile(file, agentName) {
   })
 }
 
-// 上传单文件
-export function uploadModifyFile(topicId, file, agentName, chapterNumber) {
+// 上传修改文件
+export function uploadModifyFile(topicId, file, agentName, selectedNode) {
   const formData = new FormData()
   formData.append('topicId', topicId)
   formData.append('file', file)
   formData.append('agentName', agentName)
-  if (chapterNumber) {
-    formData.append('chapterNumber', chapterNumber)
+  if (selectedNode) {
+    formData.append('selectedNode', selectedNode)
   }
   return request({
     url: '/llm/agent/modifyFile',
@@ -125,3 +126,151 @@ export function writeTitles(data) {
     data: data
   })
 }
+
+/**
+ * 解析SSE事件 - 优化版
+ * 正确处理多行data和事件类型
+ */
+function parseSseEvents(buffer) {
+  const events = []
+  const lines = buffer.split(/\r?\n/)
+  let currentEvent = null
+  let currentData = []
+  let remaining = []
+  let i = 0
+  
+  for (; i < lines.length; i++) {
+    const line = lines[i]
+    
+    // 空行表示事件结束
+    if (line === '') {
+      if (currentData.length > 0) {
+        const data = currentData.join('\n')
+        events.push({
+          event: currentEvent || 'message',
+          data: data
+        })
+        currentEvent = null
+        currentData = []
+      }
+      continue
+    }
+    
+    // 解析事件类型
+    if (line.startsWith('event:')) {
+      currentEvent = line.slice(6).trim()
+      continue
+    }
+    
+    // 解析数据行
+    if (line.startsWith('data:')) {
+      const dataValue = line.slice(5).trimStart()
+      currentData.push(dataValue)
+      continue
+    }
+  }
+  
+  // 保存剩余未完成的数据
+  if (i < lines.length) {
+    remaining = lines.slice(i).join('\n')
+  } else if (currentData.length > 0) {
+    // 如果最后没有空行，但数据已完整，也要保留
+    remaining = lines.slice(i - currentData.length - (currentEvent ? 1 : 0)).join('\n')
+  } else {
+    remaining = ''
+  }
+  
+  return { events, rest: remaining }
+}
+
+/**
+ * 规范化SSE数据
+ */
+function normalizeSseData(data) {
+  if (data == null) return ''
+  let s = String(data)
+  // 移除可能的 "data:" 前缀
+  while (s.startsWith('data:')) {
+    s = s.slice(5)
+    if (s.startsWith(' ')) s = s.slice(1)
+  }
+  return s
+}
+
+/**
+ * 流式生成标书章节（SSE）- 支持逐字输出
+ * @param {FormData} formData - 包含 topicId, agentName, file, selectedNode
+ * @param {Function} onMessage - 收到消息时的回调，接收解析后的JSON对象
+ * @param {Function} onError - 错误回调
+ * @param {Function} onComplete - 完成回调
+ * @returns {AbortController} 用于取消请求的控制器
+ */
+export function streamGenerateChapters(formData, onMessage, onError, onComplete) {
+  const controller = new AbortController()
+
+  fetch(process.env.VUE_APP_BASE_API + '/llm/agent/streamGenerate', {
+    method: 'POST',
+    body: formData,
+    headers: {
+      'Accept': 'text/event-stream',
+      'Authorization': 'Bearer ' + getToken(),
+      'Cache-Control': 'no-cache'
+    },
+    signal: controller.signal
+  })
+    .then(async (response) => {
+      if (!response.ok) {
+        throw new Error(`HTTP error! status: ${response.status}`)
+      }
+      if (!response.body) {
+        throw new Error('ReadableStream not supported')
+      }
+
+      const reader = response.body.getReader()
+      const decoder = new TextDecoder('utf-8')
+      let buffer = ''
+
+      while (true) {
+        const { done, value } = await reader.read()
+        if (done) break
+
+        buffer += decoder.decode(value, { stream: true })
+        
+        // 解析SSE事件
+        const { events, rest } = parseSseEvents(buffer)
+        buffer = rest
+
+        for (const event of events) {
+          const normalizedData = normalizeSseData(event.data)
+          
+          // 检查是否结束
+          if (normalizedData === '[DONE]') {
+            onComplete && onComplete()
+            controller.abort()
+            return
+          }
+          
+          try {
+            const jsonData = JSON.parse(normalizedData)
+            // 根据事件名称分发
+            onMessage && onMessage(jsonData, event.event)
+          } catch (e) {
+            console.warn('解析JSON失败，作为普通文本处理:', normalizedData)
+            onMessage && onMessage({ content: normalizedData }, event.event)
+          }
+        }
+      }
+      
+      onComplete && onComplete()
+    })
+    .catch((error) => {
+      if (error.name === 'AbortError') {
+        console.log('请求已取消')
+        return
+      }
+      console.error('流式请求错误:', error)
+      onError && onError(error)
+    })
+
+  return controller
+}

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

@@ -163,4 +163,4 @@ export function getAnswerWithDocumentStream(params, onMessage, onError, onComple
   const baseURL = process.env.VUE_APP_BASE_API
   const url = `${baseURL}/llm/session/answerWithDocument?topicId=${params.topicId}&chatId=${params.chatId}&question=${encodeURIComponent(params.question)}`
   return streamFetchSse(url, onMessage, onError, onComplete)
-}
+}

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

@@ -8,10 +8,13 @@
   <div class="app-container">
     <div class="main-content">
       <!-- 左侧智能体列表 -->
-      <div class="left-panel">
+      <div v-if="showLeftPanel" class="left-panel">
         <div class="panel-header">
-          <h3>智能体列表</h3>
-          <span class="agent-count">{{ agentList.length }} 个智能体</span>
+          <div class="header-left">
+  
+            <h3>智能体列表</h3>
+            <span class="agent-count">{{ agentList.length }} 个智能体</span>
+          </div>
           <el-button type="primary" size="small" @click="openAddDialog" icon="el-icon-plus">
             新增智能体
           </el-button>
@@ -69,8 +72,10 @@
       </div>
 
       <!-- 右侧详细内容 -->
-      <div class="agent-detail">
-        <AgentDetail :agent-id="selectedAgentId" />
+      <div class="right-panel">
+        <div class="agent-detail">
+          <AgentDetail :agent-id="selectedAgentId" :show-left-panel="showLeftPanel" @toggle-left-panel="showLeftPanel = !showLeftPanel" />
+        </div>
       </div>
 
       <el-dialog :visible.sync="agentDialogVisible" :title="dialogTitle" width="500px" @close="resetForm">
@@ -123,6 +128,7 @@ export default {
       agentList: [],
       total: 0,
       selectedAgentId: null,
+      showLeftPanel: true,
       dialogVisible: false,
       form: {
         agentName: '',
@@ -310,9 +316,7 @@ export default {
 }
 
 .left-panel {
-  width: 25%;
-  min-width: 260px;
-  max-width: 320px;
+  width: 20%;
   background: white;
   border-radius: 8px;
   box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1);
@@ -329,6 +333,7 @@ export default {
   display: flex;
   flex-direction: column;
   overflow: hidden;
+
 }
 
 .panel-header {
@@ -339,6 +344,12 @@ export default {
   align-items: center;
   background: #fafbfc;
 
+  .header-left {
+    display: flex;
+    align-items: center;
+    gap: 8px;
+  }
+
   h3 {
     margin: 0;
     font-size: 16px;

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

@@ -3,10 +3,12 @@
     <!-- 主要内容区域 -->
     <div class="main-content">
       <!-- 左侧知识库列表 -->
-      <div class="left-panel">
+      <div v-if="showLeftPanel" class="left-panel">
         <div class="panel-header">
-          <h3>知识库列表</h3>
-          <span class="knowledge-count">{{ knowledgeList.length }} 个知识库</span>
+          <div class="header-left">
+            <h3>知识库列表</h3>
+            <span class="knowledge-count">{{ knowledgeList.length }} 个知识库</span>
+          </div>
           <el-button type="primary" size="small" @click="handleAdd" v-hasPermi="['llm:knowledge:add']"
             icon="el-icon-plus">
             新增知识库
@@ -71,7 +73,10 @@
       <!-- 右侧面板 -->
       <div class="right-panel">
         <div class="panel-header">
-          <h3>{{ isChatMode ? '知识库对话' : (isScanMode ? '内容列表' : '文件列表') }}</h3>
+          <div class="header-left">
+            <el-button class="collapse-icon-btn" :icon="showLeftPanel ? 'el-icon-s-fold' : 'el-icon-s-unfold'" @click="showLeftPanel = !showLeftPanel" />
+            <h3>{{ isChatMode ? '知识库对话' : (isScanMode ? '内容列表' : '文件列表') }}</h3>
+          </div>
           <div class="header-actions">
             <el-button v-if="!isChatMode && !isScanMode" type="primary" icon="el-icon-chat-round" @click="handleChat(selectedKnowledge)"
               :disabled="!selectedKnowledge" v-hasPermi="['llm:knowledge:chat']">
@@ -441,6 +446,7 @@ export default {
   data() {
     return {
       knowledgeList: [],
+      showLeftPanel: true,
       open: false,
       uploadOpen: false,
       loading: true,
@@ -1277,7 +1283,7 @@ export default {
 }
 
 .left-panel {
-  width: 250px;
+  width: 20%;
   background: white;
   border-radius: 8px;
   box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1);
@@ -1304,6 +1310,27 @@ export default {
   align-items: center;
   background: #fafbfc;
 
+  .header-left {
+    display: flex;
+    align-items: center;
+    gap: 8px;
+  }
+
+  .collapse-icon-btn {
+    padding: 0;
+    width: 24px;
+    height: 24px;
+    border: none;
+    background: transparent;
+    color: #6c757d;
+    transition: all 0.3s;
+
+    &:hover {
+      background: #e9ecef;
+      color: #495057;
+    }
+  }
+
   h3 {
     margin: 0;
     font-size: 16px;