批量导入、批量删除文件

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

@@ -1,97 +0,0 @@
-package com.ruoyi.web.llm.controller;
-
-import java.util.List;
-import javax.servlet.http.HttpServletResponse;
-import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.PostMapping;
-import org.springframework.web.bind.annotation.PutMapping;
-import org.springframework.web.bind.annotation.DeleteMapping;
-import org.springframework.web.bind.annotation.PathVariable;
-import org.springframework.web.bind.annotation.RequestBody;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
-import com.ruoyi.common.annotation.Log;
-import com.ruoyi.common.core.controller.BaseController;
-import com.ruoyi.common.core.domain.AjaxResult;
-import com.ruoyi.common.enums.BusinessType;
-import com.ruoyi.llm.domain.CmcAgentTool;
-import com.ruoyi.llm.service.ICmcAgentToolService;
-import com.ruoyi.common.utils.poi.ExcelUtil;
-import com.ruoyi.common.core.page.TableDataInfo;
-
-/**
- * 智能体工具Controller
- * 
- * @author ruoyi
- * @date 2025-07-17
- */
-@RestController
-@RequestMapping("/llm/tool")
-public class CmcAgentToolController extends BaseController
-{
-    @Autowired
-    private ICmcAgentToolService cmcAgentToolService;
-
-    /**
-     * 查询智能体工具列表
-     */
-    @GetMapping("/list")
-    public TableDataInfo list(CmcAgentTool cmcAgentTool)
-    {
-        startPage();
-        List<CmcAgentTool> list = cmcAgentToolService.selectCmcAgentToolList(cmcAgentTool);
-        return getDataTable(list);
-    }
-
-    /**
-     * 导出智能体工具列表
-     */
-    @Log(title = "智能体工具", businessType = BusinessType.EXPORT)
-    @PostMapping("/export")
-    public void export(HttpServletResponse response, CmcAgentTool cmcAgentTool)
-    {
-        List<CmcAgentTool> list = cmcAgentToolService.selectCmcAgentToolList(cmcAgentTool);
-        ExcelUtil<CmcAgentTool> util = new ExcelUtil<CmcAgentTool>(CmcAgentTool.class);
-        util.exportExcel(response, list, "智能体工具数据");
-    }
-
-    /**
-     * 获取智能体工具详细信息
-     */
-    @GetMapping(value = "/{agentToolId}")
-    public AjaxResult getInfo(@PathVariable("agentToolId") Integer agentToolId)
-    {
-        return success(cmcAgentToolService.selectCmcAgentToolByAgentToolId(agentToolId));
-    }
-
-    /**
-     * 新增智能体工具
-     */
-    @Log(title = "智能体工具", businessType = BusinessType.INSERT)
-    @PostMapping
-    public AjaxResult add(@RequestBody CmcAgentTool cmcAgentTool)
-    {
-        return toAjax(cmcAgentToolService.insertCmcAgentTool(cmcAgentTool));
-    }
-
-    /**
-     * 修改智能体工具
-     */
-    @Log(title = "智能体工具", businessType = BusinessType.UPDATE)
-    @PutMapping
-    public AjaxResult edit(@RequestBody CmcAgentTool cmcAgentTool)
-    {
-        return toAjax(cmcAgentToolService.updateCmcAgentTool(cmcAgentTool));
-    }
-
-    /**
-     * 删除智能体工具
-     */
-    @Log(title = "智能体工具", businessType = BusinessType.DELETE)
-	@DeleteMapping("/{agentToolIds}")
-    public AjaxResult remove(@PathVariable Integer[] agentToolIds)
-    {
-        return success(cmcAgentToolService.deleteCmcAgentToolByAgentToolIds(agentToolIds));
-    }
-}

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

@@ -79,6 +79,7 @@ public class KnowLedgeController extends BaseController
     @DeleteMapping("/remove")
     public AjaxResult removeKnowLedgeCollection(String collectionName)
     {
+        milvusService.removeAllDocument(milvusClient, collectionName);
         milvusService.deleteCollectionName(milvusClient, collectionName);
         return success();
     }
@@ -97,10 +98,12 @@ public class KnowLedgeController extends BaseController
      * 导入知识库文件
      */
     @PostMapping("/insertDocument")
