优化 API 配置:一个弹窗完成接口、Key 与模型选择
- 去掉预设供应商,直接选择接口类型并填写 Base URL、API Key - 可在保存前从接口拉取模型,搜索并勾选多个;也可手动输入模型名 - 适配中转站:漏写 /v1(Gemini 为 /v1beta)时自动补全,同时发送 Bearer 与官方鉴权头 - 保存时服务端同步连接下的模型列表;编辑时不填 Key 也可拉取 - 单模型测试与参数设置;手机端布局优化 Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
This commit is contained in:
+50
-22
@@ -59,29 +59,57 @@ export function modelSettings(row: ModelRow) {
|
||||
return s
|
||||
}
|
||||
|
||||
/** 从提供方拉取可用模型列表 */
|
||||
export async function listRemoteModels(type: ProviderType, baseUrl: string, apiKey: string): Promise<string[]> {
|
||||
const base = trimUrl(baseUrl) || DEFAULT_BASE_URL[type]
|
||||
if (type === 'anthropic') {
|
||||
const res = await $fetch<{ data: { id: string }[] }>(`${base}/models`, {
|
||||
headers: { 'x-api-key': apiKey, 'anthropic-version': '2023-06-01' },
|
||||
query: { limit: 1000 }
|
||||
})
|
||||
return res.data.map(m => m.id)
|
||||
}
|
||||
if (type === 'google') {
|
||||
const res = await $fetch<{ models: { name: string, supportedGenerationMethods?: string[] }[] }>(`${base}/models`, {
|
||||
headers: { 'x-goog-api-key': apiKey },
|
||||
query: { pageSize: 1000 }
|
||||
})
|
||||
return res.models
|
||||
.filter(m => !m.supportedGenerationMethods || m.supportedGenerationMethods.includes('generateContent'))
|
||||
.map(m => m.name.replace(/^models\//, ''))
|
||||
}
|
||||
const res = await $fetch<{ data: { id: string }[] }>(`${base}/models`, {
|
||||
headers: { Authorization: `Bearer ${apiKey}` }
|
||||
/** 各接口类型的版本路径:中转站地址经常漏填,拉取失败时自动补上再试 */
|
||||
const VERSION_SUFFIX: Record<ProviderType, string> = {
|
||||
'openai-compatible': '/v1',
|
||||
'anthropic': '/v1',
|
||||
'google': '/v1beta'
|
||||
}
|
||||
|
||||
type ModelListResponse = {
|
||||
data?: { id?: string, name?: string }[]
|
||||
models?: { name?: string, id?: string, supportedGenerationMethods?: string[] }[]
|
||||
}
|
||||
|
||||
async function fetchModelIds(type: ProviderType, base: string, apiKey: string): Promise<string[]> {
|
||||
// 同时带上官方鉴权头与 Bearer,兼容各种中转站
|
||||
const headers: Record<string, string> = { Authorization: `Bearer ${apiKey}` }
|
||||
if (type === 'anthropic') Object.assign(headers, { 'x-api-key': apiKey, 'anthropic-version': '2023-06-01' })
|
||||
if (type === 'google') headers['x-goog-api-key'] = apiKey
|
||||
const res = await $fetch<ModelListResponse>(`${base}/models`, {
|
||||
headers,
|
||||
query: type === 'google' ? { pageSize: 1000 } : type === 'anthropic' ? { limit: 1000 } : undefined,
|
||||
timeout: 20_000
|
||||
})
|
||||
return res.data.map(m => m.id).sort()
|
||||
const ids = [
|
||||
...(res.data ?? []).map(m => m.id || m.name || ''),
|
||||
...(res.models ?? [])
|
||||
.filter(m => !m.supportedGenerationMethods || m.supportedGenerationMethods.includes('generateContent'))
|
||||
.map(m => (m.name || m.id || '').replace(/^models\//, ''))
|
||||
].filter(Boolean)
|
||||
if (!Array.isArray(res.data) && !Array.isArray(res.models)) throw new Error('接口返回的不是模型列表,请检查 Base URL')
|
||||
return [...new Set(ids)].sort((a, b) => a.localeCompare(b))
|
||||
}
|
||||
|
||||
/**
|
||||
* 从接口拉取可用模型列表。
|
||||
* 返回实际可用的 baseUrl:用户填的地址缺少 /v1 等版本路径时,会自动补全并返回补全后的地址。
|
||||
*/
|
||||
export async function listRemoteModels(type: ProviderType, baseUrl: string, apiKey: string): Promise<{ models: string[], baseUrl: string }> {
|
||||
const base = trimUrl(baseUrl) || DEFAULT_BASE_URL[type]
|
||||
try {
|
||||
return { models: await fetchModelIds(type, base, apiKey), baseUrl: trimUrl(baseUrl) }
|
||||
}
|
||||
catch (err) {
|
||||
const suffix = VERSION_SUFFIX[type]
|
||||
if (base.endsWith(suffix) || /\/v\d+(beta)?$/.test(base)) throw err
|
||||
try {
|
||||
return { models: await fetchModelIds(type, base + suffix, apiKey), baseUrl: base + suffix }
|
||||
}
|
||||
catch {
|
||||
throw err
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/** 把 AI SDK / HTTP 错误整理成用户能看懂的信息 */
|
||||
|
||||
Reference in New Issue
Block a user