From f1806477c4972c2d09f32df848216bcb69f4d680 Mon Sep 17 00:00:00 2001
From: Administrator <15274802129@163.com>
Date: Mon, 29 Sep 2025 14:00:09 +0800
Subject: [PATCH] feat(ai): 新增阿里云百炼知识库工具类 - 添加 KnowledgeBaseUtil 工具类，封装阿里云百炼知识库操作 - 实现文件上传、知识库创建与更新等核心功能 - 支持文件MD5计算、分类管理、索引任务提交与状态查询 - 集成阿里云百炼SDK依赖，版本为2.5.0 - 提供完整的知识库初始化与文件更新流程示例 - 添加文件解析状态轮询与错误处理机制 - 支持知识库中文档的追加导入与旧文档删除功能
---
src/main/java/cc/mrbird/febs/ai/strategy/Impl/AliApplicationLlmStrategyServiceImpl.java | 37 +++++++++++++++++--------------------
1 files changed, 17 insertions(+), 20 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 7a158ef..95441da 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
@@ -39,6 +39,7 @@
private final String bizParam_1 = "prompt_ai_system";
private final String bizParam_2 = "question";
private final String bizParam_3 = "query";
+ private final String bizParam_4 = "messages";
private HashMap getPrompt(List<LlmStrategyDto> dto) {
HashMap<String, String> bizParamsMap = new HashMap<>();
@@ -52,25 +53,11 @@
if (StrUtil.equals(dtoItem.getRole(),Role.USER.getValue())){
bizParamsMap.put(bizParam_3, dtoItem.getContent());
}
- }
- return bizParamsMap;
- }
-
- private List<Message> getMessages(List<LlmStrategyDto> dto) {
- List<Message> messages = new ArrayList<>();
- for (LlmStrategyDto dtoItem : dto){
- if (StrUtil.equals(AiTypeEnum.MESSAGES.getName(), dtoItem.getRole())){
- List<LlmStrategyDto> messages1 = dtoItem.getMessages();
- for (LlmStrategyDto dtoItem1 : messages1){
- messages.add(Message.builder()
- .role(dtoItem1.getRole())
- .content(dtoItem1.getContent())
- .build());
- }
- break;
+ if (StrUtil.equals(dtoItem.getRole(),AiTypeEnum.MESSAGES.getName())){
+ bizParamsMap.put(bizParam_4, JSONUtil.toJsonStr(dtoItem.getMessages()));
}
}
- return messages;
+ return bizParamsMap;
}
private String getQuery(List<LlmStrategyDto> dto) {
@@ -90,6 +77,17 @@
if (StrUtil.equals(dtoItem.getRole(),Role.TOOL.getValue())){
int code = Integer.parseInt(dtoItem.getContent());
appId = LlmApplicationAppIdEnum.HIGH_LIGHT.getAppIdByCode(code);
+ break;
+ }
+ }
+ return appId;
+ }
+
+ private String getAppIdV2(List<LlmStrategyDto> dto) {
+ String appId = null;
+ for (LlmStrategyDto dtoItem : dto){
+ if (StrUtil.equals(dtoItem.getRole(),Role.TOOL.getValue())){
+ appId = dtoItem.getContent();
break;
}
}
@@ -178,9 +176,9 @@
throw new FebsException("百炼工作流初始化异常");
}
HashMap prompt = getPrompt(dto);
- List<Message> messages = getMessages(dto);
String query = getQuery(dto);
- String appId = getAppId(dto);
+// String appId = getAppId(dto);
+ String appId = getAppIdV2(dto);
if (prompt == null || prompt.size() == 0){
throw new FebsException("百炼工作流初始化异常");
}
@@ -197,7 +195,6 @@
.appId(appId) //替换为实际的应用 ID
.flowStreamMode(FlowStreamMode.MESSAGE_FORMAT)
.prompt(query)
- .messages( messages)
.bizParams(JsonUtils.toJsonObject( prompt))
.build();
--
Gitblit v1.9.1