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

调试

Browse Source
This commit is contained in:
Liuyang 2025-02-08 19:33:19 +08:00
parent edf1befda1
commit 942f320fe2
3 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncKnowledgeBase.java
View File

@ -66,7 +66,7 @@ public class AsyncKnowledgeBase {
* @param knowledgeList 文件列表
* @param id
*/
@Async
// @Async
public void knowledgeEmbed (List<KnowledgeDocumentsDO> knowledgeList, Long id) {
if (!CollectionUtils.isAnyEmpty(knowledgeList)) {
knowledgeList.forEach(knowledge -> {

9
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/RagHttpService.java
View File

@ -225,15 +225,20 @@ public class RagHttpService {
// 获取知识库向量嵌入的url
String ragEmbed = llmBackendProperties.getEmbed();
log.info("url : {}", ragEmbed);
aDo.setKnowledgeBaseIntro("8-- url: "+ragEmbed);
knowledgeBaseMapper.updateById(aDo);
// 构建请求参数
HttpResponse<String> response = Unirest.post(ragEmbed)
.field("file_id", reqVO.getFileId())
.field("file", reqVO.getFileInputStream(), reqVO.getFileName())
.asString();
aDo.setKnowledgeBaseIntro("9-- url: "+ragEmbed);
knowledgeBaseMapper.updateById(aDo);
String body = response.getBody();
log.info("body : {}", body);
aDo.setKnowledgeBaseIntro("10-- url: "+ragEmbed);
knowledgeBaseMapper.updateById(aDo);
JSONObject jsonObject = JSON.parseObject(body);

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

@ -95,12 +95,14 @@ public class KnowledgeBaseServiceImpl implements KnowledgeBaseService {
knowledgeDocumentsMapper.insertOrUpdate(knowledgeDocumentsDO);
}
);
// 4.5 异步处理知识库外挂
asyncKnowledgeBase.knowledgeEmbed(knowledgeDocumentsList, updateReqVO.getId());
List<Long> deleteIds = knowledgeDocumentsMapper.selectDeleteIds(updateReqVO.getId());
asyncKnowledgeBase.createKnowledgeBase(knowledgeDocumentsList, deleteIds);
updateObj.setKnowledgeBaseIntro("4");
knowledgeBaseMapper.updateById(updateObj);
// 4.5 异步处理知识库外挂
asyncKnowledgeBase.knowledgeEmbed(knowledgeDocumentsList, updateReqVO.getId());
// asyncKnowledgeBase.knowledgeEmbed(knowledgeDocumentsList, updateReqVO.getId());
} else {
knowledgeDocumentsMapper.delete(new LambdaQueryWrapperX<KnowledgeDocumentsDO>()
.eq(KnowledgeDocumentsDO::getKnowledgeBaseId, updateReqVO.getId()));