Something went wrong. Try again.
sloproxy
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101package 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) }}