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 | 13 +++++++------
src/main/java/cc/mrbird/febs/ai/service/impl/AiServiceImpl.java | 4 +++-
src/main/resources/templates/febs/views/modules/ai/workflow/list.html | 2 ++
src/main/java/cc/mrbird/febs/ai/enums/LlmApplicationAppIdEnum.java | 31 +++++++++++++++++++++++++++++++
4 files changed, 43 insertions(+), 7 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/enums/LlmApplicationAppIdEnum.java b/src/main/java/cc/mrbird/febs/ai/enums/LlmApplicationAppIdEnum.java
new file mode 100644
index 0000000..75dbed6
--- /dev/null
+++ b/src/main/java/cc/mrbird/febs/ai/enums/LlmApplicationAppIdEnum.java
@@ -0,0 +1,31 @@
+package cc.mrbird.febs.ai.enums;
+
+import lombok.Getter;
+
+@Getter
+public enum LlmApplicationAppIdEnum {
+
+ ADMIN_QUESTION(7,"生成问题"),
+
+ REPORT(6,"生成报告"),
+
+ NORMAL(5,"通用"),
+
+ KEY_KNOWLEDGE(4,"知识点总结"),
+
+ REFERENCE_ANSWER(3,"标准答案"),
+
+ SUGGESTION(2,"建议"),
+
+ HIGH_LIGHT(1,"亮点");
+
+ private final int code;
+ private final String value;
+
+ LlmApplicationAppIdEnum(int code, String value) {
+
+ this.code = code;
+ this.value = value;
+ }
+
+}
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..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,12 +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(),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("操作成功");
}
diff --git a/src/main/java/cc/mrbird/febs/ai/service/impl/AiServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/service/impl/AiServiceImpl.java
index 931084e..3b6013e 100644
--- a/src/main/java/cc/mrbird/febs/ai/service/impl/AiServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/service/impl/AiServiceImpl.java
@@ -1,6 +1,7 @@
package cc.mrbird.febs.ai.service.impl;
import cc.mrbird.febs.ai.entity.AiCompanyWorkflow;
+import cc.mrbird.febs.ai.enums.LlmApplicationAppIdEnum;
import cc.mrbird.febs.ai.mapper.AiCompanyWorkflowMapper;
import cc.mrbird.febs.ai.req.AiProductQuestionAiDto;
import cc.mrbird.febs.ai.service.AiCompanyService;
@@ -45,7 +46,8 @@
String companyAppId = appId;
String companyId = dto.getCompanyId();
if (StrUtil.isNotEmpty(companyId)){
- AiCompanyWorkflow workflowByTypeAndCompanyId = aiCompanyService.getWorkflowByTypeAndCompanyId(5, companyId);
+ AiCompanyWorkflow workflowByTypeAndCompanyId =
+ aiCompanyService.getWorkflowByTypeAndCompanyId(LlmApplicationAppIdEnum.ADMIN_QUESTION.getCode(), companyId);
companyAppId = workflowByTypeAndCompanyId.getCode();
}
diff --git a/src/main/resources/templates/febs/views/modules/ai/workflow/list.html b/src/main/resources/templates/febs/views/modules/ai/workflow/list.html
index 842095c..9b9d0d6 100644
--- a/src/main/resources/templates/febs/views/modules/ai/workflow/list.html
+++ b/src/main/resources/templates/febs/views/modules/ai/workflow/list.html
@@ -64,6 +64,8 @@
<span>知识点引用</span>
{{# }else if(d.type == 6) { }}
<span>生成报告</span>
+ {{# }else if(d.type == 7) { }}
+ <span>生成题目</span>
{{# } else { }}
<span>-</span>
{{# } }}
--
Gitblit v1.9.1