113 lines
3.5 KiB
Go
113 lines
3.5 KiB
Go
package settings
|
||
|
||
import (
|
||
"context"
|
||
"strings"
|
||
|
||
"apps/backend/internal/module/ai"
|
||
"apps/backend/internal/module/member/domain"
|
||
"apps/backend/internal/svc"
|
||
"apps/backend/internal/types"
|
||
)
|
||
|
||
func resolveAPIKey(svcCtx *svc.ServiceContext, st *domain.UserSettings, provider string) (key string, platformAvailable bool) {
|
||
provider = ai.NormalizeProvider(provider)
|
||
// 1) user BYOK for this provider
|
||
if st != nil {
|
||
if k := st.KeyForProvider(provider); k != "" {
|
||
return k, platformKeyAvailable(svcCtx, provider)
|
||
}
|
||
}
|
||
// 2) platform key
|
||
return platformKey(svcCtx, provider), platformKeyAvailable(svcCtx, provider)
|
||
}
|
||
|
||
func platformKey(svcCtx *svc.ServiceContext, provider string) string {
|
||
if svcCtx == nil {
|
||
return ""
|
||
}
|
||
switch ai.NormalizeProvider(provider) {
|
||
case ai.ProviderOpenCodeGo:
|
||
return strings.TrimSpace(svcCtx.Config.Platform.OpenCodeKey)
|
||
default:
|
||
k := strings.TrimSpace(svcCtx.Config.Platform.XAIKey)
|
||
if k == "" {
|
||
k = strings.TrimSpace(svcCtx.Config.Platform.AIKey)
|
||
}
|
||
return k
|
||
}
|
||
}
|
||
|
||
func platformKeyAvailable(svcCtx *svc.ServiceContext, provider string) bool {
|
||
return platformKey(svcCtx, provider) != ""
|
||
}
|
||
|
||
// loadModelsForProvider — 真拉 + 5 分鐘快取；失敗用 fallback（不回空 list）
|
||
func loadModelsForProvider(ctx context.Context, svcCtx *svc.ServiceContext, st *domain.UserSettings, provider string) (list []string, fromCache bool, modelsErr string) {
|
||
provider = ai.NormalizeProvider(provider)
|
||
key, _ := resolveAPIKey(svcCtx, st, provider)
|
||
if key == "" {
|
||
return ai.FallbackModels(provider), false, "尚未設定 API key（會員 BYOK 或平台 key）"
|
||
}
|
||
if svcCtx.AIRegistry == nil {
|
||
return ai.FallbackModels(provider), false, "AI registry not configured"
|
||
}
|
||
list, fromCache, err := svcCtx.AIRegistry.ListModelsCached(ctx, provider, key)
|
||
if err != nil {
|
||
return ai.FallbackModels(provider), false, err.Error()
|
||
}
|
||
return list, fromCache, ""
|
||
}
|
||
|
||
// pickModel：優先用會員已存模型；即使暫時不在列表也保留（不寫死改成別台）
|
||
func pickModel(preferred string, list []string) string {
|
||
preferred = strings.TrimSpace(preferred)
|
||
if preferred != "" {
|
||
return preferred
|
||
}
|
||
if len(list) > 0 {
|
||
return list[0]
|
||
}
|
||
return ""
|
||
}
|
||
|
||
func buildAiSettingsData(ctx context.Context, svcCtx *svc.ServiceContext, st *domain.UserSettings, modelsProvider string) *types.AiSettingsData {
|
||
if st == nil {
|
||
st = domain.DefaultSettings(0)
|
||
}
|
||
savedProvider := ai.NormalizeProvider(st.Provider)
|
||
if savedProvider == "" {
|
||
savedProvider = ai.ProviderXAI
|
||
}
|
||
modelsProvider = ai.NormalizeProvider(modelsProvider)
|
||
if modelsProvider == "" {
|
||
modelsProvider = savedProvider
|
||
}
|
||
|
||
list, fromCache, modelsErr := loadModelsForProvider(ctx, svcCtx, st, modelsProvider)
|
||
// 顯示／回傳都以會員存檔為準
|
||
selected := pickModel(st.Model, list)
|
||
savedModel := strings.TrimSpace(st.Model)
|
||
if modelsProvider == savedProvider && selected != "" {
|
||
savedModel = selected
|
||
}
|
||
|
||
keyConfigured := st.HasKeyForProvider(savedProvider)
|
||
platformAvail := platformKeyAvailable(svcCtx, savedProvider)
|
||
|
||
return &types.AiSettingsData{
|
||
Provider: savedProvider,
|
||
Model: savedModel,
|
||
ResearchProvider: savedProvider,
|
||
ResearchModel: savedModel,
|
||
ApiKeyConfigured: keyConfigured,
|
||
ResearchApiKeyConfigured: keyConfigured || st.ResearchAPIKeyConfigured,
|
||
PlatformKeyAvailable: platformAvail,
|
||
Models: list,
|
||
ModelsProvider: modelsProvider,
|
||
SelectedModel: selected,
|
||
ModelsFromCache: fromCache,
|
||
ModelsError: modelsErr,
|
||
}
|
||
}
|