cneicy/new-api-custom
1
0 Fork 0
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

feat: publish sanitized custom.97 source

Browse Source
This commit is contained in:
cneicy committed 2026-10-27 20:43:39 +08:00
1 parent 8e6ed0317b
commit 86c0276156
1022 files changed
+69903 -19207
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

service/image_generation_group_test.go
+6 -6
View File
@@ -18,11 +18,11 @@ func TestDedicatedImageGenerationPoolSelection(t *testing.T) {
cfg := model_setting.GetGlobalSettings()
oldGroup, oldModels := cfg.ImageGenerationGroup, cfg.ImageGenerationModels
t.Cleanup(func() { cfg.ImageGenerationGroup, cfg.ImageGenerationModels = oldGroup, oldModels })
cfg.ImageGenerationGroup, cfg.ImageGenerationModels = "example-images", []string{"gpt-image-2"}
require.NoError(t, setting.UpdateUserUsableGroupsByJSONString(`{"default":"Default","vip":"VIP","example-images":"Images"}`))
require.NoError(t, ratio_setting.UpdateGroupRatioByJSONString(`{"default":1,"vip":2,"example-images":1.25}`))
cfg.ImageGenerationGroup, cfg.ImageGenerationModels = "gpt-images", []string{"gpt-image-2"}
require.NoError(t, setting.UpdateUserUsableGroupsByJSONString(`{"default":"Default","vip":"VIP","gpt-images":"Images"}`))
require.NoError(t, ratio_setting.UpdateGroupRatioByJSONString(`{"default":1,"vip":2,"gpt-images":1.25}`))
createChannelSelectAutoGroupsChannel(t, db, 8001, "default", "gpt-image-2")
createChannelSelectAutoGroupsChannel(t, db, 8002, "example-images", "gpt-image-2")
createChannelSelectAutoGroupsChannel(t, db, 8002, "gpt-images", "gpt-image-2")
model.InitChannelCache()
ResetGroupAvailability()
t.Cleanup(ResetGroupAvailability)
@@ -33,12 +33,12 @@ func TestDedicatedImageGenerationPoolSelection(t *testing.T) {
common.SetContextKey(c, constant.ContextKeyUserGroup, "default")
c.Set(model.TokenRoutingContextKey, &model.TokenRoutingPolicy{Version: 1, Mode: mode, Groups: []model.TokenGroupEntry{{Group: "default", Priority: 0}}})
entries := ResolveRequestPool(c, "gpt-image-2")
require.Equal(t, []model.TokenGroupEntry{{Group: "example-images", Priority: 0}}, entries)
require.Equal(t, []model.TokenGroupEntry{{Group: "gpt-images", Priority: 0}}, entries)
channel, group, err := CacheGetRandomSatisfiedChannel(&RetryParam{Ctx: c, TokenGroup: "default", ModelName: "gpt-image-2", Retry: common.GetPointer(0)})
require.NoError(t, err)
require.NotNil(t, channel)
require.Equal(t, 8002, channel.Id)
require.Equal(t, "example-images", group)
require.Equal(t, "gpt-images", group)
})
}
c, _ := gin.CreateTestContext(httptest.NewRecorder())