From c8e3460c5aefedfa45dc19a7a8d5720a10d0f058 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 23 Sep 2025 15:38:44 +0800
Subject: [PATCH] feat(ai): 引入LlmApplicationAppIdEnum枚举优化工作流类型管理
---
src/main/java/cc/mrbird/febs/ai/service/impl/AiCompanyServiceImpl.java | 22 +++++++++++++++++-----
1 files changed, 17 insertions(+), 5 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 5a1a242..9c311a9 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,6 +2,7 @@
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;
@@ -53,11 +54,12 @@
- 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(),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("操作成功");
}
@@ -92,6 +94,16 @@
}
@Override
+ public AiCompanyWorkflow getWorkflowByTypeAndCompanyId(int type, String companyId) {
+
+ LambdaQueryWrapper<AiCompanyWorkflow> query = Wrappers.lambdaQuery(AiCompanyWorkflow.class);
+ query.eq(AiCompanyWorkflow::getCompanyId, companyId);
+ query.eq(AiCompanyWorkflow::getType, type);
+ query.last("limit 1");
+ return aiCompanyWorkflowMapper.selectOne(query);
+ }
+
+ @Override
public IPage<AiCompanyWorkflow> workflowListInPage(AiCompanyWorkflow dto, QueryRequest request) {
Page<AiCompanyWorkflow> page = new Page<>(request.getPageNum(), request.getPageSize());
LambdaQueryWrapper<AiCompanyWorkflow> query = Wrappers.lambdaQuery(AiCompanyWorkflow.class);
--
Gitblit v1.9.1