优化三重一大查询接口

src/main/java/com/gxwebsoft/ai/controller/AuditContent3Controller.java

@@ -11,6 +11,8 @@ import com.gxwebsoft.ai.utils.ExcelExportTool;
 import com.gxwebsoft.common.core.web.ApiResult;
 import com.gxwebsoft.common.core.web.BaseController;
 import com.gxwebsoft.common.system.entity.User;
+import com.gxwebsoft.pwl.entity.PwlProjectLibrary;
+import com.gxwebsoft.pwl.service.PwlProjectLibraryService;
 import com.gxwebsoft.ai.service.AiCloudDocService;
 import com.gxwebsoft.ai.service.AiCloudFileService;
 import com.gxwebsoft.ai.service.AuditContent3TripleService;
@@ -51,6 +53,9 @@ public class AuditContent3Controller extends BaseController {
     @Autowired
     private KnowledgeBaseService knowledgeBaseService;
     
+    @Autowired
+    private PwlProjectLibraryService pwlProjectLibraryService;
+    
     /**
      * 生成三重一大制度对比分析表数据
      */
@@ -59,6 +64,7 @@ public class AuditContent3Controller extends BaseController {
     public ApiResult<?> generateTripleOneTable(@RequestBody AuditContentRequest request) {
         final User loginUser = getLoginUser();
         String kbIdTmp = "";
+        String libraryKbIds = "";
         
         try {
             // 创建临时知识库（如果需要）
@@ -66,11 +72,19 @@ public class AuditContent3Controller extends BaseController {
                 kbIdTmp = createTempKnowledgeBase(request);
             }
             
+            // 提前查询项目库信息
+            if (StrUtil.isNotBlank(request.getLibraryIds())) {
+                List<String> idList = StrUtil.split(request.getLibraryIds(), ',');
+                List<PwlProjectLibrary> ret = pwlProjectLibraryService.list(
+                    new LambdaQueryWrapper<PwlProjectLibrary>().in(PwlProjectLibrary::getId, idList));
+                libraryKbIds = ret.stream().map(PwlProjectLibrary::getKbId).collect(Collectors.joining(","));
+            }
+            
             // 生成三重一大制度对比分析表数据
             String knowledgeBaseId = StrUtil.isNotBlank(kbIdTmp) ? kbIdTmp : request.getKbIds();
             JSONObject result = auditContent3TripleService.generateTripleOneTableData(
                 knowledgeBaseId, 
-                request.getLibraryIds(), 
+                libraryKbIds, 
                 request.getProjectLibrary(), 
                 loginUser.getUsername(), 
                 request.getHistory(),