From 537cea53b6e54fdcf1b178f1ef84d5fd07e27f57 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Mon, 01 Sep 2025 17:48:13 +0800
Subject: [PATCH] refactor(ai): 重构 AI模块代码
---
src/main/java/cc/mrbird/febs/ai/strategy/enumerates/LlmStrategyContextEnum.java | 16 ++++++++++++++++
src/main/java/cc/mrbird/febs/ai/strategy/enumerates/LlmStrategyEnum.java | 2 +-
src/main/java/cc/mrbird/febs/ai/controller/TestController.java | 4 +---
src/main/java/cc/mrbird/febs/ai/strategy/Impl/AliLlmStrategyServiceImpl.java | 5 +++--
4 files changed, 21 insertions(+), 6 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/controller/TestController.java b/src/main/java/cc/mrbird/febs/ai/controller/TestController.java
index 7bc45f1..16f0d63 100644
--- a/src/main/java/cc/mrbird/febs/ai/controller/TestController.java
+++ b/src/main/java/cc/mrbird/febs/ai/controller/TestController.java
@@ -4,7 +4,7 @@
import cc.mrbird.febs.ai.mapper.AiMemberMapper;
import cc.mrbird.febs.ai.req.talk.AiTalkAnswerStream;
import cc.mrbird.febs.ai.res.memberTalk.ApiMemberTalkStreamVo;
-import cc.mrbird.febs.ai.strategy.LlmStrategyEnum;
+import cc.mrbird.febs.ai.strategy.enumerates.LlmStrategyEnum;
import cc.mrbird.febs.ai.strategy.LlmStrategyFactory;
import cc.mrbird.febs.ai.strategy.param.LlmStrategyDto;
import cc.mrbird.febs.ai.utils.UUID;
@@ -18,8 +18,6 @@
import cn.hutool.core.util.StrUtil;
import cn.hutool.crypto.asymmetric.KeyType;
import cn.hutool.crypto.asymmetric.RSA;
-import cn.hutool.json.JSONUtil;
-import com.alibaba.dashscope.aigc.multimodalconversation.MultiModalConversationResult;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import io.reactivex.Flowable;
diff --git a/src/main/java/cc/mrbird/febs/ai/strategy/Impl/AliLlmStrategyServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/strategy/Impl/AliLlmStrategyServiceImpl.java
index 9f6f69e..2f4c68f 100644
--- a/src/main/java/cc/mrbird/febs/ai/strategy/Impl/AliLlmStrategyServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/strategy/Impl/AliLlmStrategyServiceImpl.java
@@ -1,6 +1,7 @@
package cc.mrbird.febs.ai.strategy.Impl;
import cc.mrbird.febs.ai.strategy.LlmStrategyService;
+import cc.mrbird.febs.ai.strategy.enumerates.LlmStrategyContextEnum;
import cc.mrbird.febs.ai.strategy.param.LlmStrategyDto;
import cc.mrbird.febs.common.entity.FebsResponse;
import cc.mrbird.febs.common.exception.FebsException;
@@ -114,10 +115,10 @@
.map(message -> {
HashMap<String, String> stringStringHashMap = new HashMap<>();
if (StrUtil.isNotEmpty(message.getOutput().getChoices().get(0).getMessage().getReasoningContent())){
- stringStringHashMap.put("reasoningContent",message.getOutput().getChoices().get(0).getMessage().getReasoningContent());
+ stringStringHashMap.put(LlmStrategyContextEnum.THINK.name(),message.getOutput().getChoices().get(0).getMessage().getReasoningContent());
}
if (StrUtil.isNotEmpty(message.getOutput().getChoices().get(0).getMessage().getContent())){
- stringStringHashMap.put("content",message.getOutput().getChoices().get(0).getMessage().getReasoningContent());
+ stringStringHashMap.put(LlmStrategyContextEnum.CONTENT.name(),message.getOutput().getChoices().get(0).getMessage().getReasoningContent());
}
return new FebsResponse().success().data(stringStringHashMap);
})
diff --git a/src/main/java/cc/mrbird/febs/ai/strategy/enumerates/LlmStrategyContextEnum.java b/src/main/java/cc/mrbird/febs/ai/strategy/enumerates/LlmStrategyContextEnum.java
new file mode 100644
index 0000000..31254c0
--- /dev/null
+++ b/src/main/java/cc/mrbird/febs/ai/strategy/enumerates/LlmStrategyContextEnum.java
@@ -0,0 +1,16 @@
+package cc.mrbird.febs.ai.strategy.enumerates;
+
+public enum LlmStrategyContextEnum {
+ THINK("思考过程","THINK"),
+ CONTENT("响应内容","CONTENT");
+
+ private final String code;
+ private final String name;
+
+ LlmStrategyContextEnum(String code,String name) {
+
+ this.code = code;
+ this.name = name;
+ }
+
+}
diff --git a/src/main/java/cc/mrbird/febs/ai/strategy/LlmStrategyEnum.java b/src/main/java/cc/mrbird/febs/ai/strategy/enumerates/LlmStrategyEnum.java
similarity index 93%
rename from src/main/java/cc/mrbird/febs/ai/strategy/LlmStrategyEnum.java
rename to src/main/java/cc/mrbird/febs/ai/strategy/enumerates/LlmStrategyEnum.java
index 5a3c2d2..fe13c1d 100644
--- a/src/main/java/cc/mrbird/febs/ai/strategy/LlmStrategyEnum.java
+++ b/src/main/java/cc/mrbird/febs/ai/strategy/enumerates/LlmStrategyEnum.java
@@ -1,4 +1,4 @@
-package cc.mrbird.febs.ai.strategy;
+package cc.mrbird.febs.ai.strategy.enumerates;
import lombok.Getter;
--
Gitblit v1.9.1