jqb/agent_9art
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

知识库内容返回多条

Browse Source
This commit is contained in:
qinyong@9artedu.com 2026-07-13 17:52:29 +08:00
parent f29efa8988
commit 0c00d161d2
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
backend/generation-service/src/main/java/com/artedu/generation/controller/GenerationController.java
View File

@ -10,6 +10,8 @@ import com.artedu.generation.service.LlmModelService;
import com.artedu.generation.service.PromptEngine;
import com.artedu.generation.service.QianwenClient;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.MediaType;
@ -329,10 +331,13 @@ public class GenerationController {
String answer = getAnswer(top15, external_str);
log.info("原问题：{},命中知识库结果:{}", external_str, answer);
if(!StringUtils.isBlank(answer)){
answer = answer.replaceAll("\"", "").replace("[","").replace("]","");
if(!StringUtils.isBlank(answer)){
String promptNew = promptEngine.buildAnswerPrompt(answer);
generated = qianwenClient.generate(promptNew, 800);
ObjectMapper mapper = new ObjectMapper();
List<String> list = mapper.readValue(answer, new TypeReference<List<String>>(){});
//answer = answer.replaceAll("\"", "").replace("[","").replace("]","");
for (String li : list){
String promptNew = promptEngine.buildAnswerPrompt(li);
String res = qianwenClient.generate(promptNew, 800);
generated = generated + res + "|||";
}
}
}