知识库、智能体功能完善

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

@@ -48,11 +48,11 @@ public class McpServiceImpl implements IMcpService {
 
     private static final EmbeddingModel embeddingModel = new BgeSmallZhV15EmbeddingModel();
 
-    private static final String llmServiceUrl = "http://192.168.28.188:8000/v1/chat/completions";
+    private static final String llmServiceUrl = "http://192.168.28.196:8000/v1/chat/completions";
 
     private static final MilvusClientV2 milvusClient = new MilvusClientV2(
             ConnectConfig.builder()
-                    .uri("http://192.168.28.188:19530")
+                    .uri("http://192.168.28.196:19530")
                     .build());
 
     /**
@@ -65,7 +65,9 @@ public class McpServiceImpl implements IMcpService {
                                            @Param(description = "技术文件地址") String templatePath) throws IOException
     {
             try {
-                templatePath = templatePath.replace("/dev-api/profile", Solon.cfg().getProperty("cmc.profile"));
+                if (templatePath.startsWith("/dev-api"))
+                    templatePath = templatePath.replace("/dev-api/profile", Solon.cfg().getProperty("cmc.profile"));
+                templatePath = Solon.cfg().getProperty("cmc.profile") + templatePath;
                 List<String> subTitles = extractSubTitles(templatePath, title);
                 List<JSONObject> contexts = retrieveFromMilvus(collectionName, title, 10);
                 return generateAnswerWithDocumentAndCollection(agentName, templatePath, subTitles, contexts);
@@ -144,7 +146,7 @@ public class McpServiceImpl implements IMcpService {
      */
     public String generateAnswer(String prompt) throws IOException {
         ChatModel chatModel = ChatModel.of(llmServiceUrl)
-                .model("Qwen2.5-1.5B-Instruct")
+                .model("Qwen2.5-7B-Instruct")
                 .build();
 
         List<ChatMessage> messages = new ArrayList<>();

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

@@ -41,6 +41,15 @@ public class KnowLedgeController extends BaseController
         JSONArray collectionNames = milvusService.getCollectionNames();
         return success(collectionNames);
     }
+    /**
+     * 新建知识库
+     */
+    @GetMapping("/listByName")
+    public AjaxResult listKnowLedgeByCollectionName(String collectionName)
+    {
+        JSONArray collectionNames = milvusService.listKnowLedgeByCollectionName(collectionName);
+        return success(collectionNames);
+    }
 
     /**
      * 新建知识库

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

@@ -1,6 +1,7 @@
 package com.ruoyi.web.llm.controller;
 
 import com.alibaba.fastjson2.JSONObject;
+import com.ruoyi.common.config.RuoYiConfig;
 import com.ruoyi.common.core.controller.BaseController;
 import com.ruoyi.llm.domain.CmcChat;
 import com.ruoyi.llm.service.ICmcAgentService;
@@ -11,6 +12,7 @@ import dev.langchain4j.model.embedding.EmbeddingModel;
 import dev.langchain4j.model.embedding.onnx.bgesmallzhv15.BgeSmallZhV15EmbeddingModel;
 import io.milvus.client.MilvusServiceClient;
 import io.milvus.param.ConnectParam;
+import org.noear.solon.Solon;
 import org.noear.solon.ai.chat.ChatModel;
 import org.noear.solon.ai.chat.ChatResponse;
 import org.noear.solon.ai.chat.ChatSession;
@@ -52,11 +54,11 @@ public class McpController extends BaseController
 
     private static final EmbeddingModel embeddingModel = new BgeSmallZhV15EmbeddingModel();
 
-    private static final String llmServiceUrl = "http://192.168.28.188:8000/v1/chat/completions";
+    private static final String llmServiceUrl = "http://192.168.28.196:8000/v1/chat/completions";
 
     private static final MilvusServiceClient milvusClient = new MilvusServiceClient(
             ConnectParam.newBuilder()
-                    .withHost("192.168.28.188")
+                    .withHost("192.168.28.196")
                     .withPort(19530)
                     .build());
 
@@ -71,7 +73,7 @@ public class McpController extends BaseController
                 .apiUrl("http://localhost:8080/mcp/sse")
                 .build();
         ChatModel chatModel = ChatModel.of(llmServiceUrl)
-                .model("Qwen2.5-1.5B-Instruct")
+                .model("Qwen2.5-7B-Instruct")
                 .defaultToolsAdd(clientProvider)
                 .build();
 

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

@@ -32,7 +32,7 @@ public class RagController extends BaseController
     public Flux<AssistantMessage> answerWithCollection(String collectionName, String topicId, String question)
     {
         List<JSONObject> contexts = langChainMilvusService.retrieveFromMilvus(collectionName, question, 10);
-        return langChainMilvusService.generateAnswerWithCollection(topicId, question, contexts, "http://192.168.28.188:8000/v1/chat/completions");
+        return langChainMilvusService.generateAnswerWithCollection(topicId, question, contexts, "http://192.168.28.196:8000/v1/chat/completions");
     }
 
     /**

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

@@ -29,7 +29,7 @@ public class SessionController extends BaseController
      */
     @GetMapping("/answer")
     public Flux<AssistantMessage> answer(String topicId, String question) {
-        return langChainMilvusService.generateAnswer(topicId, question, "http://192.168.28.188:8000/v1/chat/completions");
+        return langChainMilvusService.generateAnswer(topicId, question, "http://192.168.28.196:8000/v1/chat/completions");
     }
 
     /**
@@ -38,7 +38,7 @@ public class SessionController extends BaseController
     @GetMapping("/answerWithDocument")
     public Flux<AssistantMessage> answerWithDocument(String topicId, String chatId, String question) throws IOException
     {
-        return langChainMilvusService.generateAnswerWithDocument(topicId, chatId, question, "http://192.168.28.188:8000/v1/chat/completions");
+        return langChainMilvusService.generateAnswerWithDocument(topicId, chatId, question, "http://192.168.28.196:8000/v1/chat/completions");
     }
 
 }

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

@@ -16,6 +16,11 @@ public interface IMilvusService {
      */
     public JSONArray getCollectionNames();
 
+    /**
+     * 查询知识库Collection
+     */
+    public JSONArray listKnowLedgeByCollectionName(String collectionName);
+
     /**
      * 修改知识库Collection
      */

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

@@ -61,7 +61,7 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
 
     private static final MilvusClientV2 milvusClient = new MilvusClientV2(
             ConnectConfig.builder()
-                    .uri("http://192.168.28.188:19530")
+                    .uri("http://192.168.28.196:19530")
                     .build());
 
     private static final EmbeddingModel embeddingModel = new BgeSmallZhV15EmbeddingModel();
@@ -155,7 +155,7 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
     public Flux<AssistantMessage> generateAnswer(String topicId, String prompt, String llmServiceUrl) {
         ChatModel chatModel = ChatModel.of(llmServiceUrl)
                 
-                .model("Qwen2.5-1.5B-Instruct")
+                .model("Qwen2.5-7B-Instruct")
                 .build();
 
         List<ChatMessage> messages = new ArrayList<>();

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

@@ -24,7 +24,7 @@ public class MilvusServiceImpl implements IMilvusService {
 
     private static final MilvusClientV2 milvusClient = new MilvusClientV2(
             ConnectConfig.builder()
-                    .uri("http://192.168.28.188:19530")
+                    .uri("http://192.168.28.196:19530")
                     .build());
 
     /**
@@ -88,6 +88,7 @@ public class MilvusServiceImpl implements IMilvusService {
     /**
      * 查询知识库Collection
      */
+    @Override
     public JSONArray getCollectionNames() {
         JSONArray jsonArray = new JSONArray();
         ListCollectionsResp listResponse = milvusClient.listCollections();
@@ -112,9 +113,42 @@ public class MilvusServiceImpl implements IMilvusService {
         return jsonArray;
     }
 
+    /**
+     * 根据名称查询知识库Collection
+     * 返回指定名称的集合信息
+     */
+    @Override
+    public JSONArray listKnowLedgeByCollectionName(String collectionName) {
+        JSONArray jsonArray = new JSONArray();
+        ListCollectionsResp listResponse = milvusClient.listCollections();
+        
+        if (listResponse != null) {
+            List<String> allCollectionNames = listResponse.getCollectionNames();
+            for (String name : allCollectionNames) {
+                JSONObject jsonObject = new JSONObject();
+                DescribeCollectionReq request = DescribeCollectionReq.builder()
+                        .collectionName(name)
+                        .build();
+                DescribeCollectionResp describeResponse = milvusClient.describeCollection(request);
+                if (describeResponse.getDescription().contains(collectionName)) {
+                    jsonObject.put("collectionId", describeResponse.getCollectionID());
+                    jsonObject.put("collectionName", name);
+                    SimpleDateFormat beijingFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
+                    beijingFormat.setTimeZone(TimeZone.getTimeZone("Asia/Shanghai"));
+                    String beijingTime = beijingFormat.format(describeResponse.getCreateUtcTime());
+                    jsonObject.put("createdTime", beijingTime);
+                    jsonObject.put("description", describeResponse.getDescription());
+                    jsonArray.add(jsonObject);
+                }
+            }
+        }
+        return jsonArray;
+    }
+
     /**
      * 修改知识库Collection
      */
+    @Override
     public void collectionRename(String collectionName, String newCollectionName) {
         RenameCollectionReq renameCollectionReq = RenameCollectionReq.builder()
                 .collectionName(collectionName)
@@ -127,6 +161,7 @@ public class MilvusServiceImpl implements IMilvusService {
     /**
      * 删除知识库Collection
      */
+    @Override
     public void deleteCollectionName(String collectionName) {
         DropCollectionReq dropCollectionReq = DropCollectionReq.builder()
                 .collectionName(collectionName)
@@ -137,6 +172,7 @@ public class MilvusServiceImpl implements IMilvusService {
     /**
      * 查询知识库文件
      */
+    @Override
     public List<String> listDocument(String collectionName, String fileType) {
         List<String> documentList = new ArrayList<>();
         loadCollectionName(collectionName);
@@ -168,6 +204,7 @@ public class MilvusServiceImpl implements IMilvusService {
     /**
      * 删除知识库文件
      */
+    @Override
     public void removeDocument(String collectionName, String fileName) {
         loadCollectionName(collectionName);
         DeleteReq deleteReq = DeleteReq.builder()
@@ -180,6 +217,7 @@ public class MilvusServiceImpl implements IMilvusService {
     /**
      * 删除知识库所有文件
      */
+    @Override
     public void removeAllDocument(String collectionName) {
         loadCollectionName(collectionName);
         DeleteReq deleteReq = DeleteReq.builder()

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

@@ -74,11 +74,11 @@ public class CmcAgentServiceImpl implements ICmcAgentService
 
     private static final EmbeddingModel embeddingModel = new BgeSmallZhV15EmbeddingModel();
 
-    private static final String llmServiceUrl = "http://192.168.28.188:8000/v1/chat/completions";
+    private static final String llmServiceUrl = "http://192.168.28.196:8000/v1/chat/completions";
 
     private static final MilvusClientV2 milvusClient = new MilvusClientV2(
             ConnectConfig.builder()
-                    .uri("http://192.168.28.188:19530")
+                    .uri("http://192.168.28.196:19530")
                     .build());
 
     /**
@@ -116,6 +116,8 @@ public class CmcAgentServiceImpl implements ICmcAgentService
         String content = "";
         if (agentName.contains("技术"))
             content = "我是投标文件写作助手，我将助您完成技术文件部分撰写。请上传招标询价文件，分析后将依招标服务要求，运用技术方案知识库，为您提供参考。";
+        else if (agentName.contains("检查"))
+            content = "我是文档检查助手，我将助您完成错别字检查。请上传文件。";
         return content;
     }
 
@@ -142,20 +144,24 @@ public class CmcAgentServiceImpl implements ICmcAgentService
         cmcDocument.setDocumentId(new SnowFlake().generateId());
         cmcDocument.setChatId(chatId);
         cmcDocument.setPath(prefixPath + "/" + file.getOriginalFilename());
+        if (file.getOriginalFilename().endsWith(".pdf"))
+            cmcDocument.setPath(prefixPath + "/" + file.getOriginalFilename().replace(".pdf", ".docx"));
         cmcDocumentMapper.insertCmcDocument(cmcDocument);
         String message = "";
+        CmcChat cmcChat = new CmcChat();
+        cmcChat.setChatId(jsonObject.getString("chatId"));
+        cmcChat.setInputTime(new Date());
+        cmcChat.setInput("招标文件地址：" + prefixPath + "/" + file.getOriginalFilename());
+        cmcChat.setUserId(SecurityUtils.getUserId());
+        cmcChatMapper.insertCmcChat(cmcChat);
         if (agentName.contains("技术")) {
-            CmcChat cmcChat = new CmcChat();
-            cmcChat.setChatId(jsonObject.getString("chatId"));
-            cmcChat.setInputTime(new Date());
-            cmcChat.setInput("招标文件地址：" + prefixPath + "/" + file.getOriginalFilename());
-            cmcChat.setUserId(SecurityUtils.getUserId());
-            cmcChatMapper.insertCmcChat(cmcChat);
             String[] filenameSplit = file.getOriginalFilename().split("\\.");
             String outputFilename = prefixPath + "/" + file.getOriginalFilename()
                     .replace(filenameSplit[filenameSplit.length - 2], filenameSplit[filenameSplit.length - 2] + "_" + agentName);
             if (file.getOriginalFilename().endsWith(".doc"))
                 outputFilename = outputFilename.replace(".doc", ".docx");
+            if (file.getOriginalFilename().endsWith(".pdf"))
+                outputFilename = outputFilename.replace(".pdf", ".docx");
             Path outputFilePath = Paths.get(RuoYiConfig.getProfile() + outputFilename);
             Files.deleteIfExists(outputFilePath);
             InputStream fileInputStream = new FileInputStream(RuoYiConfig.getProfile() + "/upload/agent/template/technical.docx");
@@ -165,12 +171,16 @@ public class CmcAgentServiceImpl implements ICmcAgentService
                     doc.write(out);
                 }
             }
+            String question = "工作大纲/工作范围/招标范围/服务范围";
             String chapters = generateAnswerWithDocument(profilePath + "/" + file.getOriginalFilename(),
-                    RuoYiConfig.getProfile() + outputFilename, "工作大纲");
-            message = "好的，我已经收到您上传的招标文件。\n\n"+ chapters + "\n\n" +
+                    RuoYiConfig.getProfile() + outputFilename, question);
+            message = "好的，我已经收到您上传的招标文件。以下为根据招标文件生成的章节大纲：\n\n"+ chapters + "\n\n" +
                     "若您对章节标题有异议，请打开" + "【<a href='/profile" + outputFilename + "'> 技术文件 " + "</a>】" + "进行修改，后续将根据修改后的章节标题，帮您生成对应章节内容。\n\n" +
                     "思考时间可能较长，请耐心等待！\n";
         }
+        else if (agentName.contains("检查")) {
+            message = generateAnswerWithDocumentContent(profilePath + "/" + file.getOriginalFilename());
+        }
         jsonObject.put("assistantMessage", message);
         return jsonObject;
     }
@@ -203,6 +213,8 @@ public class CmcAgentServiceImpl implements ICmcAgentService
         cmcDocument.setDocumentId(new SnowFlake().generateId());
         cmcDocument.setChatId(chatId);
         cmcDocument.setPath(prefixPath + "/" + file.getOriginalFilename());
+        if (file.getOriginalFilename().endsWith(".pdf"))
+            cmcDocument.setPath(prefixPath + "/" + file.getOriginalFilename().replace(".pdf", ".docx"));
         cmcDocumentMapper.insertCmcDocument(cmcDocument);
         String message = "";
         if (agentName.contains("技术")) {
@@ -303,6 +315,22 @@ public class CmcAgentServiceImpl implements ICmcAgentService
         return cmcAgentMapper.deleteCmcAgentByAgentId(agentId);
     }
 
+    /**
+     * 调用LLM生成回答 - 检查文档内容
+     */
+    public String generateAnswerWithDocumentContent(String documentPath) throws IOException {
+        File profilePath = new File(documentPath);
+        List<TextSegment> segments = splitDocument(profilePath, 10000, 0);
+        StringBuilder sb = new StringBuilder();
+        for (TextSegment segment : segments) {
+            String prompt = "文档内容如下，帮我检查文档中用字、语言表达是否准确：\n" + segment.text();
+            sb.append("片段").append(segments.indexOf(segment) + 1).append("：\n").append(generateAnswer(prompt));
+            processValue = "检查中: " + Double.parseDouble(String.format("%.2f%n", (double) (segments.indexOf(segment) + 1)  / segments.size() * 100)) + "%";
+        }
+        sb.append("综合以上各片段检查内容修改建议，给出完整修改意见");
+        return generateAnswer(sb.toString());
+    }
+
     /**
      * 调用LLM生成回答 - 针对技术文件章节内容生成
      */
@@ -311,7 +339,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService
 
         // 构建招标文件内容的嵌入向量存储
         InMemoryEmbeddingStore<TextSegment> embeddingStore = new InMemoryEmbeddingStore<>();
-        List<TextSegment> segments = splitDocument(profilePath);
+        List<TextSegment> segments = splitDocument(profilePath, 300, 50);
         List<Embedding> embeddings = embeddingModel.embedAll(segments).content();
         embeddingStore.addAll(embeddings, segments);
 
@@ -504,7 +532,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService
         StringBuilder sb = new StringBuilder("招标文件内容：\n\n");
         File profilePath = new File(uploadFilePath);
         InMemoryEmbeddingStore<TextSegment> embeddingStore = new InMemoryEmbeddingStore<>();
-        List<TextSegment> segments = splitDocument(profilePath);
+        List<TextSegment> segments = splitDocument(profilePath, 300, 50);
         List<Embedding> embeddings = embeddingModel.embedAll(segments).content();
         embeddingStore.addAll(embeddings, segments);
         Embedding queryEmbedding = embeddingModel.embed(question).content();
@@ -519,10 +547,15 @@ public class CmcAgentServiceImpl implements ICmcAgentService
             sb.append(requests).append("\n\n");
         }
         //根据招标文件工作大纲要求生成二级标题
-        sb.append("请基于上述招标文件中提到的工作大纲内容，先列出二级章节标题，严格按以下格式，仅输出标题列表：\n")
+        sb.append("请基于上述招标文件中提到的")
+                .append(question)
+                .append("，先列出二级章节标题，严格按以下格式，仅输出标题列表：\n")
                 .append("6.1 XX\n" +
                 "6.2 XX\n" +
-                "6.3 XX");
+                "6.3 XX\n" +
+                "......\n" +
+                "6.n-1 XX\n" +
+                "6.n XX\n");
         return writeChapters(sb.toString(), templatePath);
     }
 
@@ -569,7 +602,11 @@ public class CmcAgentServiceImpl implements ICmcAgentService
                 "6.2.2 XX\n" +
                 "6.3 XX\n" +
                 "6.3.1 XX\n" +
-                "6.3.2 XX\n";
+                "6.3.2 XX\n" +
+                "...... \n" +
+                "6.n XX\n" +
+                "6.n.1 XX\n" +
+                "6.n 2 XX\n";
         String content = generateAnswer(sb);
         writeTitles(content, templatePath);
         return content;
@@ -581,7 +618,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService
      */
     public String generateAnswer(String prompt) throws IOException {
         ChatModel chatModel = ChatModel.of(llmServiceUrl)
-                .model("Qwen2.5-1.5B-Instruct")
+                .model("Qwen2.5-7B-Instruct")
                 .build();
 
         List<ChatMessage> messages = new ArrayList<>();
@@ -658,19 +695,24 @@ public class CmcAgentServiceImpl implements ICmcAgentService
         File file = new File(templatePath);
         FileInputStream fileInputStream = new FileInputStream(file);
         try (XWPFDocument document = new XWPFDocument(fileInputStream)) {
-           for (int i = 0; i < chapters.size(); i++) {
-               XWPFParagraph contentParagraph = document.createParagraph();
-               int dotNum = chapters.get(i).split("\\.").length - 1;
-               contentParagraph.setStyle(String.valueOf(dotNum + 2));
-
-               XWPFRun run = contentParagraph.createRun();
-               if (chapters.get(i).split(" ").length > 1)
-                   run.setText(chapters.get(i).split(" ")[1]);
-               if (dotNum > 1)
-                   run.addBreak();
-               if (i < chapters.size() - 1 && chapters.get(i + 1).indexOf(".") == chapters.get(i + 1).lastIndexOf("."))
-                   run.addBreak(BreakType.PAGE);
-           }
+            String otherTitles = "";
+            for (int i = 0; i < chapters.size(); i++) {
+                if (otherTitles.equals("") && chapters.get(i).split(" ")[1].startsWith("其它"))
+                    otherTitles = chapters.get(i).split(" ")[0];
+                if (!otherTitles.equals("") && chapters.get(i).split(" ")[0].startsWith(otherTitles) && !chapters.get(i).split(" ")[0].equals(otherTitles))
+                    continue;
+                XWPFParagraph contentParagraph = document.createParagraph();
+                int dotNum = chapters.get(i).split("\\.").length - 1;
+                contentParagraph.setStyle(String.valueOf(dotNum + 2));
+
+                XWPFRun run = contentParagraph.createRun();
+                if (chapters.get(i).split(" ").length > 1)
+                    run.setText(chapters.get(i).split(" ")[1]);
+                if (dotNum > 1)
+                    run.addBreak();
+                if (i < chapters.size() - 1 && chapters.get(i + 1).indexOf(".") == chapters.get(i + 1).lastIndexOf("."))
+                    run.addBreak(BreakType.PAGE);
+            }
 
             try (FileOutputStream out = new FileOutputStream(templatePath)) {
                 document.write(out);
@@ -821,7 +863,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService
     /**
      * 分割文档
      */
-    private List<TextSegment> splitDocument(File transferFile) throws IOException {
+    private List<TextSegment> splitDocument(File transferFile, int maxSegmentSizeInChars, int maxOverlapSizeInChars) throws IOException {
         // 加载文档
         Document document;
         InputStream fileInputStream = new FileInputStream(transferFile);
@@ -844,7 +886,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService
         else {
             throw new UnsupportedOperationException("不支持文件类型: " + filename);
         }
-        DocumentByParagraphSplitter splitter = new DocumentByParagraphSplitter(300,50);
+        DocumentByParagraphSplitter splitter = new DocumentByParagraphSplitter(maxSegmentSizeInChars,  maxOverlapSizeInChars);
         return splitter.split(document);
     }
 
@@ -872,17 +914,26 @@ public class CmcAgentServiceImpl implements ICmcAgentService
 
             // 将内容解析为块（普通段落/表格/标题/列表项等），然后倒序插入，确保顺序正确
             class Block {
-                String type; // para | table | h4 | list1 | list2 | imageText
+                String type; // para | table | h4 | list1 | list2 | imageText | boldOnly | numberedBold
                 String text;
                 List<String> table;
                 Integer h4Index;       // 预计算的四级标题编号
                 Integer list1Index;    // 预计算的一级列表编号 1),2),3)
                 Integer list2Index;    // 预计算的二级列表编号 (1),(2),(3)
+                String boldText;      // 加粗文本部分
+                String suffixText;    // 后缀文本（如冒号及后续内容）
+                String numberPrefix;  // 数字前缀，如 "1. "
                 Block(String t, String x) { type = t; text = x; }
                 Block(List<String> tbl) { type = "table"; table = new ArrayList<>(tbl); }
                 Block(String t, String x, Integer h4Idx, Integer l1Idx, Integer l2Idx) {
                     type = t; text = x; h4Index = h4Idx; list1Index = l1Idx; list2Index = l2Idx;
                 }
+                Block(String t, String bold, String suffix) {
+                    type = t; boldText = bold; suffixText = suffix;
+                }
+                Block(String t, String numPrefix, String bold, String suffix) {
+                    type = t; numberPrefix = numPrefix; boldText = bold; suffixText = suffix;
+                }
             }
             List<Block> blocks = new ArrayList<>();
 
@@ -890,16 +941,25 @@ public class CmcAgentServiceImpl implements ICmcAgentService
             int h4Counter = 0;
             int list1Counter = 0;
             int list2Counter = 0;
+            String previousLine = ""; // 记录上一行内容
 
             for (String raw : lines) {
                 String line = raw.trim();
                 if (line.isEmpty()) {
                     blocks.add(new Block("para", "")); // 空段落
+                    previousLine = line;
                     continue;
                 }
 
+                // 检查上一行是否以中文冒号结尾，如果是则重置列表编号
+                if (previousLine.endsWith("：") || previousLine.endsWith(":")) {
+                    list1Counter = 0;
+                    list2Counter = 0;
+                }
+
                 if (line.startsWith("#")) {
                     // 跳过前三级标题
+                    previousLine = line;
                     continue;
                 } else if (line.startsWith("####")) {
                     // 四级标题，编号采用预计算，保证倒序插入后视觉顺序仍递增
@@ -930,7 +990,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService
                     tableLines.add(line);
                 } else if (line.startsWith("- **")) {
                     // 一级加粗列表项，编号为 1),2),3) —— 预计算编号
-                    String listItem = line.replaceFirst("- \\\\*\\\\*", "").replace("**", "");
+                    String listItem = line.replaceFirst("- \\*\\*", "").replace("**", "");
                     listItem = listItem.replaceFirst("^\\*\\*", "").replace("**：", "：");
                     list1Counter++;
                     // 一级列表开始时，重置二级列表编号
@@ -939,20 +999,49 @@ public class CmcAgentServiceImpl implements ICmcAgentService
                     blocks.add(b);
                 } else if (line.startsWith("  - **")) {
                     // 二级加粗列表项，编号为 (1),(2),(3) —— 预计算编号
-                    String listItem = line.replaceFirst("  - \\\\*\\\\*", "").replace("**", "");
+                    String listItem = line.replaceFirst("  - \\*\\*", "").replace("**", "");
                     listItem = listItem.replaceFirst("^\\*\\*", "").replace("**：", "：");
                     list2Counter++;
                     Block b = new Block("list2", listItem, null, null, list2Counter);
                     blocks.add(b);
-                } else if (line.startsWith("![") && line.contains("](") && line.contains(")")) {
-                    // 图片 markdown，转成说明文本
-                    String imageText = extractImageDescription(line);
-                    blocks.add(new Block("imageText", imageText));
-                } else if (line.contains("http") && (line.contains(".png") || line.contains(".jpg") || line.contains(".jpeg") || line.contains(".gif"))) {
-                    blocks.add(new Block("imageText", "图片链接: " + line));
                 } else {
-                    blocks.add(new Block("para", line));
+                    // 匹配格式：1. **文本**（数字前缀+加粗）
+                    Pattern numberedBoldPattern = Pattern.compile("^(\\d+\\.\\s+)\\*\\*([^*]+)\\*\\*(.*)$");
+                    java.util.regex.Matcher nbm = numberedBoldPattern.matcher(line);
+                    if (nbm.matches()) {
+                        String numPrefix = nbm.group(1);
+                        String bold = nbm.group(2);
+                        String suf = nbm.group(3);
+                        if (suf == null) suf = "";
+                        blocks.add(new Block("numberedBold", numPrefix, bold, suf));
+                        previousLine = line;
+                        continue;
+                    }
+                    // 匹配格式：**文本**
+                    Pattern boldOnlyPattern = Pattern.compile("^\\*\\*([^*]+)\\*\\*(.*)$");
+                    java.util.regex.Matcher boldMatcher = boldOnlyPattern.matcher(line);
+                    if (boldMatcher.matches()) {
+                        String boldText = boldMatcher.group(1);      // "数据校验"
+                        String suffix = boldMatcher.group(2);        // "："或后续文本，可能为空
+                        if (suffix == null) suffix = "";
+                        blocks.add(new Block("boldOnly", boldText, suffix));
+                        previousLine = line;
+                        continue;
+                    }
+                    // 如果不是 boldOnly 格式，继续后续处理
+                    if (line.startsWith("![") && line.contains("](") && line.contains(")")) {
+                        // 图片 markdown，转成说明文本
+                        String imageText = extractImageDescription(line);
+                        blocks.add(new Block("imageText", imageText));
+                    } else if (line.contains("http") && (line.contains(".png") || line.contains(".jpg") || line.contains(".jpeg") || line.contains(".gif"))) {
+                        blocks.add(new Block("imageText", "图片链接: " + line));
+                    } else {
+                        blocks.add(new Block("para", line));
+                    }
                 }
+                
+                // 更新上一行记录
+                previousLine = line;
             }
             // 收尾：如果文件最后是表格仍未输出
             if (inTable && !tableLines.isEmpty()) {
@@ -998,6 +1087,45 @@ public class CmcAgentServiceImpl implements ICmcAgentService
                         run.setFontSize(12);
                         break;
                     }
+                    case "numberedBold": {
+                        XWPFParagraph p = document.insertNewParagraph(cursor);
+                        // 数字前缀（不加粗）
+                        if (b.numberPrefix != null) {
+                            XWPFRun r1 = p.createRun();
+                            r1.setText(b.numberPrefix);
+                            r1.setFontSize(12);
+                        }
+                        // 加粗文本
+                        XWPFRun r2 = p.createRun();
+                        r2.setText(b.boldText != null ? b.boldText : "");
+                        r2.setFontSize(12);
+                        r2.setBold(true);
+                        // 后缀（不加粗）
+                        if (b.suffixText != null && !b.suffixText.isEmpty()) {
+                            XWPFRun r3 = p.createRun();
+                            r3.setText(b.suffixText);
+                            r3.setFontSize(12);
+                        }
+                        break;
+                    }
+                    case "boldOnly": {
+                        // 处理格式：**文本** 或 **文本**： 转换为 Word 中的部分加粗
+                        XWPFParagraph p = document.insertNewParagraph(cursor);
+                        
+                        // 加粗文本部分
+                        XWPFRun run1 = p.createRun();
+                        run1.setText(b.boldText);
+                        run1.setFontSize(12);
+                        run1.setBold(true);
+                        
+                        // 后缀部分（不加粗，如冒号及后续内容，如果有的话）
+                        if (b.suffixText != null && !b.suffixText.isEmpty()) {
+                            XWPFRun run2 = p.createRun();
+                            run2.setText(b.suffixText);
+                            run2.setFontSize(12);
+                        }
+                        break;
+                    }
                     case "imageText": {
                         XWPFParagraph p = document.insertNewParagraph(cursor);
                         XWPFRun run = p.createRun();

llm-back/vllm_server.py

@@ -1,117 +1,227 @@
 from vllm import LLM, SamplingParams
+
 from fastapi import FastAPI, Request
+
 from fastapi.responses import StreamingResponse
+
 import uvicorn
+
 import time
+
 import json
 
+
+
 def create_vllm_server(
+
     model: str,
+
     served_model_name: str,
+
+    tensor_parallel_size: int,
+
     host: str,
+
     port: int,
-    tensor_parallel_size: int,
+
     top_p: float,
+
     temperature: float,
+
     max_tokens: int,
+
     gpu_memory_utilization: float,
+
     dtype: str,
+
 ) -> FastAPI:
+
     # 只初始化 LLM
+
     llm = LLM(
+
         model=model,
+
         tensor_parallel_size=tensor_parallel_size,
+
         gpu_memory_utilization=gpu_memory_utilization,
+
         dtype=dtype,
+
     )
 
+
+
     sampling_params = SamplingParams(
+
         temperature=temperature,
+
         top_p=top_p,
+
         max_tokens=max_tokens,
+
     )
 
+
+
     app = FastAPI()
 
+
+
     @app.post("/v1/chat/completions")
+
     async def chat_completions(request: Request):
+
         try:
+
             data = await request.json()
+
             messages = data["messages"]
+
             tools = data.get("tools")  # 支持 tools 参数
+
             created_time = time.time()
+
             request_id = f"chatcmpl-{int(time.time())}"
 
+
+
             # 调用 llm.chat()，传入 tools
+
             outputs = llm.chat(
+
                 messages=messages,
+
                 sampling_params=sampling_params,
+
                 tools=tools,
+
             )
+
             if data.get("stream"):
+
                 def generate():
+
                     full_text = ""
+
                     for output in outputs:
+
                         new_text = output.outputs[0].text[len(full_text):]
+
                         full_text = output.outputs[0].text
+
                         response_data = {
+
                             "id": request_id,
+
                             "model": served_model_name,
+
                             "created": created_time,
+
                             "choices": [{
+
                                 "index": 0,
+
                                 "delta": {"content": new_text},
+
                                 "finish_reason": output.outputs[0].finish_reason,
+
                             }],
+
                         }
+
                         yield f"data: {json.dumps(response_data)}\n\n"
+
                     yield "data: [DONE]\n\n"
 
+
+
                 return StreamingResponse(generate(), media_type="text/event-stream")
+
             else:
+
                 return {
+
                     "id": request_id,
+
                     "model": served_model_name,
+
                     "created": created_time,
+
                     "choices": [{
+
                         "index": 0,
+
                         "message": {
+
                             "role": "assistant",
+
                             "content": outputs[0].outputs[0].text,
+
                         },
+
                         "finish_reason": outputs[0].outputs[0].finish_reason,
+
                     }],
+
                 }
 
+
+
         except Exception as e:
+
             return {"error": str(e)}, 400
 
+
+
     return app
 
+
+
 if __name__ == "__main__":
+
     # 配置参数
+
     CONFIG = {
-        "model": "/mnt/d/Qwen/Qwen2.5-1.5B-Instruct",
-        "served_model_name": "Qwen2.5-1.5B-Instruct",
-        # "model": "/mnt/d/Deepseek/DeepSeek-R1-Distill-Qwen-1.5B",
-        # "served_model_name": "DeepSeek-R1-Distill-Qwen-1.5B",
-        "host": "172.25.231.226",
+
+        "model": "/home/Qwen/Qwen2.5-7B-Instruct",
+
+        "served_model_name": "Qwen2.5-7B-Instruct",
+
+        "host": "192.168.28.196",
+
         "port": 8000,
-        "tensor_parallel_size": 1,
+
+        "tensor_parallel_size": 4,
+
         "top_p": 0.9,
+
         "temperature": 0.7,
+
         "max_tokens": 8192,
+
         "gpu_memory_utilization": 0.9,
+
         "dtype": "float16"
+
     }
-    
+
     # 创建应用
+
     app = create_vllm_server(**CONFIG)
+
     
+
     # 启动服务器
+
     uvicorn.run(
+
         app,
+
         host=CONFIG["host"],
+
         port=CONFIG["port"],
+
         workers= 1,
-    )
+
+    )

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

@@ -7,11 +7,19 @@
 import request from '@/utils/request'
 
 // 查询知识库列表
-export function listKnowledge(query) {
+export function listKnowledge() {
   return request({
     url: '/llm/knowledge/list',
     method: 'get',
-    params: query
+  })
+}
+
+// 根据collectionName查询知识库列表
+export function listKnowLedgeByCollectionName(collectionName) {
+  return request({
+    url: '/llm/knowledge/listByName',
+    method: 'get',
+    params: collectionName
   })
 }
 

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

@@ -2,13 +2,13 @@
  * @Author: wrh
  * @Date: 2025-01-01 00:00:00
  * @LastEditors: wrh
- * @LastEditTime: 2025-09-11 15:58:35
+ * @LastEditTime: 2025-11-03 11:28:40
 -->
 <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,7 +21,7 @@
             </span>
           </div>
         </div>
-      </div>
+      </div> -->
 
       <!-- 对话区域 -->
       <div class="chat-section">
@@ -81,7 +81,7 @@
               </div>
               <div class="message-content">
                 <div class="message-upload-area">
-                  <p class="upload-tip">请上传您需要分析的招标文件（单个文件）：</p>
+                  <p class="upload-tip"> {{ selectDocumentTip }}</p>
                   <el-upload class="inline-upload" :action="uploadAction" :multiple="false" :auto-upload="false"
                     :file-list="chatFileList" :on-change="handleChatFileChange" :before-upload="beforeUpload"
                     :show-file-list="true" :limit="1">
@@ -89,7 +89,7 @@
                       <el-icon>
                         <Upload />
                       </el-icon>
-                      选择招标文件
+                      {{ selectDocument }}
                     </el-button>
                   </el-upload>
                   <div v-if="chatFileList.length > 0" class="chat-upload-actions">
@@ -249,6 +249,8 @@ const percentage = ref(0);
 const progress = ref("");
 const percentageDisplay = ref(false);
 const techUploadDisplay = ref(false)
+const selectDocument = ref('')
+const selectDocumentTip = ref('')
 
 // 聊天内文件上传相关
 const chatFileList = ref([]) // 聊天内的文件列表
@@ -284,6 +286,14 @@ const loadAgentDetail = async (agentId) => {
     if (agentInfo.value?.agentName) {
       const res = await opening(agentInfo.value.agentName)
       openingMessage.value = res.data.resultContent;
+      if (agentInfo.value.agentName.includes('技术文件')) {
+        selectDocument.value = '选择招标文件'
+        selectDocumentTip.value = '请上传您需要分析的招标文件（单个文件）：'
+      }
+      else if (agentInfo.value.agentName.includes('文档检查')) {
+        selectDocument.value = '选择检查文件'
+        selectDocumentTip.value = '请上传您需要检查的文件（单个文件）：'
+      }
     }
     // 查询当前智能体的话题列表
     await loadTopic()
@@ -422,6 +432,7 @@ const handleDeleteTopic = async (topic) => {
     // 重新加载话题列表
     await loadTopic()
 
+    techUploadDisplay.value = false;
     ElMessage.success('话题删除成功')
   } catch (error) {
     if (error !== 'cancel') {
@@ -609,7 +620,8 @@ const submitChatUpload = async () => {
           isHtml: true // 标记这是HTML内容
         }
         chatMessages.value.push(uploadMessage);
-        techUploadDisplay.value = true;
+        if (agentInfo.value.agentName.includes('技术文件'))
+          techUploadDisplay.value = true;
         // ElMessage.success('文件上传成功');
         let topicRes = await addTopic({ agentId: props.agentId, topic: fileName });
         const topicId = topicRes.msg;
@@ -748,7 +760,7 @@ const startNewChat = () => {
       messagesContainer.value.scrollTop = 0
     }
   })
-
+  techUploadDisplay.value = false
   ElMessage.success('已切换到新对话模式')
 }
 
@@ -862,7 +874,7 @@ const formatContentLinks = (content) => {
   border: 1px solid #e4e4e4;
   overflow: hidden;
   display: flex;
-  height: calc(100vh - 200px);
+  height: calc(100vh - 150px);
 
   // 话题列表样式
   .topic-list {

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

@@ -1,108 +1,143 @@
 <!--
  * @Author: ysh
  * @Date: 2025-07-17 18:16:50
- * @LastEditors: Please set LastEditors
- * @LastEditTime: 2025-07-28 14:46:01
+ * @LastEditors: wrh
+ * @LastEditTime: 2025-11-13 16:24:00
 -->
 <template>
-  <div class="agent-container">
-    <!-- 左侧智能体列表 -->
-    <div class="agent-list">
-      <div class="list-header">
-        <h3>智能体列表</h3>
-        <el-button type="primary" size="small" @click="openAddDialog">
-          <el-icon>
-            <Plus />
-          </el-icon>
-          新增
-        </el-button>
-      </div>
+  <div class="app-container">
+    <div class="main-content">
+      <!-- 左侧智能体列表 -->
+      <div class="left-panel">
+        <div class="panel-header">
+          <h3>智能体列表</h3>
+          <span class="agent-count">{{ agentList.length }} 个智能体</span>
+          <el-button type="primary" size="small" @click="openAddDialog">
+            <el-icon>
+              <Plus />
+            </el-icon>
+            新增智能体
+          </el-button>
+        </div>
 
-      <!-- 搜索框 -->
-      <div class="search-box">
-        <el-input v-model="queryParams.agentName" placeholder="请输入智能体名称" clearable @clear="handleQuery"
-          @keyup.enter="handleQuery">
-          <template #append>
-            <el-button @click="handleQuery">
-              <el-icon>
-                <Search />
-              </el-icon>
-            </el-button>
-          </template>
-        </el-input>
-      </div>
+        <!-- 搜索框 -->
+        <div class="search-box">
+          <el-input v-model="queryParams.agentName" placeholder="请输入智能体名称" clearable @clear="handleQuery"
+            @keyup.enter="handleQuery">
+            <template #append>
+              <el-button @click="handleQuery">
+                <el-icon>
+                  <Search />
+                </el-icon>
+              </el-button>
+            </template>
+          </el-input>
+        </div>
 
-      <!-- 智能体列表 -->
-      <div class="list-content" v-loading="loading">
-        <div v-for="agent in agentList" :key="agent.agentId" class="agent-item"
-          :class="{ 'active': selectedAgentId === agent.agentId }" @click="selectAgent(agent)">
-          <div class="agent-info">
-            <div class="agent-name">{{ agent.agentName }}</div>
-            <div class="agent-desc">{{ agent.description || '暂无描述' }}</div>
-            <div class="agent-meta">
-              <span class="create-time">{{ agent.createTime }}</span>
+        <!-- 智能体列表 -->
+        <div class="agent-cards" v-loading="loading">
+          <div v-for="agent in agentList" :key="agent.agentId" class="agent-card"
+            :class="{ 'active': selectedAgentId === agent.agentId }" @click="selectAgent(agent)">
+            <div class="card-header">
+              <div class="card-title">
+                <el-icon class="folder-icon">
+                  <Folder />
+                </el-icon>
+                <span class="title-text">{{ agent.agentName }}</span>
+              </div>
+              <div class="card-actions">
+                <el-dropdown @command="handleCommand" trigger="click">
+                  <el-button type="text" size="small">
+                    <el-icon>
+                      <MoreFilled />
+                    </el-icon>
+                  </el-button>
+                  <template #dropdown>
+                    <el-dropdown-menu>
+                      <el-dropdown-item :command="{ action: 'edit', agent }">编辑</el-dropdown-item>
+                      <el-dropdown-item :command="{ action: 'delete', agent }" divided>删除</el-dropdown-item>
+                    </el-dropdown-menu>
+                  </template>
+                </el-dropdown>
+              </div>
+            </div>
+            <div class="card-content">
+              <p class="description">{{ agent.description || '暂无描述' }}</p>
+              <div class="meta-info">
+                <span class="create-time">{{ agent.createTime }}</span>
+              </div>
             </div>
           </div>
-          <div class="agent-actions">
-            <el-dropdown @command="handleCommand" trigger="click">
-              <el-icon class="action-icon">
-                <More />
-              </el-icon>
-              <template #dropdown>
-                <el-dropdown-menu>
-                  <el-dropdown-item :command="{ action: 'edit', agent }">编辑</el-dropdown-item>
-                  <el-dropdown-item :command="{ action: 'delete', agent }" divided>删除</el-dropdown-item>
-                </el-dropdown-menu>
-              </template>
-            </el-dropdown>
-          </div>
-        </div>
 
-        <!-- 空状态 -->
-        <div v-if="agentList.length === 0 && !loading" class="empty-state">
-          <el-empty description="暂无智能体数据">
-            <el-button type="primary" @click="dialogVisible = true">创建智能体</el-button>
-          </el-empty>
+
+          <!-- 空状态 -->
+          <div v-if="agentList.length === 0 && !loading" class="empty-state">
+            <el-icon class="empty-icon">
+              <FolderOpened />
+            </el-icon>
+            <p>暂无智能体</p>
+            <el-button type="primary" @click="openAddDialog">创建第一个智能体</el-button>
+          </div>
         </div>
       </div>
 
-      <!-- 分页 -->
-      <div class="pagination-wrapper" v-if="total > 0">
-        <el-pagination v-model:current-page="queryParams.pageNum" v-model:page-size="queryParams.pageSize"
-          :total="total" :page-sizes="[10, 20, 50]" layout="sizes, prev, pager, next" small @size-change="handleQuery"
-          @current-change="handleQuery" />
+      <!-- 右侧详细内容 -->
+      <div class="agent-detail">
+        <AgentDetail :agent-id="selectedAgentId" />
       </div>
-    </div>
 
-    <!-- 右侧详细内容 -->
-    <div class="agent-detail">
-      <AgentDetail :agent-id="selectedAgentId" />
+      <el-dialog
+        v-model="agentDialogVisible"
+        :title="dialogTitle"
+        width="500px"
+        @close="resetForm">
+        <el-form
+          :model="form"
+          :rules="rules"
+          label-width="80px">
+          <el-form-item label="智能体名称" prop="agentName">
+            <el-input v-model="form.agentName" placeholder="请输入智能体名称" maxlength="20" show-word-limit />
+          </el-form-item>
+          <el-form-item label="描述" prop="description">
+            <el-input v-model="form.description" type="textarea" placeholder="请输入描述" :rows="2" maxlength="200" show-word-limit />
+          </el-form-item>
+          <el-form-item label="角色" prop="role">
+            <el-input v-model="form.role" placeholder="请输入角色" maxlength="50" show-word-limit />
+          </el-form-item>
+          <el-form-item label="初始提示词" prop="initPrompt">
+            <el-input v-model="form.initPrompt" type="textarea" placeholder="请输入初始提示词" :rows="4" maxlength="1000" show-word-limit />
+          </el-form-item>
+          <el-form-item label="模型" prop="modelName">
+            <el-select v-model="form.modelName" placeholder="请选择模型" clearable>
+              <el-option
+                v-for="model in modelList"
+                :key="model.name"
+                :label="model.name"
+                :value="model.name">
+              </el-option>
+            </el-select>
+          </el-form-item>
+        </el-form>
+        <template #footer>
+          <span class="dialog-footer">
+            <el-button @click="closeDialog">取消</el-button>
+            <el-button type="primary" @click="submitForm">确定</el-button>
+          </span>
+        </template>
+      </el-dialog>
     </div>
-
-    <el-dialog v-model="dialogVisible" :title="isEdit ? '编辑智能体' : '新增智能体'" width="30%" @close="resetForm">
-      <el-form :model="form" :rules="rules" label-width="120px">
-        <el-form-item label="智能体名称" prop="agentName">
-          <el-input v-model="form.agentName" placeholder="请输入智能体名称" />
-        </el-form-item>
-        <el-form-item label="智能体描述" prop="description">
-          <el-input type="textarea" v-model="form.description" placeholder="请输入智能体描述" />
-        </el-form-item>
-      </el-form>
-      <template #footer>
-        <el-button @click="() => { dialogVisible = false; resetForm(); }">取消</el-button>
-        <el-button type="primary" @click="handleSubmit">{{ isEdit ? '保存' : '确定' }}</el-button>
-      </template>
-    </el-dialog>
   </div>
 </template>
 
 <script setup>
 import { ref, reactive, onMounted } from 'vue'
 import { ElMessage, ElMessageBox } from 'element-plus'
+import { Search, Plus, MoreFilled, Folder, FolderOpened } from '@element-plus/icons-vue'
 import { listAgent, addAgent, updateAgent, delAgent, opening } from '@/api/llm/agent'
 import { answer } from '@/api/llm/mcp'
 import AgentDetail from './AgentDetail.vue'
 
+const open = ref(false);
 // 响应式数据
 const loading = ref(false)
 const agentList = ref([])
@@ -158,33 +193,62 @@ const resetForm = () => {
   editAgentId.value = null
   form.agentName = ''
   form.description = ''
+  form.avatar = '';
+  form.role = '';
+  form.initPrompt = '';
+  form.modelName = '';
 }
 
-const openAddDialog = () => {
-  resetForm()
-  dialogVisible.value = true
-}
+// 新增智能体对话框控制变量
+const agentDialogVisible = ref(false);
+const dialogTitle = ref('');
+const isModifyAgent = ref(false);
 
-const handleSubmit = () => {
-  if (isEdit.value) {
-    updateAgent({ agentId: editAgentId.value, ...form }).then(res => {
-      if (res.code === 200) {
-        ElMessage.success('编辑成功')
-        dialogVisible.value = false
-        isEdit.value = false
-        editAgentId.value = null
-        getList()
-      }
-    })
-  } else {
-    addAgent(form).then(res => {
-      if (res.code === 200) {
-        ElMessage.success('新增成功')
-        dialogVisible.value = false
-        getList()
+// 打开新增智能体对话框
+const openAddDialog = () => {
+  agentDialogVisible.value = true;
+  dialogTitle.value = "添加智能体";
+  isModifyAgent.value = false;
+  // 重置表单
+  resetForm();
+};
+
+// 关闭对话框
+const closeDialog = () => {
+  agentDialogVisible.value = false;
+  resetForm();
+};
+
+// 提交表单
+const submitForm = () => {
+  const formRef = { value: { validate: (callback) => callback(true) } };
+  formRef.value.validate((valid) => {
+    if (valid) {
+      loading.value = true
+      const agentForm = { ...form }
+      if (isModifyAgent.value) {
+        updateAgent(agentForm).then(() => {
+          ElMessage.success('修改成功')
+          loading.value = false
+          agentDialogVisible.value = false
+          getList()
+        }).catch(() => {
+          loading.value = false
+          ElMessage.error('修改失败')
+        })
+      } else {
+        addAgent(agentForm).then(() => {
+          ElMessage.success('新增成功')
+          loading.value = false
+          agentDialogVisible.value = false
+          getList()
+        }).catch(() => {
+          loading.value = false
+          ElMessage.error('新增失败')
+        })
       }
-    })
-  }
+    }
+  })
 }
 
 // 操作菜单处理
@@ -234,114 +298,181 @@ onMounted(() => {
 </script>
 
 <style lang="scss" scoped>
-.agent-container {
+.app-container {
+  height: 100vh;
+  display: flex;
+  flex-direction: column;
+  background: #f5f7fa;
+}
+
+.main-content {
+  flex: 1;
   display: flex;
-  height: calc(100vh - 120px);
-  background: #f5f5f5;
+  gap: 16px;
+  padding: 16px;
+  overflow: hidden;
 }
 
-.agent-list {
-  width: 300px;
+.left-panel {
+  width: 400px;
   background: white;
-  border-right: 1px solid #e4e4e4;
+  border-radius: 8px;
+  box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1);
   display: flex;
   flex-direction: column;
+  overflow: hidden;
+}
 
-  .list-header {
-    padding: 16px;
-    border-bottom: 1px solid #e4e4e4;
-    display: flex;
-    justify-content: space-between;
-    align-items: center;
+.right-panel {
+  flex: 1;
+  background: white;
+  border-radius: 8px;
+  box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1);
+  display: flex;
+  flex-direction: column;
+  overflow: hidden;
+}
 
-    h3 {
-      margin: 0;
-      font-size: 16px;
-      font-weight: 500;
-    }
+.panel-header {
+  padding: 20px 24px;
+  border-bottom: 1px solid #e4e7ed;
+  display: flex;
+  justify-content: space-between;
+  align-items: center;
+  background: #fafbfc;
+
+  h3 {
+    margin: 0;
+    font-size: 16px;
+    font-weight: 600;
+    color: #303133;
+  }
+
+  .agent-count {
+    font-size: 12px;
+    color: #909399;
+    background: #f0f2f5;
+    padding: 4px 8px;
+    border-radius: 4px;
   }
 
   .search-box {
     padding: 16px;
     border-bottom: 1px solid #e4e4e4;
   }
+}
+
+.agent-cards {
+  flex: 1;
+  padding: 16px;
+  overflow-y: auto;
+  display: flex;
+  flex-direction: column;
+  gap: 12px;
+}
+
+.agent-card {
+  background: white;
+  border: 1px solid #e4e7ed;
+  border-radius: 8px;
+  padding: 16px;
+  cursor: pointer;
+  transition: all 0.3s ease;
+  position: relative;
+
+  &:hover {
+    border-color: #409eff;
+    box-shadow: 0 4px 12px rgba(64, 158, 255, 0.15);
+    transform: translateY(-2px);
+  }
 
-  .list-content {
-    flex: 1;
-    overflow-y: auto;
+  &.active {
+    border-color: #409eff;
+    background: #f0f9ff;
+    box-shadow: 0 4px 12px rgba(64, 158, 255, 0.2);
+  }
 
-    .agent-item {
-      padding: 12px 16px;
-      border-bottom: 1px solid #f0f0f0;
-      cursor: pointer;
-      transition: all 0.2s;
+  .card-header {
+    display: flex;
+    justify-content: space-between;
+    align-items: flex-start;
+    margin-bottom: 12px;
+
+    .card-title {
       display: flex;
-      justify-content: space-between;
       align-items: center;
+      gap: 8px;
+      flex: 1;
 
-      &:hover {
-        background: #f8f9fa;
+      .folder-icon {
+        color: #409eff;
+        font-size: 18px;
       }
 
-      &.active {
-        background: #e6f7ff;
-        border-right: 3px solid #1890ff;
+      .title-text {
+        font-weight: 600;
+        color: #303133;
+        font-size: 14px;
+        line-height: 1.4;
       }
+    }
 
-      .agent-info {
-        flex: 1;
-
-        .agent-name {
-          font-size: 14px;
-          font-weight: 500;
-          color: #333;
-          margin-bottom: 4px;
-        }
+    .card-actions {
+      opacity: 0;
+      transition: opacity 0.3s ease;
+    }
+  }
 
-        .agent-desc {
-          font-size: 12px;
-          color: #666;
-          margin-bottom: 4px;
-          overflow: hidden;
-          text-overflow: ellipsis;
-          white-space: nowrap;
-        }
+  &:hover .card-actions {
+    opacity: 1;
+  }
 
-        .agent-meta {
-          font-size: 11px;
-          color: #999;
+  .card-content {
+    .description {
+      color: #606266;
+      font-size: 13px;
+      line-height: 1.5;
+      margin: 0 0 12px 0;
+      display: -webkit-box;
+      -webkit-line-clamp: 2;
+      -webkit-box-orient: vertical;
+      overflow: hidden;
+    }
 
-          .create-time {
-            margin-right: 8px;
-          }
-        }
-      }
+    .meta-info {
+      display: flex;
+      justify-content: space-between;
+      align-items: center;
+      font-size: 12px;
+      color: #909399;
 
-      .agent-actions {
-        .action-icon {
-          color: #999;
-          font-size: 16px;
-          padding: 4px;
-          border-radius: 4px;
-
-          &:hover {
-            background: #f0f0f0;
-            color: #666;
-          }
-        }
+      .create-time {
+        background: #f0f2f5;
+        padding: 2px 6px;
+        border-radius: 3px;
       }
     }
+  }
+}
 
-    .empty-state {
-      padding: 40px 20px;
-      text-align: center;
-    }
+.empty-state {
+  display: flex;
+  flex-direction: column;
+  align-items: center;
+  justify-content: center;
+  padding: 60px 20px;
+  text-align: center;
+  color: #909399;
+
+  .empty-icon {
+    font-size: 48px;
+    margin-bottom: 16px;
+    opacity: 0.5;
   }
 
-  .pagination-wrapper {
-    padding: 16px;
-    border-top: 1px solid #e4e4e4;
-    background: white;
+  p {
+    margin: 0 0 16px 0;
+    font-size: 14px;
   }
 }
 
@@ -349,12 +480,4 @@ onMounted(() => {
   flex: 1;
   background: white;
 }
-
-:deep(.el-pagination) {
-  justify-content: center;
-
-  .el-pagination__sizes {
-    margin-right: 8px;
-  }
-}
 </style>

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

@@ -1,25 +1,5 @@
 <template>
   <div class="app-container">
-    <!-- 顶部搜索和操作栏 -->
-    <div class="top-toolbar">
-      <el-form :model="queryParams" ref="queryForm" :inline="true" v-show="showSearch" label-width="100px">
-        <el-form-item label="知识库名称" prop="collectionName">
-          <el-input v-model="queryParams.collectionName" placeholder="请输入知识库名称" clearable @keyup.enter="handleQuery" />
-        </el-form-item>
-        <el-form-item>
-          <el-button type="primary" :icon="Search" @click="handleQuery">搜索</el-button>
-          <el-button :icon="Refresh" @click="resetQuery">重置</el-button>
-        </el-form-item>
-      </el-form>
-
-      <div class="toolbar-actions">
-        <el-button type="primary" plain :icon="Plus" @click="handleAdd" v-hasPermi="['llm:knowledge:add']">
-          新增知识库
-        </el-button>
-        <right-toolbar v-model:showSearch="showSearch" @queryTable="getList"></right-toolbar>
-      </div>
-    </div>
-
     <!-- 主要内容区域 -->
     <div class="main-content">
       <!-- 左侧知识库列表 -->
@@ -27,6 +7,26 @@
         <div class="panel-header">
           <h3>知识库列表</h3>
           <span class="knowledge-count">{{ knowledgeList.length }} 个知识库</span>
+          <el-button type="primary" size="small" @click="handleAdd" v-hasPermi="['llm:knowledge:add']">
+            <el-icon>
+              <Plus />
+            </el-icon>
+            新增知识库
+          </el-button>
+        </div>
+
+        <!-- 搜索框 -->
+        <div class="search-box">
+          <el-input v-model="queryParams.collectionName" placeholder="请输入知识库名称" clearable @clear="handleQuery"
+            @keyup.enter="handleQuery">
+            <template #append>
+              <el-button @click="handleQuery">
+                <el-icon>
+                  <Search />
+                </el-icon>
+              </el-button>
+            </template>
+          </el-input>
         </div>
 
         <div class="knowledge-cards" v-loading="loading">
@@ -69,13 +69,13 @@
             </div>
           </div>
 
-          <!-- 空状态 -->
+          <!-- 空状态 - 无论是否搜索，当列表为空且未加载时都显示创建按钮 -->
           <div v-if="knowledgeList.length === 0 && !loading" class="empty-state">
             <el-icon class="empty-icon">
               <FolderOpened />
             </el-icon>
-            <p>暂无知识库</p>
-            <el-button type="primary" @click="handleAdd">创建第一个知识库</el-button>
+            <p>{{ queryParams.collectionName ? '搜索结果为空' : '暂无知识库' }}</p>
+            <el-button type="primary" @click="handleAdd">{{ queryParams.collectionName ? '创建知识库' : '创建第一个知识库' }}</el-button>
           </div>
         </div>
       </div>
@@ -333,7 +333,7 @@
 <script setup>
 import { ref, reactive, getCurrentInstance, onMounted, onUnmounted, nextTick, watch, computed } from 'vue'
 import { Search, Refresh, Plus, Edit, Delete, Upload, UploadFilled, Folder, MoreFilled, FolderOpened, Document, ChatRound, User, ChatDotRound, Promotion } from '@element-plus/icons-vue'
-import { listKnowledge, addKnowledge, updateKnowledge, delKnowledge, insertKnowledgeFile, listKnowledgeDocument, deleteKnowledgeFile } from "@/api/llm/knowLedge";
+import { listKnowledge, listKnowLedgeByCollectionName, addKnowledge, updateKnowledge, delKnowledge, insertKnowledgeFile, listKnowledgeDocument, deleteKnowledgeFile } from "@/api/llm/knowLedge";
 import { getAnswer, getAnswerStream,getContextFile } from '@/api/llm/rag';
 import { getToken } from "@/utils/auth";
 
@@ -418,25 +418,46 @@ const uploadRules = reactive({
 /** 查询知识库列表 */
 function getList() {
   loading.value = true;
-  listKnowledge().then(response => {
-    // 确保返回的数据是数组格式
-    if (Array.isArray(response.data)) {
-      knowledgeList.value = response.data;
-      knowledgeList.value.map(item => {
-        listKnowledgeDocument(item.collectionName).then(response => {
-          item.fileCount = response.data.length;
+  // 将查询参数传递给API调用
+  if (queryParams.collectionName == '') {
+    listKnowledge().then(response => {
+      // 确保返回的数据是数组格式
+      if (Array.isArray(response.data)) {
+        knowledgeList.value = response.data;
+        knowledgeList.value.map(item => {
+          listKnowledgeDocument(item.collectionName).then(response => {
+            item.fileCount = response.data.length;
+          })
         })
-      })
-    } else {
-      knowledgeList.value = [];
-    }
-    loading.value = false;
-  }).catch(error => {
-    loading.value = false;
-    proxy.$modal.msgError("获取知识库列表失败：" + error.message);
-  });
+      } else {
+        knowledgeList.value = [];
+      }
+      loading.value = false;
+    }).catch(error => {
+      loading.value = false;
+      proxy.$modal.msgError("获取知识库列表失败：" + error.message);
+    });
+  }
+  else {
+    listKnowLedgeByCollectionName({collectionName: queryParams.collectionName}).then(response => {
+      // 确保返回的数据是数组格式
+      if (Array.isArray(response.data)) {
+        knowledgeList.value = response.data;
+        knowledgeList.value.map(item => {
+          listKnowledgeDocument(item.collectionName).then(response => {
+            item.fileCount = response.data.length;
+          })
+        })
+      } else {
+        knowledgeList.value = [];
+      }
+      loading.value = false;
+    }).catch(error => {
+      loading.value = false;
+      proxy.$modal.msgError("获取知识库列表失败：" + error.message);
+    });
+  }
 }
-
 /** 选择知识库 */
 function selectKnowledge(knowledge) {
   // 如果正在生成回答，先停止
@@ -832,6 +853,8 @@ function reset() {
 
 /** 搜索按钮操作 */
 function handleQuery() {
+  // 重置页码
+  queryParams.pageNum = 1;
   getList();
 }
 
@@ -1006,20 +1029,6 @@ onUnmounted(() => {
   background: #f5f7fa;
 }
 
-.top-toolbar {
-  background: white;
-  padding: 16px 24px;
-  border-bottom: 1px solid #e4e7ed;
-  box-shadow: 0 2px 4px rgba(0, 0, 0, 0.05);
-
-  .toolbar-actions {
-    margin-top: 12px;
-    display: flex;
-    gap: 8px;
-    align-items: center;
-  }
-}
-
 .main-content {
   flex: 1;
   display: flex;
@@ -1071,6 +1080,11 @@ onUnmounted(() => {
     border-radius: 4px;
   }
 
+  .search-box {
+    padding: 16px;
+    border-bottom: 1px solid #e4e4e4;
+  }
+  
   .header-actions {
     display: flex;
     gap: 8px;