From 3d3fa2c75ac3ed9439b94ae2bdd034403839af34 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Thu, 30 Oct 2025 10:12:43 +0800
Subject: [PATCH] feat(ai): 添加上下文生成工作流功能
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiKnowledgeFileServiceImpl.java | 17 +++++++++++------
1 files changed, 11 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..221403a 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
@@ -122,6 +122,7 @@
Wrappers.lambdaUpdate(AiKnowledgeFile.class)
.set(AiKnowledgeFile::getFileId, fileId)
.set(AiKnowledgeFile::getJobId, jobId)
+ .set(AiKnowledgeFile::getUpdatedTime, new Date())
.eq(AiKnowledgeFile::getId, aiKnowledgeFile.getId())
);
@@ -138,10 +139,11 @@
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)
+ .set(AiKnowledgeFile::getUpdatedTime, new Date())
.eq(AiKnowledgeFile::getId, aiKnowledgeFile.getId())
);
}
@@ -157,11 +159,12 @@
}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)
+ .set(AiKnowledgeFile::getUpdatedTime, new Date())
.eq(AiKnowledgeFile::getId, aiKnowledgeFile.getId())
);
}
@@ -181,12 +184,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