mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-02-03 04:50:52 +08:00
374faa2640
Ensure thinking settings translate correctly across providers: - Only apply reasoning_effort to level-based models and derive it from numeric budget suffixes when present - Strip effort string fields for budget-based models and skip Claude/Gemini budget resolution for level-based or unsupported models - Default Gemini include_thoughts when a nonzero budget override is set - Add cross-protocol conversion and budget range tests
50 lines
1.5 KiB
Go
50 lines
1.5 KiB
Go
package util
|
|
|
|
import (
|
|
"github.com/tidwall/gjson"
|
|
"github.com/tidwall/sjson"
|
|
)
|
|
|
|
// ApplyClaudeThinkingConfig applies thinking configuration to a Claude API request payload.
|
|
// It sets the thinking.type to "enabled" and thinking.budget_tokens to the specified budget.
|
|
// If budget is nil or the payload already has thinking config, it returns the payload unchanged.
|
|
func ApplyClaudeThinkingConfig(body []byte, budget *int) []byte {
|
|
if budget == nil {
|
|
return body
|
|
}
|
|
if gjson.GetBytes(body, "thinking").Exists() {
|
|
return body
|
|
}
|
|
if *budget <= 0 {
|
|
return body
|
|
}
|
|
updated := body
|
|
updated, _ = sjson.SetBytes(updated, "thinking.type", "enabled")
|
|
updated, _ = sjson.SetBytes(updated, "thinking.budget_tokens", *budget)
|
|
return updated
|
|
}
|
|
|
|
// ResolveClaudeThinkingConfig resolves thinking configuration from metadata for Claude models.
|
|
// It uses the unified ResolveThinkingConfigFromMetadata and normalizes the budget.
|
|
// Returns the normalized budget (nil if thinking should not be enabled) and whether it matched.
|
|
func ResolveClaudeThinkingConfig(modelName string, metadata map[string]any) (*int, bool) {
|
|
if !ModelSupportsThinking(modelName) {
|
|
return nil, false
|
|
}
|
|
budget, include, matched := ResolveThinkingConfigFromMetadata(modelName, metadata)
|
|
if !matched {
|
|
return nil, false
|
|
}
|
|
if include != nil && !*include {
|
|
return nil, true
|
|
}
|
|
if budget == nil {
|
|
return nil, true
|
|
}
|
|
normalized := NormalizeThinkingBudget(modelName, *budget)
|
|
if normalized <= 0 {
|
|
return nil, true
|
|
}
|
|
return &normalized, true
|
|
}
|