feat: publish sanitized custom.97 source

controller/model_pricing_conversion_endpoints_test.go

@@ -0,0 +1,71 @@
+package controller
+
+import (
+	"net/http"
+	"testing"
+
+	"github.com/QuantumNous/new-api/common"
+	"github.com/QuantumNous/new-api/model"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+// Model metadata may declare endpoints either as a type array or as a path map.
+// Conversion must understand both; reading only the map form made every model
+// with array-form metadata look like a broken routing configuration.
+func TestModelPricingConversionAcceptsDeclaredEndpointArray(t *testing.T) {
+	db := modelManagementDB(t, "sqlite", "")
+	previousQuota := common.QuotaPerUnit
+	common.QuotaPerUnit = 500000
+	t.Cleanup(func() { common.QuotaPerUnit = previousQuota })
+
+	for _, tc := range []struct {
+		name        string
+		endpoints   string
+		draft       model.PricingValues
+		wantImageCt bool
+	}{
+		{"array-endpoints-image", `["image-generation","openai"]`,
+			model.PricingValues{"ModelRatio": 1.7875, "CompletionRatio": 6, "ImageRatio": 1.6}, false},
+		{"array-endpoints-image-fixed", `["image-generation","openai"]`,
+			model.PricingValues{"ModelPrice": 1}, true},
+		{"array-endpoints-gemini", `["gemini","openai"]`,
+			model.PricingValues{"ModelRatio": 0.10725, "CompletionRatio": 100}, false},
+		{"array-endpoints-gemini-cache", `["gemini","openai"]`,
+			model.PricingValues{"ModelRatio": 0.715, "CompletionRatio": 60, "CacheRatio": 0.1}, false},
+		{"array-endpoints-openai-response", `["openai","openai-response","openai-response-compact","gemini"]`,
+			model.PricingValues{"ModelRatio": 0.17875, "CompletionRatio": 120}, false},
+		{"map-endpoints-image", `{"image-generation": {"path": "/v1/images/generations", "method": "POST"}}`,
+			model.PricingValues{"ModelPrice": 1}, true},
+		// Endpoint names must be trimmed before the switch matches them: a
+		// padded name would otherwise match nothing and silently downgrade a
+		// fixed-price image model to per-request billing.
+		{"array-endpoints-image-padded", `[" image-generation ","openai"]`,
+			model.PricingValues{"ModelPrice": 1}, true},
+		{"map-endpoints-image-padded", `{"image-generation ": {"path": "/v1/images/generations", "method": "POST"}}`,
+			model.PricingValues{"ModelPrice": 1}, true},
+	} {
+		t.Run(tc.name, func(t *testing.T) {
+			require.NoError(t, db.Create(&model.Model{
+				ModelName: tc.name, Endpoints: tc.endpoints, Status: 1, NameRule: model.NameRuleExact,
+			}).Error)
+			var response struct {
+				Success bool
+				Data    model.ModelPricingConversion
+			}
+			modelManagementRequest(t, PreviewModelPricingConversion, http.MethodPost,
+				"/api/option/model_pricing/convert",
+				map[string]any{"model_name": tc.name, "pricing": tc.draft}, &response)
+			require.True(t, response.Success)
+			t.Logf("PROBE %-34s expression=%q unsupported=%q image_count=%v",
+				tc.name, response.Data.Expression, response.Data.UnsupportedReason,
+				response.Data.BillingDetails.ImageCount)
+			require.Empty(t, response.Data.UnsupportedReason)
+			require.NotEmpty(t, response.Data.Expression)
+			// Declared endpoint types must actually reach the endpoint switch;
+			// an array of names read as array indices would silently skip it.
+			assert.Equal(t, tc.wantImageCt, response.Data.BillingDetails.ImageCount,
+				"declared image-generation endpoint must drive image-count billing")
+		})
+	}
+}