From a649df4aa1cbadb102692134b7ff4845531b196c Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 30 Sep 2025 13:06:27 +0800
Subject: [PATCH] fix(ai):修复知识库文件删除逻辑
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiKnowledgeFileServiceImpl.java | 14 ++++++++------
1 files changed, 8 insertions(+), 6 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiKnowledgeFileServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiKnowledgeFileServiceImpl.java
index 7d7e8f7..41b0733 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiKnowledgeFileServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiKnowledgeFileServiceImpl.java
@@ -138,10 +138,10 @@
if (state == 1){
state = KnowledgeBaseUtil.getFileJobStatus(aiKnowledgeFile.getFileId());
- if ( state == 1) {
+ if ( state == 2) {
aiKnowledgeFileMapper.update(null,
Wrappers.lambdaUpdate(AiKnowledgeFile.class)
- .set(AiKnowledgeFile::getState, 2)
+ .set(AiKnowledgeFile::getState, state)
.eq(AiKnowledgeFile::getId, aiKnowledgeFile.getId())
);
}
@@ -157,11 +157,11 @@
}else{
knowledgeId = KnowledgeBaseUtil.DEFAULT_KNOWLEDGE_ID;
}
- state = KnowledgeBaseUtil.getIndexKnowledgeJobStatus(aiKnowledgeFile.getFileId(),knowledgeId);
- if ( state == 1) {
+ state = KnowledgeBaseUtil.getIndexKnowledgeJobStatus(aiKnowledgeFile.getJobId(),knowledgeId);
+ if ( state == 3) {
aiKnowledgeFileMapper.update(null,
Wrappers.lambdaUpdate(AiKnowledgeFile.class)
- .set(AiKnowledgeFile::getState, 3)
+ .set(AiKnowledgeFile::getState, state)
.eq(AiKnowledgeFile::getId, aiKnowledgeFile.getId())
);
}
@@ -181,12 +181,14 @@
Path filePath = Paths.get(aiKnowledgeFile.getSavePath());
boolean deleted = Files.deleteIfExists(filePath);
if (!deleted) {
- throw new FebsException("文件删除成功");
+ throw new FebsException("文件删除失败");
}
} catch (Exception e) {
throw new FebsException("删除文件时发生错误: " + e.getMessage());
}
+ aiKnowledgeFileMapper.deleteById(id);
+
//知识库删除
String knowledgeId = null;
if (StrUtil.isNotEmpty(aiKnowledgeFile.getCompanyId())){
--
Gitblit v1.9.1