DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

refactor(yudao-module-llm): 优化知识库测试接口的参数处理逻辑

Browse Source 
- 设置 topK 和 score 的默认值
- 优化 topK 和 score 的处理逻辑，确保使用合适的值
- 移除不必要的条件判断，简化代码结构
This commit is contained in:
Liuyang 2025-03-15 12:24:20 +08:00
parent 387861888c
commit 25ed9735bf
1 changed files with 8 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/knowledgebase/KnowledgeBaseServiceImpl.java
View File

@ -295,25 +295,23 @@ public class KnowledgeBaseServiceImpl implements KnowledgeBaseService {
if (baseDO == null) {
throw exception(KNOWLEDGE_BASE_NOT_EXISTS);
}
// 设置默认值
Integer topK = 4;
Double score = 0.2;
// 处理 topK 的逻辑
if (baseDO.getTopK() == null || baseDO.getTopK() <= 0) {
testReqVO.setTopK(topK);
if (testReqVO.getTopK() == null){
testReqVO.setTopK(baseDO.getTopK());
}
} else {
topK = testReqVO.getTopK();
topK = testReqVO.getTopK() != null ? testReqVO.getTopK() : baseDO.getTopK();
testReqVO.setTopK(topK);
if (testReqVO.getScore() == null){
testReqVO.setScore(baseDO.getScore());
}
}
Double score = 0.2;
if (baseDO.getScore() == null || baseDO.getTopK() <= 0.0 || baseDO.getScore() > 1) {
// 处理 score 的逻辑
if (baseDO.getScore() == null || baseDO.getScore() <= 0.0 || baseDO.getScore() > 1) {
testReqVO.setScore(score);
} else {
score = testReqVO.getScore();
score = testReqVO.getScore() != null ? testReqVO.getScore() : baseDO.getScore();
testReqVO.setScore(score);
}