提交 bad76444 编写于 作者: 小傅哥's avatar 小傅哥

feat:chatglm_turbo 新增模型

上级 5ea80c6b
...@@ -13,11 +13,14 @@ import lombok.Getter; ...@@ -13,11 +13,14 @@ import lombok.Getter;
@AllArgsConstructor @AllArgsConstructor
public enum Model { public enum Model {
@Deprecated
CHATGLM_6B_SSE("chatGLM_6b_SSE", "ChatGLM-6B 测试模型"), CHATGLM_6B_SSE("chatGLM_6b_SSE", "ChatGLM-6B 测试模型"),
CHATGLM_LITE("chatglm_lite", "轻量版模型,适用对推理速度和成本敏感的场景"), CHATGLM_LITE("chatglm_lite", "轻量版模型,适用对推理速度和成本敏感的场景"),
CHATGLM_LITE_32K("chatglm_lite_32k", "标准版模型,适用兼顾效果和成本的场景"), CHATGLM_LITE_32K("chatglm_lite_32k", "标准版模型,适用兼顾效果和成本的场景"),
CHATGLM_STD("chatglm_std", "适用于对知识量、推理能力、创造力要求较高的场景"), CHATGLM_STD("chatglm_std", "适用于对知识量、推理能力、创造力要求较高的场景"),
CHATGLM_PRO("chatglm_pro", "适用于对知识量、推理能力、创造力要求较高的场景"), CHATGLM_PRO("chatglm_pro", "适用于对知识量、推理能力、创造力要求较高的场景"),
/** 智谱AI最新模型 */
CHATGLM_TURBO("chatglm_turbo", "适用于对知识量、推理能力、创造力要求较高的场景"),
; ;
private final String code; private final String code;
......
...@@ -51,7 +51,7 @@ public class ApiTest { ...@@ -51,7 +51,7 @@ public class ApiTest {
public void test_completions() throws JsonProcessingException, InterruptedException { public void test_completions() throws JsonProcessingException, InterruptedException {
// 入参;模型、请求信息 // 入参;模型、请求信息
ChatCompletionRequest request = new ChatCompletionRequest(); ChatCompletionRequest request = new ChatCompletionRequest();
request.setModel(Model.CHATGLM_LITE); // chatGLM_6b_SSE、chatglm_lite、chatglm_lite_32k、chatglm_std、chatglm_pro request.setModel(Model.CHATGLM_TURBO); // chatGLM_6b_SSE、chatglm_lite、chatglm_lite_32k、chatglm_std、chatglm_pro
request.setIncremental(false); request.setIncremental(false);
request.setPrompt(new ArrayList<ChatCompletionRequest.Prompt>() { request.setPrompt(new ArrayList<ChatCompletionRequest.Prompt>() {
private static final long serialVersionUID = -7988151926241837899L; private static final long serialVersionUID = -7988151926241837899L;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册