initialize Electron desktop app and remove unused files

2026-10-25 01:09:01 +08:00
commit 01d9e301fc
877 changed files with 262513 additions and 0 deletions
@@ -0,0 +1,175 @@
+// @vitest-environment jsdom
+import { beforeEach, describe, expect, it, vi } from 'vitest'
+import { createPinia, setActivePinia } from 'pinia'
+
+const mockSystemApi = vi.hoisted(() => ({
+  fetchAvailableModels: vi.fn(),
+  fetchAvailableModelsForProfile: vi.fn(),
+  updateDefaultModel: vi.fn(),
+  addCustomProvider: vi.fn(),
+  removeCustomProvider: vi.fn(),
+}))
+
+vi.mock('@/api/hermes/system', () => mockSystemApi)
+vi.mock('@/api/client', () => ({ hasApiKey: () => true }))
+
+import { useAppStore } from '@/stores/hermes/app'
+import { useModelsStore } from '@/stores/hermes/models'
+
+describe('Models Store', () => {
+  beforeEach(() => {
+    setActivePinia(createPinia())
+    vi.clearAllMocks()
+    window.localStorage.clear()
+  })
+
+  it('keeps the sidebar model picker in sync after provider model visibility changes', async () => {
+    const visibleGroups = [
+      {
+        provider: 'deepseek',
+        label: 'DeepSeek',
+        base_url: 'https://api.deepseek.com/v1',
+        api_key: 'sk-test',
+        models: ['deepseek-v4-flash', 'deepseek-v4-pro'],
+        available_models: ['deepseek-v4-flash', 'deepseek-v4-pro'],
+        model_meta: {
+          'deepseek-v4-pro': { preview: true },
+        },
+      },
+    ]
+    const availableModelsResponse = {
+      default: 'deepseek-v4-flash',
+      default_provider: 'deepseek',
+      groups: visibleGroups,
+      allProviders: visibleGroups,
+      model_visibility: {
+        deepseek: { mode: 'include', models: ['deepseek-v4-flash', 'deepseek-v4-pro'] },
+      },
+      profiles: [
+        {
+          profile: 'default',
+          default: 'deepseek-v4-flash',
+          default_provider: 'deepseek',
+          groups: visibleGroups,
+        },
+      ],
+    }
+    mockSystemApi.fetchAvailableModelsForProfile.mockResolvedValue(availableModelsResponse)
+    mockSystemApi.fetchAvailableModels.mockResolvedValue(availableModelsResponse)
+    mockSystemApi.addCustomProvider.mockResolvedValue(undefined)
+
+    const appStore = useAppStore()
+    appStore.modelGroups = [
+      {
+        provider: 'deepseek',
+        label: 'DeepSeek',
+        base_url: 'https://api.deepseek.com/v1',
+        api_key: 'sk-test',
+        models: ['deepseek-v4-flash'],
+        available_models: ['deepseek-v4-flash', 'deepseek-v4-pro'],
+      },
+    ]
+
+    const modelsStore = useModelsStore()
+    await modelsStore.addProvider({
+      name: 'deepseek',
+      base_url: 'https://api.deepseek.com/v1',
+      api_key: 'sk-test',
+      model: 'deepseek-v4-flash',
+    })
+
+    expect(mockSystemApi.fetchAvailableModelsForProfile).toHaveBeenCalledWith('default')
+    expect(mockSystemApi.fetchAvailableModels).toHaveBeenCalled()
+    expect(modelsStore.providers[0].models).toEqual(['deepseek-v4-flash', 'deepseek-v4-pro'])
+    expect(appStore.modelGroups[0].models).toEqual(['deepseek-v4-flash', 'deepseek-v4-pro'])
+    expect(appStore.modelGroups[0].available_models).toEqual(['deepseek-v4-flash', 'deepseek-v4-pro'])
+    expect(appStore.modelGroups[0].model_meta).toEqual({
+      'deepseek-v4-pro': { preview: true },
+    })
+    expect(appStore.modelVisibility).toEqual({
+      deepseek: { mode: 'include', models: ['deepseek-v4-flash', 'deepseek-v4-pro'] },
+    })
+    expect(appStore.selectedModel).toBe('deepseek-v4-flash')
+    expect(appStore.selectedProvider).toBe('deepseek')
+  })
+
+  it('sets the default provider to the first visible model when the current default is not available there', async () => {
+    const deepseekGroup = {
+      provider: 'deepseek',
+      label: 'DeepSeek',
+      base_url: 'https://api.deepseek.com/v1',
+      api_key: 'sk-test',
+      models: ['deepseek-chat'],
+    }
+    const openaiGroup = {
+      provider: 'openai',
+      label: 'OpenAI',
+      base_url: 'https://api.openai.com/v1',
+      api_key: 'sk-openai',
+      models: ['gpt-4.1', 'gpt-4.1-mini'],
+    }
+    const availableModelsResponse = {
+      default: 'gpt-4.1',
+      default_provider: 'openai',
+      groups: [deepseekGroup, openaiGroup],
+      allProviders: [deepseekGroup, openaiGroup],
+    }
+
+    mockSystemApi.fetchAvailableModels.mockResolvedValue(availableModelsResponse)
+    mockSystemApi.updateDefaultModel.mockResolvedValue(undefined)
+
+    const modelsStore = useModelsStore()
+    modelsStore.providers = [deepseekGroup, openaiGroup]
+    modelsStore.defaultModel = 'deepseek-chat'
+    modelsStore.defaultProvider = 'deepseek'
+
+    await modelsStore.setDefaultProvider('openai')
+
+    expect(mockSystemApi.updateDefaultModel).toHaveBeenCalledWith({
+      default: 'gpt-4.1',
+      provider: 'openai',
+    })
+    expect(modelsStore.defaultModel).toBe('gpt-4.1')
+    expect(modelsStore.defaultProvider).toBe('openai')
+  })
+
+  it('keeps the current default model when another provider exposes the same model id', async () => {
+    const providerA = {
+      provider: 'provider-a',
+      label: 'Provider A',
+      base_url: 'https://provider-a.example/v1',
+      api_key: 'sk-a',
+      models: ['shared-model'],
+    }
+    const providerB = {
+      provider: 'provider-b',
+      label: 'Provider B',
+      base_url: 'https://provider-b.example/v1',
+      api_key: 'sk-b',
+      models: ['shared-model', 'provider-b-only'],
+    }
+    const availableModelsResponse = {
+      default: 'shared-model',
+      default_provider: 'provider-b',
+      groups: [providerA, providerB],
+      allProviders: [providerA, providerB],
+    }
+
+    mockSystemApi.fetchAvailableModels.mockResolvedValue(availableModelsResponse)
+    mockSystemApi.updateDefaultModel.mockResolvedValue(undefined)
+
+    const modelsStore = useModelsStore()
+    modelsStore.providers = [providerA, providerB]
+    modelsStore.defaultModel = 'shared-model'
+    modelsStore.defaultProvider = 'provider-a'
+
+    await modelsStore.setDefaultProvider('provider-b')
+
+    expect(mockSystemApi.updateDefaultModel).toHaveBeenCalledWith({
+      default: 'shared-model',
+      provider: 'provider-b',
+    })
+    expect(modelsStore.defaultModel).toBe('shared-model')
+    expect(modelsStore.defaultProvider).toBe('provider-b')
+  })
+})