Add projects

backend/internal/runtime/executor/helps/thinking_test.go

@@ -0,0 +1,100 @@
+package helps_test
+
+import (
+	"context"
+	"testing"
+
+	helps "github.com/router-for-me/CLIProxyAPI/v7/internal/runtime/executor/helps"
+	_ "github.com/router-for-me/CLIProxyAPI/v7/internal/thinking/provider/gemini"
+	_ "github.com/router-for-me/CLIProxyAPI/v7/internal/translator"
+	sdktranslator "github.com/router-for-me/CLIProxyAPI/v7/sdk/translator"
+	"github.com/tidwall/gjson"
+	"github.com/tidwall/sjson"
+)
+
+type summaryRemovingPluginHooks struct {
+	t *testing.T
+}
+
+func (h *summaryRemovingPluginHooks) NormalizeRequest(_ context.Context, _, _ sdktranslator.Format, _ string, body []byte, _ bool) []byte {
+	h.t.Helper()
+	const path = "generationConfig.thinkingConfig.includeThoughts"
+	if !gjson.GetBytes(body, path).Bool() {
+		h.t.Fatalf("request normalizer did not receive enabled summary: %s", body)
+	}
+	out, _ := sjson.DeleteBytes(body, path)
+	return out
+}
+
+func (*summaryRemovingPluginHooks) TranslateRequest(context.Context, sdktranslator.Format, sdktranslator.Format, string, []byte, bool) ([]byte, bool) {
+	return nil, false
+}
+
+func (*summaryRemovingPluginHooks) NormalizeResponseBefore(context.Context, sdktranslator.Format, sdktranslator.Format, string, []byte, []byte, []byte, bool) []byte {
+	return nil
+}
+
+func (*summaryRemovingPluginHooks) TranslateResponse(context.Context, sdktranslator.Format, sdktranslator.Format, string, []byte, []byte, []byte, bool) ([]byte, bool) {
+	return nil, false
+}
+
+func (*summaryRemovingPluginHooks) NormalizeResponseAfter(context.Context, sdktranslator.Format, sdktranslator.Format, string, []byte, []byte, []byte, bool) []byte {
+	return nil
+}
+
+func TestApplyThinkingWithSourcePayloadPreservesNormalizerSummaryRemoval(t *testing.T) {
+	hooks := &summaryRemovingPluginHooks{t: t}
+	sdktranslator.SetPluginHooks(hooks)
+	t.Cleanup(func() { sdktranslator.SetPluginHooks(nil) })
+
+	source := []byte(`{"model":"gemini-3.6-flash","reasoning":{"effort":"high","summary":"auto"},"input":"hi"}`)
+	translated := sdktranslator.TranslateRequest(
+		sdktranslator.FormatOpenAIResponse,
+		sdktranslator.FormatGemini,
+		"gemini-3.6-flash",
+		source,
+		false,
+	)
+	const summaryPath = "generationConfig.thinkingConfig.includeThoughts"
+	if gjson.GetBytes(translated, summaryPath).Exists() {
+		t.Fatalf("request normalizer did not remove summary: %s", translated)
+	}
+
+	out, err := helps.ApplyThinkingWithSourcePayload(
+		translated,
+		source,
+		source,
+		"gemini-3.6-flash",
+		sdktranslator.FormatOpenAIResponse.String(),
+		sdktranslator.FormatGemini.String(),
+		"gemini",
+	)
+	if err != nil {
+		t.Fatalf("ApplyThinkingWithSourcePayload() error = %v", err)
+	}
+	if gjson.GetBytes(out, summaryPath).Exists() {
+		t.Fatalf("executor restored summary removed by request normalizer: %s", out)
+	}
+}
+
+func TestApplyThinkingWithSourcePayloadPreservesOriginalOnlySummary(t *testing.T) {
+	currentSource := []byte(`{"model":"gemini-3.6-flash","input":"hi"}`)
+	originalSource := []byte(`{"model":"gemini-3.6-flash","reasoning":{"summary":null},"input":"hi"}`)
+	body := []byte(`{"generationConfig":{"thinkingConfig":{"thinkingLevel":"high"}}}`)
+
+	out, err := helps.ApplyThinkingWithSourcePayload(
+		body,
+		currentSource,
+		originalSource,
+		"gemini-3.6-flash",
+		sdktranslator.FormatOpenAIResponse.String(),
+		sdktranslator.FormatGemini.String(),
+		"gemini",
+	)
+	if err != nil {
+		t.Fatalf("ApplyThinkingWithSourcePayload() error = %v", err)
+	}
+	if include := gjson.GetBytes(out, "generationConfig.thinkingConfig.includeThoughts"); !include.Exists() || include.Bool() {
+		t.Fatalf("original disabled summary was not preserved: %s", out)
+	}
+}