Commit 33fbbe6e authored by alex yao's avatar alex yao

fix: 修改部分BUG

parent 43f4d770
...@@ -108,7 +108,8 @@ public class DatabaseRestImpl implements DatabaseRest { ...@@ -108,7 +108,8 @@ public class DatabaseRestImpl implements DatabaseRest {
List<BizKnowledgeDatabaseDto> result = new ArrayList<>(); List<BizKnowledgeDatabaseDto> result = new ArrayList<>();
for (Integer id : ids) { for (Integer id : ids) {
BizKnowledgeDatabaseEntity bizKnowledgeDatabaseEntity = bizKnowledgeDatabaseService.get(id.longValue()); 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; continue;
} }
BizKnowledgeDatabaseDto bizKnowledgeDatabaseDto = BizKnowledgeDatabaseConvert.entityToDto(bizKnowledgeDatabaseEntity); BizKnowledgeDatabaseDto bizKnowledgeDatabaseDto = BizKnowledgeDatabaseConvert.entityToDto(bizKnowledgeDatabaseEntity);
......
...@@ -23,10 +23,6 @@ public class ChainServiceImpl implements ChainService { ...@@ -23,10 +23,6 @@ public class ChainServiceImpl implements ChainService {
private Logger logger = LoggerFactory.getLogger(ChainServiceImpl.class); private Logger logger = LoggerFactory.getLogger(ChainServiceImpl.class);
@Value(value = "${large-model.apikey}")
private String API_KEY;
@Resource @Resource
private AIDialogueService aiDialogueService; private AIDialogueService aiDialogueService;
...@@ -53,7 +49,7 @@ public class ChainServiceImpl implements ChainService { ...@@ -53,7 +49,7 @@ public class ChainServiceImpl implements ChainService {
@Override @Override
public CSVChainResult csvChain(CSVChainResponse response) { public CSVChainResult csvChain(CSVChainResponse response) {
logger.info("csvChain response : {}", response); logger.info("csvChain response : {}", response);
response.setApiKey(API_KEY); try {
CSVChainResult csvChainResult = aiDialogueService.csvChain(response); CSVChainResult csvChainResult = aiDialogueService.csvChain(response);
if (csvChainResult == null || csvChainResult.getStatus().equals("error")) { if (csvChainResult == null || csvChainResult.getStatus().equals("error")) {
logger.error("csvChain result error : {} , response:{}", csvChainResult, response); logger.error("csvChain result error : {} , response:{}", csvChainResult, response);
...@@ -65,5 +61,9 @@ public class ChainServiceImpl implements ChainService { ...@@ -65,5 +61,9 @@ public class ChainServiceImpl implements ChainService {
} }
return csvChainResult; return csvChainResult;
} catch (Exception e) {
logger.error("csv chain error:{}", e.getMessage());
return null;
}
} }
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment