diff --git a/model/ability.go b/model/ability.go index 08519de0..2df45917 100644 --- a/model/ability.go +++ b/model/ability.go @@ -301,7 +301,7 @@ func FixAbility() (int, int, error) { } var channels []*Channel // Find all channels - err = DB.Model(&Channel{}).Find(&channels).Error + err := DB.Model(&Channel{}).Find(&channels).Error if err != nil { return 0, 0, err } diff --git a/relay/gemini_handler.go b/relay/gemini_handler.go index 862630ea..42b695b7 100644 --- a/relay/gemini_handler.go +++ b/relay/gemini_handler.go @@ -80,7 +80,11 @@ func getGeminiInputTokens(req *dto.GeminiChatRequest, info *relaycommon.RelayInf func isNoThinkingRequest(req *dto.GeminiChatRequest) bool { if req.GenerationConfig.ThinkingConfig != nil && req.GenerationConfig.ThinkingConfig.ThinkingBudget != nil { - return *req.GenerationConfig.ThinkingConfig.ThinkingBudget == 0 + configBudget := req.GenerationConfig.ThinkingConfig.ThinkingBudget + if configBudget != nil && *configBudget == 0 { + // 如果思考预算为 0,则认为是非思考请求 + return true + } } return false }