Pre-release 0.44.151

Author: actions-user

Core/Package.swift

@@ -254,6 +254,7 @@ let package = Package(
         .target(
             name: "GitHubCopilotViewModel",
             dependencies: [
+                "Client",
                 .product(name: "GitHubCopilotService", package: "Tool"),
                 .product(name: "ComposableArchitecture", package: "swift-composable-architecture"),
                 .product(name: "Status", package: "Tool"),

Core/Sources/ConversationTab/ModeAndModelPicker/ModelManagerUtils.swift

@@ -179,7 +179,7 @@ public extension CopilotModelManager {
     static func getDefaultChatModel(scope: PromptTemplateScope = .chatPanel) -> LLMModel? {
         let LLMs = CopilotModelManager.getAvailableLLMs()
         let LLMsInScope = LLMs.filter({ $0.scopes.contains(scope) })
-        let defaultModel = LLMsInScope.first(where: { $0.isChatDefault })
+        let defaultModel = LLMsInScope.first(where: { $0.isChatDefault && !$0.isAutoModel })
         // If a default model is found, return it
         if let defaultModel = defaultModel {
             return LLMModel(
@@ -202,7 +202,7 @@ public extension CopilotModelManager {
         }
 
         // If no default model is found, fallback to the first available model
-        if let firstModel = LLMsInScope.first {
+        if let firstModel = LLMsInScope.first(where: { !$0.isAutoModel }) {
             return LLMModel(
                 modelName: firstModel.modelName,
                 modelFamily: firstModel.modelFamily,
@@ -270,3 +270,7 @@ public extension LLMModel {
     /// Apply to `Copilot Models`
     var isAutoModel: Bool { isStandardModel && modelName == "Auto" }
 }
+
+extension CopilotModel {
+    var isAutoModel: Bool { modelName == "Auto" }
+}

Core/Sources/GitHubCopilotViewModel/GitHubCopilotViewModel.swift

@@ -4,6 +4,7 @@ import ComposableArchitecture
 import Status
 import SwiftUI
 import Cache
+import Client
 
 public struct SignInResponse {
     public let status: SignInInitiateStatus
@@ -126,7 +127,7 @@ public class GitHubCopilotViewModel: ObservableObject {
         waitingForSignIn = false
     }
 
-    public func copyAndOpen() {
+    public func copyAndOpen(fromHostApp: Bool = false) {
         waitingForSignIn = true
         guard let signInResponse else {
             toast("Missing sign in details.", .error)
@@ -137,10 +138,10 @@ public class GitHubCopilotViewModel: ObservableObject {
         pasteboard.setString(signInResponse.userCode, forType: NSPasteboard.PasteboardType.string)
         toast("Sign-in code \(signInResponse.userCode) copied", .info)
         NSWorkspace.shared.open(signInResponse.verificationURL)
-        waitForSignIn()
+        waitForSignIn(fromHostApp: fromHostApp)
     }
 
-    public func waitForSignIn() {
+    public func waitForSignIn(fromHostApp: Bool = false) {
         Task {
             do {
                 guard waitingForSignIn else { return }
@@ -155,14 +156,19 @@ public class GitHubCopilotViewModel: ObservableObject {
                 self.status = status
                 await Status.shared.updateAuthStatus(.loggedIn, username: username)
                 broadcastStatusChange()
-                let models = try? await service.models()
-                if let models = models, !models.isEmpty {
-                    CopilotModelManager.updateLLMs(models)
+                if !fromHostApp {
+                    let models = try? await service.models()
+                    if let models = models, !models.isEmpty {
+                        CopilotModelManager.updateLLMs(models)
+                    }
+                } else {
+                    let xpcService = try getService()
+                    _ = try? await xpcService.updateCopilotModels()
                 }
             } catch let error as GitHubCopilotError {
                 switch error {
                 case .languageServerError(.timeout):
-                    waitForSignIn()
+                    waitForSignIn(fromHostApp: fromHostApp)
                     return
                 case .languageServerError(
                     .serverError(

Core/Sources/HostApp/AdvancedSettings/SuggestionSection.swift

@@ -7,6 +7,7 @@ struct SuggestionSection: View {
     @AppStorage(\.realtimeNESToggle) var realtimeNESToggle
     @State var isSuggestionFeatureDisabledLanguageListViewOpen = false
     @State private var shouldPresentTurnoffSheet = false
+    @ObservedObject private var featureFlags = FeatureFlagManager.shared
 
     var realtimeSuggestionBinding : Binding<Bool> {
         Binding(
@@ -27,11 +28,15 @@ struct SuggestionSection: View {
                 title: "Enable completions while typing",
                 isOn: realtimeSuggestionBinding
             )
-            Divider()
-            SettingsToggle(
-                title: "Enable Next Edit Suggestions (NES)",
-                isOn: $realtimeNESToggle
-            )
+            
+            if featureFlags.isEditorPreviewEnabled {
+                Divider()
+                SettingsToggle(
+                    title: "Enable Next Edit Suggestions (NES)",
+                    isOn: $realtimeNESToggle
+                )
+            }
+            
             Divider()
             SettingsToggle(
                 title: "Accept suggestions with Tab",

Core/Sources/HostApp/GeneralSettings/CopilotConnectionView.swift

@@ -57,7 +57,10 @@ struct CopilotConnectionView: View {
                     isPresented: $viewModel.isSignInAlertPresented,
                     presenting: viewModel.signInResponse) { _ in
                         Button("Cancel", role: .cancel, action: {})
-                        Button("Copy Code and Open", action: viewModel.copyAndOpen)
+                        Button(
+                            "Copy Code and Open",
+                            action: { viewModel.copyAndOpen(fromHostApp: true) }
+                        )
                     } message: { response in
                         Text("""
                                Please enter the above code in the \

Core/Sources/Service/SuggestionCommandHandler/PseudoCommandHandler.swift

@@ -10,6 +10,7 @@ import WorkspaceSuggestionService
 import XcodeInspector
 import XPCShared
 import AXHelper
+import GitHubCopilotService
 
 /// It's used to run some commands without really triggering the menu bar item.
 ///
@@ -59,7 +60,8 @@ struct PseudoCommandHandler {
         if Task.isCancelled { return }
 
         let codeCompletionEnabled = UserDefaults.shared.value(for: \.realtimeSuggestionToggle)
-        let nesEnabled = UserDefaults.shared.value(for: \.realtimeNESToggle)
+        // Enabled both by Feature Flag and User.
+        let nesEnabled = FeatureFlagNotifierImpl.shared.featureFlags.editorPreviewFeatures && UserDefaults.shared.value(for: \.realtimeNESToggle)
         guard codeCompletionEnabled || nesEnabled else {
             cleanupAllSuggestions(filespace: filespace, presenter: nil)
             return

Core/Sources/Service/XPCService.swift

@@ -594,6 +594,21 @@ public class XPCService: NSObject, XPCServiceProtocol {
         }
     }
 
+    public func updateCopilotModels(withReply reply: @escaping (Data?, Error?) -> Void) {
+        Task { @MainActor in
+            do {
+                let service = try GitHubCopilotViewModel.shared.getGitHubCopilotAuthService()
+                let models = try await service.models()
+                CopilotModelManager.updateLLMs(models)
+                let data = try JSONEncoder().encode(models)
+                reply(data, nil)
+            } catch {
+                Logger.service.error("Failed to get models: \(error.localizedDescription)")
+                reply(nil, NSError.from(error))
+            }
+        }
+    }
+    
     // MARK: - BYOK
     public func saveBYOKApiKey(_ params: Data, withReply reply: @escaping (Data?) -> Void) {
         let decoder = JSONDecoder()

Core/Sources/SuggestionWidget/ChatWindow/ChatLoginView.swift

@@ -77,7 +77,7 @@ struct ChatLoginView: View {
                 Button("Cancel", role: .cancel, action: {})
                     .scaledFont(.body)
 
-                Button("Copy Code and Open", action: viewModel.copyAndOpen)
+                Button("Copy Code and Open", action: { viewModel.copyAndOpen(fromHostApp: false) })
                     .scaledFont(.body)
             } message: { response in
                 Text("""

Core/Sources/SuggestionWidget/Extensions/WidgetWindowsController+FeatureFlags.swift

@@ -0,0 +1,27 @@
+import GitHubCopilotService
+
+extension WidgetWindowsController {
+    
+    @MainActor
+    var isNESFeatureFlagEnabled: Bool {
+        FeatureFlagNotifierImpl.shared.featureFlags.editorPreviewFeatures
+    }
+    
+    func setupFeatureFlagObservers() {
+        Task { @MainActor in
+            let sinker = FeatureFlagNotifierImpl.shared.featureFlagsDidChange
+                .sink(receiveValue: { [weak self] _ in
+                    self?.onFeatureFlagChanged()
+                })
+            
+            await self.storeCancellables([sinker])
+        }
+    }
+    
+    @MainActor
+    func onFeatureFlagChanged() {
+        if !isNESFeatureFlagEnabled {
+            hideAllNESWindows()
+        }
+    }
+}

Core/Sources/SuggestionWidget/Extensions/WidgetWindowsController+NES.swift

@@ -1,4 +1,5 @@
 import AppKit
+import GitHubCopilotService
 
 extension WidgetWindowsController {
     func setupNESSuggestionPanelObservers() {
@@ -16,6 +17,40 @@ extension WidgetWindowsController {
         }
     }
 
+    @MainActor
+    func applyOpacityForNESWindows(by noFocus: Bool) {
+        guard !noFocus, isNESFeatureFlagEnabled
+        else {
+            hideAllNESWindows()
+            return
+        }
+        
+        displayAllNESWindows()
+    }
+    
+    @MainActor
+    func hideAllNESWindows() {
+        windows.nesMenuWindow.alphaValue = 0
+        windows.nesDiffWindow.setIsVisible(false)
+        
+        hideNESDiffWindow()
+        
+        windows.nesNotificationWindow.alphaValue = 0
+        windows.nesNotificationWindow.setIsVisible(false)
+    }
+    
+    @MainActor
+    func displayAllNESWindows() {
+        windows.nesMenuWindow.alphaValue = 1
+        windows.nesDiffWindow.setIsVisible(true)
+        
+        windows.nesDiffWindow.alphaValue = 1
+        windows.nesDiffWindow.setIsVisible(true)
+        
+        windows.nesNotificationWindow.alphaValue = 1
+        windows.nesNotificationWindow.setIsVisible(true)
+    }
+    
     @MainActor
     func hideNESDiffWindow() {
         NSAnimationContext.runAnimationGroup { context in