BJ_ZTEITS-Developers / parking-cloud-yudao

Commit 794d4c3195434a530230c56a14d7755cefd2c175

Authored by cherishsince
1 parent fb82dd2f

【调整】合并 AI 模型到一个 enums里面

Inline Side-by-side
Showing 2 changed files with 67 additions and 30 deletions
yudao-module-ai/yudao-module-ai-api/src/main/java/cn/iocoder/yudao/module/ai/enums/AiModelEnum.java 0 → 100644
  View file @794d4c3
  1 +package cn.iocoder.yudao.module.ai.enums;
  2 +
  3 +import lombok.AllArgsConstructor;
  4 +import lombok.Getter;
  5 +
  6 +// TODO done @fansili：1）类注释要加下；2）author 和 time 用 javadoc，@author 和 @since；3）@AllArgsConstructor 使用这个注解，去掉构造方法；4）value 改成 model 字段，然后注释都写下哈；5）message 改成 name，然后注释都写下哈
  7 +// TODO @fan: AiModelEnum 是不是可以缩写成这个哈；所有的模型，都写在这里枚举；
  8 +/**
  9 + * @author: fansili
  10 + * @time: 2024/3/4 12:36
  11 + */
  12 +@Getter
  13 +@AllArgsConstructor
  14 +public enum AiModelEnum {
  15 +
  16 + // open ai
  17 + OPEN_AI_GPT_3_5( "GPT3.5", "gpt-3.5-turbo",null),
  18 + OPEN_AI_GPT_4("GPT4", "gpt-4-turbo",null),
  19 +
  20 + // 千问付费模型
  21 + QWEN_TURBO("通义千问超大规模语言模型", "qwen-turbo", null),
  22 + QWEN_PLUS("通义千问超大规模语言模型增强版", "qwen-plus", null),
  23 + QWEN_MAX("通义千问千亿级别超大规模语言模型", "qwen-max", null),
  24 + QWEN_MAX_0403("通义千问千亿级别超大规模语言模型-0403", "qwen-max-0403", null),
  25 + QWEN_MAX_0107("通义千问千亿级别超大规模语言模型-0107", "qwen-max-0107", null),
  26 + QWEN_MAX_1201("通义千问千亿级别超大规模语言模型-1201", "qwen-max-1201", null),
  27 + QWEN_MAX_LONGCONTEXT("通义千问千亿级别超大规模语言模型-28k tokens", "qwen-max-longcontext", null),
  28 +
  29 + // 千问开源模型
  30 + // https://help.aliyun.com/document_detail/2666503.html?spm=a2c4g.2701795.0.0.26eb34dfKzcWN4
  31 + QWEN_72B_CHAT("通义千问1.5对外开源的72B规模参数量的经过人类指令对齐的chat模型", "qwen-72b-chat", null),
  32 +
  33 + // 一言模型
  34 + ERNIE4_0("ERNIE 4.0", "ERNIE 4.0", "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions_pro"),
  35 + ERNIE4_3_5_8K("ERNIE-3.5-8K", "ERNIE-3.5-8K", "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions"),
  36 + ERNIE4_3_5_8K_0205("ERNIE-3.5-8K-0205", "ERNIE-3.5-8K-0205", "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ernie-3.5-8k-0205"),
  37 +
  38 + ERNIE4_3_5_8K_1222("ERNIE-3.5-8K-1222", "ERNIE-3.5-8K-1222", "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ernie-3.5-8k-1222"),
  39 + ERNIE4_BOT_8K("ERNIE-Bot-8K", "ERNIE-Bot-8K", "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ernie_bot_8k"),
  40 + ERNIE4_3_5_4K_0205("ERNIE-3.5-4K-0205", "ERNIE-3.5-4K-0205", "/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ernie-3.5-4k-0205"),
  41 +
  42 +// 文档地址：https://www.xfyun.cn/doc/spark/Web.html#_1-%E6%8E%A5%E5%8F%A3%E8%AF%B4%E6%98%8E
  43 +// general指向V1.5版本;
  44 +// generalv2指向V2版本;
  45 +// generalv3指向V3版本;
  46 +// generalv3.5指向V3.5版本;
  47 +
  48 + XING_HUO_1_5("星火大模型1.5", "general", "/v1.1/chat"),
  49 + XING_HUO_2_0("星火大模型2.0", "generalv2", "/v2.1/chat"),
  50 + XING_HUO_3_0("星火大模型3.0", "generalv3", "/v3.1/chat"),
  51 + XING_HUO_3_5("星火大模型3.5", "generalv3.5", "/v3.5/chat"),
  52 + ;
  53 +
  54 + /**
  55 + * 模型名字 - 用于展示
  56 + */
  57 + private final String name;
  58 + /**
  59 + * 模型标志 - 用于参数传递
  60 + */
  61 + private final String model;
  62 + /**
  63 + * uri地址
  64 + */
  65 + private final String uri;
  66 +
  67 +}
yudao-module-ai/yudao-module-ai-api/src/main/java/cn/iocoder/yudao/module/ai/enums/AiOpenAiModelEnum.java deleted
View file @fb82dd2
1 -package cn.iocoder.yudao.module.ai.enums;  
2 -  
3 -import lombok.AllArgsConstructor;  
4 -import lombok.Getter;  
5 -  
6 -// TODO done @fansili：1）类注释要加下；2）author 和 time 用 javadoc，@author 和 @since；3）@AllArgsConstructor 使用这个注解，去掉构造方法；4）value 改成 model 字段，然后注释都写下哈；5）message 改成 name，然后注释都写下哈  
7 -// TODO @fan: AiModelEnum 是不是可以缩写成这个哈；所有的模型，都写在这里枚举；  
8 -/**  
9 - * @author: fansili  
10 - * @time: 2024/3/4 12:36  
11 - */  
12 -@Getter  
13 -@AllArgsConstructor  
14 -public enum AiOpenAiModelEnum {  
15 -  
16 -  
17 - OPEN_AI_GPT_3_5("gpt-3.5-turbo", "GPT3.5"),  
18 - OPEN_AI_GPT_4("gpt-4-turbo", "GPT4")  
19 - ;  
20 -  
21 - /**  
22 - * 模型标志 - 用于参数传递  
23 - */  
24 - private final String model;  
25 - /**  
26 - * 模型名字 - 用于展示  
27 - */  
28 - private final String name;  
29 -  
30 -}