From af0f57dfc905fc84df833a230dadf4749267779c Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 30 Sep 2025 13:02:01 +0800
Subject: [PATCH] fix(ai):修正文件删除失败时的异常信息- 将文件删除失败时抛出的异常信息从"文件删除成功"更正为"文件删除失败" - 保持异常处理逻辑的一致性，确保错误信息准确反映问题
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java | 36 ++++++++++++++++++++++++++++++++++++
1 files changed, 36 insertions(+), 0 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java
index 9c311a9..1bae742 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java
@@ -6,11 +6,16 @@
import cc.mrbird.febs.ai.mapper.AiCompanyMapper;
import cc.mrbird.febs.ai.mapper.AiCompanyWorkflowMapper;
import cc.mrbird.febs.ai.service.AiCompanyService;
+import cc.mrbird.febs.ai.util.KnowledgeBaseUtil;
import cc.mrbird.febs.ai.util.UUID;
import cc.mrbird.febs.common.entity.FebsResponse;
import cc.mrbird.febs.common.entity.QueryRequest;
+import cc.mrbird.febs.common.exception.FebsException;
+import cc.mrbird.febs.common.utils.ShareCodeUtil;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjectUtil;
+import com.aliyun.bailian20231229.Client;
+import com.aliyun.bailian20231229.models.CreateIndexResponse;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
@@ -19,6 +24,7 @@
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
import java.util.*;
import java.util.stream.Collectors;
@@ -44,11 +50,13 @@
}
@Override
+ @Transactional
public FebsResponse add(AiCompany dto) {
Date createTime = new Date();
AiCompany entity = new AiCompany();
entity.setId(UUID.getSimpleUUIDString());
entity.setName(dto.getName());
+ entity.setCode(ShareCodeUtil.toSerialCode(1L));
entity.setCreatedTime(createTime);
this.baseMapper.insert( entity);
@@ -61,9 +69,37 @@
this.addWorkflow(entity.getId(), LlmApplicationAppIdEnum.REPORT.getCode(),null,createTime);
this.addWorkflow(entity.getId(), LlmApplicationAppIdEnum.ADMIN_QUESTION.getCode(),null,createTime);
+ String categoryId = "";
+ try {
+ categoryId = KnowledgeBaseUtil.AddCategory(entity.getName(), null);
+ } catch (Exception e) {
+ throw new FebsException("初始化分类失败");
+ }
+
+ String knowledgeId = "";
+ try {
+ String name = entity.getName();
+ String fileId = "file_b7d0a5b2df1745b9bfb7402f5fe1054f_12629554";
+ knowledgeId = KnowledgeBaseUtil.createKnowledgeBase(fileId, name);
+ } catch (Exception e) {
+ throw new FebsException("初始化知识库失败");
+ }
+
+ this.update(null,
+ Wrappers.lambdaUpdate(AiCompany.class)
+ .set(AiCompany::getKnowledgeId, knowledgeId)
+ .set(AiCompany::getCategoryId, categoryId)
+ .eq(AiCompany::getId, entity.getId()));
+
return new FebsResponse().success().message("操作成功");
}
+ public static void main(String[] args) {
+ for (int i = 0; i < 10; i++){
+ System.out.println(ShareCodeUtil.toSerialCode(1L));
+ }
+ }
+
@Override
public FebsResponse update(AiCompany dto) {
String id = dto.getId();
--
Gitblit v1.9.1