From 360984bc4b071d6550933ed37384004ce2278ed9 Mon Sep 17 00:00:00 2001 From: winkey Date: Mon, 7 Apr 2025 21:54:57 +0800 Subject: [PATCH] =?UTF-8?q?=E5=BA=94=E7=94=A8=E5=A2=9E=E5=8A=A0=E7=B3=BB?= =?UTF-8?q?=E7=BB=9F=E8=A7=92=E8=89=B2=E6=8F=90=E7=A4=BA=E8=AF=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../main/java/org/ruoyi/system/domain/ChatGpts.java | 5 +++++ .../java/org/ruoyi/system/domain/bo/ChatGptsBo.java | 10 ++++++++++ .../java/org/ruoyi/system/domain/vo/ChatGptsVo.java | 5 +++++ .../org/ruoyi/system/service/impl/SseServiceImpl.java | 2 +- 4 files changed, 21 insertions(+), 1 deletion(-) diff --git a/ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/domain/ChatGpts.java b/ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/domain/ChatGpts.java index a1cb2b48..8734eaa0 100644 --- a/ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/domain/ChatGpts.java +++ b/ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/domain/ChatGpts.java @@ -103,4 +103,9 @@ public class ChatGpts extends BaseEntity { private String modelName; + /** + * 模型system + */ + private String systemPrompt; + } diff --git a/ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/domain/bo/ChatGptsBo.java b/ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/domain/bo/ChatGptsBo.java index 01920938..dd08528f 100644 --- a/ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/domain/bo/ChatGptsBo.java +++ b/ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/domain/bo/ChatGptsBo.java @@ -84,4 +84,14 @@ public class ChatGptsBo extends BaseEntity { */ private String updateIp; + /** + * 模型名称 + */ + private String modelName; + + /** + * 模型system + */ + private String systemPrompt; + } diff --git a/ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/domain/vo/ChatGptsVo.java b/ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/domain/vo/ChatGptsVo.java index d1b17d1c..b6331798 100644 --- a/ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/domain/vo/ChatGptsVo.java +++ b/ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/domain/vo/ChatGptsVo.java @@ -102,5 +102,10 @@ public class ChatGptsVo implements Serializable { @ExcelProperty(value = "模型名称") private String modelName; + /** + * 模型system + */ + private String systemPrompt; + } diff --git a/ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/service/impl/SseServiceImpl.java b/ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/service/impl/SseServiceImpl.java index adb9ccff..7310b233 100644 --- a/ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/service/impl/SseServiceImpl.java +++ b/ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/service/impl/SseServiceImpl.java @@ -143,7 +143,7 @@ public class SseServiceImpl implements ISseService { openAiStreamClient = chatConfig.createOpenAiStreamClient(sysModel.getApiHost(), sysModel.getApiKey()); if (StringUtils.isNotEmpty(chatRequest.getAppId())) { // 设置应用的系统角色为描述 ChatGptsVo chatGptsVo = chatGptsService.queryById(Long.valueOf(chatRequest.getAppId())); - Message sysMessage = Message.builder().content(chatGptsVo.getInfo()).role(Message.Role.SYSTEM).build(); + Message sysMessage = Message.builder().content(chatGptsVo.getSystemPrompt()).role(Message.Role.SYSTEM).build(); messages.add(0,sysMessage); } else { // 模型设置默认提示词