Update preferences

Author: intitni

Core/Sources/HostApp/AccountSettings/AzureView.swift

@@ -9,6 +9,7 @@ final class AzureViewSettings: ObservableObject {
     @AppStorage(\.azureOpenAIAPIKey) var azureOpenAIAPIKey: String
     @AppStorage(\.azureOpenAIBaseURL) var azureOpenAIBaseURL: String
     @AppStorage(\.azureChatGPTDeployment) var azureChatGPTDeployment: String
+    @AppStorage(\.azureEmbeddingDeployment) var azureEmbeddingDeployment: String
     init() {}
 }
 
@@ -58,6 +59,15 @@ struct AzureView: View {
                 }
                 .disabled(isTesting)
             }
+            
+            HStack {
+                TextField(
+                    text: $settings.azureEmbeddingDeployment,
+                    prompt: Text("")
+                ) {
+                    Text("Embedding Model Deployment Name")
+                }.textFieldStyle(.roundedBorder)
+            }
         }
     }
 }

Core/Sources/HostApp/AccountSettings/OpenAIView.swift

@@ -9,6 +9,7 @@ struct OpenAIView: View {
     final class Settings: ObservableObject {
         @AppStorage(\.openAIAPIKey) var openAIAPIKey: String
         @AppStorage(\.chatGPTModel) var chatGPTModel: String
+        @AppStorage(\.embeddingModel) var embeddingModel: String
         @AppStorage(\.openAIBaseURL) var openAIBaseURL: String
         init() {}
     }
@@ -82,6 +83,26 @@ struct OpenAIView: View {
                     Image(systemName: "questionmark.circle.fill")
                 }.buttonStyle(.plain)
             }
+            
+            HStack {
+                Picker(selection: $settings.embeddingModel) {
+                    if !settings.embeddingModel.isEmpty,
+                       OpenAIEmbeddingModel(rawValue: settings.embeddingModel) == nil
+                    {
+                        Text(settings.embeddingModel).tag(settings.embeddingModel)
+                    }
+                    ForEach(OpenAIEmbeddingModel.allCases, id: \.self) { model in
+                        Text(model.rawValue).tag(model.rawValue)
+                    }
+                } label: {
+                    Text("Embedding Model")
+                }.pickerStyle(.menu)
+                Button(action: {
+                    openURL(modelURL)
+                }) {
+                    Image(systemName: "questionmark.circle.fill")
+                }.buttonStyle(.plain)
+            }
         }
     }
 }

Core/Sources/HostApp/FeatureSettings/ChatSettingsView.swift

@@ -21,6 +21,8 @@ struct ChatSettingsView: View {
         @AppStorage(\.chatGPTModel) var chatGPTModel
         @AppStorage(\.defaultChatSystemPrompt) var defaultChatSystemPrompt
         @AppStorage(\.chatSearchPluginMaxIterations) var chatSearchPluginMaxIterations
+        
+        @AppStorage(\.embeddingFeatureProvider) var embeddingFeatureProvider
 
         init() {}
     }
