Make chat with selection and prompt to code return immediately

Author: intitni

EditorExtension/ChatWithSelection.swift

@@ -10,20 +10,10 @@ class ChatWithSelectionCommand: NSObject, XCSourceEditorCommand, CommandType {
         with invocation: XCSourceEditorCommandInvocation,
         completionHandler: @escaping (Error?) -> Void
     ) {
+        completionHandler(nil)
         Task {
-            do {
-                let service = try getService()
-                if let content = try await service.chatWithSelection(
-                    editorContent: .init(invocation)
-                ) {
-                    invocation.accept(content)
-                }
-                completionHandler(nil)
-            } catch is CancellationError {
-                completionHandler(nil)
-            } catch {
-                completionHandler(error)
-            }
+            let service = try getService()
+            _ = try await service.chatWithSelection(editorContent: .init(invocation))
         }
     }
 }

EditorExtension/PromptToCodeCommand.swift

@@ -10,20 +10,10 @@ class PromptToCodeCommand: NSObject, XCSourceEditorCommand, CommandType {
         with invocation: XCSourceEditorCommandInvocation,
         completionHandler: @escaping (Error?) -> Void
     ) {
+        completionHandler(nil)
         Task {
-            do {
-                let service = try getService()
-                if let content = try await service.promptToCode(
-                    editorContent: .init(invocation)
-                ) {
-                    invocation.accept(content)
-                }
-                completionHandler(nil)
-            } catch is CancellationError {
-                completionHandler(nil)
-            } catch {
-                completionHandler(error)
-            }
+            let service = try getService()
+            _ = try await service.promptToCode(editorContent: .init(invocation))
         }
     }
 }