oadmin/new-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: improve error handling and readability in ability.go

Browse Source
This commit is contained in:
CaIon
2025-08-02 14:06:12 +08:00
parent 7188749cb3
commit 74ec34da67
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
model/ability.go
View File

@@ -301,7 +301,7 @@ func FixAbility() (int, int, error) {
} }
var channels []*Channel var channels []*Channel
// Find all channels // Find all channels
err = DB.Model(&Channel{}).Find(&channels).Error err := DB.Model(&Channel{}).Find(&channels).Error
if err != nil { if err != nil {
return 0, 0, err return 0, 0, err
} }

6
relay/gemini_handler.go
View File

@@ -80,7 +80,11 @@ func getGeminiInputTokens(req *dto.GeminiChatRequest, info *relaycommon.RelayInf
func isNoThinkingRequest(req *dto.GeminiChatRequest) bool { func isNoThinkingRequest(req *dto.GeminiChatRequest) bool {
if req.GenerationConfig.ThinkingConfig != nil && req.GenerationConfig.ThinkingConfig.ThinkingBudget != nil { 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 return false
} }