资料管理界面新增查询编辑功能添加了确定所属知识库的功能，按钮管理界面的增删改查及界面按钮开关实时操作开启和关闭功能

+package org.jeecg.modules.airag.app.controller;
+
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import io.swagger.v3.oas.annotations.Operation;
+import io.swagger.v3.oas.annotations.tags.Tag;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.shiro.authz.annotation.RequiresPermissions;
+import org.jeecg.common.api.vo.Result;
+import org.jeecg.common.aspect.annotation.AutoLog;
+import org.jeecg.common.system.base.controller.JeecgController;
+import org.jeecg.common.system.query.QueryGenerator;
+import org.jeecg.modules.airag.app.entity.AiragButton;
+import org.jeecg.modules.airag.app.service.IAiragButtonService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.*;
+import org.springframework.web.servlet.ModelAndView;
+
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.util.Arrays;
+
+/**
+* @Description: 按钮表单
+* @Author: jeecg-boot
+* @Date:   2025-06-05
+* @Version: V1.0
+*/
+@Tag(name="按钮表单")
+@RestController
+@RequestMapping("/airagbutton/airagButton")
+@Slf4j
+public class AiragButtonController extends JeecgController<AiragButton, IAiragButtonService> {
+   @Autowired
+   private IAiragButtonService airagButtonService;
+
+   /**
+    * 分页列表查询
+    *
+    * @param airagButton
+    * @param pageNo
+    * @param pageSize
+    * @param req
+    * @return
+    */
+   //@AutoLog(value = "按钮表单-分页列表查询")
+   @Operation(summary="按钮表单-分页列表查询")
+   @GetMapping(value = "/list")
+   public Result<IPage<AiragButton>> queryPageList(AiragButton airagButton,
+                                  @RequestParam(name="pageNo", defaultValue="1") Integer pageNo,
+                                  @RequestParam(name="pageSize", defaultValue="10") Integer pageSize,
+                                  HttpServletRequest req) {
+       QueryWrapper<AiragButton> queryWrapper = QueryGenerator.initQueryWrapper(airagButton, req.getParameterMap());
+       Page<AiragButton> page = new Page<AiragButton>(pageNo, pageSize);
+       IPage<AiragButton> pageList = airagButtonService.page(page, queryWrapper);
+//       pageList.setRecords(airagButtonService.  findAll(airagButton));
+       return Result.OK(pageList);
+   }
+
+   /**
+    *   添加
+    *
+    * @param airagButton
+    * @return
+    */
+   @AutoLog(value = "按钮表单-添加")
+   @Operation(summary="按钮表单-添加")
+   @RequiresPermissions("airagbutton:airag_button:add")
+   @PostMapping(value = "/add")
+   public Result<String> add(@RequestBody AiragButton airagButton) {
+       // 检查按钮开关是否为开启状态
+       if ("Y".equals(airagButton.getButtonSwitch())) {
+           // 查询当前已开启的按钮数量
+           QueryWrapper<AiragButton> queryWrapper = new QueryWrapper<>();
+           queryWrapper.eq("button_switch", "Y");
+           long openedButtonCount = airagButtonService.count(queryWrapper);
+
+           // 如果已开启的按钮数量超过或等于5个，则返回错误提示
+           if (openedButtonCount >= 5) {
+               return Result.error("按钮超过五个，请先关闭一个按钮");
+           }
+       }
+       airagButtonService.save(airagButton);
+       return Result.OK("添加成功！");
+   }
+
+   /**
+    *  编辑
+    *
+    * @param airagButton
+    * @return
+    */
+   @AutoLog(value = "按钮表单-编辑")
+   @Operation(summary="按钮表单-编辑")
+   @RequiresPermissions("airagbutton:airag_button:edit")
+   @RequestMapping(value = "/edit", method = {RequestMethod.PUT,RequestMethod.POST})
+   public Result<String> edit(@RequestBody AiragButton airagButton) {
+       // 检查按钮开关是否为开启状态
+       if ("Y".equals(airagButton.getButtonSwitch())) {
+           // 查询当前已开启的按钮数量（排除自身）
+           QueryWrapper<AiragButton> queryWrapper = new QueryWrapper<>();
+           queryWrapper.eq("button_switch", "Y")
+                   .ne("id", airagButton.getId()); // 排除当前正在编辑的按钮
+           long openedButtonCount = airagButtonService.count(queryWrapper);
+
+           // 如果已开启的按钮数量超过或等于5个，则返回错误提示
+           if (openedButtonCount >= 5) {
+               return Result.error("按钮超过五个，请先关闭一个按钮");
+           }
+       }
+       airagButtonService.updateById(airagButton);
+       return Result.OK("编辑成功!");
+   }
+
+   /**
+    *   通过id删除
+    *
+    * @param id
+    * @return
+    */
+   @AutoLog(value = "按钮表单-通过id删除")
+   @Operation(summary="按钮表单-通过id删除")
+   @RequiresPermissions("airagbutton:airag_button:delete")
+   @DeleteMapping(value = "/delete")
+   public Result<String> delete(@RequestParam(name="id",required=true) String id) {
+       airagButtonService.removeById(id);
+       return Result.OK("删除成功!");
+   }
+
+   /**
+    *  批量删除
+    *
+    * @param ids
+    * @return
+    */
+   @AutoLog(value = "按钮表单-批量删除")
+   @Operation(summary="按钮表单-批量删除")
+   @RequiresPermissions("airagbutton:airag_button:deleteBatch")
+   @DeleteMapping(value = "/deleteBatch")
+   public Result<String> deleteBatch(@RequestParam(name="ids",required=true) String ids) {
+       this.airagButtonService.removeByIds(Arrays.asList(ids.split(",")));
+       return Result.OK("批量删除成功!");
+   }
+
+   /**
+    * 通过id查询
+    *
+    * @param id
+    * @return
+    */
+   //@AutoLog(value = "按钮表单-通过id查询")
+   @Operation(summary="按钮表单-通过id查询")
+   @GetMapping(value = "/queryById")
+   public Result<AiragButton> queryById(@RequestParam(name="id",required=true) String id) {
+       AiragButton airagButton = airagButtonService.getById(id);
+       if(airagButton==null) {
+           return Result.error("未找到对应数据");
+       }
+       return Result.OK(airagButton);
+   }
+
+   /**
+   * 导出excel
+   *
+   * @param request
+   * @param airagButton
+   */
+   @RequiresPermissions("airagbutton:airag_button:exportXls")
+   @RequestMapping(value = "/exportXls")
+   public ModelAndView exportXls(HttpServletRequest request, AiragButton airagButton) {
+       return super.exportXls(request, airagButton, AiragButton.class, "按钮表单");
+   }
+
+   /**
+     * 通过excel导入数据
+   *
+   * @param request
+   * @param response
+   * @return
+   */
+   @RequiresPermissions("airagbutton:airag_button:importExcel")
+   @RequestMapping(value = "/importExcel", method = RequestMethod.POST)
+   public Result<?> importExcel(HttpServletRequest request, HttpServletResponse response) {
+       return super.importExcel(request, response, AiragButton.class);
+   }
+
+}

