feat: publish sanitized custom.97 source

model/task_plugin_channel_select_test.go

@@ -80,3 +80,60 @@ func TestSharedPluginKeysFilterBothChannelSources(t *testing.T) {
 	assert.Equal(t, []int{910001, 910002}, kept)
 	assert.Empty(t, emptied)
 }
+
+func TestNewAPIChannelServesExtendedTaskPlugins(t *testing.T) {
+	truncateTables(t)
+	priority := int64(0)
+	weight := uint(1)
+	baseURL := "https://gateway.example"
+	extended := `{"task_extend_plugin_keys":["alpha","gamma"]}`
+	single := `{"task_plugin_key":"delta"}`
+	channels := []Channel{
+		{Id: 920001, Type: constant.ChannelTypeNewAPI, Status: common.ChannelStatusEnabled, Name: "gateway", Models: "shared,chat", Group: "default", Priority: &priority, Weight: &weight, BaseURL: &baseURL, Setting: &extended},
+		{Id: 920002, Type: constant.ChannelTypeNewAPI, Status: common.ChannelStatusEnabled, Name: "gateway-single", Models: "shared", Group: "default", Priority: &priority, Weight: &weight, BaseURL: &baseURL, Setting: &single},
+	}
+	for i := range channels {
+		require.NoError(t, channels[i].Insert())
+	}
+
+	selected, err := GetChannel("default", "shared", 0, identityFilters("alpha", nil))
+	require.NoError(t, err)
+	require.NotNil(t, selected)
+	assert.Equal(t, "gateway", selected.Name)
+	selected, err = GetChannel("default", "shared", 0, identityFilters("delta", nil))
+	require.NoError(t, err)
+	require.NotNil(t, selected)
+	assert.Equal(t, "gateway-single", selected.Name, "a single task_plugin_key still binds a New API channel")
+	selected, err = GetChannel("default", "shared", 0, identityFilters("beta", nil))
+	require.NoError(t, err)
+	assert.Nil(t, selected, "unbound plugins never reach the gateway channel")
+	shared := []dto.ChannelFilter{{Kind: dto.FilterTaskPluginIdentity, TaskPluginKey: "beta", TaskPluginKeys: []string{"beta", "gamma"}}}
+	selected, err = GetChannel("default", "shared", 0, shared)
+	require.NoError(t, err)
+	require.NotNil(t, selected)
+	assert.Equal(t, "gateway", selected.Name, "a bound shared-model candidate admits the channel")
+	selected, err = GetChannel("default", "chat", 0, identityFilters("", nil))
+	require.NoError(t, err)
+	require.NotNil(t, selected)
+	assert.Equal(t, "gateway", selected.Name, "requests without a pinned plugin keep using the gateway")
+}
+
+func TestModelModifierNamesRouteWithAndWithoutMemoryCache(t *testing.T) {
+	truncateTables(t)
+	originalMemoryCache := common.MemoryCacheEnabled
+	t.Cleanup(func() { common.MemoryCacheEnabled = originalMemoryCache; InitChannelCache() })
+	channel := Channel{Id: 930001, Type: constant.ChannelTypeDeepSeek, Status: common.ChannelStatusEnabled, Name: "deepseek", Models: "deepseek-flash", Group: "default"}
+	require.NoError(t, channel.Insert())
+
+	for _, memoryCache := range []bool{false, true} {
+		common.MemoryCacheEnabled = memoryCache
+		InitChannelCache()
+		selected, err := GetRandomSatisfiedChannel("default", "deepseek-flash@effort:max", 0, nil)
+		require.NoError(t, err)
+		require.NotNil(t, selected, "memory cache %v", memoryCache)
+		assert.Equal(t, "deepseek", selected.Name)
+		selected, err = GetRandomSatisfiedChannel("default", "deepseek-pro@effort:max", 0, nil)
+		require.NoError(t, err)
+		assert.Nil(t, selected, "memory cache %v", memoryCache)
+	}
+}