xieyinlu/ruoyi-ai
1
0
Fork 0
mirror of https://gitcode.com/ageerle/ruoyi-ai.git synced 2026-03-15 13:43:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

增加知识库配置属性

Browse Source
This commit is contained in:
ageerle
2025-03-07 09:56:15 +08:00
parent d1b6a13a58
commit ea09421b0e
40 changed files with 332 additions and 236 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ruoyi-admin/src/main/java/org/ruoyi/controller/KnowledgeController.java
View File

@@ -81,7 +81,7 @@ public class KnowledgeController extends BaseController {
List<Message> messages = chatRequest.getMessages();
String content = messages.get(messages.size() - 1).getContent().toString();
List<String> nearestList;
List<Double> queryVector = embeddingService.getQueryVector(content);
List<Double> queryVector = embeddingService.getQueryVector(content, chatRequest.getKid());
nearestList = vectorStore.nearest(queryVector,chatRequest.getKid());
for (String prompt : nearestList) {
Message sysMessage = Message.builder().content(prompt).role(Message.Role.USER).build();