103 lines
3.2 KiB
Swift
103 lines
3.2 KiB
Swift
//
|
|
// ProviderRegistry.swift
|
|
// oAI
|
|
//
|
|
// Registry for managing multiple AI providers
|
|
//
|
|
|
|
import Foundation
|
|
import os
|
|
|
|
class ProviderRegistry {
|
|
static let shared = ProviderRegistry()
|
|
|
|
private var providers: [Settings.Provider: AIProvider] = [:]
|
|
private let settings = SettingsService.shared
|
|
|
|
private init() {}
|
|
|
|
// MARK: - Get Provider
|
|
|
|
func getProvider(for providerType: Settings.Provider) -> AIProvider? {
|
|
// Return cached provider if exists
|
|
if let provider = providers[providerType] {
|
|
return provider
|
|
}
|
|
|
|
// Create new provider based on type
|
|
let provider: AIProvider?
|
|
|
|
switch providerType {
|
|
case .openrouter:
|
|
guard let apiKey = settings.openrouterAPIKey, !apiKey.isEmpty else {
|
|
Log.api.warning("No API key configured for OpenRouter")
|
|
return nil
|
|
}
|
|
provider = OpenRouterProvider(apiKey: apiKey)
|
|
|
|
case .anthropic:
|
|
if AnthropicOAuthService.shared.isAuthenticated {
|
|
// OAuth (Pro/Max subscription) takes precedence
|
|
provider = AnthropicProvider(oauth: true)
|
|
} else if let apiKey = settings.anthropicAPIKey, !apiKey.isEmpty {
|
|
provider = AnthropicProvider(apiKey: apiKey)
|
|
} else {
|
|
Log.api.warning("No API key or OAuth configured for Anthropic")
|
|
return nil
|
|
}
|
|
|
|
case .openai:
|
|
guard let apiKey = settings.openaiAPIKey, !apiKey.isEmpty else {
|
|
Log.api.warning("No API key configured for OpenAI")
|
|
return nil
|
|
}
|
|
provider = OpenAIProvider(apiKey: apiKey)
|
|
|
|
case .ollama:
|
|
provider = OllamaProvider(baseURL: settings.ollamaEffectiveURL)
|
|
}
|
|
|
|
// Cache and return
|
|
if let provider = provider {
|
|
Log.api.info("Created \(providerType.rawValue) provider")
|
|
providers[providerType] = provider
|
|
}
|
|
|
|
return provider
|
|
}
|
|
|
|
// MARK: - Current Provider
|
|
|
|
func getCurrentProvider() -> AIProvider? {
|
|
let currentProviderType = settings.defaultProvider
|
|
return getProvider(for: currentProviderType)
|
|
}
|
|
|
|
// MARK: - Clear Cache
|
|
|
|
func clearCache() {
|
|
providers.removeAll()
|
|
}
|
|
|
|
// MARK: - Validate API Key
|
|
|
|
func hasValidAPIKey(for providerType: Settings.Provider) -> Bool {
|
|
switch providerType {
|
|
case .openrouter:
|
|
return settings.openrouterAPIKey != nil && !settings.openrouterAPIKey!.isEmpty
|
|
case .anthropic:
|
|
return AnthropicOAuthService.shared.isAuthenticated
|
|
|| (settings.anthropicAPIKey != nil && !settings.anthropicAPIKey!.isEmpty)
|
|
case .openai:
|
|
return settings.openaiAPIKey != nil && !settings.openaiAPIKey!.isEmpty
|
|
case .ollama:
|
|
return settings.ollamaConfigured
|
|
}
|
|
}
|
|
|
|
/// Providers that have credentials configured (API key or, for Ollama, a saved URL)
|
|
var configuredProviders: [Settings.Provider] {
|
|
Settings.Provider.allCases.filter { hasValidAPIKey(for: $0) }
|
|
}
|
|
}
|