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

refactor(llm): 知识库文档嵌入逻辑

Browse Source 
- 注释掉了原有的针对 txt 文件的特殊处理逻辑
This commit is contained in:
Liuyang 2025-02-26 13:44:55 +08:00
parent 3be8b1e869
commit 3a6362bab1
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -75,14 +75,14 @@ public class AsyncKnowledgeBase {
if (lastIndex != -1) {
String extension = knowledge.getDocumentName().substring(lastIndex + 1).toLowerCase();
log.info("文档扩展名: {}", extension);
if ("txt".equals(extension)) {
log.info("文档为 txt 文件，直接上传嵌入，文档 ID: {}", knowledge.getId());
ragHttpService.embedUploadFile(regUploadReqVO);
} else {
log.info("文档为非 txt 文件，调用知识嵌入方法，文档 ID: {}", knowledge.getId());
knowledgeEmbed(knowledge, knowledge.getKnowledgeBaseId());
}
knowledgeEmbed(knowledge, knowledge.getKnowledgeBaseId());
// if ("txt".equals(extension)) {
// log.info("文档为 txt 文件，直接上传嵌入，文档 ID: {}", knowledge.getId());
// ragHttpService.embedUploadFile(regUploadReqVO);
// } else {
// log.info("文档为非 txt 文件，调用知识嵌入方法，文档 ID: {}", knowledge.getId());
// knowledgeEmbed(knowledge, knowledge.getKnowledgeBaseId());
// }
} else {
log.warn("文档无扩展名，跳过处理，文档 ID: {}", knowledge.getId());
}