seewe/sub2api
1
0
Fork 0
mirror of https://github.com/Wei-Shaw/sub2api.git synced 2026-10-16 07:33:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: allow composite grok messages routing

Browse Source
This commit is contained in:
Heatherm Huang
2026-07-23 09:20:52 +08:00
parent 06e7d12640
commit 1d2dfab86d
3 changed files with 32 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/internal/handler/composite_platform_test.go
+12
View File
@@ -22,6 +22,18 @@ func TestCompositeTargetPlatformAllowedResolvesKnownAllowedModel(t *testing.T) {
require.Equal(t, service.PlatformOpenAI, platform)
}
func TestOpenAICompatibleMessagesTargetAllowsCompositeGrokModel(t *testing.T) {
gin.SetMode(gin.TestMode)
c, _ := gin.CreateTestContext(httptest.NewRecorder())
c.Request = httptest.NewRequest("POST", "/v1/messages", nil)
apiKey := &service.APIKey{Group: &service.Group{Platform: service.PlatformComposite}}
require.True(t, openAICompatibleMessagesTargetAllowed(c, apiKey, "grok-4.3"))
platform, ok := service.ResolvedTargetPlatformFromContext(c.Request.Context())
require.True(t, ok)
require.Equal(t, service.PlatformGrok, platform)
}
func TestCompositeTargetPlatformAllowedRejectsWrongOrUnknownModel(t *testing.T) {
gin.SetMode(gin.TestMode)