feat: sync deployed v1.0.0-rc.38-custom.67

controller/astra_sampling_test.go

@@ -5,11 +5,14 @@ import (
 	"io"
 	"net/http"
 	"net/http/httptest"
+	"strings"
 	"testing"
 
 	"github.com/QuantumNous/new-api/common"
 	"github.com/QuantumNous/new-api/model"
 	fingerprint "github.com/QuantumNous/new-api/pkg/model_fingerprint"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 	"github.com/tidwall/gjson"
 )
 
@@ -17,6 +20,13 @@ func TestAstraSamplingUsesIsolatedResponsesWithoutGuard(t *testing.T) {
 	requests := 0
 	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		requests++
+		bodyPreview, readErr := io.ReadAll(r.Body)
+		require.NoError(t, readErr)
+		r.Body = io.NopCloser(strings.NewReader(string(bodyPreview)))
+		assert.EqualValues(t, fingerprint.OutputTokens, gjson.GetBytes(bodyPreview, "max_output_tokens").Int())
+		assert.Equal(t, "low", gjson.GetBytes(bodyPreview, "reasoning.effort").String())
+		assert.Contains(t, gjson.GetBytes(bodyPreview, "input.0.content").String(), "355")
+
 		body, _ := io.ReadAll(r.Body)
 		if r.URL.Path != "/v1/responses" || gjson.GetBytes(body, "model").String() != fingerprint.Model || !gjson.GetBytes(body, "stream").Bool() {
 			t.Errorf("wrong probe request %s %s", r.URL.Path, body)
@@ -30,8 +40,8 @@ func TestAstraSamplingUsesIsolatedResponsesWithoutGuard(t *testing.T) {
 	defer server.Close()
 	url := server.URL
 	channel := &model.Channel{Id: 1, Type: 1, Key: "test-only", BaseURL: &url, Status: common.ChannelStatusEnabled, Models: fingerprint.Model}
-	r, count, err := runAstraFingerprintProbe(context.Background(), channel)
-	if err != nil || r.DeclaredModel != "gpt-5.6-luna" || count < 292 || count > 332 || requests != 1 {
+	r, count, err := runAstraFingerprintProbe(context.Background(), channel, 0)
+	if err != nil || r.DeclaredModel != "gpt-5.6-luna" || count < 292 || count > 304 || requests != 1 {
 		t.Fatalf("probe %+v count=%d requests=%d err=%v", r, count, requests, err)
 	}
 }