From 2465c228997816822eaeb621556f52173643e7c9 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Mon, 29 Sep 2025 15:11:40 +0800
Subject: [PATCH] feat(ai):为AiCompany实体添加知识库和分类字段
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java | 21 +++++++++++++++------
1 files changed, 15 insertions(+), 6 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 f04f657..1be9722 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
@@ -2,12 +2,14 @@
import cc.mrbird.febs.ai.entity.AiCompany;
import cc.mrbird.febs.ai.entity.AiCompanyWorkflow;
+import cc.mrbird.febs.ai.enums.LlmApplicationAppIdEnum;
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.UUID;
import cc.mrbird.febs.common.entity.FebsResponse;
import cc.mrbird.febs.common.entity.QueryRequest;
+import cc.mrbird.febs.common.utils.ShareCodeUtil;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjectUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
@@ -48,21 +50,28 @@
AiCompany entity = new AiCompany();
entity.setId(UUID.getSimpleUUIDString());
entity.setName(dto.getName());
+ entity.setCode(ShareCodeUtil.toSerialCode(1L));
entity.setCreatedTime(createTime);
this.baseMapper.insert( entity);
- this.addWorkflow(entity.getId(),1,null,createTime);
- this.addWorkflow(entity.getId(),2,null,createTime);
- this.addWorkflow(entity.getId(),3,null,createTime);
- this.addWorkflow(entity.getId(),4,null,createTime);
- this.addWorkflow(entity.getId(),5,null,createTime);
- this.addWorkflow(entity.getId(),6,null,createTime);
+ this.addWorkflow(entity.getId(), LlmApplicationAppIdEnum.HIGH_LIGHT.getCode(),null,createTime);
+ this.addWorkflow(entity.getId(), LlmApplicationAppIdEnum.SUGGESTION.getCode(),null,createTime);
+ this.addWorkflow(entity.getId(), LlmApplicationAppIdEnum.REFERENCE_ANSWER.getCode(),null,createTime);
+ this.addWorkflow(entity.getId(), LlmApplicationAppIdEnum.KEY_KNOWLEDGE.getCode(),null,createTime);
+ this.addWorkflow(entity.getId(), LlmApplicationAppIdEnum.REPORT.getCode(),null,createTime);
+ this.addWorkflow(entity.getId(), LlmApplicationAppIdEnum.ADMIN_QUESTION.getCode(),null,createTime);
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