Prechádzať zdrojové kódy

应用增加系统角色提示词

winkey 3 týždňov pred
rodič
commit
360984bc4b

+ 5 - 0
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;
+
 }

+ 10 - 0
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;
+
 }

+ 5 - 0
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;
+
 
 }

+ 1 - 1
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 {
                         // 模型设置默认提示词