diff --git a/src/main/java/cn/bugstack/chatglm/model/Role.java b/src/main/java/cn/bugstack/chatglm/model/Role.java index df0bf7e8d042b37c15cfd7f5db0205d70f531e56..2b7aa9af7b51e288e81ddfd5122ff5a3e05af049 100644 --- a/src/main/java/cn/bugstack/chatglm/model/Role.java +++ b/src/main/java/cn/bugstack/chatglm/model/Role.java @@ -20,6 +20,10 @@ public enum Role { * 模型生成的内容,role位assistant */ assistant("assistant"), + + /** + * 系统 + */ system("system"), ; diff --git a/src/test/java/cn/bugstack/chatglm/test/ApiTest.java b/src/test/java/cn/bugstack/chatglm/test/ApiTest.java index bdd1d8801c8f57597a94ea711040d1f50f50de43..5e40c45d0b4ac10d85db05a2100da468f2073bd4 100644 --- a/src/test/java/cn/bugstack/chatglm/test/ApiTest.java +++ b/src/test/java/cn/bugstack/chatglm/test/ApiTest.java @@ -62,6 +62,7 @@ public class ApiTest { "要求语言通俗易懂、幽默有趣,并且要以第一人称的口吻。") .build()); + /* system 和 user 为一组出现 */ add(ChatCompletionRequest.Prompt.builder() .role(Role.system.getCode()) .content("请问有什么需要帮助的吗")