fix: 修改部分BUG

src/main/java/cn/com/poc/knowledge/rest/impl/DatabaseRestImpl.java

@@ -108,7 +108,8 @@ public class DatabaseRestImpl implements DatabaseRest {
         List<BizKnowledgeDatabaseDto> result = new ArrayList<>();
         for (Integer id : ids) {
             BizKnowledgeDatabaseEntity bizKnowledgeDatabaseEntity = bizKnowledgeDatabaseService.get(id.longValue());
-            if (bizKnowledgeDatabaseEntity == null || !bizKnowledgeDatabaseEntity.getMemberId().equals(userId.intValue())) {
+            if (bizKnowledgeDatabaseEntity == null || (!bizKnowledgeDatabaseEntity.getMemberId().equals(0)
+                    && !bizKnowledgeDatabaseEntity.getMemberId().equals(userId.intValue()))) {
                 continue;
             }
             BizKnowledgeDatabaseDto bizKnowledgeDatabaseDto = BizKnowledgeDatabaseConvert.entityToDto(bizKnowledgeDatabaseEntity);

src/main/java/cn/com/poc/thirdparty/service/impl/ChainServiceImpl.java

@@ -23,10 +23,6 @@ public class ChainServiceImpl implements ChainService {
 
     private Logger logger = LoggerFactory.getLogger(ChainServiceImpl.class);
 
-
-    @Value(value = "${large-model.apikey}")
-    private String API_KEY;
-
     @Resource
     private AIDialogueService aiDialogueService;
 
@@ -53,17 +49,21 @@ public class ChainServiceImpl implements ChainService {
     @Override
     public CSVChainResult csvChain(CSVChainResponse response) {
         logger.info("csvChain response : {}", response);
-        response.setApiKey(API_KEY);
-        CSVChainResult csvChainResult = aiDialogueService.csvChain(response);
-        if (csvChainResult == null || csvChainResult.getStatus().equals("error")) {
-            logger.error("csvChain result error : {} , response:{}", csvChainResult, response);
-            return null;
-        }
-        if (StringUtils.isBlank(csvChainResult.getResult())) {
-            logger.warn("csvChain result sqlResult is blank : {} , response:{}", csvChainResult, response);
+        try {
+            CSVChainResult csvChainResult = aiDialogueService.csvChain(response);
+            if (csvChainResult == null || csvChainResult.getStatus().equals("error")) {
+                logger.error("csvChain result error : {} , response:{}", csvChainResult, response);
+                return null;
+            }
+            if (StringUtils.isBlank(csvChainResult.getResult())) {
+                logger.warn("csvChain result sqlResult is blank : {} , response:{}", csvChainResult, response);
+                return null;
+            }
+
+            return csvChainResult;
+        } catch (Exception e) {
+            logger.error("csv chain error:{}", e.getMessage());
             return null;
         }
-
-        return csvChainResult;
     }
 }