-    public AjaxResult insertKnowledgeDocument(MultipartFile file, String collectionName) throws IOException {
-        R<MutationResult> insertResult = langChainMilvusService.insertLangchainEmbeddingDocument(milvusClient, file, collectionName, embeddingModel);
+    public AjaxResult insertKnowledgeDocument(MultipartFile[] fileList, String collectionName) throws IOException {
+        R<MutationResult> insertResult = null;
+        for (MultipartFile file : fileList)
+            insertResult = langChainMilvusService.insertLangchainEmbeddingDocument(milvusClient, file, collectionName, embeddingModel);
         String message = "文件导入成功";
-        if (insertResult.getStatus() != R.Status.Success.getCode()) {
+        if (insertResult != null && insertResult.getStatus() != R.Status.Success.getCode()) {
             message = "文件导入失败：" + insertResult.getMessage();
             return error(message);
         }

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

@@ -38,4 +38,9 @@ public interface IMilvusService {
      * 删除知识库文件
      */
     public void removeDocument(MilvusClient milvusClient, String collectionName, String fileName);
+
+    /**
+     * 删除知识库所有文件
+     */
+    public void removeAllDocument(MilvusClient milvusClient, String collectionName);
 }

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

@@ -52,7 +52,7 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
     @Override
     public R<MutationResult> insertLangchainEmbeddingDocument(MilvusClient milvusClient, MultipartFile file, String collectionName, EmbeddingModel embeddingModel) throws IOException
     {
-        File profilePath = new File( RuoYiConfig.getProfile() + "/upload/knowledge");
+        File profilePath = new File( RuoYiConfig.getProfile() + "/upload/knowledge/" + collectionName);
         if (!profilePath.exists())
             profilePath.mkdirs();
         File transferFile = new File( profilePath + File.separator + file.getOriginalFilename());
@@ -75,7 +75,7 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
         else {
             throw new UnsupportedOperationException("不支持文件类型: " + filename);
         }
-        DocumentByParagraphSplitter splitter = new DocumentByParagraphSplitter(1024,100);;
+        DocumentByParagraphSplitter splitter = new DocumentByParagraphSplitter(1000,200);;
         List<TextSegment> segments = splitter.split(document);
 
         // 提取文本和生成嵌入
@@ -208,7 +208,7 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
                 .withOutFields(Arrays.asList("file_name", "file_type", "content"))
                 .withVectorFieldName("embedding")
                 .withMetricType(MetricType.COSINE)
-                .withParams("{\"nprobe\": 1}")
+                .withParams("{\"nprobe\": 8}")
                 .build();
 
         R<SearchResults> response = milvusClient.search(searchParam);

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

@@ -180,6 +180,18 @@ public class MilvusServiceImpl implements IMilvusService {
         milvusClient.delete(deleteParam);
     }
 
+    /**
+     * 删除知识库所有文件
+     */
+    public void removeAllDocument(MilvusClient milvusClient, String collectionName) {
+        loadCollectionName(milvusClient, collectionName);
+        DeleteParam deleteParam = DeleteParam.newBuilder()
+                .withCollectionName(collectionName)
+                .withExpr("id > 0")
+                .build();
+        milvusClient.delete(deleteParam);
+    }
+
     /**
      * 加载知识库Collection
      */

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

@@ -1,79 +0,0 @@
-package com.ruoyi.llm.domain;
-
-import org.apache.commons.lang3.builder.ToStringBuilder;
-import org.apache.commons.lang3.builder.ToStringStyle;
-import com.ruoyi.common.annotation.Excel;
-import com.ruoyi.common.core.domain.BaseEntity;
-
-/**
- * 智能体工具对象 cmc_agent_tool
- * 
- * @author ruoyi
- * @date 2025-07-17
- */
-public class CmcAgentTool extends BaseEntity
-{
-    private static final long serialVersionUID = 1L;
-
-    /** 智能体工具id */
-    private Integer agentToolId;
-
-    /** 工具描述 */
-    @Excel(name = "工具描述")
-    private String toolName;
-
-    /** 智能体id */
-    @Excel(name = "智能体id")
-    private Integer agentId;
-
-    /** 顺序 */
-    @Excel(name = "顺序")
-    private String toolSort;
-
-    public void setAgentToolId(Integer agentToolId) 
-    {
-        this.agentToolId = agentToolId;
-    }
-
-    public Integer getAgentToolId() 
-    {
-        return agentToolId;
-    }
-    public void setToolName(String toolName) 
-    {
-        this.toolName = toolName;
-    }
-
-    public String getToolName() 
-    {
-        return toolName;
-    }
-    public void setAgentId(Integer agentId) 
-    {
-        this.agentId = agentId;
-    }
-
-    public Integer getAgentId() 
-    {
-        return agentId;
-    }
-    public void setToolSort(String toolSort) 
-    {
-        this.toolSort = toolSort;
-    }
-
-    public String getToolSort() 
-    {
-        return toolSort;
-    }
-
-    @Override
-    public String toString() {
-        return new ToStringBuilder(this,ToStringStyle.MULTI_LINE_STYLE)
-            .append("agentToolId", getAgentToolId())
-            .append("toolName", getToolName())
-            .append("agentId", getAgentId())
-            .append("toolSort", getToolSort())
-            .toString();
-    }
-}

llm-back/ruoyi-system/src/main/java/com/ruoyi/llm/mapper/CmcAgentToolMapper.java

@@ -1,61 +0,0 @@
-package com.ruoyi.llm.mapper;
-
-import java.util.List;
-import com.ruoyi.llm.domain.CmcAgentTool;
-
-/**
- * 智能体工具Mapper接口
- * 
- * @author ruoyi
- * @date 2025-07-17
- */
-public interface CmcAgentToolMapper 
-{
-    /**
-     * 查询智能体工具
-     * 
-     * @param agentToolId 智能体工具主键
-     * @return 智能体工具
-     */
-    public CmcAgentTool selectCmcAgentToolByAgentToolId(Integer agentToolId);
-
-    /**
-     * 查询智能体工具列表
-     * 
-     * @param cmcAgentTool 智能体工具
-     * @return 智能体工具集合
-     */
-    public List<CmcAgentTool> selectCmcAgentToolList(CmcAgentTool cmcAgentTool);
-
-    /**
-     * 新增智能体工具
-     * 
-     * @param cmcAgentTool 智能体工具
-     * @return 结果
-     */
-    public int insertCmcAgentTool(CmcAgentTool cmcAgentTool);
-
-    /**
-     * 修改智能体工具
-     * 
-     * @param cmcAgentTool 智能体工具
-     * @return 结果
-     */
-    public int updateCmcAgentTool(CmcAgentTool cmcAgentTool);
-
-    /**
-     * 删除智能体工具
-     * 
-     * @param agentToolId 智能体工具主键
-     * @return 结果
-     */
-    public int deleteCmcAgentToolByAgentToolId(Integer agentToolId);
-
-    /**
-     * 批量删除智能体工具
-     * 
-     * @param agentToolIds 需要删除的数据主键集合
-     * @return 结果
-     */
-    public int deleteCmcAgentToolByAgentToolIds(Integer[] agentToolIds);
-}

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

@@ -1,61 +0,0 @@
-package com.ruoyi.llm.service;
-
-import java.util.List;
-import com.ruoyi.llm.domain.CmcAgentTool;
-
-/**
- * 智能体工具Service接口
- * 
- * @author ruoyi
- * @date 2025-07-17
- */
-public interface ICmcAgentToolService 
-{
-    /**
-     * 查询智能体工具
-     * 
-     * @param agentToolId 智能体工具主键
-     * @return 智能体工具
-     */
-    public CmcAgentTool selectCmcAgentToolByAgentToolId(Integer agentToolId);
-
-    /**
-     * 查询智能体工具列表
-     * 
-     * @param cmcAgentTool 智能体工具
-     * @return 智能体工具集合
-     */
-    public List<CmcAgentTool> selectCmcAgentToolList(CmcAgentTool cmcAgentTool);
-
-    /**
-     * 新增智能体工具
-     * 
-     * @param cmcAgentTool 智能体工具
-     * @return 结果
-     */
-    public int insertCmcAgentTool(CmcAgentTool cmcAgentTool);
-
-    /**
-     * 修改智能体工具
-     * 
-     * @param cmcAgentTool 智能体工具
-     * @return 结果
-     */
-    public int updateCmcAgentTool(CmcAgentTool cmcAgentTool);
-
-    /**
-     * 批量删除智能体工具
-     * 
-     * @param agentToolIds 需要删除的智能体工具主键集合
-     * @return 结果
-     */
-    public int deleteCmcAgentToolByAgentToolIds(Integer[] agentToolIds);
-
-    /**
-     * 删除智能体工具信息
-     * 
-     * @param agentToolId 智能体工具主键
-     * @return 结果
-     */
-    public int deleteCmcAgentToolByAgentToolId(Integer agentToolId);
-}

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

@@ -1,93 +0,0 @@
-package com.ruoyi.llm.service.impl;
-
-import java.util.List;
-import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.stereotype.Service;
-import com.ruoyi.llm.mapper.CmcAgentToolMapper;
-import com.ruoyi.llm.domain.CmcAgentTool;
-import com.ruoyi.llm.service.ICmcAgentToolService;
-
-/**
- * 智能体工具Service业务层处理
- * 
- * @author ruoyi
- * @date 2025-07-17
- */
-@Service
-public class CmcAgentToolServiceImpl implements ICmcAgentToolService 
-{
-    @Autowired
-    private CmcAgentToolMapper cmcAgentToolMapper;
-
-    /**
-     * 查询智能体工具
-     * 
-     * @param agentToolId 智能体工具主键
-     * @return 智能体工具
-     */
-    @Override
-    public CmcAgentTool selectCmcAgentToolByAgentToolId(Integer agentToolId)
-    {
-        return cmcAgentToolMapper.selectCmcAgentToolByAgentToolId(agentToolId);
-    }
-
-    /**
-     * 查询智能体工具列表
-     * 
-     * @param cmcAgentTool 智能体工具
-     * @return 智能体工具
-     */
-    @Override
-    public List<CmcAgentTool> selectCmcAgentToolList(CmcAgentTool cmcAgentTool)
-    {
-        return cmcAgentToolMapper.selectCmcAgentToolList(cmcAgentTool);
-    }
-
-    /**
-     * 新增智能体工具
-     * 
-     * @param cmcAgentTool 智能体工具
-     * @return 结果
-     */
-    @Override
-    public int insertCmcAgentTool(CmcAgentTool cmcAgentTool)
-    {
-        return cmcAgentToolMapper.insertCmcAgentTool(cmcAgentTool);
-    }
-
-    /**
-     * 修改智能体工具
-     * 
-     * @param cmcAgentTool 智能体工具
-     * @return 结果
-     */
-    @Override
-    public int updateCmcAgentTool(CmcAgentTool cmcAgentTool)
-    {
-        return cmcAgentToolMapper.updateCmcAgentTool(cmcAgentTool);
-    }
-
-    /**
-     * 批量删除智能体工具
-     * 
-     * @param agentToolIds 需要删除的智能体工具主键
-     * @return 结果
-     */
-    @Override
-    public int deleteCmcAgentToolByAgentToolIds(Integer[] agentToolIds)
-    {
-        return cmcAgentToolMapper.deleteCmcAgentToolByAgentToolIds(agentToolIds);
-    }
-
-    /**
-     * 删除智能体工具信息
-     * 
-     * @param agentToolId 智能体工具主键
-     * @return 结果
-     */
-    @Override
-    public int deleteCmcAgentToolByAgentToolId(Integer agentToolId)
-    {
-        return cmcAgentToolMapper.deleteCmcAgentToolByAgentToolId(agentToolId);
-    }
-}

llm-back/ruoyi-system/src/main/resources/mapper/llm/CmcAgentToolMapper.xml

@@ -1,68 +0,0 @@
-<?xml version="1.0" encoding="UTF-8" ?>
-<!DOCTYPE mapper
-PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
-"http://mybatis.org/dtd/mybatis-3-mapper.dtd">
-<mapper namespace="com.ruoyi.llm.mapper.CmcAgentToolMapper">
-    
-    <resultMap type="CmcAgentTool" id="CmcAgentToolResult">
-        <result property="agentToolId"    column="agent_tool_id"    />
-        <result property="toolName"    column="tool_name"    />
-        <result property="agentId"    column="agent_id"    />
-        <result property="toolSort"    column="tool_sort"    />
-    </resultMap>
-
-    <sql id="selectCmcAgentToolVo">
-        select agent_tool_id, tool_name, agent_id, tool_sort from cmc_agent_tool
-    </sql>
-
-    <select id="selectCmcAgentToolList" parameterType="CmcAgentTool" resultMap="CmcAgentToolResult">
-        <include refid="selectCmcAgentToolVo"/>
-        <where>  
-            <if test="toolName != null  and toolName != ''"> and tool_name like concat('%', #{toolName}, '%')</if>
-            <if test="agentId != null "> and agent_id = #{agentId}</if>
-            <if test="toolSort != null  and toolSort != ''"> and tool_sort = #{toolSort}</if>
-        </where>
-    </select>
-    
-    <select id="selectCmcAgentToolByAgentToolId" parameterType="Integer" resultMap="CmcAgentToolResult">
-        <include refid="selectCmcAgentToolVo"/>
-        where agent_tool_id = #{agentToolId}
-    </select>
-
-    <insert id="insertCmcAgentTool" parameterType="CmcAgentTool">
-        insert into cmc_agent_tool
-        <trim prefix="(" suffix=")" suffixOverrides=",">
-            <if test="agentToolId != null">agent_tool_id,</if>
-            <if test="toolName != null">tool_name,</if>
-            <if test="agentId != null">agent_id,</if>
-            <if test="toolSort != null">tool_sort,</if>
-         </trim>
-        <trim prefix="values (" suffix=")" suffixOverrides=",">
-            <if test="agentToolId != null">#{agentToolId},</if>
-            <if test="toolName != null">#{toolName},</if>
-            <if test="agentId != null">#{agentId},</if>
-            <if test="toolSort != null">#{toolSort},</if>
-         </trim>
-    </insert>
-
-    <update id="updateCmcAgentTool" parameterType="CmcAgentTool">
-        update cmc_agent_tool
-        <trim prefix="SET" suffixOverrides=",">
-            <if test="toolName != null">tool_name = #{toolName},</if>
-            <if test="agentId != null">agent_id = #{agentId},</if>
-            <if test="toolSort != null">tool_sort = #{toolSort},</if>
-        </trim>
-        where agent_tool_id = #{agentToolId}
-    </update>
-
-    <delete id="deleteCmcAgentToolByAgentToolId" parameterType="Integer">
-        delete from cmc_agent_tool where agent_tool_id = #{agentToolId}
-    </delete>
-
-    <delete id="deleteCmcAgentToolByAgentToolIds" parameterType="String">
-        delete from cmc_agent_tool where agent_tool_id in 
-        <foreach item="agentToolId" collection="array" open="(" separator="," close=")">
-            #{agentToolId}
-        </foreach>
-    </delete>
-</mapper>

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

@@ -1,8 +1,8 @@
 /*
  * @Author: ysh
  * @Date: 2025-06-30 09:56:10
- * @LastEditors: Please set LastEditors
- * @LastEditTime: 2025-07-03 10:47:54
+ * @LastEditors: wrh
+ * @LastEditTime: 2025-07-18 15:40:00
  */
 import request from '@/utils/request'
 
@@ -43,9 +43,11 @@ export function delKnowledge(collectionName) {
 }
 
 // 插入知识库文件
-export function insertKnowledgeFile(file, collectionName) {
+export function insertKnowledgeFile(fileList, collectionName) {
   const formData = new FormData()
-  formData.append('file', file)
+  for(let f of fileList){
+    formData.append('fileList', f)
+  }
   formData.append('collectionName', collectionName)
   return request({
     url: '/llm/knowledge/insertDocument',

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

@@ -266,7 +266,7 @@
           <el-input v-model="uploadForm.collectionName" disabled />
         </el-form-item>
         <el-form-item label="选择文件" prop="file">
-          <el-upload ref="knowledgeUpload" :limit="1" accept=".doc,.docx,.pdf" :headers="upload.headers" :action="''"
+          <el-upload ref="knowledgeUpload" multiple accept=".docx,.pdf" :headers="upload.headers" :action="''"
             :disabled="upload.isUploading" :on-progress="handleFileUploadProgress" :on-success="handleFileSuccess"
             :auto-upload="false" drag :on-change="handleFileChange">
             <el-icon class="el-icon--upload"><upload-filled /></el-icon>
@@ -277,7 +277,7 @@
                   <el-checkbox v-model="upload.updateSupport" />
                   是否更新已经存在的文件数据
                 </div>
-                <span>支持 .doc、.docx、.pdf 格式文件</span>
+                <span>支持 .docx、.pdf 格式文件</span>
               </div>
             </template>
           </el-upload>
@@ -638,8 +638,6 @@ function getFileType(fileName) {
   switch (extension) {
     case 'pdf':
       return 'PDF文档';
-    case 'doc':
-      return 'Word文档';
     case 'docx':
       return 'Word文档';
     default:
@@ -772,7 +770,8 @@ function handleFileUploadProgress(event, file, fileList) {
 }
 
 function handleFileChange(file, fileList) {
-  upload.file = file.raw;
+  upload.fileList = fileList.map(f => f.raw)
+  console.log(upload.fileList)
 }
 
 // 文件上传成功处理
@@ -797,7 +796,7 @@ function handleFileSuccess(response, file, fileList) {
 
 // 提交上传
 function submitUpload() {
-  insertKnowledgeFile(upload.file, uploadForm.collectionName).then(response => {
+  insertKnowledgeFile(upload.fileList, uploadForm.collectionName).then(response => {
     proxy.$modal.msgSuccess("上传成功");
     uploadOpen.value = false;
     proxy.$refs.knowledgeUpload.clearFiles();