diff --git a/backend/ruoyi-modules/ruoyi-aihr/src/main/java/org/dromara/aihr/service/AihrModelSeedService.java b/backend/ruoyi-modules/ruoyi-aihr/src/main/java/org/dromara/aihr/service/AihrModelSeedService.java
index fb7709e7..87daca2d 100644
--- a/backend/ruoyi-modules/ruoyi-aihr/src/main/java/org/dromara/aihr/service/AihrModelSeedService.java
+++ b/backend/ruoyi-modules/ruoyi-aihr/src/main/java/org/dromara/aihr/service/AihrModelSeedService.java
@@ -474,8 +474,8 @@ public class AihrModelSeedService {
private static String category(String value) {
String category = required(value, "模型类型不能为空");
- if (!List.of("chat", "vector", "rerank", "asr", "tts").contains(category)) {
- throw new IllegalArgumentException("模型类型只支持 chat/vector/rerank/asr/tts");
+ if (!List.of("chat", "vector", "rerank", "asr", "tts", "vision").contains(category)) {
+ throw new IllegalArgumentException("模型类型只支持 chat/vector/rerank/asr/tts/vision");
}
return category;
}
diff --git a/frontend/src/views/system/model/index.vue b/frontend/src/views/system/model/index.vue
index 9d760afc..c4debd32 100644
--- a/frontend/src/views/system/model/index.vue
+++ b/frontend/src/views/system/model/index.vue
@@ -209,6 +209,7 @@
+