From 74ec34da674470c5dc657b15816829da570eb2b6 Mon Sep 17 00:00:00 2001
From: CaIon <i@caion.me>
Date: Sat, 2 Aug 2025 14:06:12 +0800
Subject: [PATCH] fix: improve error handling and readability in ability.go

---
 model/ability.go        | 2 +-
 relay/gemini_handler.go | 6 +++++-
 2 files changed, 6 insertions(+), 2 deletions(-)

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
 }