| | |
| | | import cn.iocoder.yudao.framework.common.core.ArrayValuable; |
| | | import lombok.Getter; |
| | | import lombok.RequiredArgsConstructor; |
| | | |
| | | import java.util.Arrays; |
| | | |
| | | /** |
| | | * AI 模型类型的枚举 |
| | | * |
| | | * @author 芋道源码 |
| | | */ |
| | | @Getter |
| | | @RequiredArgsConstructor |
| | |
| | | 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; |
| | | } |
| | | public Integer[] array() { return ARRAYS; } |
| | | |
| | | } |