From 95ae0c83d2aa58d86b4039e3f7f16ee93afb8f00 Mon Sep 17 00:00:00 2001
From: 云 <2163098428@qq.com>
Date: 星期五, 23 十月 2026 17:01:00 +0800
Subject: [PATCH] feat(ai): 完善模型管理和智能体导入功能
---
yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/model/AiModelTypeEnum.java | 18 +++++++-----------
1 files changed, 7 insertions(+), 11 deletions(-)
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/model/AiModelTypeEnum.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/model/AiModelTypeEnum.java
index bdba3e8..a6ef4f9 100644
--- a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/model/AiModelTypeEnum.java
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/model/AiModelTypeEnum.java
@@ -3,13 +3,10 @@
import cn.iocoder.yudao.framework.common.core.ArrayValuable;
import lombok.Getter;
import lombok.RequiredArgsConstructor;
-
import java.util.Arrays;
/**
* AI 妯″瀷绫诲瀷鐨勬灇涓�
- *
- * @author 鑺嬮亾婧愮爜
*/
@Getter
@RequiredArgsConstructor
@@ -22,20 +19,19 @@
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; }
+
+ public static String getName(Integer type) {
+ for (AiModelTypeEnum e : values()) {
+ if (e.getType().equals(type)) return e.getName();
+ }
+ return String.valueOf(type);
}
}
--
Gitblit v1.9.3