Administrator
yesterday 98b037ec600f2543d2690195bd2e08b879de2014
fix(ai):修复知识库文件状态更新逻辑

- 调整文件解析成功状态值从1为2- 调整索引任务完成状态值从1为3
- 修复状态更新条件判断逻辑
- 使用jobId替代fileId进行索引状态查询
- 统一状态更新时的条件判断值
2 files modified
14 ■■■■ changed files
src/main/java/cc/mrbird/febs/ai/service/impl/AiKnowledgeFileServiceImpl.java 10 ●●●● patch | view | raw | blame | history
src/main/java/cc/mrbird/febs/ai/util/KnowledgeBaseUtil.java 4 ●●●● patch | view | raw | blame | history
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())
                );
            }
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();