Files
sub2api/backend/internal/service/anthropic_apikey_auth.go
T
李建琦 6d655c9903
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
Sub2API v1.0 - AI API 网关(二开初始版本,基于上游 Wei-Shaw/sub2api)
2026-08-21 18:30:13 +08:00

43 lines
1.4 KiB
Go

package service
import (
"net/http"
"strings"
)
const (
anthropicAPIKeyAuthSchemeExtraKey = "anthropic_apikey_auth_scheme"
AnthropicAPIKeyAuthSchemeXAPIKey = "x_api_key"
AnthropicAPIKeyAuthSchemeAuthorizationBearer = "authorization_bearer"
)
// GetAnthropicAPIKeyAuthScheme returns the upstream authentication scheme for
// Anthropic API-key accounts. Missing or invalid values keep the historical
// x-api-key behavior. CN providers using their native Anthropic endpoints
// (api_protocol=anthropic) share the same override knob — Kimi/DeepSeek default
// to x-api-key, Zhipu can opt into Authorization: Bearer.
func (a *Account) GetAnthropicAPIKeyAuthScheme() string {
if a == nil || a.Type != AccountTypeAPIKey {
return AnthropicAPIKeyAuthSchemeXAPIKey
}
if a.Platform != PlatformAnthropic && !a.IsCNProvider() {
return AnthropicAPIKeyAuthSchemeXAPIKey
}
switch strings.TrimSpace(a.GetExtraString(anthropicAPIKeyAuthSchemeExtraKey)) {
case AnthropicAPIKeyAuthSchemeAuthorizationBearer:
return AnthropicAPIKeyAuthSchemeAuthorizationBearer
default:
return AnthropicAPIKeyAuthSchemeXAPIKey
}
}
func setAnthropicAPIKeyAuthHeader(header http.Header, account *Account, token string) {
if account.GetAnthropicAPIKeyAuthScheme() == AnthropicAPIKeyAuthSchemeAuthorizationBearer {
header.Set("Authorization", "Bearer "+token)
return
}
header.Set("x-api-key", token)
}