diff --git a/backend/internal/service/account_test_service.go b/backend/internal/service/account_test_service.go index 900d3f8e15..49d5c76295 100644 --- a/backend/internal/service/account_test_service.go +++ b/backend/internal/service/account_test_service.go @@ -285,9 +285,13 @@ func (s *AccountTestService) TestAccountConnection(c *gin.Context, accountID int } // Route to platform-specific test method - if account.IsCNProvider() && - (account.GetAPIProtocol() == APIProtocolChatCompletions || account.IsAdaptiveAPIProtocol()) { - return s.testCNProviderChatCompletionsConnection(c, account, modelID, prompt) + if account.IsCNProvider() { + switch account.GetAPIProtocol() { + case APIProtocolAdaptive: + return s.testCNProviderAdaptiveConnection(c, account, modelID, prompt) + case APIProtocolChatCompletions: + return s.testCNProviderChatCompletionsConnection(c, account, modelID, prompt) + } } if account.IsOpenAI() { @@ -3064,6 +3068,13 @@ func (s *AccountTestService) testOpenAIImageOAuth(c *gin.Context, ctx context.Co } func (s *AccountTestService) sendEvent(c *gin.Context, event TestEvent) { + if event.Type == "test_complete" { + if suppress, ok := c.Get(accountTestSuppressCompletionContextKey); ok { + if suppressCompletion, _ := suppress.(bool); suppressCompletion { + return + } + } + } eventJSON, _ := json.Marshal(event) if _, err := fmt.Fprintf(c.Writer, "data: %s\n\n", eventJSON); err != nil { log.Printf("failed to write SSE event: %v", err) diff --git a/backend/internal/service/account_test_service_cn_adaptive.go b/backend/internal/service/account_test_service_cn_adaptive.go new file mode 100644 index 0000000000..ae52e4f117 --- /dev/null +++ b/backend/internal/service/account_test_service_cn_adaptive.go @@ -0,0 +1,206 @@ +package service + +import ( + "bufio" + "bytes" + "encoding/json" + "fmt" + "io" + "net/http" + "strings" + + "github.com/Wei-Shaw/sub2api/internal/pkg/claude" + "github.com/Wei-Shaw/sub2api/internal/pkg/openai" + "github.com/gin-gonic/gin" +) + +const accountTestSuppressCompletionContextKey = "account_test_suppress_completion" + +// testCNProviderAdaptiveConnection verifies every native endpoint used by an +// adaptive CN-provider account. Kimi and Zhipu use Chat Completions plus +// Anthropic; DeepSeek additionally uses its native Responses endpoint. +func (s *AccountTestService) testCNProviderAdaptiveConnection(c *gin.Context, account *Account, modelID string, prompt string) error { + testModelID := strings.TrimSpace(modelID) + if testModelID == "" { + testModelID = openai.DefaultTestModel + } + testModelID = account.GetMappedModel(testModelID) + + authToken := strings.TrimSpace(account.GetOpenAIProtocolAPIKey()) + if authToken == "" { + return s.sendErrorAndEnd(c, "No API key available") + } + + // The existing Chat probe owns the SSE lifecycle. Suppress intermediate + // completion events until every native adaptive endpoint has passed. + c.Set(accountTestSuppressCompletionContextKey, true) + defer c.Set(accountTestSuppressCompletionContextKey, false) + if err := s.testCNProviderChatCompletionsConnection(c, account, modelID, prompt); err != nil { + return err + } + + if err := s.testCNProviderAdaptiveAnthropicConnection(c, account, testModelID, authToken); err != nil { + return err + } + + if account.Platform == PlatformDeepseek { + if err := s.testCNProviderAdaptiveResponsesConnection(c, account, testModelID, authToken); err != nil { + return err + } + } + + c.Set(accountTestSuppressCompletionContextKey, false) + s.sendEvent(c, TestEvent{Type: "test_complete", Success: true}) + return nil +} + +func (s *AccountTestService) testCNProviderAdaptiveAnthropicConnection(c *gin.Context, account *Account, testModelID string, authToken string) error { + ctx := c.Request.Context() + baseURL, err := s.validateUpstreamBaseURL(account.GetCNProtocolBaseURL(APIProtocolAnthropic)) + if err != nil { + return s.sendErrorAndEnd(c, fmt.Sprintf("Invalid adaptive Anthropic base URL: %s", err.Error())) + } + apiURL := strings.TrimRight(baseURL, "/") + "/v1/messages" + + payload, err := createTestPayload(testModelID) + if err != nil { + return s.sendErrorAndEnd(c, "Failed to create adaptive Anthropic test payload") + } + payloadBytes, _ := json.Marshal(payload) + + s.sendEvent(c, TestEvent{Type: "status", Text: "正在通过原生 /v1/messages 测试自适应 Anthropic 端点"}) + req, err := http.NewRequestWithContext(ctx, http.MethodPost, apiURL, bytes.NewReader(payloadBytes)) + if err != nil { + return s.sendErrorAndEnd(c, "Failed to create adaptive Anthropic request") + } + req.Header.Set("Content-Type", "application/json") + req.Header.Set("Accept", "text/event-stream") + req.Header.Set("anthropic-version", "2023-06-01") + for key, value := range claude.DefaultHeaders { + req.Header.Set(key, value) + } + req.Header.Set("anthropic-beta", claude.APIKeyBetaHeader) + setAnthropicAPIKeyAuthHeader(req.Header, account, authToken) + account.ApplyHeaderOverrides(req.Header) + + resp, err := s.doCNProviderAdaptiveRequest(req, account) + if err != nil { + return s.sendErrorAndEnd(c, fmt.Sprintf("Adaptive Anthropic endpoint request failed: %s", err.Error())) + } + defer func() { _ = resp.Body.Close() }() + if resp.StatusCode != http.StatusOK { + body, _ := io.ReadAll(resp.Body) + errMsg := fmt.Sprintf("Adaptive Anthropic endpoint returned %d: %s", resp.StatusCode, string(body)) + if resp.StatusCode == http.StatusUnauthorized && s.accountRepo != nil { + _ = s.accountRepo.SetError(ctx, account.ID, errMsg) + } + return s.sendErrorAndEnd(c, errMsg) + } + + if err := s.processCNProviderAdaptiveAnthropicStream(c, resp.Body); err != nil { + return err + } + s.sendEvent(c, TestEvent{Type: "status", Text: "已通过原生 /v1/messages 验证"}) + return nil +} + +func (s *AccountTestService) processCNProviderAdaptiveAnthropicStream(c *gin.Context, body io.Reader) error { + reader := bufio.NewReader(body) + for { + line, err := reader.ReadString('\n') + if err != nil { + if err == io.EOF { + return s.sendErrorAndEnd(c, "Adaptive Anthropic stream ended before message_stop") + } + return s.sendErrorAndEnd(c, fmt.Sprintf("Adaptive Anthropic stream read error: %s", err.Error())) + } + + line = strings.TrimSpace(line) + if line == "" || !sseDataPrefix.MatchString(line) { + continue + } + jsonStr := sseDataPrefix.ReplaceAllString(line, "") + if jsonStr == "[DONE]" { + return nil + } + + var data map[string]any + if err := json.Unmarshal([]byte(jsonStr), &data); err != nil { + continue + } + switch eventType, _ := data["type"].(string); eventType { + case "content_block_delta": + if delta, ok := data["delta"].(map[string]any); ok { + if text, ok := delta["text"].(string); ok && text != "" { + s.sendEvent(c, TestEvent{Type: "content", Text: text}) + } + } + case "message_stop": + return nil + case "error": + errorMsg := "Unknown error" + if errData, ok := data["error"].(map[string]any); ok { + if message, ok := errData["message"].(string); ok && message != "" { + errorMsg = message + } + } + return s.sendErrorAndEnd(c, fmt.Sprintf("Adaptive Anthropic endpoint error: %s", errorMsg)) + } + } +} + +func (s *AccountTestService) testCNProviderAdaptiveResponsesConnection(c *gin.Context, account *Account, testModelID string, authToken string) error { + ctx := c.Request.Context() + baseURL, err := s.validateUpstreamBaseURL(account.GetCNProtocolBaseURL(APIProtocolResponses)) + if err != nil { + return s.sendErrorAndEnd(c, fmt.Sprintf("Invalid adaptive Responses base URL: %s", err.Error())) + } + apiURL := buildOpenAIResponsesURLForPlatform(account.Platform, baseURL) + + payload := createOpenAITestPayload(testModelID, false) + // DeepSeek's native Responses endpoint is stateless and does not need the + // OpenAI probe's synthetic instructions. + delete(payload, "instructions") + payloadBytes, _ := json.Marshal(payload) + payloadBytes = normalizeDeepSeekResponsesRequestBody(account, payloadBytes) + + s.sendEvent(c, TestEvent{Type: "status", Text: "正在通过原生 /responses 测试自适应 Responses 端点"}) + req, err := http.NewRequestWithContext(ctx, http.MethodPost, apiURL, bytes.NewReader(payloadBytes)) + if err != nil { + return s.sendErrorAndEnd(c, "Failed to create adaptive Responses request") + } + req = req.WithContext(WithHTTPUpstreamProfile(req.Context(), HTTPUpstreamProfileOpenAI)) + req.Header.Set("Content-Type", "application/json") + req.Header.Set("Accept", "text/event-stream") + req.Header.Set("Authorization", "Bearer "+authToken) + applyOpenAICodexProbeHeaders(req.Header) + account.ApplyHeaderOverrides(req.Header) + + resp, err := s.doCNProviderAdaptiveRequest(req, account) + if err != nil { + return s.sendErrorAndEnd(c, fmt.Sprintf("Adaptive Responses endpoint request failed: %s", err.Error())) + } + defer func() { _ = resp.Body.Close() }() + if resp.StatusCode != http.StatusOK { + body, _ := io.ReadAll(resp.Body) + errMsg := fmt.Sprintf("Adaptive Responses endpoint returned %d: %s", resp.StatusCode, string(body)) + if resp.StatusCode == http.StatusUnauthorized && s.accountRepo != nil { + _ = s.accountRepo.SetError(ctx, account.ID, errMsg) + } + return s.sendErrorAndEnd(c, errMsg) + } + + if err := s.processOpenAIStream(c, resp.Body); err != nil { + return err + } + s.sendEvent(c, TestEvent{Type: "status", Text: "已通过原生 /responses 验证"}) + return nil +} + +func (s *AccountTestService) doCNProviderAdaptiveRequest(req *http.Request, account *Account) (*http.Response, error) { + proxyURL := "" + if account.ProxyID != nil && account.Proxy != nil { + proxyURL = account.Proxy.URL() + } + return s.httpUpstream.DoWithTLS(req, proxyURL, account.ID, account.Concurrency, s.tlsFPProfileService.ResolveTLSProfile(account)) +} diff --git a/backend/internal/service/account_test_service_cn_adaptive_test.go b/backend/internal/service/account_test_service_cn_adaptive_test.go new file mode 100644 index 0000000000..808aec33b4 --- /dev/null +++ b/backend/internal/service/account_test_service_cn_adaptive_test.go @@ -0,0 +1,191 @@ +//go:build unit + +package service + +import ( + "io" + "net/http" + "strings" + "testing" + + "github.com/stretchr/testify/require" + "github.com/tidwall/gjson" +) + +func adaptiveCNAccountTestAccount(id int64, platform string) *Account { + return &Account{ + ID: id, + Name: "adaptive-cn-test", + Platform: platform, + Type: AccountTypeAPIKey, + Status: StatusActive, + Concurrency: 1, + Credentials: map[string]any{ + "api_key": "sk-adaptive-test", + "api_protocol": APIProtocolAdaptive, + "api_base_urls": map[string]any{ + APIProtocolChatCompletions: "http://chat.example/v1", + APIProtocolAnthropic: "http://anthropic.example", + APIProtocolResponses: "http://responses.example", + }, + }, + } +} + +func adaptiveCNAccountTestService(account *Account, responses ...*http.Response) (*AccountTestService, *httpUpstreamRecorder) { + repo := &openAIAccountTestRepo{ + mockAccountRepoForGemini: mockAccountRepoForGemini{ + accountsByID: map[int64]*Account{account.ID: account}, + }, + } + upstream := &httpUpstreamRecorder{responses: responses} + return &AccountTestService{ + accountRepo: repo, + httpUpstream: upstream, + cfg: rawChatCompletionsTestConfig(), + }, upstream +} + +func adaptiveCNChatTestResponse() *http.Response { + return &http.Response{ + StatusCode: http.StatusOK, + Header: http.Header{"Content-Type": []string{"text/event-stream"}}, + Body: io.NopCloser(strings.NewReader(`data: {"choices":[{"delta":{"content":"chat ok"},"finish_reason":"stop"}]} + +data: [DONE] + +`)), + } +} + +func adaptiveCNAnthropicTestResponse() *http.Response { + return &http.Response{ + StatusCode: http.StatusOK, + Header: http.Header{"Content-Type": []string{"text/event-stream"}}, + Body: io.NopCloser(strings.NewReader(`data: {"type":"content_block_delta","delta":{"text":"anthropic ok"}} + +data: {"type":"message_stop"} + +`)), + } +} + +func adaptiveCNResponsesTestResponse() *http.Response { + return &http.Response{ + StatusCode: http.StatusOK, + Header: http.Header{"Content-Type": []string{"text/event-stream"}}, + Body: io.NopCloser(strings.NewReader(`data: {"type":"response.output_text.delta","delta":"responses ok"} + +data: {"type":"response.completed"} + +`)), + } +} + +func TestAccountTestService_AdaptiveChatOnlyProvidersTestChatAndAnthropicEndpoints(t *testing.T) { + for index, testCase := range []struct { + name string + platform string + model string + }{ + {name: "Kimi", platform: PlatformKimi, model: "kimi-k2.5"}, + {name: "Zhipu", platform: PlatformZhipu, model: "glm-4.7"}, + } { + t.Run(testCase.name, func(t *testing.T) { + account := adaptiveCNAccountTestAccount(int64(301+index), testCase.platform) + svc, upstream := adaptiveCNAccountTestService( + account, + adaptiveCNChatTestResponse(), + adaptiveCNAnthropicTestResponse(), + ) + c, recorder := newTestContext() + + err := svc.TestAccountConnection(c, account.ID, testCase.model, "hello", AccountTestModeDefault) + + require.NoError(t, err) + require.Len(t, upstream.requests, 2) + require.Equal(t, "http://chat.example/v1/chat/completions", upstream.requests[0].URL.String()) + require.Equal(t, "http://anthropic.example/v1/messages", upstream.requests[1].URL.String()) + require.Equal(t, "Bearer sk-adaptive-test", upstream.requests[0].Header.Get("Authorization")) + require.Equal(t, "sk-adaptive-test", upstream.requests[1].Header.Get("x-api-key")) + require.Equal(t, 1, strings.Count(recorder.Body.String(), `"type":"test_start"`)) + require.Equal(t, 1, strings.Count(recorder.Body.String(), `"type":"test_complete"`)) + require.Contains(t, recorder.Body.String(), "已通过原生 /v1/messages 验证") + }) + } +} + +func TestAccountTestService_AdaptiveDeepSeekAlsoTestsResponsesEndpoint(t *testing.T) { + account := adaptiveCNAccountTestAccount(302, PlatformDeepseek) + svc, upstream := adaptiveCNAccountTestService( + account, + adaptiveCNChatTestResponse(), + adaptiveCNAnthropicTestResponse(), + adaptiveCNResponsesTestResponse(), + ) + c, recorder := newTestContext() + + err := svc.TestAccountConnection(c, account.ID, "deepseek-chat", "", AccountTestModeDefault) + + require.NoError(t, err) + require.Len(t, upstream.requests, 3) + require.Equal(t, "http://responses.example/responses", upstream.requests[2].URL.String()) + require.Equal(t, HTTPUpstreamProfileOpenAI, HTTPUpstreamProfileFromContext(upstream.requests[2].Context())) + require.Equal(t, "Bearer sk-adaptive-test", upstream.requests[2].Header.Get("Authorization")) + require.True(t, gjson.GetBytes(upstream.bodies[2], "stream").Bool()) + require.False(t, gjson.GetBytes(upstream.bodies[2], "store").Bool()) + require.False(t, gjson.GetBytes(upstream.bodies[2], "instructions").Exists()) + require.Equal(t, 1, strings.Count(recorder.Body.String(), `"type":"test_complete"`)) + require.Contains(t, recorder.Body.String(), "已通过原生 /responses 验证") +} + +func TestAccountTestService_AdaptiveStopsAndNamesFailingEndpoint(t *testing.T) { + account := adaptiveCNAccountTestAccount(303, PlatformDeepseek) + svc, upstream := adaptiveCNAccountTestService( + account, + adaptiveCNChatTestResponse(), + newJSONResponse(http.StatusNotFound, `{"error":{"message":"missing messages route"}}`), + ) + c, recorder := newTestContext() + + err := svc.TestAccountConnection(c, account.ID, "deepseek-chat", "", AccountTestModeDefault) + + require.Error(t, err) + require.Contains(t, err.Error(), "Adaptive Anthropic endpoint returned 404") + require.Len(t, upstream.requests, 2) + require.Contains(t, recorder.Body.String(), `"type":"error"`) + require.NotContains(t, recorder.Body.String(), `"type":"test_complete"`) +} + +func TestAccountTestService_AdaptiveRejectsInvalidAnthropicSuccessBody(t *testing.T) { + account := adaptiveCNAccountTestAccount(305, PlatformKimi) + svc, upstream := adaptiveCNAccountTestService( + account, + adaptiveCNChatTestResponse(), + newJSONResponse(http.StatusOK, `not an Anthropic stream`), + ) + c, recorder := newTestContext() + + err := svc.TestAccountConnection(c, account.ID, "kimi-k2.5", "", AccountTestModeDefault) + + require.Error(t, err) + require.Contains(t, err.Error(), "Adaptive Anthropic stream ended before message_stop") + require.Len(t, upstream.requests, 2) + require.Contains(t, recorder.Body.String(), `"type":"error"`) + require.NotContains(t, recorder.Body.String(), `"type":"test_complete"`) +} + +func TestAccountTestService_FixedCNChatProtocolStillTestsOnlyChatEndpoint(t *testing.T) { + account := adaptiveCNAccountTestAccount(304, PlatformZhipu) + account.Credentials["api_protocol"] = APIProtocolChatCompletions + account.Credentials["base_url"] = "http://fixed-chat.example/v1" + svc, upstream := adaptiveCNAccountTestService(account, adaptiveCNChatTestResponse()) + c, recorder := newTestContext() + + err := svc.TestAccountConnection(c, account.ID, "glm-4.7", "", AccountTestModeDefault) + + require.NoError(t, err) + require.Len(t, upstream.requests, 1) + require.Equal(t, "http://fixed-chat.example/v1/chat/completions", upstream.requests[0].URL.String()) + require.Equal(t, 1, strings.Count(recorder.Body.String(), `"type":"test_complete"`)) +} diff --git a/backend/internal/service/adaptive_api_protocol_test.go b/backend/internal/service/adaptive_api_protocol_test.go index 1da65203c6..210caa0695 100644 --- a/backend/internal/service/adaptive_api_protocol_test.go +++ b/backend/internal/service/adaptive_api_protocol_test.go @@ -10,6 +10,7 @@ import ( "net/http/httptest" "testing" + "github.com/Wei-Shaw/sub2api/internal/pkg/openai_compat" "github.com/gin-gonic/gin" "github.com/stretchr/testify/require" "github.com/tidwall/gjson" @@ -39,6 +40,45 @@ func adaptiveProtocolTestContext(path string, body []byte) *gin.Context { return c } +type cnProtocolIngressCase struct { + name string + path string + body []byte + forward func(*OpenAIGatewayService, *gin.Context, *Account, []byte) error +} + +func cnProtocolIngressCases() []cnProtocolIngressCase { + return []cnProtocolIngressCase{ + { + name: "chat completions", + path: "/v1/chat/completions", + body: []byte(`{"model":"deepseek-chat","messages":[{"role":"user","content":"hello"}],"stream":false}`), + forward: func(svc *OpenAIGatewayService, c *gin.Context, account *Account, body []byte) error { + _, err := svc.ForwardAsChatCompletions(context.Background(), c, account, body, "", "") + return err + }, + }, + { + name: "messages", + path: "/v1/messages", + body: []byte(`{"model":"deepseek-chat","max_tokens":32,"messages":[{"role":"user","content":"hello"}],"stream":false}`), + forward: func(svc *OpenAIGatewayService, c *gin.Context, account *Account, body []byte) error { + _, err := svc.ForwardAsAnthropic(context.Background(), c, account, body, "", "") + return err + }, + }, + { + name: "responses", + path: "/v1/responses", + body: []byte(`{"model":"deepseek-chat","input":"hello","stream":false}`), + forward: func(svc *OpenAIGatewayService, c *gin.Context, account *Account, body []byte) error { + _, err := svc.Forward(context.Background(), c, account, body) + return err + }, + }, + } +} + func TestAdaptiveProtocolRoutesChatCompletionsToNativeChat(t *testing.T) { gin.SetMode(gin.TestMode) body := []byte(`{"model":"glm-4.7","messages":[{"role":"user","content":"hello"}],"stream":false}`) @@ -56,6 +96,41 @@ func TestAdaptiveProtocolRoutesChatCompletionsToNativeChat(t *testing.T) { require.False(t, gjson.GetBytes(upstream.lastBody, "input").Exists()) } +func TestAdaptiveProtocolRoutesResponsesShapedChatToNativeResponses(t *testing.T) { + gin.SetMode(gin.TestMode) + body := []byte(`{"model":"deepseek-v4","input":"hello","max_output_tokens":32,"stream":false}`) + upstream := &httpUpstreamRecorder{err: errors.New("stop after capture")} + svc := &OpenAIGatewayService{cfg: rawChatCompletionsTestConfig(), httpUpstream: upstream} + account := adaptiveProtocolTestAccount(PlatformDeepseek, map[string]any{ + APIProtocolChatCompletions: "http://chat.example", + APIProtocolAnthropic: "http://anthropic.example", + APIProtocolResponses: "http://responses.example", + }) + + _, err := svc.ForwardAsChatCompletions(context.Background(), adaptiveProtocolTestContext("/v1/chat/completions", body), account, body, "", "") + require.Error(t, err) + require.Equal(t, "http://responses.example/responses", upstream.lastReq.URL.String()) + require.True(t, gjson.GetBytes(upstream.lastBody, "input").Exists()) + require.False(t, gjson.GetBytes(upstream.lastBody, "messages").Exists()) +} + +func TestAdaptiveProtocolConvertsResponsesShapedChatForChatOnlyProvider(t *testing.T) { + gin.SetMode(gin.TestMode) + body := []byte(`{"model":"kimi-k2.5","input":"hello","max_output_tokens":32,"stream":false}`) + upstream := &httpUpstreamRecorder{err: errors.New("stop after capture")} + svc := &OpenAIGatewayService{cfg: rawChatCompletionsTestConfig(), httpUpstream: upstream} + account := adaptiveProtocolTestAccount(PlatformKimi, map[string]any{ + APIProtocolChatCompletions: "http://chat.example", + APIProtocolAnthropic: "http://anthropic.example", + }) + + _, err := svc.ForwardAsChatCompletions(context.Background(), adaptiveProtocolTestContext("/v1/chat/completions", body), account, body, "", "") + require.Error(t, err) + require.Equal(t, "http://chat.example/v1/chat/completions", upstream.lastReq.URL.String()) + require.True(t, gjson.GetBytes(upstream.lastBody, "messages").IsArray()) + require.False(t, gjson.GetBytes(upstream.lastBody, "input").Exists()) +} + func TestAdaptiveProtocolRoutesMessagesToNativeAnthropic(t *testing.T) { gin.SetMode(gin.TestMode) body := []byte(`{"model":"glm-4.7","max_tokens":32,"messages":[{"role":"user","content":"hello"}],"stream":false}`) @@ -91,7 +166,7 @@ func TestAdaptiveProtocolConvertsKimiResponsesToChatCompletions(t *testing.T) { func TestAdaptiveProtocolRoutesDeepSeekResponsesToNativeResponses(t *testing.T) { gin.SetMode(gin.TestMode) - body := []byte(`{"model":"deepseek-v4","input":"hello","store":true,"previous_response_id":"resp_old","stream":false}`) + body := []byte(`{"model":"deepseek-v4","input":"hello","max_output_tokens":32,"store":true,"previous_response_id":"resp_old","stream":false}`) upstream := &httpUpstreamRecorder{err: errors.New("stop after capture")} svc := &OpenAIGatewayService{cfg: rawChatCompletionsTestConfig(), httpUpstream: upstream} account := adaptiveProtocolTestAccount(PlatformDeepseek, map[string]any{ @@ -105,20 +180,48 @@ func TestAdaptiveProtocolRoutesDeepSeekResponsesToNativeResponses(t *testing.T) require.Equal(t, "http://responses.example/responses", upstream.lastReq.URL.String()) require.False(t, gjson.GetBytes(upstream.lastBody, "store").Bool()) require.False(t, gjson.GetBytes(upstream.lastBody, "previous_response_id").Exists()) + require.Equal(t, int64(32), gjson.GetBytes(upstream.lastBody, "max_output_tokens").Int()) + require.False(t, gjson.GetBytes(upstream.lastBody, "instructions").Exists()) } -func TestAdaptiveProtocolConvertsDeepSeekResponsesCompactToChatCompletions(t *testing.T) { +func TestFixedCNChatProtocolOverridesStaleResponsesMode(t *testing.T) { gin.SetMode(gin.TestMode) - body := []byte(`{"model":"deepseek-v4","input":"hello","stream":false}`) - upstream := &httpUpstreamRecorder{err: errors.New("stop after capture")} - svc := &OpenAIGatewayService{cfg: rawChatCompletionsTestConfig(), httpUpstream: upstream} - account := adaptiveProtocolTestAccount(PlatformDeepseek, map[string]any{ - APIProtocolChatCompletions: "http://chat.example", - APIProtocolAnthropic: "http://anthropic.example", - APIProtocolResponses: "http://responses.example", - }) + for _, tc := range cnProtocolIngressCases() { + t.Run(tc.name, func(t *testing.T) { + upstream := &httpUpstreamRecorder{err: errors.New("stop after capture")} + svc := &OpenAIGatewayService{cfg: rawChatCompletionsTestConfig(), httpUpstream: upstream} + account := adaptiveProtocolTestAccount(PlatformDeepseek, nil) + account.Credentials["api_protocol"] = APIProtocolChatCompletions + account.Credentials["base_url"] = "http://chat.example" + account.Extra = map[string]any{ + openai_compat.ExtraKeyResponsesMode: string(openai_compat.ResponsesSupportModeForceResponses), + } - _, err := svc.Forward(context.Background(), adaptiveProtocolTestContext("/v1/responses/compact", body), account, body) - require.Error(t, err) - require.Equal(t, "http://chat.example/v1/chat/completions", upstream.lastReq.URL.String()) + err := tc.forward(svc, adaptiveProtocolTestContext(tc.path, tc.body), account, tc.body) + + require.Error(t, err) + require.Equal(t, "http://chat.example/v1/chat/completions", upstream.lastReq.URL.String()) + }) + } +} + +func TestFixedCNResponsesProtocolOverridesStaleChatMode(t *testing.T) { + gin.SetMode(gin.TestMode) + for _, tc := range cnProtocolIngressCases() { + t.Run(tc.name, func(t *testing.T) { + upstream := &httpUpstreamRecorder{err: errors.New("stop after capture")} + svc := &OpenAIGatewayService{cfg: rawChatCompletionsTestConfig(), httpUpstream: upstream} + account := adaptiveProtocolTestAccount(PlatformDeepseek, nil) + account.Credentials["api_protocol"] = APIProtocolResponses + account.Credentials["base_url"] = "http://responses.example" + account.Extra = map[string]any{ + openai_compat.ExtraKeyResponsesMode: string(openai_compat.ResponsesSupportModeForceChatCompletions), + } + + err := tc.forward(svc, adaptiveProtocolTestContext(tc.path, tc.body), account, tc.body) + + require.Error(t, err) + require.Equal(t, "http://responses.example/responses", upstream.lastReq.URL.String()) + }) + } } diff --git a/backend/internal/service/openai_apikey_responses_probe.go b/backend/internal/service/openai_apikey_responses_probe.go index 43ad7c82d7..329cfebad4 100644 --- a/backend/internal/service/openai_apikey_responses_probe.go +++ b/backend/internal/service/openai_apikey_responses_probe.go @@ -125,8 +125,8 @@ func (s *AccountTestService) ProbeOpenAIAPIKeyResponsesSupport(ctx context.Conte if account.IsCNProvider() { // 国产 OpenAI 兼容上游（kimi/zhipu/deepseek）普遍仅支持 /v1/chat/completions， // 不存在 /v1/responses 端点。直接落标 false 走 Chat Completions 直转，跳过网络探测。 - // 例外：deepseek 的 responses 协议账号（api_protocol=responses）使用官方原生 - // /responses 端点，落标 force_responses 强制走 Responses 路径。 + // 例外：deepseek 的固定 responses 和 adaptive 账号使用官方原生 /responses + // 端点，落标 force_responses；其余协议显式重置为 auto，避免切换后残留强制模式。 if account.GetAPIProtocol() == APIProtocolResponses || (account.Platform == PlatformDeepseek && account.IsAdaptiveAPIProtocol()) { _ = s.accountRepo.UpdateExtra(ctx, account.ID, map[string]any{ @@ -136,6 +136,7 @@ func (s *AccountTestService) ProbeOpenAIAPIKeyResponsesSupport(ctx context.Conte return } _ = s.accountRepo.UpdateExtra(ctx, account.ID, map[string]any{ + openai_compat.ExtraKeyResponsesMode: string(openai_compat.ResponsesSupportModeAuto), openai_compat.ExtraKeyResponsesSupported: false, }) return diff --git a/backend/internal/service/openai_apikey_responses_probe_test.go b/backend/internal/service/openai_apikey_responses_probe_test.go index 1b9cdd65e6..f3b4eb0e6d 100644 --- a/backend/internal/service/openai_apikey_responses_probe_test.go +++ b/backend/internal/service/openai_apikey_responses_probe_test.go @@ -49,17 +49,19 @@ func TestProbeOpenAIAPIKeyResponsesSupportUsesCodexProbeHeaders(t *testing.T) { require.Equal(t, true, updates[openai_compat.ExtraKeyResponsesSupported]) } -func TestProbeOpenAIAPIKeyResponsesSupportAdaptiveCNProviders(t *testing.T) { +func TestProbeOpenAIAPIKeyResponsesSupportCNProviders(t *testing.T) { tests := []struct { - name string - id int64 - platform string - wantSupport bool - wantForceMode bool + name string + id int64 + platform string + protocol string + wantSupport bool + wantMode string }{ - {name: "deepseek adaptive supports responses", id: 201, platform: PlatformDeepseek, wantSupport: true, wantForceMode: true}, - {name: "kimi adaptive falls back to chat", id: 202, platform: PlatformKimi, wantSupport: false}, - {name: "zhipu adaptive falls back to chat", id: 203, platform: PlatformZhipu, wantSupport: false}, + {name: "deepseek adaptive supports responses", id: 201, platform: PlatformDeepseek, protocol: APIProtocolAdaptive, wantSupport: true, wantMode: string(openai_compat.ResponsesSupportModeForceResponses)}, + {name: "deepseek chat clears forced responses", id: 202, platform: PlatformDeepseek, protocol: APIProtocolChatCompletions, wantSupport: false, wantMode: string(openai_compat.ResponsesSupportModeAuto)}, + {name: "kimi adaptive falls back to chat", id: 203, platform: PlatformKimi, protocol: APIProtocolAdaptive, wantSupport: false, wantMode: string(openai_compat.ResponsesSupportModeAuto)}, + {name: "zhipu adaptive falls back to chat", id: 204, platform: PlatformZhipu, protocol: APIProtocolAdaptive, wantSupport: false, wantMode: string(openai_compat.ResponsesSupportModeAuto)}, } for _, tc := range tests { @@ -67,7 +69,10 @@ func TestProbeOpenAIAPIKeyResponsesSupportAdaptiveCNProviders(t *testing.T) { updateCalls := make(chan map[string]any, 1) account := Account{ ID: tc.id, Platform: tc.platform, Type: AccountTypeAPIKey, - Credentials: map[string]any{"api_key": "sk-test", "api_protocol": APIProtocolAdaptive}, + Credentials: map[string]any{"api_key": "sk-test", "api_protocol": tc.protocol}, + Extra: map[string]any{ + openai_compat.ExtraKeyResponsesMode: string(openai_compat.ResponsesSupportModeForceResponses), + }, } repo := &snapshotUpdateAccountRepo{ stubOpenAIAccountRepo: stubOpenAIAccountRepo{accounts: []Account{account}}, @@ -79,9 +84,7 @@ func TestProbeOpenAIAPIKeyResponsesSupportAdaptiveCNProviders(t *testing.T) { updates := <-updateCalls require.Equal(t, tc.wantSupport, updates[openai_compat.ExtraKeyResponsesSupported]) - if tc.wantForceMode { - require.Equal(t, string(openai_compat.ResponsesSupportModeForceResponses), updates[openai_compat.ExtraKeyResponsesMode]) - } + require.Equal(t, tc.wantMode, updates[openai_compat.ExtraKeyResponsesMode]) }) } } diff --git a/backend/internal/service/openai_gateway_chat_completions.go b/backend/internal/service/openai_gateway_chat_completions.go index 5dac6e9268..9c413f0bc6 100644 --- a/backend/internal/service/openai_gateway_chat_completions.go +++ b/backend/internal/service/openai_gateway_chat_completions.go @@ -46,11 +46,11 @@ var cursorResponsesUnsupportedFields = []string{ // 正确的，但 sub2api 接入 DeepSeek/Kimi/GLM 等第三方 OpenAI 兼容上游后假设破裂： // 这些上游普遍只支持 /v1/chat/completions，无 /v1/responses 端点。 // -// 当前路由策略（基于账号覆盖模式/探测标记，详见 openai_compat.ShouldUseResponsesAPI）： -// - APIKey 账号 + 强制或探测确认不支持 Responses → 走 forwardAsRawChatCompletions -// 直转上游 /v1/chat/completions，不做协议转换 -// - 其他所有情况（OAuth、APIKey 强制/探测确认支持、未探测）→ 走原有 CC→Responses -// 转换路径（保留旧行为，存量未探测账号零兼容破坏） +// 当前路由策略： +// - CN 账号以 credentials.api_protocol 为权威；adaptive/chat_completions 入站 Chat +// 直转原生 CC，anthropic 走原生 Anthropic，responses 走 Responses +// - 其他 APIKey 账号仍按覆盖模式/探测标记分流（详见 +// openai_compat.ShouldUseResponsesAPI） func (s *OpenAIGatewayService) ForwardAsChatCompletions( ctx context.Context, c *gin.Context, @@ -88,9 +88,37 @@ func (s *OpenAIGatewayService) ForwardAsChatCompletions( return s.forwardAsRawChatCompletions(ctx, c, account, body, defaultMappedModel) } - // 自适应账号的 Chat Completions 入站始终使用供应商原生 CC 端点。 + // Cursor compatibility: some clients send a Responses-shaped body to the + // /v1/chat/completions URL. Detect it before adaptive routing so adaptive + // accounts never forward the body unchanged to a Chat Completions endpoint. + isResponsesShape := !gjson.GetBytes(body, "messages").Exists() && gjson.GetBytes(body, "input").Exists() + + // 自适应账号的标准 Chat Completions 入站使用供应商原生 CC 端点。 + // Responses 形状下，DeepSeek 继续走下方原生 Responses 链；Kimi/GLM + // 没有 Responses 端点，先转换成 Chat Completions 再直转。 if account.IsAdaptiveAPIProtocol() { - return s.forwardAsRawChatCompletions(ctx, c, account, body, defaultMappedModel) + if !isResponsesShape { + return s.forwardAsRawChatCompletions(ctx, c, account, body, defaultMappedModel) + } + if account.Platform != PlatformDeepseek { + var responsesReq apicompat.ResponsesRequest + if err := json.Unmarshal(body, &responsesReq); err != nil { + return nil, fmt.Errorf("parse responses-shaped chat completions request: %w", err) + } + chatReq, err := apicompat.ResponsesToChatCompletionsRequestWithOptions( + &responsesReq, + &apicompat.ResponsesToChatOptions{ReasoningContentByID: s.reasoningContentByID}, + ) + if err != nil { + return nil, fmt.Errorf("convert responses-shaped chat completions request: %w", err) + } + chatBody, err := json.Marshal(chatReq) + if err != nil { + return nil, fmt.Errorf("marshal converted chat completions request: %w", err) + } + return s.forwardAsRawChatCompletions(ctx, c, account, chatBody, defaultMappedModel) + } + // DeepSeek 原生 Responses 请求继续走下方 Responses→Chat 回程转换。 } // 入口分流（国产供应商 Anthropic 协议）：上游为供应商原生 Anthropic 端点， @@ -101,9 +129,9 @@ func (s *OpenAIGatewayService) ForwardAsChatCompletions( return s.forwardChatCompletionsViaNativeAnthropic(ctx, c, account, body, defaultMappedModel) } - // 入口分流：APIKey 账号 + 强制或已探测确认上游不支持 Responses，走 CC 直转。 - // 自动模式下标记缺失（未探测）按"现状即证据"原则继续走下方原 Responses 转换路径。 - if account.Type == AccountTypeAPIKey && !openai_compat.ShouldUseResponsesAPI(account.Extra) { + // 固定 chat_completions 的 CN 账号，以及强制或已探测确认不支持 Responses + // 的其他 APIKey 账号，均走 CC 直转。 + if shouldForwardOpenAIResponsesViaRawChatCompletions(account) { return s.forwardAsRawChatCompletions(ctx, c, account, body, defaultMappedModel) } @@ -138,11 +166,9 @@ func (s *OpenAIGatewayService) ForwardAsChatCompletions( // and produce `input: null`, which Codex upstreams reject with // "Invalid type for 'input': expected a string, but got an object". // - // Detect that shape and forward the raw body as-is, only rewriting `model` + // Forward that shape as-is, only rewriting `model` // to the resolved upstream model. The downstream codex OAuth transform will // still normalize store/stream/instructions/etc. - isResponsesShape := !gjson.GetBytes(body, "messages").Exists() && gjson.GetBytes(body, "input").Exists() - var ( responsesReq *apicompat.ResponsesRequest responsesBody []byte diff --git a/backend/internal/service/openai_gateway_forward.go b/backend/internal/service/openai_gateway_forward.go index 415164cc8e..ec5fafdb72 100644 --- a/backend/internal/service/openai_gateway_forward.go +++ b/backend/internal/service/openai_gateway_forward.go @@ -104,6 +104,8 @@ func (s *OpenAIGatewayService) Forward(ctx context.Context, c *gin.Context, acco requestView := newOpenAIRequestView(body) reqModel, reqStream, promptCacheKey := requestView.Model, requestView.Stream, requestView.PromptCacheKey originalModel := reqModel + nativeDeepSeekResponses := account.Platform == PlatformDeepseek && + (account.GetAPIProtocol() == APIProtocolResponses || account.IsAdaptiveAPIProtocol()) if account.Platform == PlatformGrok { return s.forwardGrokResponses(ctx, c, account, body, originalModel, reqStream, startTime) @@ -115,10 +117,6 @@ func (s *OpenAIGatewayService) Forward(ctx context.Context, c *gin.Context, acco if account.IsAnthropicProtocol() { return s.forwardResponsesViaNativeAnthropic(ctx, c, account, body, reqModel) } - if account.IsAdaptiveAPIProtocol() && - (account.Platform != PlatformDeepseek || isOpenAIResponsesCompactPath(c)) { - return s.forwardResponsesViaRawChatCompletions(ctx, c, account, body) - } if shouldForwardOpenAIResponsesViaRawChatCompletions(account) { return s.forwardResponsesViaRawChatCompletions(ctx, c, account, body) @@ -290,7 +288,7 @@ func (s *OpenAIGatewayService) Forward(ctx context.Context, c *gin.Context, acco instructions := gjson.GetBytes(body, "instructions") instructionsEmpty := !instructions.Exists() || instructions.Type != gjson.String || strings.TrimSpace(instructions.String()) == "" - if instructionsEmpty && !compatMessagesBridge { + if instructionsEmpty && !compatMessagesBridge && !nativeDeepSeekResponses { markPatchSet("instructions", defaultCodexSynthInstructions(reqModel)) } @@ -463,7 +461,7 @@ func (s *OpenAIGatewayService) Forward(ctx context.Context, c *gin.Context, acco maxOutputTokens := gjson.GetBytes(body, "max_output_tokens") if maxOutputTokens.Exists() { switch account.Platform { - case PlatformOpenAI: + case PlatformOpenAI, PlatformDeepseek: // Preserve Responses-native output limits unless the selected upstream // explicitly rejects the field in the bounded HTTP retry loop below. case PlatformAnthropic: @@ -1044,9 +1042,22 @@ func (s *OpenAIGatewayService) Forward(ctx context.Context, c *gin.Context, acco } func shouldForwardOpenAIResponsesViaRawChatCompletions(account *Account) bool { - return account != nil && - account.Type == AccountTypeAPIKey && - !openai_compat.ShouldUseResponsesAPI(account.Extra) + if account == nil || account.Type != AccountTypeAPIKey { + return false + } + if account.IsCNProvider() { + // CN 的显式协议配置优先于异步探针 Extra；adaptive 仅 DeepSeek 有原生 + // Responses，Kimi/GLM 回退 Chat Completions。 + switch account.GetAPIProtocol() { + case APIProtocolChatCompletions: + return true + case APIProtocolAdaptive: + return account.Platform != PlatformDeepseek + default: + return false + } + } + return !openai_compat.ShouldUseResponsesAPI(account.Extra) } func (s *OpenAIGatewayService) buildUpstreamRequest(ctx context.Context, c *gin.Context, account *Account, body []byte, token string, isStream bool, promptCacheKey string, isCodexCLI bool) (*http.Request, error) { diff --git a/backend/internal/service/openai_gateway_messages.go b/backend/internal/service/openai_gateway_messages.go index 99c74bedd2..f1ae729ad1 100644 --- a/backend/internal/service/openai_gateway_messages.go +++ b/backend/internal/service/openai_gateway_messages.go @@ -15,7 +15,6 @@ import ( "github.com/Wei-Shaw/sub2api/internal/pkg/apicompat" "github.com/Wei-Shaw/sub2api/internal/pkg/claude" "github.com/Wei-Shaw/sub2api/internal/pkg/logger" - "github.com/Wei-Shaw/sub2api/internal/pkg/openai_compat" "github.com/Wei-Shaw/sub2api/internal/util/responseheaders" "github.com/gin-gonic/gin" "github.com/tidwall/gjson" @@ -45,11 +44,9 @@ func (s *OpenAIGatewayService) ForwardAsAnthropic( return s.forwardAnthropicViaNativeAnthropicEndpoint(ctx, c, account, body, defaultMappedModel) } - // 入口分流：APIKey 账号 + 上游不支持 Responses API → 走 CC 直转（与 - // ForwardAsChatCompletions 对称）。缺少此分流时，/v1/messages 入站请求 - // 会被无条件转为 Responses 格式发往上游 /v1/responses，导致只支持 - // /v1/chat/completions 的第三方 OpenAI 兼容上游全部 400。 - if account.Type == AccountTypeAPIKey && !openai_compat.ShouldUseResponsesAPI(account.Extra) { + // 固定 chat_completions 的 CN 账号，以及不支持 Responses 的其他 APIKey + // 账号，均将 Messages 转为 CC；固定 responses 的 CN 账号不受探针旧值覆盖。 + if shouldForwardOpenAIResponsesViaRawChatCompletions(account) { return s.forwardAnthropicViaRawChatCompletions(ctx, c, account, body, defaultMappedModel) } diff --git a/backend/internal/service/upstream_billing_probe.go b/backend/internal/service/upstream_billing_probe.go index c6a62bab09..7255464c03 100644 --- a/backend/internal/service/upstream_billing_probe.go +++ b/backend/internal/service/upstream_billing_probe.go @@ -597,6 +597,9 @@ func (s *UpstreamBillingProbeService) probeLoadedAccount(ctx context.Context, ac return s.persistProbeFailure(ctx, account, intervalMinutes, now, 0, "missing_api_key", 0) } baseURL := account.GetCredential("base_url") + if account.IsCNProvider() && account.IsAdaptiveAPIProtocol() { + baseURL = account.GetCNProtocolBaseURL(APIProtocolChatCompletions) + } if account.Platform == PlatformOpenAI { if baseURL == "" { // 保持官方语义：OpenAI 账号无自定义 base 时探官方域（404 → unsupported）。 diff --git a/backend/internal/service/upstream_billing_probe_test.go b/backend/internal/service/upstream_billing_probe_test.go index e032e9aa16..9a5f774f4f 100644 --- a/backend/internal/service/upstream_billing_probe_test.go +++ b/backend/internal/service/upstream_billing_probe_test.go @@ -340,6 +340,38 @@ func TestUpstreamBillingProbeSuccessPersistsSanitizedSnapshot(t *testing.T) { require.Equal(t, snapshot.Status, persisted.Status) } +func TestUpstreamBillingProbeAdaptiveCNUsesChatProtocolBaseURL(t *testing.T) { + account := &Account{ + ID: 18, + Platform: PlatformKimi, + Type: AccountTypeAPIKey, + Status: StatusActive, + Concurrency: 1, + Credentials: map[string]any{ + "api_key": "sk-sensitive", + "api_protocol": APIProtocolAdaptive, + "base_url": "https://legacy-relay.example/v1", + "api_base_urls": map[string]any{ + APIProtocolChatCompletions: "https://chat-relay.example/v1", + }, + }, + Extra: map[string]any{UpstreamBillingProbeEnabledExtraKey: true}, + } + repo := &upstreamBillingProbeAccountRepo{accounts: map[int64]*Account{account.ID: account}} + upstream := &httpUpstreamRecorder{resp: &http.Response{ + StatusCode: http.StatusOK, + Header: http.Header{"Content-Type": []string{"application/json"}}, + Body: upstreamBillingProbeValidBody(), + }} + svc := newUpstreamBillingProbeTestService(repo, upstream, &upstreamBillingProbeSettingRepo{}) + + snapshot, err := svc.ProbeAccount(context.Background(), account.ID) + + require.NoError(t, err) + require.Equal(t, UpstreamBillingProbeStatusOK, snapshot.Status) + require.Equal(t, "https://chat-relay.example/v1/sub2api/billing", upstream.lastReq.URL.String()) +} + func TestUpstreamBillingProbeSyncsResolvedRateForAllAPIKeyPlatforms(t *testing.T) { for _, platform := range []string{ PlatformOpenAI, diff --git a/frontend/src/components/account/CreateAccountModal.vue b/frontend/src/components/account/CreateAccountModal.vue index 546c5f0066..2c9621f17f 100644 --- a/frontend/src/components/account/CreateAccountModal.vue +++ b/frontend/src/components/account/CreateAccountModal.vue @@ -1281,7 +1281,12 @@ - +

