fix(thinking): use static lookup to avoid alias issues

This commit is contained in:
hkfires
2026-01-14 15:44:08 +08:00
parent a75fb6af90
commit 40ee065eff
11 changed files with 52 additions and 28 deletions

View File

@@ -60,7 +60,7 @@ func init() {
// }
func (a *Applier) Apply(body []byte, config thinking.ThinkingConfig, modelInfo *registry.ModelInfo) ([]byte, error) {
if modelInfo == nil {
return body, nil
return a.applyCompatible(body, config)
}
if modelInfo.Thinking == nil {
if modelInfo.Type == "" {

View File

@@ -450,8 +450,9 @@ func TestGeminiApplyNilModelInfo(t *testing.T) {
if err != nil {
t.Fatalf("Apply() with nil modelInfo should not error, got: %v", err)
}
if string(result) != string(body) {
t.Fatalf("Apply() with nil modelInfo should return original body, got: %s", result)
// nil modelInfo now applies compatible config
if !gjson.GetBytes(result, "generationConfig.thinkingConfig.thinkingBudget").Exists() {
t.Fatalf("Apply() with nil modelInfo should apply thinking config, got: %s", result)
}
}