refactor: centralize AI model defaults via YAML catalog

Author: KariHall619

.gitignore

@@ -77,6 +77,9 @@ temp/
 
 # AI model files and caches
 models/
+!pkg/ai/models/
+!pkg/ai/models/*.yaml
+!pkg/ai/models/*.go
 *.model
 *.cache
 *.pkl

frontend/src/components/AISettingsPanel.vue

@@ -144,11 +144,7 @@
                           </el-option>
                         </template>
                         <template v-else>
-                          <el-option value="gpt-5" label="GPT-5 ⭐ Recommended" />
-                          <el-option value="gpt-5-mini" label="GPT-5 Mini" />
-                          <el-option value="gpt-5-nano" label="GPT-5 Nano" />
-                          <el-option value="gpt-5-pro" label="GPT-5 Pro" />
-                          <el-option value="gpt-4.1" label="GPT-4.1" />
+                          <el-option disabled value="no-models" :label="t('ai.welcome.noModels')" />
                         </template>
                       </el-select>
                       <el-button
@@ -218,8 +214,7 @@
                           </el-option>
                         </template>
                         <template v-else>
-                          <el-option value="deepseek-reasoner" label="DeepSeek Reasoner ⭐" />
-                          <el-option value="deepseek-chat" label="DeepSeek Chat" />
+                          <el-option disabled value="no-models" :label="t('ai.welcome.noModels')" />
                         </template>
                       </el-select>
                       <el-button

frontend/src/composables/useAIChat.ts

@@ -125,8 +125,24 @@ export function useAIChat(_context: AppContext) {
       }
     } catch (error) {
       console.error('Failed to fetch models:', error)
-      // Use mock models as fallback for this provider
-      modelsByProvider.value[storeKey] = getMockModels(storeKey)
+      const cachedFallback = catalogCache.value[storeKey]
+      if (cachedFallback && cachedFallback.length) {
+        modelsByProvider.value[storeKey] = cachedFallback
+        return
+      }
+
+      try {
+        const catalog = await aiService.fetchModelCatalog(storeKey)
+        const entry = catalog[storeKey]
+        const fallbackModels = entry?.models ?? []
+        modelsByProvider.value[storeKey] = fallbackModels
+        if (fallbackModels.length) {
+          catalogCache.value[storeKey] = fallbackModels
+        }
+      } catch (catalogError) {
+        console.error('Failed to fetch catalog fallback:', catalogError)
+        modelsByProvider.value[storeKey] = []
+      }
     }
   }
 

frontend/src/types/index.ts

@@ -42,7 +42,9 @@ export interface AIConfig {
 export interface Model {
   id: string
   name: string
-  size: string
+  size?: string
+  description?: string
+  maxTokens?: number
 }
 
 /**

frontend/src/utils/config.ts

@@ -1,4 +1,4 @@
-import type { AIConfig, Model, DatabaseDialect } from '@/types'
+import type { AIConfig, DatabaseDialect } from '@/types'
 
 export type Provider = 'ollama' | 'openai' | 'deepseek' | 'local'
 
@@ -109,30 +109,6 @@ export function getDefaultConfig(provider: string): Partial<AIConfig> {
   }
 }
 
-/**
- * Get mock models when API fails
- */
-export function getMockModels(provider: string): Model[] {
-  const mocks: Record<string, Model[]> = {
-    ollama: [
-      { id: 'llama3.2:3b', name: 'Llama 3.2 3B', size: '2GB' },
-      { id: 'gemma2:9b', name: 'Gemma 2 9B', size: '5GB' }
-    ],
-    openai: [
-      { id: 'gpt-5', name: 'GPT-5 ⭐', size: 'Cloud' },
-      { id: 'gpt-5-mini', name: 'GPT-5 Mini', size: 'Cloud' },
-      { id: 'gpt-5-nano', name: 'GPT-5 Nano', size: 'Cloud' },
-      { id: 'gpt-5-pro', name: 'GPT-5 Pro', size: 'Cloud' },
-      { id: 'gpt-4.1', name: 'GPT-4.1', size: 'Cloud' }
-    ],
-    deepseek: [
-      { id: 'deepseek-chat', name: 'DeepSeek Chat', size: 'Cloud' },
-      { id: 'deepseek-reasoner', name: 'DeepSeek Reasoner', size: 'Cloud' }
-    ]
-  }
-  return mocks[provider] || []
-}
-
 /**
  * Generate unique ID
  */

frontend/tests/composables/useAIChat.test.ts

