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

Sync production source including routing, billing, Astra detection and public uptime. Update custom build instructions and license packaging.

service/channel_select_test.go

@@ -22,9 +22,15 @@ func TestPinnedTaskPluginChannelTypesUsesPinnedGenerationIndex(t *testing.T) {
 		Plugin:     plugin,
 	})
 
-	assert.Equal(t, []int{constant.ChannelTypeKling}, pinnedTaskPluginChannelTypes(c, "legacy-select"))
-	assert.Empty(t, pinnedTaskPluginChannelTypes(c, "another-plugin"))
-	assert.Empty(t, pinnedTaskPluginChannelTypes(nil, "legacy-select"))
+	types, keys := pinnedTaskPluginIdentities(c, "legacy-select")
+	assert.Equal(t, []int{constant.ChannelTypeKling}, types)
+	assert.Equal(t, []string{"legacy-select"}, keys)
+	types, keys = pinnedTaskPluginIdentities(c, "another-plugin")
+	assert.Empty(t, types)
+	assert.Empty(t, keys)
+	types, keys = pinnedTaskPluginIdentities(nil, "legacy-select")
+	assert.Empty(t, types)
+	assert.Empty(t, keys)
 }
 
 func TestPinnedTaskPluginChannelTypesLeavesGenericChannelsKeyed(t *testing.T) {
@@ -38,7 +44,9 @@ func TestPinnedTaskPluginChannelTypesLeavesGenericChannelsKeyed(t *testing.T) {
 		Plugin:     plugin,
 	})
 
-	assert.Empty(t, pinnedTaskPluginChannelTypes(c, "generic-select"))
+	types, keys := pinnedTaskPluginIdentities(c, "generic-select")
+	assert.Empty(t, types)
+	assert.Equal(t, []string{"generic-select"}, keys)
 }
 
 func TestPinnedTaskPluginChannelTypesIncludesSharedEndpointProviders(t *testing.T) {
@@ -64,7 +72,11 @@ func TestPinnedTaskPluginChannelTypesIncludesSharedEndpointProviders(t *testing.
 		Candidates: candidates,
 	})
 
-	assert.Equal(t, []int{constant.ChannelTypeGemini, constant.ChannelTypeVertexAi}, pinnedTaskPluginChannelTypes(c, candidates[0].Plugin.Meta.Key))
+	AppendTaskPluginIdentityFilter(c, candidates[0].Plugin.Meta.Key)
+	filters := GetChannelConstraints(c).Filters
+	require.Len(t, filters, 1)
+	assert.Equal(t, []int{constant.ChannelTypeGemini, constant.ChannelTypeVertexAi}, filters[0].TaskPluginChannelTypes)
+	assert.Equal(t, []string{"gemini-select", "vertex-select"}, filters[0].TaskPluginKeys)
 }
 
 func channelSelectTaskPluginSource(key string, channelType int) string {
@@ -129,7 +141,9 @@ func TestPinnedTaskPluginChannelTypesIncludesCompatibleTypes(t *testing.T) {
 		Plugin:     plugin,
 	})
 
-	assert.Equal(t, []int{constant.ChannelTypeSora, constant.ChannelTypeOpenAI}, pinnedTaskPluginChannelTypes(c, "sora-select"))
+	types, keys := pinnedTaskPluginIdentities(c, "sora-select")
+	assert.Equal(t, []int{constant.ChannelTypeSora, constant.ChannelTypeOpenAI}, types)
+	assert.Equal(t, []string{"sora-select"}, keys)
 }
 
 func channelSelectCompatiblePluginSource(key string, channelType, compatibleType int) string {
@@ -150,3 +164,22 @@ export function buildQueryRequest() { return {}; }
 export function parseTaskResult() { return {status: "SUCCESS"}; }
 `, key, key, channelType, compatibleType)
 }
+
+func TestSharedType61IdentityFilterContainsAllCandidateKeys(t *testing.T) {
+	registry := jsplugin.NewRegistry()
+	for _, key := range []string{"alpha", "beta"} {
+		_, err := registry.Register(channelSelectEndpointPluginSource(key, 0), jsplugin.Options{})
+		require.NoError(t, err)
+	}
+	generation := registry.Generation()
+	candidates := generation.LookupEndpointCandidates("POST", "/v1/responses", "task-model")
+	require.Len(t, candidates, 2)
+	c, _ := gin.CreateTestContext(nil)
+	c.Set(jsplugin.ContextKeyPinnedEndpoint, jsplugin.PinnedEndpoint{Generation: generation, Plugin: candidates[0].Plugin, Candidates: candidates})
+	AppendTaskPluginIdentityFilter(c, "alpha")
+	filters := GetChannelConstraints(c).Filters
+	require.Len(t, filters, 1)
+	assert.Equal(t, "alpha", filters[0].TaskPluginKey)
+	assert.Equal(t, []string{"alpha", "beta"}, filters[0].TaskPluginKeys)
+	assert.Empty(t, filters[0].TaskPluginChannelTypes)
+}