feat: publish sanitized custom.97 source

relay/common/relay_info_gemini_settings_test.go

@@ -3,6 +3,7 @@ package common
 import (
 	"testing"
 
+	"github.com/QuantumNous/new-api/constant"
 	"github.com/QuantumNous/new-api/setting/model_setting"
 	"github.com/stretchr/testify/assert"
 )
@@ -23,3 +24,35 @@ func TestRelayInfoConvOptionsUsesNormalizedGeminiSafetySettings(t *testing.T) {
 	assert.Equal(t, "OFF", options.Gemini.SafetySetting("HARM_CATEGORY_HATE_SPEECH"))
 	assert.Equal(t, "BLOCK_ONLY_HIGH", options.Gemini.SafetySetting("HARM_CATEGORY_DANGEROUS_CONTENT"))
 }
+
+// TestRelayInfoConvOptionsGeminiJSONSchemaSupport covers which upstreams get
+// parametersJsonSchema and responseJsonSchema: every channel except one whose
+// Gemini API version for the model is v1, and Vertex AI regardless of it.
+func TestRelayInfoConvOptionsGeminiJSONSchemaSupport(t *testing.T) {
+	settings := model_setting.GetGeminiSettings()
+	original := settings.VersionSettings
+	t.Cleanup(func() {
+		settings.VersionSettings = original
+	})
+	settings.VersionSettings = map[string]string{"default": "v1beta", "gemini-v1-model": "v1"}
+
+	tests := []struct {
+		name        string
+		channelType int
+		model       string
+		want        bool
+	}{
+		{name: "Gemini on the default version", channelType: constant.ChannelTypeGemini, model: "gemini-3.7-flash", want: true},
+		{name: "Gemini on v1", channelType: constant.ChannelTypeGemini, model: "gemini-v1-model", want: false},
+		{name: "Vertex AI", channelType: constant.ChannelTypeVertexAi, model: "gemini-v1-model", want: true},
+		{name: "other channel on v1", channelType: constant.ChannelTypeAdvancedCustom, model: "gemini-v1-model", want: false},
+		{name: "other channel on the default version", channelType: constant.ChannelTypeAdvancedCustom, model: "gemini-3.7-flash", want: true},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			info := &RelayInfo{ChannelMeta: &ChannelMeta{ChannelType: tt.channelType}}
+
+			assert.Equal(t, tt.want, info.ConvOptions().Gemini.SupportsJSONSchemaModel(tt.model))
+		})
+	}
+}