baison/ruoyi-vue-pro
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1.AI助手模版、样式修改成谢提供的对话模版（有特效）。

Browse Source 
2.点击AI分析后，对话弹出类似商品链接引入，用户点击确认后进行分析。
3.AI分析对话新增提示词编辑，根据组件名称：模块编号进行获取提示词进行模块分析。
4.商品大盘首页字体调整，数据卡片中的字体调小一点，文字描述超长不显示，鼠标放上去显示全量。
This commit is contained in:
ouhaolan
2026-03-20 08:41:45 +08:00
parent 0bfd22ccca
commit 5bdcdf205a
14 changed files with 628 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
sql/mysql/ai_image_chat_menu.sql Normal file
View File

@@ -0,0 +1,4 @@
-- AI 图片对话菜单（插入到 AI 大模型 2758 下）
-- 执行前请确认 id 5100 未被占用，若冲突可改为其他值
INSERT INTO system_menu (id, name, permission, type, sort, parent_id, path, icon, component, component_name, status, visible, keep_alive, always_show, creator, create_time, updater, update_time, deleted)
VALUES (5100, 'AI 图片对话', '', 2, 0, 2758, 'image-chat', 'ep:chat-dot-round', 'ydoyun/aichat/index.vue', 'AiImageChat', 0, '1', '1', '1', '1', NOW(), '1', NOW(), '0');

24
sql/mysql/ydoyun_ai_module_prompt.sql Normal file
View File

