feat: change ai menu

Author: zhengkunwang223

core/constant/common.go

@@ -57,17 +57,14 @@ var WebUrlMap = map[string]struct{}{
 	"/apps/upgrade":   {},
 	"/apps/setting":   {},
 
-	"/ai":                {},
-	"/ai/model":          {},
-	"/ai/gpu":            {},
-	"/ai/gpu/current":    {},
-	"/ai/gpu/history":    {},
-	"/ai/mcp":            {},
-	"/ai/model/tensorrt": {},
-	"/ai/model/vllm":     {},
-	"/ai/model/ollama":   {},
-	"/ai/agents/agent":   {},
-	"/ai/agents/model":   {},
+	"/ai":               {},
+	"/ai/model/account": {},
+	"/ai/model/local":   {},
+	"/ai/gpu":           {},
+	"/ai/gpu/current":   {},
+	"/ai/gpu/history":   {},
+	"/ai/mcp":           {},
+	"/ai/agents/agent":  {},
 
 	"/containers":                   {},
 	"/containers/container/operate": {},

core/init/migration/helper/menu.go

@@ -14,10 +14,10 @@ func LoadMenus() string {
 	item := []dto.ShowMenu{
 		{ID: "1", Disabled: true, Title: "menu.home", IsShow: true, Label: "Home-Menu", Path: "/", Sort: 100},
 		{ID: "2", Disabled: true, Title: "menu.apps", IsShow: true, Label: "App-Menu", Path: "/apps/all", Sort: 200},
-		{ID: "3", Disabled: false, Title: "menu.aiTools", IsShow: true, Label: "AI-Menu", Path: "/ai/model", Sort: 300,
+		{ID: "3", Disabled: false, Title: "menu.aiTools", IsShow: true, Label: "AI-Menu", Path: "/ai/model/account", Sort: 300,
 			Children: []dto.ShowMenu{
 				{ID: "44", Disabled: false, Title: "aiTools.agents.agent", IsShow: true, Label: "Agents", Path: "/ai/agents/agent", Sort: 50},
-				{ID: "41", Disabled: false, Title: "aiTools.model.localModel", IsShow: true, Label: "OllamaModel", Path: "/ai/model", Sort: 100},
+				{ID: "41", Disabled: false, Title: "aiTools.model.model", IsShow: true, Label: "AIModel", Path: "/ai/model/account", Sort: 100},
 				{ID: "42", Disabled: false, Title: "menu.mcp", IsShow: true, Label: "MCPServer", Path: "/ai/mcp", Sort: 200},
 				{ID: "43", Disabled: false, Title: "aiTools.gpu.gpu", IsShow: true, Label: "GPU", Path: "/ai/gpu", Sort: 300},
 			}},
@@ -66,7 +66,7 @@ func MenuSort() []dto.MenuLabelSort {
 		{Label: "App-Menu", Sort: 200},
 		{Label: "AI-Menu", Sort: 300},
 		{Label: "Agents", Sort: 50},
-		{Label: "OllamaModel", Sort: 100},
+		{Label: "AIModel", Sort: 100},
 		{Label: "MCPServer", Sort: 200},
 		{Label: "GPU", Sort: 300},
 		{Label: "Website-Menu", Sort: 400},

core/init/migration/migrate.go

@@ -35,6 +35,7 @@ func Init() {
 		migrations.AddEditionSetting,
 		migrations.UpdateAiLocalModelMenuTitle,
 		migrations.UpdateAiAgentsHideMenuTitle,
+		migrations.UpdateAiModelMenuStructure,
 		migrations.AddDocSourceSetting,
 	})
 	if err := m.Migrate(); err != nil {

core/init/migration/migrations/init.go

@@ -877,3 +877,96 @@ var UpdateAiAgentsHideMenuTitle = &gormigrate.Migration{
 		).Error
 	},
 }
+
+var UpdateAiModelMenuStructure = &gormigrate.Migration{
+	ID: "20260403-update-ai-model-menu-structure",
+	Migrate: func(tx *gorm.DB) error {
+		var menuJSON string
+		if err := tx.Model(&model.Setting{}).Where("key = ?", "HideMenu").Pluck("value", &menuJSON).Error; err != nil {
+			return err
+		}
+		if menuJSON == "" {
+			return tx.Model(&model.Setting{}).Where("key = ?", "HideMenu").Update("value", helper.LoadMenus()).Error
+		}
+
+		var menus []dto.ShowMenu
+		if err := json.Unmarshal([]byte(menuJSON), &menus); err != nil {
+			return tx.Model(&model.Setting{}).Where("key = ?", "HideMenu").Update("value", helper.LoadMenus()).Error
+		}
+
+		for i := range menus {
+			if menus[i].Label != "AI-Menu" {
+				continue
+			}
+			menus[i].Path = "/ai/model/account"
+			menus[i].Children = buildAiMenuChildren(menus[i].Children)
+			break
+		}
+
+		updatedJSON, err := json.Marshal(menus)
+		if err != nil {
+			return tx.Model(&model.Setting{}).Where("key = ?", "HideMenu").Update("value", helper.LoadMenus()).Error
+		}
+		return tx.Model(&model.Setting{}).Where("key = ?", "HideMenu").Update("value", string(updatedJSON)).Error
+	},
+}
+
+func buildAiMenuChildren(children []dto.ShowMenu) []dto.ShowMenu {
+	return []dto.ShowMenu{
+		normalizeAiMenuChild(children, dto.ShowMenu{
+			ID:       "44",
+			Label:    "Agents",
+			Disabled: false,
+			IsShow:   true,
+			Title:    "aiTools.agents.agent",
+			Path:     "/ai/agents/agent",
+			Sort:     50,
+		}, "Agents"),
+		normalizeAiMenuChild(children, dto.ShowMenu{
+			ID:       "41",
+			Label:    "AIModel",
+			Disabled: false,
+			IsShow:   true,
+			Title:    "aiTools.model.model",
+			Path:     "/ai/model/account",
+			Sort:     100,
+		}, "AIModel", "OllamaModel"),
+		normalizeAiMenuChild(children, dto.ShowMenu{
+			ID:       "42",
+			Label:    "MCPServer",
+			Disabled: false,
+			IsShow:   true,
+			Title:    "menu.mcp",
+			Path:     "/ai/mcp",
+			Sort:     200,
+		}, "MCPServer"),
+		normalizeAiMenuChild(children, dto.ShowMenu{
+			ID:       "43",
+			Label:    "GPU",
+			Disabled: false,
+			IsShow:   true,
+			Title:    "aiTools.gpu.gpu",
+			Path:     "/ai/gpu",
+			Sort:     300,
+		}, "GPU"),
+	}
+}
+
+func normalizeAiMenuChild(children []dto.ShowMenu, fallback dto.ShowMenu, labels ...string) dto.ShowMenu {
+	for _, child := range children {
+		for _, label := range labels {
+			if child.Label != label {
+				continue
+			}
+			child.ID = fallback.ID
+			child.Label = fallback.Label
+			child.Disabled = fallback.Disabled
+			child.Title = fallback.Title
+			child.Path = fallback.Path
+			child.Sort = fallback.Sort
+			child.Children = nil
+			return child
+		}
+	}
+	return fallback
+}

frontend/src/routers/modules/ai.ts

@@ -1,20 +1,11 @@
 import { Layout } from '@/routers/constant';
 
-const vllmModules = import.meta.glob('@/xpack/views/vllm/index.vue');
-const loadVllmView = () => {
-    const loader = vllmModules['/src/xpack/views/vllm/index.vue'];
-    if (loader) {
-        return loader();
-    }
-    return import('@/views/ai/model/vllm/index.vue');
-};
-
 const databaseRouter = {
     sort: 4,
     path: '/ai',
     name: 'AI-Menu',
     component: Layout,
-    redirect: '/ai/model',
+    redirect: '/ai/model/account',
     meta: {
         icon: 'p-jiqiren2',
         title: 'menu.aiTools',
@@ -31,23 +22,23 @@ const databaseRouter = {
             },
         },
         {
-            path: '/ai/agents/model',
-            name: 'AgentsModel',
-            component: () => import('@/views/ai/agents/model/index.vue'),
+            path: '/ai/model/account',
+            name: 'AIModel',
+            component: () => import('@/views/ai/model/index.vue'),
             meta: {
-                title: 'aiTools.agents.account',
-                activeMenu: '/ai/agents/agent',
+                icon: 'p-moxing-menu',
+                title: 'aiTools.model.model',
                 requiresAuth: true,
             },
         },
         {
-            path: '/ai/model/ollama',
-            name: 'OllamaModel',
-            component: () => import('@/views/ai/model/ollama/index.vue'),
+            path: '/ai/model/local',
+            hidden: true,
+            name: 'LocalModel',
+            component: () => import('@/views/ai/model/index.vue'),
             meta: {
-                icon: 'p-moxing-menu',
                 title: 'aiTools.model.localModel',
-                activeMenu: '/ai/model/ollama',
+                activeMenu: '/ai/model/account',
                 requiresAuth: true,
             },
         },
@@ -61,28 +52,6 @@ const databaseRouter = {
                 requiresAuth: true,
             },
         },
-        {
-            path: '/ai/model/tensorrt',
-            hidden: true,
-            name: 'TensorRTLLm',
-            component: () => import('@/views/ai/model/tensorrt/index.vue'),
-            meta: {
-                title: 'aiTools.tensorRT.llm',
-                activeMenu: '/ai/model/ollama',
-                requiresAuth: true,
-            },
-        },
-        {
-            path: '/ai/model/vllm',
-            hidden: true,
-            name: 'Vllm',
-            component: loadVllmView,
-            meta: {
-                title: 'vLLM',
-                activeMenu: '/ai/model/ollama',
-                requiresAuth: true,
-            },
-        },
         {
             path: '/ai/gpu/current',
             name: 'GPU',

frontend/src/utils/app.ts

@@ -32,8 +32,9 @@ export const jumpToInstall = (type: string, key: string) => {
         case 'vllm':
             if (globalStore.isProductPro) {
                 router.push({
-                    path: '/ai/model/vllm',
+                    path: '/ai/model/local',
                     query: {
+                        tab: 'vllm',
                         uncached: 'true',
                         open: 'create',
                     },

frontend/src/views/ai/agents/agent/index.vue

@@ -1,6 +1,6 @@
 <template>
     <div>
-        <RouterMenu />
+        <RouterButton :buttons="headerButtons" />
         <DockerStatus v-model:isActive="isActive" v-model:isExist="isExist" />
         <LayoutContent v-loading="loading" v-if="isExist" :class="{ mask: !isActive }">
             <template #leftToolBar>
@@ -173,7 +173,6 @@ import { SearchWithPage } from '@/api/interface';
 import { dateFormat, newUUID } from '@/utils/util';
 import { MsgSuccess } from '@/utils/message';
 
-import RouterMenu from '@/views/ai/agents/index.vue';
 import AddDialog from '@/views/ai/agents/agent/add/index.vue';
 import DeleteDialog from '@/views/ai/agents/agent/delete/index.vue';
 import ConfigDrawer from '@/views/ai/agents/agent/config/index.vue';
@@ -209,6 +208,13 @@ const isExist = ref(false);
 const noApp = ref(false);
 const searchName = ref('');
 
+const headerButtons = [
+    {
+        label: i18n.global.t('aiTools.agents.agent'),
+        path: '/ai/agents/agent',
+    },
+];
+
 const buttons = [
     {
         label: i18n.global.t('menu.config'),

frontend/src/views/ai/agents/index.vue

@@ -1,6 +1,5 @@
 <template>
     <div>
-        <RouterMenu />
         <LayoutContent>
             <template #leftToolBar>
                 <el-button type="primary" @click="openCreate">{{ $t('commons.button.create') }}</el-button>
@@ -54,7 +53,6 @@
 import { onMounted, reactive, ref } from 'vue';
 import { deleteAgentAccount, pageAgentAccounts } from '@/api/modules/ai';
 import { AI } from '@/api/interface/ai';
-import RouterMenu from '@/views/ai/agents/index.vue';
 import AddDialog from '@/views/ai/agents/model/add/index.vue';
 import ModelPoolDialog from '@/views/ai/agents/model/pool/index.vue';
 import { ElMessageBox } from 'element-plus';

frontend/src/views/ai/agents/model/index.vue

@@ -1,25 +1,31 @@
 <template>
     <div>
         <RouterButton :buttons="buttons" />
-        <LayoutContent>
-            <router-view></router-view>
-        </LayoutContent>
+        <component :is="currentComponent" />
     </div>
 </template>
 
 <script lang="ts" setup>
+import { computed } from 'vue';
+import { useRoute } from 'vue-router';
+import i18n from '@/lang';
+import AccountView from '@/views/ai/agents/model/index.vue';
+import LocalView from '@/views/ai/model/local/index.vue';
+
+const route = useRoute();
+
 const buttons = [
     {
-        label: 'Ollama',
-        path: '/ai/model/ollama',
+        label: i18n.global.t('aiTools.agents.account'),
+        path: '/ai/model/account',
     },
     {
-        label: 'vLLM',
-        path: '/ai/model/vllm',
-    },
-    {
-        label: 'TensorRT LLM',
-        path: '/ai/model/tensorrt',
+        label: i18n.global.t('aiTools.model.localModel'),
+        path: '/ai/model/local',
     },
 ];
+
+const currentComponent = computed(() => {
+    return route.path === '/ai/model/local' ? LocalView : AccountView;
+});
 </script>