fix: support Azure OpenAI Foundry providers

Author: bitcavern

Packages/OsaurusCore/Managers/RemoteProviderManager.swift

@@ -237,8 +237,18 @@ public final class RemoteProviderManager: ObservableObject {
                 RemoteProviderKeychain.saveOAuthTokens(refreshed, for: provider.id)
             }
 
-            // Fetch models from the provider
-            let models = try await RemoteProviderService.fetchModels(from: provider)
+            // Fetch models from the provider and merge any manually configured deployment IDs.
+            let discoveredModels: [String]
+            do {
+                discoveredModels = try await RemoteProviderService.fetchModels(from: provider)
+            } catch {
+                if provider.providerType == .azureOpenAI && !provider.manualModelIds.isEmpty {
+                    discoveredModels = []
+                } else {
+                    throw error
+                }
+            }
+            let models = provider.mergedModelIds(discovered: discoveredModels)
 
             // Create service instance – resolve headers eagerly on @MainActor
             // so the service actor never reads from Keychain off the main thread.
@@ -432,6 +442,10 @@ public final class RemoteProviderManager: ObservableObject {
                 if testHeaders["x-goog-api-key"] == nil {
                     testHeaders["x-goog-api-key"] = apiKey
                 }
+            case .azureOpenAI:
+                if testHeaders["api-key"] == nil {
+                    testHeaders["api-key"] = apiKey
+                }
             case .openaiLegacy, .openResponses, .openAICodex, .osaurus:
                 if testHeaders["Authorization"] == nil {
                     testHeaders["Authorization"] = "Bearer \(apiKey)"

Packages/OsaurusCore/Models/Configuration/ProviderPresets.swift

@@ -12,6 +12,7 @@ import SwiftUI
 /// Unified provider presets shared across onboarding and provider management.
 enum ProviderPreset: String, CaseIterable, Identifiable {
     case anthropic
+    case azureOpenAI
     case openai
     case google
     case xai
@@ -25,6 +26,7 @@ enum ProviderPreset: String, CaseIterable, Identifiable {
     var name: String {
         switch self {
         case .anthropic: return "Anthropic"
+        case .azureOpenAI: return "Azure OpenAI Foundry"
         case .openai: return "OpenAI"
         case .google: return "Google"
         case .xai: return "xAI"
@@ -38,6 +40,7 @@ enum ProviderPreset: String, CaseIterable, Identifiable {
     var description: String {
         switch self {
         case .anthropic: return "Claude models"
+        case .azureOpenAI: return "Azure deployments"
         case .openai: return "ChatGPT/Codex or Platform API"
         case .google: return "Gemini models"
         case .xai: return "Grok models"
@@ -51,6 +54,7 @@ enum ProviderPreset: String, CaseIterable, Identifiable {
     var icon: String {
         switch self {
         case .anthropic: return "brain.head.profile"
+        case .azureOpenAI: return "cloud.fill"
         case .openai: return "sparkles"
         case .google: return "globe"
         case .xai: return "bolt.fill"
@@ -64,6 +68,7 @@ enum ProviderPreset: String, CaseIterable, Identifiable {
     var gradient: [Color] {
         switch self {
         case .anthropic: return [Color(red: 0.85, green: 0.55, blue: 0.35), Color(red: 0.75, green: 0.4, blue: 0.25)]
+        case .azureOpenAI: return [Color(red: 0.0, green: 0.47, blue: 0.84), Color(red: 0.0, green: 0.62, blue: 0.72)]
         case .openai: return [Color(red: 0.0, green: 0.65, blue: 0.52), Color(red: 0.0, green: 0.5, blue: 0.4)]
         case .google: return [Color(red: 0.26, green: 0.52, blue: 0.96), Color(red: 0.18, green: 0.38, blue: 0.85)]
         case .xai: return [Color(red: 0.1, green: 0.1, blue: 0.1), Color(red: 0.2, green: 0.2, blue: 0.2)]
@@ -77,6 +82,7 @@ enum ProviderPreset: String, CaseIterable, Identifiable {
     var consoleURL: String {
         switch self {
         case .anthropic: return "https://console.anthropic.com/settings/keys"
+        case .azureOpenAI: return "https://ai.azure.com"
         case .openai: return "https://platform.openai.com/api-keys"
         case .google: return "https://aistudio.google.com/apikey"
         case .xai: return "https://console.x.ai/"
@@ -89,6 +95,7 @@ enum ProviderPreset: String, CaseIterable, Identifiable {
     /// Optional badge label (e.g. "Privacy") shown as a highlight pill on provider cards
     var badge: String? {
         switch self {
+        case .azureOpenAI: return "Azure"
         case .venice: return "Privacy"
         default: return nil
         }
@@ -97,6 +104,7 @@ enum ProviderPreset: String, CaseIterable, Identifiable {
     /// Optional documentation URL for the provider (shown in help sections)
     var documentationURL: String? {
         switch self {
+        case .azureOpenAI: return "https://learn.microsoft.com/azure/ai-foundry/openai/"
         case .venice: return "https://docs.venice.ai"
         default: return nil
         }
@@ -114,6 +122,13 @@ enum ProviderPreset: String, CaseIterable, Identifiable {
     /// Help steps shown when guiding the user to create an API key
     var helpSteps: [String] {
         switch self {
+        case .azureOpenAI:
+            return [
+                "Open your Azure OpenAI resource in Azure AI Foundry",
+                "Copy the resource endpoint host and an API key",
+                "Add deployment names if they do not appear automatically",
+                "Paste the key here",
+            ]
         case .openai:
             return [
                 "Go to the OpenAI Platform API keys page",
@@ -161,6 +176,16 @@ enum ProviderPreset: String, CaseIterable, Identifiable {
                 authType: .apiKey,
                 providerType: .anthropic
             )
+        case .azureOpenAI:
+            return ProviderPresetConfiguration(
+                name: "Azure OpenAI Foundry",
+                host: "",
+                providerProtocol: .https,
+                port: nil,
+                basePath: "/openai/v1",
+                authType: .apiKey,
+                providerType: .azureOpenAI
+            )
         case .openai:
             return ProviderPresetConfiguration(
                 name: "OpenAI",
@@ -228,9 +253,14 @@ enum ProviderPreset: String, CaseIterable, Identifiable {
 
     /// Attempts to match an existing RemoteProvider to a known preset by host.
     static func matching(provider: RemoteProvider) -> ProviderPreset? {
+        if provider.providerType == .azureOpenAI {
+            return .azureOpenAI
+        }
+
         let host = provider.host.lowercased().trimmingCharacters(in: .whitespaces)
         return knownPresets.first { preset in
-            preset.configuration.host.lowercased() == host
+            guard !preset.configuration.host.isEmpty else { return false }
+            return preset.configuration.host.lowercased() == host
         }
     }
 }

Packages/OsaurusCore/Models/Configuration/RemoteProviderConfiguration.swift

@@ -36,6 +36,7 @@ public enum RemoteProviderAuthType: String, Codable, Sendable, CaseIterable {
 /// Type of remote provider (determines API format)
 public enum RemoteProviderType: String, Codable, Sendable, CaseIterable {
     case openaiLegacy = "openai"  // OpenAI-compatible /chat/completions (third-party servers, backward compat)
+    case azureOpenAI = "azureOpenAI"  // Azure OpenAI Foundry /openai/v1 OpenAI-compatible chat completions
     case anthropic = "anthropic"  // Anthropic Messages API
     case openResponses = "openResponses"  // Open Responses API — used for official OpenAI and any compatible provider
     case openAICodex = "openAICodex"  // ChatGPT/Codex OAuth backend
@@ -45,6 +46,7 @@ public enum RemoteProviderType: String, Codable, Sendable, CaseIterable {
     public var displayName: String {
         switch self {
         case .openaiLegacy: return L("OpenAI Compatible")
+        case .azureOpenAI: return L("Azure OpenAI Foundry")
         case .anthropic: return L("Anthropic")
         case .openResponses: return L("Open Responses")
         case .openAICodex: return L("OpenAI Codex")
@@ -55,7 +57,7 @@ public enum RemoteProviderType: String, Codable, Sendable, CaseIterable {
 
     public var chatEndpoint: String {
         switch self {
-        case .openaiLegacy: return "/chat/completions"
+        case .openaiLegacy, .azureOpenAI: return "/chat/completions"
         case .anthropic: return "/messages"
         case .openResponses: return "/responses"
         case .openAICodex: return "/codex/responses"
@@ -86,6 +88,7 @@ public struct RemoteProvider: Codable, Identifiable, Sendable, Equatable {
     public var enabled: Bool
     public var autoConnect: Bool
     public var timeout: TimeInterval
+    public var manualModelIds: [String]
 
     // Keys for headers that should be stored in Keychain (not persisted in config)
     public var secretHeaderKeys: [String]
@@ -100,6 +103,7 @@ public struct RemoteProvider: Codable, Identifiable, Sendable, Equatable {
     private enum CodingKeys: String, CodingKey {
         case id, name, host, providerProtocol, port, basePath
         case customHeaders, authType, providerType, enabled, autoConnect, timeout
+        case manualModelIds
         case secretHeaderKeys, remoteAgentId, remoteAgentAddress
     }
 
@@ -116,6 +120,7 @@ public struct RemoteProvider: Codable, Identifiable, Sendable, Equatable {
         enabled: Bool = true,
         autoConnect: Bool = true,
         timeout: TimeInterval = 60,
+        manualModelIds: [String] = [],
         secretHeaderKeys: [String] = [],
         remoteAgentId: UUID? = nil,
         remoteAgentAddress: String? = nil
@@ -132,6 +137,7 @@ public struct RemoteProvider: Codable, Identifiable, Sendable, Equatable {
         self.enabled = enabled
         self.autoConnect = autoConnect
         self.timeout = timeout
+        self.manualModelIds = manualModelIds
         self.secretHeaderKeys = secretHeaderKeys
         self.remoteAgentId = remoteAgentId
         self.remoteAgentAddress = remoteAgentAddress
@@ -154,6 +160,7 @@ public struct RemoteProvider: Codable, Identifiable, Sendable, Equatable {
         enabled = try container.decodeIfPresent(Bool.self, forKey: .enabled) ?? true
         autoConnect = try container.decodeIfPresent(Bool.self, forKey: .autoConnect) ?? true
         timeout = try container.decodeIfPresent(TimeInterval.self, forKey: .timeout) ?? 60
+        manualModelIds = try container.decodeIfPresent([String].self, forKey: .manualModelIds) ?? []
         secretHeaderKeys = try container.decodeIfPresent([String].self, forKey: .secretHeaderKeys) ?? []
         remoteAgentId = try container.decodeIfPresent(UUID.self, forKey: .remoteAgentId)
         remoteAgentAddress = try container.decodeIfPresent(String.self, forKey: .remoteAgentAddress)
@@ -262,6 +269,10 @@ public struct RemoteProvider: Codable, Identifiable, Sendable, Equatable {
                 if headers["x-goog-api-key"] == nil {
                     headers["x-goog-api-key"] = apiKey
                 }
+            case .azureOpenAI:
+                if headers["api-key"] == nil {
+                    headers["api-key"] = apiKey
+                }
             case .openaiLegacy, .openResponses, .openAICodex, .osaurus:
                 if headers["Authorization"] == nil {
                     headers["Authorization"] = "Bearer \(apiKey)"
@@ -289,6 +300,25 @@ public struct RemoteProvider: Codable, Identifiable, Sendable, Equatable {
     public func getOAuthTokens() -> RemoteProviderOAuthTokens? {
         RemoteProviderKeychain.getOAuthTokens(for: id)
     }
+
+    public func mergedModelIds(discovered: [String]) -> [String] {
+        var seen = Set<String>()
+        var merged: [String] = []
+        let sourceModels = providerType == .azureOpenAI ? manualModelIds : discovered + manualModelIds
+
+        for rawValue in sourceModels {
+            let value = rawValue.trimmingCharacters(in: .whitespacesAndNewlines)
+            guard !value.isEmpty else { continue }
+
+            let key = value.lowercased()
+            guard !seen.contains(key) else { continue }
+
+            seen.insert(key)
+            merged.append(value)
+        }
+
+        return merged
+    }
 }
 
 // MARK: - Remote Provider Runtime State