From 485c6557ae50afe6703c0b64169ce8eb634b1924 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Tue, 17 Mar 2026 13:01:36 +0800
Subject: [PATCH] feat(rabbit): 移除旧的消息队列配置并新增AI相关的实体和映射
---
src/main/java/cc/mrbird/febs/ai/strategy/Impl/AliApplicationLlmStrategyServiceImpl.java | 41 ++++++++++++++++++++++++++++++++++++++++-
1 files changed, 40 insertions(+), 1 deletions(-)
diff --git a/src/main/java/cc/mrbird/febs/ai/strategy/Impl/AliApplicationLlmStrategyServiceImpl.java b/src/main/java/cc/mrbird/febs/ai/strategy/Impl/AliApplicationLlmStrategyServiceImpl.java
index 95441da..1df1f7e 100644
--- a/src/main/java/cc/mrbird/febs/ai/strategy/Impl/AliApplicationLlmStrategyServiceImpl.java
+++ b/src/main/java/cc/mrbird/febs/ai/strategy/Impl/AliApplicationLlmStrategyServiceImpl.java
@@ -7,6 +7,7 @@
import cc.mrbird.febs.ai.strategy.param.LlmStrategyDto;
import cc.mrbird.febs.common.entity.FebsResponse;
import cc.mrbird.febs.common.exception.FebsException;
+import cc.mrbird.febs.yinhe.req.AiRequestDto;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.json.JSONUtil;
@@ -34,7 +35,7 @@
@Component("AliApplicationLlmStrategyServiceImpl")
public class AliApplicationLlmStrategyServiceImpl implements LlmStrategyService {
- private static final String apiKey = "sk-babdcf8799144134915cee2683794b2f";
+ private static final String apiKey = "sk-4b97b556ba7c4350a41f2856f75b9377";
private final String bizParam_1 = "prompt_ai_system";
private final String bizParam_2 = "question";
@@ -58,6 +59,36 @@
}
}
return bizParamsMap;
+ }
+
+ private List<Message> getMessages(List<LlmStrategyDto> dto) {
+ List<Message> messages = new ArrayList<>();
+ for (LlmStrategyDto item : dto){
+ if (StrUtil.equals(item.getRole(),AiTypeEnum.MESSAGES.getName())){
+ List<LlmStrategyDto> messagesList = item.getMessages();
+ for (LlmStrategyDto dtoItem : messagesList){
+ if (StrUtil.equals(dtoItem.getRole(),Role.SYSTEM.getValue())){
+ messages.add(Message.builder()
+ .role(Role.SYSTEM.getValue())
+ .content(dtoItem.getContent())
+ .build());
+ }
+ if (StrUtil.equals(dtoItem.getRole(),Role.USER.getValue())){
+ messages.add(Message.builder()
+ .role(Role.USER.getValue())
+ .content(dtoItem.getContent())
+ .build());
+ }
+ if (StrUtil.equals(dtoItem.getRole(),Role.ASSISTANT.getValue())){
+ messages.add(Message.builder()
+ .role(Role.ASSISTANT.getValue())
+ .content(dtoItem.getContent())
+ .build());
+ }
+ }
+ }
+ }
+ return messages;
}
private String getQuery(List<LlmStrategyDto> dto) {
@@ -188,6 +219,8 @@
if (appId == null){
throw new FebsException("百炼工作流初始化异常");
}
+
+ List<Message> messages = getMessages(dto);
long startTime = System.currentTimeMillis();
ApplicationParam param = ApplicationParam.builder()
// 若没有配置环境变量，可用百炼API Key将下行替换为：.apiKey("sk-xxx")。但不建议在生产环境中直接将API Key硬编码到代码中，以减少API Key泄露风险。
@@ -195,6 +228,7 @@
.appId(appId) //替换为实际的应用 ID
.flowStreamMode(FlowStreamMode.MESSAGE_FORMAT)
.prompt(query)
+ .messages( messages)
.bizParams(JsonUtils.toJsonObject( prompt))
.build();
@@ -222,4 +256,9 @@
throw new FebsException(StrUtil.format("百炼工作流输出失败:{}",error));
});
}
+
+ @Override
+ public Flux<FebsResponse> llmInvokeStreamingNoThink(AiRequestDto aiRequestDto) {
+ return null;
+ }
}
--
Gitblit v1.9.1