@@ -4064,10 +4069,13 @@ function onCnPresetSelect(preset: { mode: CnAccountMode; protocol: CnApiProtocol const syncPreviewCredentials = computed(() => { if (!apiKeyValue.value) return undefined + const baseUrl = isCNPlatform.value && apiProtocol.value === 'adaptive' + ? adaptiveBaseUrls.value.chat_completions.trim() || apiKeyBaseUrl.value.trim() + : apiKeyBaseUrl.value.trim() return { platform: form.platform, type: form.type, - base_url: apiKeyBaseUrl.value || undefined, + base_url: baseUrl || undefined, api_key: apiKeyValue.value } }) diff --git a/frontend/src/components/account/EditAccountModal.vue b/frontend/src/components/account/EditAccountModal.vue index 0f433cd53a..d1cb92df59 100644 --- a/frontend/src/components/account/EditAccountModal.vue +++ b/frontend/src/components/account/EditAccountModal.vue @@ -3831,27 +3831,48 @@ const syncFormFromAccount = (newAccount: Account | null) => { editAccountMode.value = credentials.account_mode === 'coding' ? 'coding' : 'payg' const storedProtocol = credentials.api_protocol editApiProtocol.value = - storedProtocol === 'adaptive' || storedProtocol === 'anthropic' || storedProtocol === 'responses' + storedProtocol === 'adaptive' || + storedProtocol === 'chat_completions' || + storedProtocol === 'anthropic' || + storedProtocol === 'responses' ? storedProtocol - : 'adaptive' + : 'chat_completions' if (newAccount.platform !== 'deepseek' && editApiProtocol.value === 'responses') { editApiProtocol.value = 'chat_completions' } const adaptiveDefaults = defaultCNAdaptiveBaseUrls(newAccount.platform, editAccountMode.value) const storedBaseUrls = (credentials.api_base_urls as Record | undefined) || {} - editAdaptiveBaseUrls.value = { - chat_completions: typeof storedBaseUrls.chat_completions === 'string' - ? storedBaseUrls.chat_completions - : adaptiveDefaults.chat_completions, - anthropic: typeof storedBaseUrls.anthropic === 'string' - ? storedBaseUrls.anthropic - : adaptiveDefaults.anthropic, - responses: typeof storedBaseUrls.responses === 'string' - ? storedBaseUrls.responses - : adaptiveDefaults.responses + const legacyBaseUrl = typeof credentials.base_url === 'string' ? credentials.base_url.trim() : '' + const storedChatBaseUrl = typeof storedBaseUrls.chat_completions === 'string' + ? storedBaseUrls.chat_completions.trim() + : '' + const storedAnthropicBaseUrl = typeof storedBaseUrls.anthropic === 'string' + ? storedBaseUrls.anthropic.trim() + : '' + const storedResponsesBaseUrl = typeof storedBaseUrls.responses === 'string' + ? storedBaseUrls.responses.trim() + : '' + const nextAdaptiveBaseUrls: Record = { + chat_completions: storedChatBaseUrl || adaptiveDefaults.chat_completions, + anthropic: storedAnthropicBaseUrl || adaptiveDefaults.anthropic, + responses: storedResponsesBaseUrl || adaptiveDefaults.responses } + const legacyProtocol: CnNativeApiProtocol = editApiProtocol.value === 'anthropic' + ? 'anthropic' + : editApiProtocol.value === 'responses' + ? 'responses' + : 'chat_completions' + const storedLegacyBaseUrl = legacyProtocol === 'anthropic' + ? storedAnthropicBaseUrl + : legacyProtocol === 'responses' + ? storedResponsesBaseUrl + : storedChatBaseUrl + if (legacyBaseUrl && !storedLegacyBaseUrl) { + nextAdaptiveBaseUrls[legacyProtocol] = legacyBaseUrl + } + editAdaptiveBaseUrls.value = nextAdaptiveBaseUrls } - const platformDefaultUrl = + const platformDefaultUrl = newAccount.platform === 'openai' ? 'https://api.openai.com' : newAccount.platform === 'gemini' @@ -3863,9 +3884,9 @@ const syncFormFromAccount = (newAccount: Account | null) => { newAccount.platform === 'deepseek' ? defaultCNBaseUrl(newAccount.platform, editAccountMode.value, editApiProtocol.value) : 'https://api.anthropic.com' - editBaseUrl.value = isCNApiKeyAccount.value && editApiProtocol.value === 'adaptive' - ? editAdaptiveBaseUrls.value.chat_completions - : (credentials.base_url as string) || platformDefaultUrl + editBaseUrl.value = isCNApiKeyAccount.value && editApiProtocol.value === 'adaptive' + ? editAdaptiveBaseUrls.value.chat_completions + : (credentials.base_url as string) || platformDefaultUrl // Load model mappings and detect mode loadModelRestrictionFromMapping(credentials.model_mapping as Record | undefined) diff --git a/frontend/src/components/account/__tests__/CreateAccountModal.spec.ts b/frontend/src/components/account/__tests__/CreateAccountModal.spec.ts index 2681958d9a..e5a0c8c9c1 100644 --- a/frontend/src/components/account/__tests__/CreateAccountModal.spec.ts +++ b/frontend/src/components/account/__tests__/CreateAccountModal.spec.ts @@ -110,6 +110,20 @@ const GroupSelectorStub = defineComponent({ `, }) +const ModelWhitelistSelectorStub = defineComponent({ + name: 'ModelWhitelistSelector', + props: { + modelValue: { + type: Array, + default: () => [], + }, + platform: String, + syncCredentials: Object, + }, + emits: ['update:modelValue'], + template: '

', +}) + function mountModal(groups: any[] = []) { return mount(CreateAccountModal, { props: { show: true, proxies: [], groups }, @@ -124,7 +138,7 @@ function mountModal(groups: any[] = []) { ProxySelector: true, ProxyAdBanner: true, GroupSelector: GroupSelectorStub, - ModelWhitelistSelector: true, + ModelWhitelistSelector: ModelWhitelistSelectorStub, QuotaLimitCard: true, }, }, @@ -272,7 +286,6 @@ describe('CreateAccountModal OpenAI long-context billing', () => { it('submits adaptive Kimi protocol endpoints', async () => { const wrapper = mountModal() await selectButtonByText(wrapper, 'Kimi') - await selectButtonByText(wrapper, 'admin.accounts.cnProviders.apiProtocol.adaptive') await wrapper.get('form#create-account-form input[type="text"]').setValue('Kimi adaptive') await wrapper.get('form#create-account-form input[type="password"]').setValue('sk-kimi') @@ -291,6 +304,22 @@ describe('CreateAccountModal OpenAI long-context billing', () => { }) }) + it('uses the edited adaptive Chat endpoint when previewing upstream models', async () => { + const wrapper = mountModal() + await selectButtonByText(wrapper, 'Kimi') + await wrapper + .get('[data-testid="cn-adaptive-base-url-chat_completions"]') + .setValue('https://relay.example.com/v1') + await wrapper.get('form#create-account-form input[type="password"]').setValue('sk-relay') + + expect(wrapper.getComponent(ModelWhitelistSelectorStub).props('syncCredentials')).toMatchObject({ + platform: 'kimi', + type: 'apikey', + base_url: 'https://relay.example.com/v1', + api_key: 'sk-relay' + }) + }) + it('exposes Agent Identity in the OpenAI authorization methods', async () => { const wrapper = mountModal() await selectButtonByText(wrapper, 'OpenAI') diff --git a/frontend/src/components/account/__tests__/EditAccountModal.spec.ts b/frontend/src/components/account/__tests__/EditAccountModal.spec.ts index 2e82bb4542..acd0436f6a 100644 --- a/frontend/src/components/account/__tests__/EditAccountModal.spec.ts +++ b/frontend/src/components/account/__tests__/EditAccountModal.spec.ts @@ -374,6 +374,147 @@ describe('EditAccountModal', () => { }) }) + it.each([ + ['explicit Chat Completions', 'chat_completions'], + ['legacy missing protocol', undefined] + ])('preserves a custom CN relay for %s accounts', async (_name, storedProtocol) => { + const account = buildAccount() + account.platform = 'zhipu' + account.credentials = { + api_key: 'sk-glm', + account_mode: 'payg', + base_url: 'https://relay.example.com/v1' + } + if (storedProtocol) { + account.credentials.api_protocol = storedProtocol + } + updateAccountMock.mockReset().mockResolvedValue(account) + checkMixedChannelRiskMock.mockReset().mockResolvedValue({ has_risk: false }) + + const wrapper = mountModal(account) + await wrapper.get('form#edit-account-form').trigger('submit.prevent') + + expect(updateAccountMock).toHaveBeenCalledTimes(1) + const submittedCredentials = updateAccountMock.mock.calls[0]?.[1]?.credentials + expect(submittedCredentials).toMatchObject({ + account_mode: 'payg', + api_protocol: 'chat_completions', + base_url: 'https://relay.example.com/v1' + }) + expect(submittedCredentials).not.toHaveProperty('api_base_urls') + }) + + it('uses the legacy base_url when adaptive endpoints are missing', async () => { + const account = buildAccount() + account.platform = 'zhipu' + account.credentials = { + api_key: 'sk-glm', + account_mode: 'payg', + api_protocol: 'adaptive', + base_url: 'https://relay.example.com/v1', + api_base_urls: { + chat_completions: ' ' + } + } + updateAccountMock.mockReset().mockResolvedValue(account) + checkMixedChannelRiskMock.mockReset().mockResolvedValue({ has_risk: false }) + + const wrapper = mountModal(account) + await wrapper.get('form#edit-account-form').trigger('submit.prevent') + + expect(updateAccountMock).toHaveBeenCalledTimes(1) + expect(updateAccountMock.mock.calls[0]?.[1]?.credentials).toMatchObject({ + api_protocol: 'adaptive', + base_url: 'https://relay.example.com/v1', + api_base_urls: { + chat_completions: 'https://relay.example.com/v1', + anthropic: 'https://open.bigmodel.cn/api/anthropic' + } + }) + }) + + it('carries a fixed Chat relay into Adaptive when the user switches protocols', async () => { + const account = buildAccount() + account.platform = 'zhipu' + account.credentials = { + api_key: 'sk-glm', + account_mode: 'payg', + api_protocol: 'chat_completions', + base_url: 'https://relay.example.com/v1' + } + updateAccountMock.mockReset().mockResolvedValue(account) + checkMixedChannelRiskMock.mockReset().mockResolvedValue({ has_risk: false }) + + const wrapper = mountModal(account) + const adaptiveButton = wrapper + .findAll('button') + .find(button => button.text().includes('admin.accounts.cnProviders.apiProtocol.adaptive')) + expect(adaptiveButton).toBeDefined() + await adaptiveButton!.trigger('click') + await wrapper.get('form#edit-account-form').trigger('submit.prevent') + + expect(updateAccountMock).toHaveBeenCalledTimes(1) + expect(updateAccountMock.mock.calls[0]?.[1]?.credentials).toMatchObject({ + api_protocol: 'adaptive', + base_url: 'https://relay.example.com/v1', + api_base_urls: { + chat_completions: 'https://relay.example.com/v1' + } + }) + }) + + it.each([ + { + name: 'Anthropic', + platform: 'zhipu', + protocol: 'anthropic', + baseUrl: 'https://relay.example.com/anthropic', + expectedBaseUrl: 'https://open.bigmodel.cn/api/paas/v4', + expectedProtocolUrls: { + chat_completions: 'https://open.bigmodel.cn/api/paas/v4', + anthropic: 'https://relay.example.com/anthropic' + } + }, + { + name: 'Responses', + platform: 'deepseek', + protocol: 'responses', + baseUrl: 'https://relay.example.com/responses', + expectedBaseUrl: 'https://api.deepseek.com', + expectedProtocolUrls: { + chat_completions: 'https://api.deepseek.com', + anthropic: 'https://api.deepseek.com/anthropic', + responses: 'https://relay.example.com/responses' + } + } + ])('keeps a fixed $name relay in its protocol slot when switching to Adaptive', async (testCase) => { + const account = buildAccount() + account.platform = testCase.platform + account.credentials = { + api_key: 'sk-cn', + account_mode: 'payg', + api_protocol: testCase.protocol, + base_url: testCase.baseUrl + } + updateAccountMock.mockReset().mockResolvedValue(account) + checkMixedChannelRiskMock.mockReset().mockResolvedValue({ has_risk: false }) + + const wrapper = mountModal(account) + const adaptiveButton = wrapper + .findAll('button') + .find(button => button.text().includes('admin.accounts.cnProviders.apiProtocol.adaptive')) + expect(adaptiveButton).toBeDefined() + await adaptiveButton!.trigger('click') + await wrapper.get('form#edit-account-form').trigger('submit.prevent') + + expect(updateAccountMock).toHaveBeenCalledTimes(1) + expect(updateAccountMock.mock.calls[0]?.[1]?.credentials).toMatchObject({ + api_protocol: 'adaptive', + base_url: testCase.expectedBaseUrl, + api_base_urls: testCase.expectedProtocolUrls + }) + }) + it('preserves model mappings when editing the whitelist', async () => { const account = buildAccount() account.credentials.model_mapping = {