@@ -7,6 +7,7 @@ import type { AppContext } from '@/types'
 vi.mock('@/services/aiService', () => ({
   aiService: {
     fetchModels: vi.fn(),
+    fetchModelCatalog: vi.fn(),
     testConnection: vi.fn(),
     generateSQL: vi.fn(),
     saveConfig: vi.fn()
@@ -22,6 +23,7 @@ describe('useAIChat', () => {
 
     // Set default mock behavior for fetchModels (called during initialization)
     vi.mocked(aiService.fetchModels).mockResolvedValue([])
+    vi.mocked(aiService.fetchModelCatalog).mockResolvedValue({})
 
     mockContext = {
       i18n: {
@@ -185,8 +187,19 @@ describe('useAIChat', () => {
       expect(availableModels.value[0].id).toBe('model1')
     })
 
-    it('should use mock models when API fails', async () => {
+    it('should use catalog models when API fails', async () => {
       vi.mocked(aiService.fetchModels).mockRejectedValue(new Error('Network error'))
+      vi.mocked(aiService.fetchModelCatalog).mockResolvedValueOnce({
+        ollama: {
+          display_name: 'Ollama',
+          category: 'local',
+          endpoint: 'http://localhost:11434',
+          requires_api_key: false,
+          models: [
+            { id: 'llama3.2:3b', name: 'Llama 3.2 3B', size: '2GB' }
+          ]
+        }
+      })
 
       const { availableModels, refreshModels } = useAIChat(mockContext)
 

frontend/tests/services/aiService.spec.ts

@@ -84,4 +84,32 @@ describe('aiService', () => {
     expect(health.healthy).toBe(true)
     expect(health.provider).toBe('ollama')
   })
+
+  it('fetchModelCatalog should normalize provider keys', async () => {
+    const catalogPayload = {
+      OpenAI: {
+        display_name: 'OpenAI',
+        category: 'cloud',
+        endpoint: 'https://api.openai.com',
+        requires_api_key: true,
+        models: [
+          { id: 'gpt-5', name: 'GPT-5', description: 'Flagship', max_tokens: 200000 }
+        ]
+      }
+    }
+
+    fetchMock.mockResolvedValueOnce(
+      createFetchResponse({
+        data: [
+          { key: 'catalog', value: JSON.stringify(catalogPayload) },
+          { key: 'success', value: true }
+        ]
+      })
+    )
+
+    const catalog = await aiService.fetchModelCatalog()
+    expect(catalog.openai).toBeDefined()
+    expect(catalog.openai.models).toHaveLength(1)
+    expect(catalog.openai.models[0].id).toBe('gpt-5')
+  })
 })

frontend/tests/utils/config.test.ts

@@ -3,7 +3,6 @@ import {
   loadConfig,
   saveConfig,
   getDefaultConfig,
-  getMockModels,
   generateId
 } from '@/utils/config'
 import type { AIConfig } from '@/types'
@@ -180,34 +179,6 @@ describe('config utils', () => {
     })
   })
 
-  describe('getMockModels', () => {
-    it('should return ollama mock models', () => {
-      const models = getMockModels('ollama')
-
-      expect(models).toHaveLength(2)
-      expect(models[0].id).toBe('llama3.2:3b')
-      expect(models[0].name).toBe('Llama 3.2 3B')
-    })
-
-    it('should return openai mock models', () => {
-      const models = getMockModels('openai')
-
-      expect(models).toHaveLength(5)
-      expect(models.map(model => model.id)).toEqual([
-        'gpt-5',
-        'gpt-5-mini',
-        'gpt-5-nano',
-        'gpt-5-pro',
-        'gpt-4.1'
-      ])
-    })
-
-    it('should return empty array for unknown provider', () => {
-      const models = getMockModels('unknown')
-      expect(models).toEqual([])
-    })
-  })
-
   describe('generateId', () => {
     it('should generate unique IDs', () => {
       const id1 = generateId()

pkg/ai/generator.go

@@ -26,6 +26,7 @@ import (
 	"sync"
 	"time"
 
+	"github.com/linuxsuren/atest-ext-ai/pkg/ai/models"
 	"github.com/linuxsuren/atest-ext-ai/pkg/ai/providers/universal"
 	"github.com/linuxsuren/atest-ext-ai/pkg/config"
 	"github.com/linuxsuren/atest-ext-ai/pkg/interfaces"
@@ -735,12 +736,9 @@ func createRuntimeClient(provider string, runtimeConfig map[string]any) (interfa
 		}
 
 		if config.Endpoint == "" {
-			switch normalizedProvider {
-			case "openai":
-				config.Endpoint = "https://api.openai.com"
-			case "deepseek":
-				config.Endpoint = "https://api.deepseek.com"
-			case "custom":
+			if endpoint := models.EndpointForProvider(normalizedProvider); endpoint != "" {
+				config.Endpoint = endpoint
+			} else if normalizedProvider == "custom" {
 				return nil, fmt.Errorf("endpoint is required for custom provider")
 			}
 		}

pkg/ai/manager.go

@@ -29,6 +29,7 @@ import (
 	"time"
 
 	"github.com/linuxsuren/atest-ext-ai/pkg/ai/discovery"
+	"github.com/linuxsuren/atest-ext-ai/pkg/ai/models"
 	"github.com/linuxsuren/atest-ext-ai/pkg/ai/providers/universal"
 	"github.com/linuxsuren/atest-ext-ai/pkg/config"
 	"github.com/linuxsuren/atest-ext-ai/pkg/interfaces"
@@ -543,12 +544,9 @@ func createOpenAICompatibleClient(provider string, cfg config.AIService) (interf
 	}
 
 	if uniCfg.Endpoint == "" {
-		switch normalized {
-		case "openai":
-			uniCfg.Endpoint = "https://api.openai.com"
-		case "deepseek":
-			uniCfg.Endpoint = "https://api.deepseek.com"
-		case "custom":
+		if endpoint := models.EndpointForProvider(normalized); endpoint != "" {
+			uniCfg.Endpoint = endpoint
+		} else if normalized == "custom" {
 			return nil, fmt.Errorf("endpoint is required for custom provider")
 		}
 	}
@@ -585,41 +583,42 @@ func normalizeProviderName(provider string) string {
 
 // getOnlineProviders returns predefined online providers
 func (m *Manager) getOnlineProviders() []*ProviderInfo {
-	return []*ProviderInfo{
-		{
-			Name:      "deepseek",
-			Type:      "online",
-			Available: true,
-			Endpoint:  "https://api.deepseek.com",
-			Models: []interfaces.ModelInfo{
-				{ID: "deepseek-chat", Name: "DeepSeek Chat", Description: "DeepSeek's flagship conversational AI model", MaxTokens: 32768},
-				{ID: "deepseek-reasoner", Name: "DeepSeek Reasoner", Description: "DeepSeek's reasoning model with thinking capabilities", MaxTokens: 32768},
-			},
-			LastChecked: time.Now(),
-			Config: map[string]interface{}{
-				"requires_api_key": true,
-				"provider_type":    "online",
-			},
-		},
-		{
-			Name:      "openai",
-			Type:      "online",
-			Available: true,
-			Endpoint:  "https://api.openai.com",
-			Models: []interfaces.ModelInfo{
-				{ID: "gpt-5", Name: "GPT-5", Description: "OpenAI's flagship GPT-5 model", MaxTokens: 200000},
-				{ID: "gpt-5-mini", Name: "GPT-5 Mini", Description: "Optimized GPT-5 model for lower latency workloads", MaxTokens: 80000},
-				{ID: "gpt-5-nano", Name: "GPT-5 Nano", Description: "Cost-efficient GPT-5 variant for lightweight tasks", MaxTokens: 40000},
-				{ID: "gpt-5-pro", Name: "GPT-5 Pro", Description: "High performance GPT-5 model with extended reasoning", MaxTokens: 240000},
-				{ID: "gpt-4.1", Name: "GPT-4.1", Description: "Balanced GPT-4 series model with strong multimodal support", MaxTokens: 128000},
-			},
+	catalog, err := models.GetCatalog()
+	if err != nil {
+		logging.Logger.Warn("Failed to load model catalog", "error", err)
+		return nil
+	}
+
+	var providers []*ProviderInfo
+	for _, name := range catalog.ProviderNames() {
+		entry, ok := catalog.Provider(name)
+		if !ok {
+			continue
+		}
+
+		providerType := entry.Category
+		if providerType == "" {
+			providerType = "cloud"
+		}
+		if providerType != "cloud" && providerType != "online" {
+			continue
+		}
+
+		providers = append(providers, &ProviderInfo{
+			Name:        entry.Name,
+			Type:        providerType,
+			Available:   true,
+			Endpoint:    entry.Endpoint,
+			Models:      entry.Models,
 			LastChecked: time.Now(),
 			Config: map[string]interface{}{
-				"requires_api_key": true,
-				"provider_type":    "online",
+				"requires_api_key": entry.RequiresAPIKey,
+				"provider_type":    providerType,
 			},
-		},
+		})
 	}
+
+	return providers
 }
 
 // ===== Retry Logic =====