feat(api): add OAuth token fields and new provider types support

Add support for OAuth-based authentication with access/refresh tokens
and expiration tracking for API keys. Extend provider groups with
static headers configuration and headers profile options.

Changes include:
- Add AccessToken, RefreshToken, ExpiresAt, AccountID, ProjectID to APIKey model
- Add StaticHeaders and HeadersProfile to ProviderGroup model
- Add TokenRefresh configuration for background token management
- Support new provider types: ClaudeCode, Codex, GeminiCLI, Antigravity
- Update sync service to include new fields in provider snapshots
This commit is contained in:
zenfun
2025-12-28 02:49:54 +08:00
parent cca0802620
commit f0fe9f0dad
8 changed files with 132 additions and 22 deletions

View File

@@ -33,16 +33,30 @@ func (m *ProviderGroupManager) NormalizeGroup(group model.ProviderGroup) (model.
group.BaseURL = strings.TrimSpace(group.BaseURL)
group.GoogleProject = strings.TrimSpace(group.GoogleProject)
group.GoogleLocation = strings.TrimSpace(group.GoogleLocation)
group.StaticHeaders = strings.TrimSpace(group.StaticHeaders)
group.HeadersProfile = strings.TrimSpace(group.HeadersProfile)
switch ptype {
case provider.TypeOpenAI:
if group.BaseURL == "" {
group.BaseURL = "https://api.openai.com/v1"
}
case provider.TypeAnthropic, provider.TypeClaude:
case provider.TypeAnthropic, provider.TypeClaude, provider.TypeClaudeCode:
if group.BaseURL == "" {
group.BaseURL = "https://api.anthropic.com"
}
case provider.TypeCodex:
if group.BaseURL == "" {
group.BaseURL = "https://chatgpt.com"
}
case provider.TypeGeminiCLI:
if group.BaseURL == "" {
group.BaseURL = "https://cloudcode-pa.googleapis.com"
}
case provider.TypeAntigravity:
if group.BaseURL == "" {
group.BaseURL = "https://daily-cloudcode-pa.googleapis.com"
}
case provider.TypeCompatible:
if group.BaseURL == "" {
return model.ProviderGroup{}, fmt.Errorf("base_url required for compatible providers")
@@ -72,8 +86,14 @@ func (m *ProviderGroupManager) ValidateAPIKey(group model.ProviderGroup, key mod
return fmt.Errorf("provider group type required")
}
apiKey := strings.TrimSpace(key.APIKey)
accessToken := strings.TrimSpace(key.AccessToken)
switch {
case ptype == provider.TypeCodex || ptype == provider.TypeGeminiCLI || ptype == provider.TypeAntigravity || ptype == provider.TypeClaudeCode:
if accessToken == "" {
return fmt.Errorf("access_token required")
}
return nil
case provider.IsVertexFamily(ptype):
// Vertex uses ADC; api_key can be empty.
return nil