wanwu/sub2api
1
0
Fork 0
mirror of https://gitee.com/wanwujie/sub2api synced 2026-04-22 15:44:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: 恢复context-1m-2025-08-07在oauth账号中传递

Browse Source
This commit is contained in:
shaw
2026-03-09 09:18:29 +08:00
parent 6cb8980404
commit 440d2e28ed
2 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/internal/pkg/claude/constants.go
View File

@@ -16,7 +16,7 @@ const (
// DroppedBetas 是转发时需要从 anthropic-beta header 中移除的 beta token 列表。 // DroppedBetas 是转发时需要从 anthropic-beta header 中移除的 beta token 列表。
// 这些 token 是客户端特有的，不应透传给上游 API。 // 这些 token 是客户端特有的，不应透传给上游 API。
var DroppedBetas = []string{BetaContext1M, BetaFastMode} var DroppedBetas = []string{BetaFastMode}
// DefaultBetaHeader Claude Code 客户端默认的 anthropic-beta header // DefaultBetaHeader Claude Code 客户端默认的 anthropic-beta header
const DefaultBetaHeader = BetaClaudeCode + "," + BetaOAuth + "," + BetaInterleavedThinking + "," + BetaFineGrainedToolStreaming const DefaultBetaHeader = BetaClaudeCode + "," + BetaOAuth + "," + BetaInterleavedThinking + "," + BetaFineGrainedToolStreaming

12
backend/internal/service/gateway_beta_test.go
View File

@@ -86,10 +86,10 @@ func TestStripBetaTokens(t *testing.T) {
want: "oauth-2025-04-20,interleaved-thinking-2025-05-14", want: "oauth-2025-04-20,interleaved-thinking-2025-05-14",
}, },
{ {
name: "DroppedBetas removes both context-1m and fast-mode", name: "DroppedBetas removes fast-mode only",
header: "oauth-2025-04-20,context-1m-2025-08-07,fast-mode-2026-02-01,interleaved-thinking-2025-05-14", header: "oauth-2025-04-20,context-1m-2025-08-07,fast-mode-2026-02-01,interleaved-thinking-2025-05-14",
tokens: claude.DroppedBetas, tokens: claude.DroppedBetas,
want: "oauth-2025-04-20,interleaved-thinking-2025-05-14", want: "oauth-2025-04-20,context-1m-2025-08-07,interleaved-thinking-2025-05-14",
}, },
} }
@@ -117,21 +117,21 @@ func TestMergeAnthropicBetaDropping_DroppedBetas(t *testing.T) {
drop := droppedBetaSet() drop := droppedBetaSet()
got := mergeAnthropicBetaDropping(required, incoming, drop) got := mergeAnthropicBetaDropping(required, incoming, drop)
require.Equal(t, "oauth-2025-04-20,interleaved-thinking-2025-05-14,foo-beta", got) require.Equal(t, "oauth-2025-04-20,interleaved-thinking-2025-05-14,context-1m-2025-08-07,foo-beta", got)
require.NotContains(t, got, "context-1m-2025-08-07") require.Contains(t, got, "context-1m-2025-08-07")
require.NotContains(t, got, "fast-mode-2026-02-01") require.NotContains(t, got, "fast-mode-2026-02-01")
} }
func TestDroppedBetaSet(t *testing.T) { func TestDroppedBetaSet(t *testing.T) {
// Base set contains DroppedBetas // Base set contains DroppedBetas
base := droppedBetaSet() base := droppedBetaSet()
require.Contains(t, base, claude.BetaContext1M) require.NotContains(t, base, claude.BetaContext1M)
require.Contains(t, base, claude.BetaFastMode) require.Contains(t, base, claude.BetaFastMode)
require.Len(t, base, len(claude.DroppedBetas)) require.Len(t, base, len(claude.DroppedBetas))
// With extra tokens // With extra tokens
extended := droppedBetaSet(claude.BetaClaudeCode) extended := droppedBetaSet(claude.BetaClaudeCode)
require.Contains(t, extended, claude.BetaContext1M) require.NotContains(t, extended, claude.BetaContext1M)
require.Contains(t, extended, claude.BetaFastMode) require.Contains(t, extended, claude.BetaFastMode)
require.Contains(t, extended, claude.BetaClaudeCode) require.Contains(t, extended, claude.BetaClaudeCode)
require.Len(t, extended, len(claude.DroppedBetas)+1) require.Len(t, extended, len(claude.DroppedBetas)+1)