@@ -0,0 +1,24 @@
-- AI 模块提示词配置表：存储各组件模块的 AI 分析提示词，key 为 组件名称:模块编号
CREATE TABLE IF NOT EXISTS `ydoyun_ai_module_prompt` (
`id` bigint NOT NULL AUTO_INCREMENT COMMENT '主键ID',
`component_name` varchar(64) NOT NULL COMMENT '组件名称，如 ProductDashboard',
`module_name` varchar(128) NOT NULL COMMENT '模块名称，如 KPI指标、供应商表现',
`module_key` varchar(128) NOT NULL COMMENT '组件:模块编号，如 ProductDashboard:kpi',
`prompt` text COMMENT 'AI 提示词，用于定义任务并传入 Dify 分析',
`creator` varchar(64) DEFAULT '' NULL COMMENT '创建者',
`create_time` datetime DEFAULT CURRENT_TIMESTAMP NOT NULL COMMENT '创建时间',
`updater` varchar(64) DEFAULT '' NULL COMMENT '更新者',
`update_time` datetime DEFAULT CURRENT_TIMESTAMP NOT NULL ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间',
`deleted` bit(1) DEFAULT b'0' NOT NULL COMMENT '是否删除',
`tenant_id` bigint DEFAULT 0 NULL COMMENT '租户ID',
PRIMARY KEY (`id`),
UNIQUE KEY `uk_tenant_module_key` (`tenant_id`, `module_key`) COMMENT '租户内组件:模块编号唯一',
KEY `idx_component_name` (`component_name`) COMMENT '按组件名称查询',
KEY `idx_tenant_id` (`tenant_id`) COMMENT '按租户查询'
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci COMMENT='AI 模块提示词配置表';
-- ProductDashboard 可分析模块说明（供 AI 定义任务时列举）：
-- ProductDashboard:kpi - KPI 指标卡片
-- ProductDashboard:supplier - 供应商表现 (Top 10)
-- ProductDashboard:pie - 中类销售排名 Top 5
-- ProductDashboard:productList - 商品明细列表

66
yudao-module-ydoyun/src/main/java/cn/iocoder/yudao/module/ydoyun/controller/admin/aichat/AiChatController.java Normal file
View File

@@ -0,0 +1,66 @@
package cn.iocoder.yudao.module.ydoyun.controller.admin.aichat;
import cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils;
import cn.iocoder.yudao.module.ydoyun.service.aichat.DifyAiChatService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.MediaType;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import javax.servlet.http.HttpServletResponse;
import java.io.OutputStream;
import java.nio.charset.StandardCharsets;
/**
* AI 对话（流式响应）
* 前端直接上传图片文件，后端上传到 Dify 后对话
* 使用登录用户 ID 作为 Dify user 参数，实现各用户对话隔离
*/
@Tag(name = "管理后台 - AI 图片对话")
@Slf4j
@RestController
@RequestMapping("/ydoyun/ai-chat")
@RequiredArgsConstructor
public class AiChatController {
private final DifyAiChatService difyAiChatService;
@Operation(summary = "流式图片对话")
@PostMapping(value = "/stream", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
public void streamChat(
@RequestParam("query") String query,
@RequestParam(value = "file", required = false) MultipartFile file,
@RequestParam(value = "task_desc", required = false) String taskDesc,
HttpServletResponse response
) {
String user = "system";
try {
Long userId = SecurityFrameworkUtils.getLoginUserId();
if (userId != null) {
user = "user_" + userId;
}
} catch (Exception e) {
log.debug("获取登录用户失败，使用默认 user: {}", e.getMessage());
}
response.setContentType("text/event-stream;charset=UTF-8");
response.setHeader("Cache-Control", "no-cache");
response.setHeader("Connection", "keep-alive");
response.setHeader("X-Accel-Buffering", "no");
try (OutputStream out = response.getOutputStream()) {
difyAiChatService.streamChat(query, file, taskDesc, user, out);
} catch (Exception e) {
log.error("AI 图片对话流式响应异常", e);
try {
String err = "data: {\"error\":\"" + e.getMessage().replace("\"", "\\\"") + "\"}\n\n";
response.getOutputStream().write(err.getBytes(StandardCharsets.UTF_8));
response.getOutputStream().flush();
} catch (Exception ex) {
log.warn("写入错误响应失败", ex);
}
}
}
}

18
yudao-module-ydoyun/src/main/java/cn/iocoder/yudao/module/ydoyun/controller/admin/aichat/vo/AiChatStreamReqVO.java Normal file
View File

@@ -0,0 +1,18 @@
package cn.iocoder.yudao.module.ydoyun.controller.admin.aichat.vo;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;
import javax.validation.constraints.NotBlank;
@Schema(description = "AI 对话流式请求")
@Data
public class AiChatStreamReqVO {
@Schema(description = "用户输入/问题", requiredMode = Schema.RequiredMode.REQUIRED, example = "你好")
@NotBlank(message = "问题不能为空")
private String query;
@Schema(description = "图片 URL（上传或粘贴后获取的完整访问地址）")
private String pic;
}

57
yudao-module-ydoyun/src/main/java/cn/iocoder/yudao/module/ydoyun/controller/admin/aimoduleprompt/AiModulePromptController.java Normal file
View File

@@ -0,0 +1,57 @@
package cn.iocoder.yudao.module.ydoyun.controller.admin.aimoduleprompt;
import cn.iocoder.yudao.framework.common.pojo.CommonResult;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.module.ydoyun.controller.admin.aimoduleprompt.vo.AiModulePromptRespVO;
import cn.iocoder.yudao.module.ydoyun.controller.admin.aimoduleprompt.vo.AiModulePromptSaveReqVO;
import cn.iocoder.yudao.module.ydoyun.dal.dataobject.aimoduleprompt.AiModulePromptDO;
import cn.iocoder.yudao.module.ydoyun.service.aimoduleprompt.AiModulePromptService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.tags.Tag;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import javax.annotation.Resource;
import javax.validation.Valid;
import java.util.List;
import java.util.Map;
import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
/**
* AI 模块提示词配置 Controller
*
* @author 衣朵云
*/
@Tag(name = "管理后台 - AI 模块提示词")
@RestController
@RequestMapping("/ydoyun/ai-module-prompt")
@Validated
public class AiModulePromptController {
@Resource
private AiModulePromptService aiModulePromptService;
@PostMapping("/save")
@Operation(summary = "保存提示词")
public CommonResult<Boolean> savePrompt(@Valid @RequestBody AiModulePromptSaveReqVO reqVO) {
aiModulePromptService.savePrompt(reqVO);
return success(true);
}
@GetMapping("/list-by-component")
@Operation(summary = "按组件名称查询提示词列表")
@Parameter(name = "componentName", description = "组件名称", required = true, example = "ProductDashboard")
public CommonResult<List<AiModulePromptRespVO>> listByComponent(@RequestParam("componentName") String componentName) {
List<AiModulePromptDO> list = aiModulePromptService.getListByComponent(componentName);
return success(BeanUtils.toBean(list, AiModulePromptRespVO.class));
}
@GetMapping("/map-by-component")
@Operation(summary = "按组件名称查询提示词映射（moduleKey -> prompt）")
@Parameter(name = "componentName", description = "组件名称", required = true, example = "ProductDashboard")
public CommonResult<Map<String, String>> getPromptMapByComponent(@RequestParam("componentName") String componentName) {
return success(aiModulePromptService.getPromptMapByComponent(componentName));
}
}

26
yudao-module-ydoyun/src/main/java/cn/iocoder/yudao/module/ydoyun/controller/admin/aimoduleprompt/vo/AiModulePromptRespVO.java Normal file
View File

@@ -0,0 +1,26 @@
package cn.iocoder.yudao.module.ydoyun.controller.admin.aimoduleprompt.vo;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;
import java.time.LocalDateTime;
@Schema(description = "管理后台 - AI 模块提示词 Response VO")
@Data
public class AiModulePromptRespVO {
@Schema(description = "主键ID", example = "1")
private Long id;
@Schema(description = "组件名称", example = "ProductDashboard")
private String componentName;
@Schema(description = "模块名称", example = "KPI指标")
private String moduleName;
@Schema(description = "组件:模块编号", example = "ProductDashboard:kpi")
private String moduleKey;
@Schema(description = "AI 提示词")
private String prompt;
@Schema(description = "创建时间")
private LocalDateTime createTime;
@Schema(description = "更新时间")
private LocalDateTime updateTime;
}

29
yudao-module-ydoyun/src/main/java/cn/iocoder/yudao/module/ydoyun/controller/admin/aimoduleprompt/vo/AiModulePromptSaveReqVO.java Normal file
View File

@@ -0,0 +1,29 @@
package cn.iocoder.yudao.module.ydoyun.controller.admin.aimoduleprompt.vo;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;
import javax.validation.constraints.NotBlank;
@Schema(description = "管理后台 - AI 模块提示词保存 Request VO")
@Data
public class AiModulePromptSaveReqVO {
@Schema(description = "主键ID，更新时必填", example = "1")
private Long id;
@Schema(description = "组件名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "ProductDashboard")
@NotBlank(message = "组件名称不能为空")
private String componentName;
@Schema(description = "模块名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "KPI指标")
@NotBlank(message = "模块名称不能为空")
private String moduleName;
@Schema(description = "组件:模块编号，唯一键", requiredMode = Schema.RequiredMode.REQUIRED, example = "ProductDashboard:kpi")
@NotBlank(message = "模块编号不能为空")
private String moduleKey;
@Schema(description = "AI 提示词")
private String prompt;
}

34
yudao-module-ydoyun/src/main/java/cn/iocoder/yudao/module/ydoyun/dal/dataobject/aimoduleprompt/AiModulePromptDO.java Normal file
View File

@@ -0,0 +1,34 @@
package cn.iocoder.yudao.module.ydoyun.dal.dataobject.aimoduleprompt;
import com.baomidou.mybatisplus.annotation.KeySequence;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import cn.iocoder.yudao.framework.tenant.core.db.TenantBaseDO;
import lombok.*;
/**
* AI 模块提示词配置 DO
*
* @author 衣朵云
*/
@TableName("ydoyun_ai_module_prompt")
@KeySequence("ydoyun_ai_module_prompt_seq")
@Data
@EqualsAndHashCode(callSuper = true)
@ToString(callSuper = true)
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class AiModulePromptDO extends TenantBaseDO {
@TableId
private Long id;
/** 组件名称，如 ProductDashboard */
private String componentName;
/** 模块名称，如 KPI指标、供应商表现 */
private String moduleName;
/** 组件:模块编号，唯一键，如 ProductDashboard:kpi */
private String moduleKey;
/** AI 提示词 */
private String prompt;
}

24
yudao-module-ydoyun/src/main/java/cn/iocoder/yudao/module/ydoyun/dal/mysql/aimoduleprompt/AiModulePromptMapper.java Normal file
View File

@@ -0,0 +1,24 @@
package cn.iocoder.yudao.module.ydoyun.dal.mysql.aimoduleprompt;
import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
import cn.iocoder.yudao.module.ydoyun.dal.dataobject.aimoduleprompt.AiModulePromptDO;
import org.apache.ibatis.annotations.Mapper;
import java.util.List;
/**
* AI 模块提示词配置 Mapper
*
* @author 衣朵云
*/
@Mapper
public interface AiModulePromptMapper extends BaseMapperX<AiModulePromptDO> {
default AiModulePromptDO selectByModuleKey(String moduleKey) {
return selectOne(AiModulePromptDO::getModuleKey, moduleKey);
}
default List<AiModulePromptDO> selectListByComponentName(String componentName) {
return selectList(AiModulePromptDO::getComponentName, componentName);
}
}

20
yudao-module-ydoyun/src/main/java/cn/iocoder/yudao/module/ydoyun/service/aichat/DifyAiChatService.java Normal file
View File

@@ -0,0 +1,20 @@
package cn.iocoder.yudao.module.ydoyun.service.aichat;
import org.springframework.web.multipart.MultipartFile;
/**
* Dify AI 对话服务（流式响应）
*/
public interface DifyAiChatService {
/**
* 流式发送对话请求，逐字返回 AI 响应
*
* @param query 用户问题
* @param file 图片文件（可选，前端直接上传）
* @param taskDesc 任务描述（可选，传给 Dify task_desc 做 AI 分析）
* @param user Dify 用户标识，用于对话隔离（不同用户对话互不污染）
* @param output 输出流，用于写入 SSE 格式的响应
*/
void streamChat(String query, MultipartFile file, String taskDesc, String user, java.io.OutputStream output) throws Exception;
}

218
yudao-module-ydoyun/src/main/java/cn/iocoder/yudao/module/ydoyun/service/aichat/DifyAiChatServiceImpl.java Normal file
View File

@@ -0,0 +1,218 @@
package cn.iocoder.yudao.module.ydoyun.service.aichat;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.*;
import org.springframework.stereotype.Service;
import org.springframework.util.LinkedMultiValueMap;
import org.springframework.util.MultiValueMap;
import org.springframework.web.client.RestTemplate;
import org.springframework.core.io.ByteArrayResource;
import org.springframework.web.multipart.MultipartFile;
import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
import java.nio.charset.StandardCharsets;
import java.util.*;
/**
* Dify AI 对话服务实现（流式响应）
* 前端直接上传图片文件，后端上传到 Dify 后对话
*/
@Slf4j
@Service
public class DifyAiChatServiceImpl implements DifyAiChatService {
private static final ObjectMapper OBJECT_MAPPER = new ObjectMapper();
@Value("${ydoyun.dify.ai-chat.base-url:http://118.253.178.8:8888/v1}")
private String baseUrl;
@Value("${ydoyun.dify.ai-chat.api-key:app-uch2HPKpicnPgNpJCnQJTehq}")
private String apiKey;
private final RestTemplate restTemplate = new RestTemplate();
@Override
public void streamChat(String query, MultipartFile file, String taskDesc, String user, java.io.OutputStream output) throws Exception {
// 构建请求体
String base = baseUrl.replaceAll("/$", "");
String chatUrl = base + "/chat-messages";
List<Map<String, String>> files = new ArrayList<>();
if (file != null && !file.isEmpty()) {
String uploadFileId = uploadImageToDify(file.getBytes(), file.getOriginalFilename(), user);
if (uploadFileId != null) {
Map<String, String> fileObj = new HashMap<>();
fileObj.put("type", "image");
fileObj.put("transfer_method", "local_file");
fileObj.put("upload_file_id", uploadFileId);
files.add(fileObj);
} else {
log.warn("图片上传到 Dify 失败，将不携带图片继续请求");
}
}
HashMap<Object, Object> objectObjectHashMap = new HashMap<>();
objectObjectHashMap.put("pic", files);
if (taskDesc != null && !taskDesc.trim().isEmpty()) {
objectObjectHashMap.put("task_desc", taskDesc.trim());
}
Map<String, Object> requestBody = new HashMap<>();
requestBody.put("query", query);
requestBody.put("inputs", objectObjectHashMap);
if (!files.isEmpty()) {
requestBody.put("files", files);
}
requestBody.put("response_mode", "streaming");
requestBody.put("user", user);
String jsonBody = OBJECT_MAPPER.writeValueAsString(requestBody);
log.info("Dify 请求体：{}", jsonBody);
HttpURLConnection conn = null;
try {
URL url = new URL(chatUrl);
conn = (HttpURLConnection) url.openConnection();
conn.setRequestMethod("POST");
conn.setDoOutput(true);
conn.setRequestProperty("Content-Type", "application/json");
conn.setRequestProperty("Authorization", "Bearer " + apiKey);
conn.setConnectTimeout(30000);
conn.setReadTimeout(300000); // 5 分钟
try (OutputStream os = conn.getOutputStream()) {
os.write(jsonBody.getBytes(StandardCharsets.UTF_8));
}
int code = conn.getResponseCode();
if (code != 200) {
String err = readStream(conn.getErrorStream());
log.error("Dify 请求失败: {} {}", code, err);
writeSseError(output, "Dify 请求失败: " + code);
return;
}
// 3. 读取 Dify SSE 流并转发给前端
int chunkCount = 0;
try (BufferedReader reader = new BufferedReader(
new InputStreamReader(conn.getInputStream(), StandardCharsets.UTF_8))) {
String line;
while ((line = reader.readLine()) != null) {
if (line.startsWith("data: ")) {
String data = line.substring(6).trim();
if ("[DONE]".equals(data)) break;
try {
JsonNode node = OBJECT_MAPPER.readTree(data);
String event = node.path("event").asText("");
if (chunkCount < 3) {
log.info("[Dify] SSE event: event={}, hasAnswer={}", event, node.has("answer"));
}
// 兼容 message / agent_message 事件，均含 answer 增量
if (node.has("answer")) {
String answer = node.get("answer").asText();
if (answer != null && !answer.isEmpty()) {
Map<String, Object> chunk = new HashMap<String, Object>();
chunk.put("text", answer);
writeSseData(output, chunk);
chunkCount++;
}
}
if ("message_end".equals(event)) {
log.info("[Dify] 流式结束, 共转发 {} 个 chunk", chunkCount);
break;
}
if (node.has("code")) {
String errMsg = node.path("message").asText("未知错误");
log.warn("[Dify] 流式响应错误: code={}, message={}", node.path("code").asText(), errMsg);
writeSseError(output, errMsg);
break;
}
} catch (Exception e) {
log.debug("解析 SSE 行: {}", e.getMessage());
}
}
}
}
} finally {
if (conn != null) conn.disconnect();
}
}
/**
* 上传图片到 Dify，返回 upload_file_id
*/
private String uploadImageToDify(byte[] imageBytes, String originalFilename, String user) {
try {
String ext = "png";
if (originalFilename != null) {
int dot = originalFilename.lastIndexOf('.');
if (dot >= 0 && dot < originalFilename.length() - 1) {
ext = originalFilename.substring(dot + 1).toLowerCase();
}
}
if (!ext.matches("png|jpg|jpeg|webp|gif")) ext = "png";
String uploadUrl = baseUrl.replaceAll("/$", "") + "/files/upload";
HttpHeaders headers = new HttpHeaders();
headers.setContentType(MediaType.MULTIPART_FORM_DATA);
headers.set("Authorization", "Bearer " + apiKey);
final String fileExt = ext;
MultiValueMap<String, Object> body = new LinkedMultiValueMap<>();
body.add("file", new ByteArrayResource(imageBytes) {
@Override
public String getFilename() {
return "image." + fileExt;
}
});
body.add("user", user);
HttpEntity<MultiValueMap<String, Object>> entity = new HttpEntity<>(body, headers);
ResponseEntity<String> resp = restTemplate.postForEntity(uploadUrl, entity, String.class);
// Dify 上传成功返回 201 CREATED，非 200 OK
if ((resp.getStatusCode() == HttpStatus.OK || resp.getStatusCode() == HttpStatus.CREATED) && resp.getBody() != null) {
JsonNode node = OBJECT_MAPPER.readTree(resp.getBody());
JsonNode idNode = node.has("id") ? node.get("id") : (node.has("data") && node.get("data").has("id") ? node.get("data").get("id") : null);
if (idNode != null) {
return idNode.asText();
}
log.warn("[Dify] 上传响应无 id 字段: {}", resp.getBody());
} else {
log.warn("[Dify] 上传失败: status={}, body={}", resp.getStatusCode(), resp.getBody());
}
} catch (Exception e) {
log.error("[Dify] 上传图片到 Dify 异常", e);
}
return null;
}
private String readStream(InputStream is) throws IOException {
if (is == null) return "";
try (BufferedReader r = new BufferedReader(new InputStreamReader(is, StandardCharsets.UTF_8))) {
StringBuilder sb = new StringBuilder();
String line;
while ((line = r.readLine()) != null) sb.append(line);
return sb.toString();
}
}
private void writeSseData(OutputStream out, Map<String, Object> data) throws IOException {
try {
String json = OBJECT_MAPPER.writeValueAsString(data);
out.write(("data: " + json + "\n\n").getBytes(StandardCharsets.UTF_8));
out.flush();
} catch (Exception e) {
log.debug("写入 SSE: {}", e.getMessage());
}
}
private void writeSseError(OutputStream out, String msg) throws IOException {
Map<String, Object> errData = new HashMap<String, Object>();
errData.put("error", msg);
writeSseData(out, errData);
}
}

36
yudao-module-ydoyun/src/main/java/cn/iocoder/yudao/module/ydoyun/service/aimoduleprompt/AiModulePromptService.java Normal file
View File

@@ -0,0 +1,36 @@
package cn.iocoder.yudao.module.ydoyun.service.aimoduleprompt;
import cn.iocoder.yudao.module.ydoyun.controller.admin.aimoduleprompt.vo.AiModulePromptSaveReqVO;
import cn.iocoder.yudao.module.ydoyun.dal.dataobject.aimoduleprompt.AiModulePromptDO;
import javax.validation.Valid;
import java.util.List;
import java.util.Map;
/**
* AI 模块提示词配置 Service
*
* @author 衣朵云
*/
public interface AiModulePromptService {
/**
* 保存提示词（按 moduleKey 存在则更新，否则新增）
*/
void savePrompt(@Valid AiModulePromptSaveReqVO reqVO);
/**
* 按组件名称查询所有模块提示词，返回 moduleKey -> prompt 映射
*/
Map<String, String> getPromptMapByComponent(String componentName);
/**
* 按组件名称查询所有模块提示词列表
*/
List<AiModulePromptDO> getListByComponent(String componentName);
/**
* 按 moduleKey 查询
*/
AiModulePromptDO getByModuleKey(String moduleKey);
}

65
yudao-module-ydoyun/src/main/java/cn/iocoder/yudao/module/ydoyun/service/aimoduleprompt/AiModulePromptServiceImpl.java Normal file
View File

@@ -0,0 +1,65 @@
package cn.iocoder.yudao.module.ydoyun.service.aimoduleprompt;
import cn.iocoder.yudao.module.ydoyun.controller.admin.aimoduleprompt.vo.AiModulePromptSaveReqVO;
import cn.iocoder.yudao.module.ydoyun.dal.dataobject.aimoduleprompt.AiModulePromptDO;
import cn.iocoder.yudao.module.ydoyun.dal.mysql.aimoduleprompt.AiModulePromptMapper;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.validation.annotation.Validated;
import javax.annotation.Resource;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
/**
* AI 模块提示词配置 Service 实现类
*
* @author 衣朵云
*/
@Service
@Validated
public class AiModulePromptServiceImpl implements AiModulePromptService {
@Resource
private AiModulePromptMapper aiModulePromptMapper;
@Override
@Transactional(rollbackFor = Exception.class)
public void savePrompt(AiModulePromptSaveReqVO reqVO) {
AiModulePromptDO existing = aiModulePromptMapper.selectByModuleKey(reqVO.getModuleKey());
AiModulePromptDO entity = BeanUtils.toBean(reqVO, AiModulePromptDO.class);
if (existing != null) {
entity.setId(existing.getId());
aiModulePromptMapper.updateById(entity);
} else {
aiModulePromptMapper.insert(entity);
}
}
@Override
public Map<String, String> getPromptMapByComponent(String componentName) {
List<AiModulePromptDO> list = aiModulePromptMapper.selectListByComponentName(componentName);
Map<String, String> map = new HashMap<>();
if (list != null) {
for (AiModulePromptDO d : list) {
if (d.getPrompt() != null && !d.getPrompt().isEmpty()) {
map.put(d.getModuleKey(), d.getPrompt());
}
}
}
return map;
}
@Override
public List<AiModulePromptDO> getListByComponent(String componentName) {
return aiModulePromptMapper.selectListByComponentName(componentName);
}
@Override
public AiModulePromptDO getByModuleKey(String moduleKey) {
return aiModulePromptMapper.selectByModuleKey(moduleKey);
}
}

7
yudao-server/src/main/resources/application.yaml
View File

@@ -363,4 +363,11 @@ ydoyun:
base-url: http://118.253.178.8:5001/v1/chat-messages
api-key1: app-a6GWULbcg8zkv8g3VVLPARPz
api-key2: app-xc2JwxgwqTRZWybnkAgyLZQV
# AI 图片对话（独立应用，流式响应）
ai-chat:
base-url: http://118.253.178.8:5001/v1
api-key: app-uch2HPKpicnPgNpJCnQJTehq
pic-param: pic # Dify 应用中的图片参数名
pic-format: object # object=单对象 | array=数组 | id=仅 upload_file_id 字符串，若 AI 看不到图可切换尝试
image-payload-mode: both # both=同时传 inputs+files | inputs=只传 inputs.pic | files=只传顶层 files