@@ -46,12 +48,20 @@ struct ChatSettingsView: View {
     var chatSettingsForm: some View {
         Form {
             Picker(
-                "Feature Provider",
+                "Chat Feature Provider",
                 selection: $settings.chatFeatureProvider
             ) {
                 Text("OpenAI").tag(ChatFeatureProvider.openAI)
                 Text("Azure OpenAI").tag(ChatFeatureProvider.azureOpenAI)
             }
+            
+            Picker(
+                "Embedding Feature Provider",
+                selection: $settings.embeddingFeatureProvider
+            ) {
+                Text("OpenAI").tag(EmbeddingFeatureProvider.openAI)
+                Text("Azure OpenAI").tag(EmbeddingFeatureProvider.azureOpenAI)
+            }
 
             if #available(macOS 13.0, *) {
                 LabeledContent("Reply in Language") {

Tool/Sources/LangChain/VectorStore/TemporaryUSearch.swift

@@ -7,6 +7,10 @@ private actor TemporaryUSearchActor {
     static let shared = TemporaryUSearchActor()
 }
 
+#warning(
+    "It's not working yet because of a bug in USearch https://github.com/unum-cloud/usearch/issues/131"
+)
+
 /// A temporary USearch index for small and temporary documents.
 public actor TemporaryUSearch: VectorStore {
     public let identifier: String

Tool/Sources/OpenAIService/Configuration/EmbeddingConfiguration.swift

@@ -1,8 +1,6 @@
 import Foundation
 import Preferences
 
-public typealias EmbeddingFeatureProvider = ChatFeatureProvider
-
 public protocol EmbeddingConfiguration {
     var featureProvider: EmbeddingFeatureProvider { get }
     var endpoint: String { get }
@@ -27,7 +25,7 @@ extension EmbeddingConfiguration {
         }
     }
 
-    func apiKey(for provider: ChatFeatureProvider) -> String {
+    func apiKey(for provider: EmbeddingFeatureProvider) -> String {
         switch provider {
         case .openAI:
             return UserDefaults.shared.value(for: \.openAIAPIKey)

Tool/Sources/OpenAIService/Configuration/UserPreferenceEmbeddingConfiguration.swift

@@ -2,12 +2,12 @@ import Foundation
 import Preferences
 
 public struct UserPreferenceEmbeddingConfiguration: EmbeddingConfiguration {
-    public var featureProvider: ChatFeatureProvider {
-        UserDefaults.shared.value(for: \.chatFeatureProvider)
+    public var featureProvider: EmbeddingFeatureProvider {
+        UserDefaults.shared.value(for: \.embeddingFeatureProvider)
     }
 
     public var model: String {
-        return "text-embedding-ada-002"
+        OpenAIEmbeddingModel.textEmbeddingAda002.rawValue
     }
 
     public var endpoint: String {
@@ -19,7 +19,7 @@ public struct UserPreferenceEmbeddingConfiguration: EmbeddingConfiguration {
     }
 
     public var maxToken: Int {
-        8191
+        OpenAIEmbeddingModel.textEmbeddingAda002.maxToken
     }
 
     public init() {}
@@ -29,15 +29,15 @@ public class OverridingEmbeddingConfiguration<
     Configuration: EmbeddingConfiguration
 >: EmbeddingConfiguration {
     public struct Overriding {
-        var featureProvider: ChatFeatureProvider?
+        var featureProvider: EmbeddingFeatureProvider?
         var model: String?
         var endPoint: String?
         var apiKey: String?
         var maxTokens: Int?
 
         public init(
             model: String? = nil,
-            featureProvider: ChatFeatureProvider? = nil,
+            featureProvider: EmbeddingFeatureProvider? = nil,
             endPoint: String? = nil,
             apiKey: String? = nil,
             maxTokens: Int? = nil
@@ -58,7 +58,7 @@ public class OverridingEmbeddingConfiguration<
         self.configuration = configuration
     }
 
-    public var featureProvider: ChatFeatureProvider {
+    public var featureProvider: EmbeddingFeatureProvider {
         overriding.featureProvider ?? configuration.featureProvider
     }
 

Tool/Sources/Preferences/Keys.swift

@@ -79,7 +79,7 @@ public extension UserDefaultPreferenceKeys {
     }
 
     var chatGPTModel: PreferenceKey<String> {
-        .init(defaultValue: Preferences.ChatGPTModel.gpt35Turbo.rawValue, key: "ChatGPTModel")
+        .init(defaultValue: ChatGPTModel.gpt35Turbo.rawValue, key: "ChatGPTModel")
     }
 
     var chatGPTMaxToken: PreferenceKey<Int> {
@@ -97,6 +97,13 @@ public extension UserDefaultPreferenceKeys {
     var chatGPTTemperature: PreferenceKey<Double> {
         .init(defaultValue: 0.7, key: "ChatGPTTemperature")
     }
+
+    var embeddingModel: PreferenceKey<String> {
+        .init(
+            defaultValue: OpenAIEmbeddingModel.textEmbeddingAda002.rawValue,
+            key: "OpenAIEmbeddingModel"
+        )
+    }
 }
 
 // MARK: - Azure OpenAI Settings
@@ -113,6 +120,10 @@ public extension UserDefaultPreferenceKeys {
     var azureChatGPTDeployment: PreferenceKey<String> {
         .init(defaultValue: "", key: "AzureChatGPTDeployment")
     }
+
+    var azureEmbeddingDeployment: PreferenceKey<String> {
+        .init(defaultValue: "", key: "AzureEmbeddingDeployment")
+    }
 }
 
 // MARK: - GitHub Copilot Settings
@@ -125,7 +136,7 @@ public extension UserDefaultPreferenceKeys {
     var gitHubCopilotProxyHost: PreferenceKey<String> {
         .init(defaultValue: "", key: "GitHubCopilotProxyHost")
     }
-    
+
     var gitHubCopilotProxyPort: PreferenceKey<String> {
         .init(defaultValue: "", key: "GitHubCopilotProxyPort")
     }
@@ -221,6 +232,10 @@ public extension UserDefaultPreferenceKeys {
     var chatFeatureProvider: PreferenceKey<ChatFeatureProvider> {
         .init(defaultValue: .openAI, key: "ChatFeatureProvider")
     }
+    
+    var embeddingFeatureProvider: PreferenceKey<EmbeddingFeatureProvider> {
+        .init(defaultValue: .openAI, key: "EmbeddingFeatureProvider")
+    }
 
     var chatFontSize: PreferenceKey<Double> {
         .init(defaultValue: 12, key: "ChatFontSize")
@@ -345,7 +360,7 @@ public extension UserDefaultPreferenceKeys {
     var triggerActionWithAccessibilityAPI: FeatureFlag {
         .init(defaultValue: true, key: "FeatureFlag-TriggerActionWithAccessibilityAPI")
     }
-    
+
     var alwaysAcceptSuggestionWithAccessibilityAPI: FeatureFlag {
         .init(defaultValue: false, key: "FeatureFlag-AlwaysAcceptSuggestionWithAccessibilityAPI")
     }
@@ -361,7 +376,7 @@ public extension UserDefaultPreferenceKeys {
     var animationCCrashSuggestion: FeatureFlag {
         .init(defaultValue: true, key: "FeatureFlag-AnimationCCrashSuggestion")
     }
-    
+
     var enableXcodeInspectorDebugMenu: FeatureFlag {
         .init(defaultValue: false, key: "FeatureFlag-EnableXcodeInspectorDebugMenu")
     }