+package org.jeecg.modules.airag.app.entity;
+
+import com.baomidou.mybatisplus.annotation.IdType;
+import com.baomidou.mybatisplus.annotation.TableId;
+import com.baomidou.mybatisplus.annotation.TableName;
+import com.fasterxml.jackson.annotation.JsonFormat;
+import io.swagger.v3.oas.annotations.media.Schema;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+import lombok.experimental.Accessors;
+import org.jeecgframework.poi.excel.annotation.Excel;
+import org.springframework.format.annotation.DateTimeFormat;
+
+import java.io.Serializable;
+import java.io.UnsupportedEncodingException;
+import java.util.Date;
+
+/**
+ * @Description: 按钮表单
+ * @Author: jeecg-boot
+ * @Date:   2025-06-05
+ * @Version: V1.0
+ */
+@Data
+@TableName("airag_button")
+@Accessors(chain = true)
+@EqualsAndHashCode(callSuper = false)
+@Schema(description="按钮表单")
+public class AiragButton implements Serializable {
+    private static final long serialVersionUID = 1L;
+
+	/**主键*/
+	@TableId(type = IdType.ASSIGN_ID)
+    @Schema(description = "主键")
+    private String id;
+	/**创建人*/
+    @Schema(description = "创建人")
+    private String createBy;
+	/**创建日期*/
+	@JsonFormat(timezone = "GMT+8",pattern = "yyyy-MM-dd HH:mm:ss")
+    @DateTimeFormat(pattern="yyyy-MM-dd HH:mm:ss")
+    @Schema(description = "创建日期")
+    private Date createTime;
+	/**更新人*/
+    @Schema(description = "更新人")
+    private String updateBy;
+	/**更新日期*/
+	@JsonFormat(timezone = "GMT+8",pattern = "yyyy-MM-dd HH:mm:ss")
+    @DateTimeFormat(pattern="yyyy-MM-dd HH:mm:ss")
+    @Schema(description = "更新日期")
+    private Date updateTime;
+	/**所属部门*/
+    @Schema(description = "所属部门")
+    private String sysOrgCode;
+	/**按钮名称*/
+	@Excel(name = "按钮名称", width = 15)
+    @Schema(description = "按钮名称")
+    private String buttonName;
+
+
+    private String buttonSwitch;
+
+	/**按钮值*/
+	@Excel(name = "按钮值", width = 15)
+    @Schema(description = "按钮值")
+    private String buttonValues;
+}

