From 98b037ec600f2543d2690195bd2e08b879de2014 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 30 Sep 2025 12:59:53 +0800
Subject: [PATCH] fix(ai):修复知识库文件状态更新逻辑
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiKnowledgeFileServiceImpl.java | 10 +++++-----
src/main/java/cc/mrbird/febs/ai/util/KnowledgeBaseUtil.java | 4 ++--
2 files changed, 7 insertions(+), 7 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..ad5f1f2 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())
);
}
diff --git a/src/main/java/cc/mrbird/febs/ai/util/KnowledgeBaseUtil.java b/src/main/java/cc/mrbird/febs/ai/util/KnowledgeBaseUtil.java
index 3ea8e89..ff05bbd 100644
--- a/src/main/java/cc/mrbird/febs/ai/util/KnowledgeBaseUtil.java
+++ b/src/main/java/cc/mrbird/febs/ai/util/KnowledgeBaseUtil.java
@@ -406,7 +406,7 @@
String status = describeResponse.getBody().getData().getStatus();
System.out.println("当前文件状态：" + status);
if ("PARSE_SUCCESS".equals(status)) {
- state = 1;
+ state = 2;
}
} catch (Exception e) {
e.printStackTrace();
@@ -426,7 +426,7 @@
String status = jobStatusResponse.getBody().getData().getStatus();
System.out.println("当前索引任务状态：" + status);
if ("COMPLETED".equals(status)) {
- state = 1;
+ state = 3;
}
} catch (Exception e) {
e.printStackTrace();
--
Gitblit v1.9.1