mirror of
https://github.com/QuantumNous/new-api.git
synced 2026-03-30 12:51:58 +00:00
77 lines
2.3 KiB
Go
77 lines
2.3 KiB
Go
package model_setting
|
|
|
|
import (
|
|
"github.com/QuantumNous/new-api/setting/config"
|
|
)
|
|
|
|
// GeminiSettings defines Gemini model configuration. 注意bool要以enabled结尾才可以生效编辑
|
|
type GeminiSettings struct {
|
|
SafetySettings map[string]string `json:"safety_settings"`
|
|
VersionSettings map[string]string `json:"version_settings"`
|
|
SupportedImagineModels []string `json:"supported_imagine_models"`
|
|
ThinkingAdapterEnabled bool `json:"thinking_adapter_enabled"`
|
|
ThinkingAdapterBudgetTokensPercentage float64 `json:"thinking_adapter_budget_tokens_percentage"`
|
|
FunctionCallThoughtSignatureEnabled bool `json:"function_call_thought_signature_enabled"`
|
|
RemoveFunctionResponseIdEnabled bool `json:"remove_function_response_id_enabled"`
|
|
}
|
|
|
|
// 默认配置
|
|
var defaultGeminiSettings = GeminiSettings{
|
|
SafetySettings: map[string]string{
|
|
"default": "OFF",
|
|
},
|
|
VersionSettings: map[string]string{
|
|
"default": "v1beta",
|
|
"gemini-1.0-pro": "v1",
|
|
},
|
|
SupportedImagineModels: []string{
|
|
"gemini-2.0-flash-exp-image-generation",
|
|
"gemini-2.0-flash-exp",
|
|
"gemini-3-pro-image-preview",
|
|
"gemini-2.5-flash-image",
|
|
"gemini-3.1-flash-image-preview",
|
|
},
|
|
ThinkingAdapterEnabled: false,
|
|
ThinkingAdapterBudgetTokensPercentage: 0.6,
|
|
FunctionCallThoughtSignatureEnabled: true,
|
|
RemoveFunctionResponseIdEnabled: true,
|
|
}
|
|
|
|
// 全局实例
|
|
var geminiSettings = defaultGeminiSettings
|
|
|
|
func init() {
|
|
// 注册到全局配置管理器
|
|
config.GlobalConfig.Register("gemini", &geminiSettings)
|
|
}
|
|
|
|
// GetGeminiSettings 获取Gemini配置
|
|
func GetGeminiSettings() *GeminiSettings {
|
|
return &geminiSettings
|
|
}
|
|
|
|
// GetGeminiSafetySetting 获取安全设置
|
|
func GetGeminiSafetySetting(key string) string {
|
|
if value, ok := geminiSettings.SafetySettings[key]; ok {
|
|
return value
|
|
}
|
|
return geminiSettings.SafetySettings["default"]
|
|
}
|
|
|
|
// GetGeminiVersionSetting 获取版本设置
|
|
func GetGeminiVersionSetting(key string) string {
|
|
if value, ok := geminiSettings.VersionSettings[key]; ok {
|
|
return value
|
|
}
|
|
return geminiSettings.VersionSettings["default"]
|
|
}
|
|
|
|
func IsGeminiModelSupportImagine(model string) bool {
|
|
for _, v := range geminiSettings.SupportedImagineModels {
|
|
if v == model {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|