@@ -25,5 +25,4 @@ public class Embeddings {
     private String docId;       // 新增文档ID字段
     private String index;       // 新增索引位置字段
 
-
 }
\ No newline at end of file

+package org.jeecg.modules.airag.app.mapper;
+
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import io.lettuce.core.dynamic.annotation.Param;
+import org.jeecg.modules.airag.app.entity.AiragButton;
+
+import java.util.List;
+
+/**
+ * @Description: 按钮表单
+ * @Author: jeecg-boot
+ * @Date:   2025-06-05
+ * @Version: V1.0
+ */
+public interface AiragButtonMapper extends BaseMapper<AiragButton> {
+}

@@ -32,15 +32,19 @@ public class PgVectorMapper {
     // 查询所有向量记录
     public List<Embeddings> findAll(Embeddings embeddings) {
         List<Embeddings> results = new ArrayList<>();
-        String sql = "SELECT * FROM embeddings";
+        String sql = "SELECT * FROM embeddings WHERE 1 =1 ";
         if (StringUtils.isNotBlank(embeddings.getId())){
-            sql += " WHERE embedding_id = '" + embeddings.getId() + "'";
+            sql += " and embedding_id = '" + embeddings.getId() + "'";
         }
 
-        if(StringUtils.isNotBlank(embeddings.getText())){
-            sql += " where text = '" + embeddings.getText() + "'";
+        if (StringUtils.isNotBlank(embeddings.getKnowledgeId())){
+            sql += " and metadata ->> 'knowledgeId' = '" + embeddings.getKnowledgeId() + "'";
         }
 
+        if(StringUtils.isNotBlank(embeddings.getText())){
+            sql += " and text = '" + embeddings.getText() + "'";
+        }
+        System.out.println("sql = " + sql);
         try (Connection conn = getConnection();
              PreparedStatement stmt = conn.prepareStatement(sql);
              ResultSet rs = stmt.executeQuery()) {
@@ -111,18 +115,15 @@ public class PgVectorMapper {
 
     // 更新向量记录
     public int update(Embeddings record) {
-        String sql = "UPDATE embeddings SET embedding = ?, metadata = ?, text = ?::jsonb WHERE embedding_id = ?";
+        String sql = "UPDATE embeddings SET embedding = ?, metadata = ?::jsonb, text = ? WHERE embedding_id = ?";
 
         try (Connection conn = getConnection();
              PreparedStatement stmt = conn.prepareStatement(sql)) {
 
             JSONObject mataData = new JSONObject();
-            SnowflakeGenerator snowflakeGenerator = new SnowflakeGenerator();
-            String knowledgeId = String.valueOf(snowflakeGenerator.next());
-            mataData.put("knowledgeId", knowledgeId); // 使用前端传入的知识库ID
+            mataData.put("knowledgeId", record.getKnowledgeId()); // 使用前端传入的知识库ID
             mataData.put("docName", record.getDocName());
-            String docId = String.valueOf(snowflakeGenerator.next());
-            mataData.put("docId", docId);      // 自动生成唯一文档ID
+            mataData.put("docId", record.getDocId());      // 自动生成唯一文档ID
             mataData.put("index", "0");
 
             PGobject jsonObject = new PGobject();

@@ -2,4 +2,6 @@
 <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
 <mapper namespace="org.jeecg.modules.airag.app.mapper.AiragAppMapper">
 
+
+
 </mapper>
\ No newline at end of file

+<?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="org.jeecg.modules.airag.app.mapper.AiragButtonMapper">
+
+
+</mapper>
\ No newline at end of file

+package org.jeecg.modules.airag.app.service;
+
+import com.baomidou.mybatisplus.extension.service.IService;
+import org.jeecg.modules.airag.app.entity.AiragButton;
+import org.jeecg.modules.airag.app.entity.Embeddings;
+
+import java.util.List;
+
+/**
+ * @Description: 按钮表单
+ * @Author: jeecg-boot
+ * @Date:   2025-06-05
+ * @Version: V1.0
+ */
+public interface IAiragButtonService extends IService<AiragButton> {
+
+}

+package org.jeecg.modules.airag.app.service.impl;
+
+import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import org.jeecg.modules.airag.app.entity.AiragButton;
+import org.jeecg.modules.airag.app.entity.Embeddings;
+import org.jeecg.modules.airag.app.mapper.AiragButtonMapper;
+import org.jeecg.modules.airag.app.mapper.PgVectorMapper;
+import org.jeecg.modules.airag.app.service.IAiragButtonService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+
+import java.util.List;
+
+/**
+ * @Description: 按钮表单
+ * @Author: jeecg-boot
+ * @Date:   2025-06-05
+ * @Version: V1.0
+ */
+@Service
+public class AiragButtonServiceImpl extends ServiceImpl<AiragButtonMapper, AiragButton> implements IAiragButtonService {
+
+
+}