refactor(yudao-module-llm): 恢复文档嵌入逻辑

- 重新启用了对上传文件进行嵌入的逻辑
- 根据文件扩展名决定使用不同的嵌入方法
This commit is contained in:
Liuyang 2025-02-21 11:20:11 +08:00
parent df92cc9ef4
commit c1a5833bec

View File

@ -62,13 +62,13 @@ public class AsyncKnowledgeBase {
int lastIndex = knowledge.getDocumentName().lastIndexOf(".");
if (lastIndex != -1) {
ragHttpService.embedUploadFile(regUploadReqVO);
// String extension = knowledge.getDocumentName().substring(lastIndex + 1).toLowerCase();
// if ("txt".equals(extension)) {
// ragHttpService.embedUploadFile(regUploadReqVO);
// } else {
// knowledgeEmbed(knowledge, knowledge.getKnowledgeBaseId());
// }
String extension = knowledge.getDocumentName().substring(lastIndex + 1).toLowerCase();
if ("txt".equals(extension)) {
ragHttpService.embedUploadFile(regUploadReqVO);
} else {
knowledgeEmbed(knowledge, knowledge.getKnowledgeBaseId());
}
}
} catch (Exception e) {
log.error("the creation of the knowledge base error {}", e.getMessage());