1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
| package cn.iocoder.yudao.module.ai.enums.model;
|
| import cn.iocoder.yudao.framework.common.core.ArrayValuable;
| import lombok.Getter;
| import lombok.RequiredArgsConstructor;
| import java.util.Arrays;
|
| /**
| * AI 模型类型的枚举
| */
| @Getter
| @RequiredArgsConstructor
| public enum AiModelTypeEnum implements ArrayValuable<Integer> {
|
| CHAT(1, "对话"),
| IMAGE(2, "图片"),
| VOICE(3, "语音"),
| VIDEO(4, "视频"),
| EMBEDDING(5, "向量"),
| RERANK(6, "重排序");
|
| private final Integer type;
| private final String name;
|
| public static final Integer[] ARRAYS = Arrays.stream(values()).map(AiModelTypeEnum::getType).toArray(Integer[]::new);
|
| @Override
| public Integer[] array() { return ARRAYS; }
|
| }
|
|
