From 835ae178d4108df9bff3b79408604d2adb9f02fd Mon Sep 17 00:00:00 2001
From: hkfires <10558748+hkfires@users.noreply.github.com>
Date: Tue, 3 Mar 2026 19:49:51 +0800
Subject: [PATCH] feat(thinking): rename isBudgetBasedProvider to
 isBudgetCapableProvider and update logic for provider checks

---
 internal/thinking/apply.go    |  2 +-
 internal/thinking/validate.go | 13 +++----------
 2 files changed, 4 insertions(+), 11 deletions(-)

diff --git a/internal/thinking/apply.go b/internal/thinking/apply.go
index 16f1a2f9..b8a0fcae 100644
--- a/internal/thinking/apply.go
+++ b/internal/thinking/apply.go
@@ -293,7 +293,7 @@ func normalizeUserDefinedConfig(config ThinkingConfig, fromFormat, toFormat stri
 	if config.Mode != ModeLevel {
 		return config
 	}
-	if !isBudgetBasedProvider(toFormat) || !isLevelBasedProvider(fromFormat) {
+	if !isBudgetCapableProvider(toFormat) {
 		return config
 	}
 	budget, ok := ConvertLevelToBudget(string(config.Level))
diff --git a/internal/thinking/validate.go b/internal/thinking/validate.go
index d1f784c5..4a3ca97c 100644
--- a/internal/thinking/validate.go
+++ b/internal/thinking/validate.go
@@ -335,7 +335,9 @@ func normalizeLevels(levels []string) []string {
 	return out
 }
 
-func isBudgetBasedProvider(provider string) bool {
+// isBudgetCapableProvider returns true if the provider supports budget-based thinking.
+// These providers may also support level-based thinking (hybrid models).
+func isBudgetCapableProvider(provider string) bool {
 	switch provider {
 	case "gemini", "gemini-cli", "antigravity", "claude":
 		return true
@@ -344,15 +346,6 @@ func isBudgetBasedProvider(provider string) bool {
 	}
 }
 
-func isLevelBasedProvider(provider string) bool {
-	switch provider {
-	case "openai", "openai-response", "codex":
-		return true
-	default:
-		return false
-	}
-}
-
 func isGeminiFamily(provider string) bool {
 	switch provider {
 	case "gemini", "gemini-cli", "antigravity":