refactor(yudao-module-llm): 恢复文档嵌入逻辑
- 重新启用了对上传文件进行嵌入的逻辑 - 根据文件扩展名决定使用不同的嵌入方法
This commit is contained in:
parent
df92cc9ef4
commit
c1a5833bec
@ -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());
|
||||
|
Loading…
x
Reference in New Issue
Block a user