feat(frontend): 支持直接配置模型API密钥

This commit is contained in:
2026-08-30 10:57:05 +08:00
parent 0836807aa2
commit 352975d753
4 changed files with 73 additions and 11 deletions
+16 -2
View File
@@ -8,6 +8,8 @@ vi.mock('@/services/providerService', () => ({
mockModels: {},
listProviders: vi.fn(),
listProviderPresets: vi.fn(),
getCredentialStatus: vi.fn(),
putCredential: vi.fn(),
listModels: vi.fn(),
createProvider: vi.fn(),
updateProvider: vi.fn(),
@@ -16,7 +18,7 @@ vi.mock('@/services/providerService', () => ({
}))
import { useProviderStore } from './provider'
import { listModels, listProviderPresets, listProviders } from '@/services/providerService'
import { getCredentialStatus, listModels, listProviderPresets, listProviders, putCredential } from '@/services/providerService'
import { ApiErrorClass } from '@/services/apiClient'
const providers: ProviderConfig[] = [
@@ -49,6 +51,8 @@ beforeEach(() => {
vi.clearAllMocks()
vi.mocked(listProviders).mockResolvedValue(providers)
vi.mocked(listProviderPresets).mockResolvedValue(presets)
vi.mocked(getCredentialStatus).mockResolvedValue(false)
vi.mocked(putCredential).mockResolvedValue(undefined)
})
describe('provider store model discovery', () => {
@@ -89,6 +93,16 @@ describe('provider store model discovery', () => {
await expect(store.loadModels('openai')).rejects.toThrow('Credential is unavailable')
expect(store.modelErrorsByProvider.openai).toContain('DEEPSEEK_API_KEY')
expect(store.modelErrorsByProvider.openai).toContain('填写并保存')
})
it('sends an API key to the credential endpoint without storing it in Pinia', async () => {
const store = useProviderStore()
await store.saveCredential('deepseek', 'sk-test-sensitive-value')
expect(putCredential).toHaveBeenCalledWith('deepseek', 'sk-test-sensitive-value')
expect(store.credentialConfiguredById.deepseek).toBe(true)
expect(JSON.stringify(store.$state)).not.toContain('sk-test-sensitive-value')
})
})
+17 -5
View File
@@ -1,7 +1,7 @@
import { defineStore } from 'pinia'
import { ref, computed } from 'vue'
import type { ProviderConfig, ModelInfo, ProviderPreset } from '@/contracts'
import { createProvider, deleteProvider as deleteProviderRequest, listModels, listProviderPresets, listProviders, mockProviders, mockModels, testProvider as testProviderRequest, updateProvider as updateProviderRequest } from '@/services/providerService'
import { createProvider, deleteProvider as deleteProviderRequest, getCredentialStatus, listModels, listProviderPresets, listProviders, mockProviders, mockModels, putCredential, testProvider as testProviderRequest, updateProvider as updateProviderRequest } from '@/services/providerService'
import { ApiErrorClass } from '@/services/apiClient'
export const useProviderStore = defineStore('provider', () => {
@@ -10,6 +10,7 @@ export const useProviderStore = defineStore('provider', () => {
const modelsByProvider = ref<Record<string, ModelInfo[]>>(mockModels)
const modelLoadingByProvider = ref<Record<string, boolean>>({})
const modelErrorsByProvider = ref<Record<string, string>>({})
const credentialConfiguredById = ref<Record<string, boolean>>({})
const defaultProviderId = ref('mock')
const isLoading = ref(false)
const error = ref<string | null>(null)
@@ -53,10 +54,7 @@ export const useProviderStore = defineStore('provider', () => {
const credentialId = provider?.credential_id
let message = reason instanceof Error ? reason.message : '模型列表获取失败'
if (reason instanceof ApiErrorClass && reason.code === 'PROVIDER_CREDENTIAL_MISSING') {
const environmentName = credentialId === 'deepseek'
? 'DEEPSEEK_API_KEY'
: `AINOTE_CREDENTIAL_${credentialId?.replace(/[^A-Za-z0-9]/g, '_').toUpperCase() || '<ID>'}`
message = `未找到凭据“${credentialId || '未设置'}”,请在启动 AI Core 前设置 ${environmentName}`
message = '尚未配置 API Key,请编辑该 Provider 后填写并保存。'
} else if (reason instanceof ApiErrorClass && reason.code === 'PROVIDER_AUTH_FAILED') {
message = `鉴权失败,请检查凭据“${credentialId || '未设置'}”对应的 API Key 是否有效。`
}
@@ -73,6 +71,17 @@ export const useProviderStore = defineStore('provider', () => {
)
}
async function loadCredentialStatus(credentialId: string): Promise<boolean> {
const configured = await getCredentialStatus(credentialId)
credentialConfiguredById.value[credentialId] = configured
return configured
}
async function saveCredential(credentialId: string, apiKey: string) {
await putCredential(credentialId, apiKey)
credentialConfiguredById.value[credentialId] = true
}
async function addProvider(data: Omit<ProviderConfig, 'provider_id'>) {
const newProvider = await createProvider(data)
providers.value.push(newProvider)
@@ -108,6 +117,7 @@ export const useProviderStore = defineStore('provider', () => {
modelsByProvider,
modelLoadingByProvider,
modelErrorsByProvider,
credentialConfiguredById,
defaultProviderId,
enabledProviders,
defaultProvider,
@@ -117,6 +127,8 @@ export const useProviderStore = defineStore('provider', () => {
loadPresets,
loadModels,
refreshEnabledModels,
loadCredentialStatus,
saveCredential,
addProvider,
updateProvider,
deleteProvider,