Fix GitHubCopilot chat error

Author: intitni

Tool/Sources/GitHubCopilotService/LanguageServer/CopilotLocalProcessServer.swift

@@ -14,6 +14,7 @@ class CopilotLocalProcessServer {
     private var wrappedServer: CustomJSONRPCLanguageServer?
     var terminationHandler: (() -> Void)?
     @MainActor var ongoingCompletionRequestIDs: [JSONId] = []
+    @MainActor var ongoingConversationRequestIDs: [String: JSONId] = [:]
 
     public convenience init(
         path: String,
@@ -58,6 +59,21 @@ class CopilotLocalProcessServer {
                 Task { @MainActor [weak self] in
                     self?.ongoingCompletionRequestIDs.append(request.id)
                 }
+            } else if request.method == "conversation/create" {
+                Task { @MainActor [weak self] in
+                    if let paramsData = try? JSONEncoder().encode(request.params) {
+                        do {
+                            let params = try JSONDecoder().decode(
+                                GitHubCopilotRequest.ConversationCreate.RequestBody.self,
+                                from: paramsData
+                            )
+                            self?.ongoingConversationRequestIDs[params.workDoneToken] = request.id
+                        } catch {
+                            // Handle decoding error
+                            print("Error decoding ConversationCreateParams: \(error)")
+                        }
+                    }
+                }
             }
         }
 
@@ -131,19 +147,35 @@ extension CopilotLocalProcessServer: LanguageServerProtocol.Server {
 
         let task = Task { @MainActor in
             for id in self.ongoingCompletionRequestIDs {
-                switch id {
-                case let .numericId(id):
-                    try? await server.sendNotification(.protocolCancelRequest(.init(id: id)))
-                case let .stringId(id):
-                    try? await server.sendNotification(.protocolCancelRequest(.init(id: id)))
-                }
+                await cancelTask(id)
             }
             self.ongoingCompletionRequestIDs = []
         }
 
         await task.value
     }
 
+    public func cancelOngoingTask(workDoneToken: String) async {
+        let task = Task { @MainActor in
+            guard let id = ongoingConversationRequestIDs[workDoneToken] else { return }
+            await cancelTask(id)
+        }
+        await task.value
+    }
+
+    public func cancelTask(_ id: JSONId) async {
+        guard let server = wrappedServer, process.isRunning else {
+            return
+        }
+
+        switch id {
+        case let .numericId(id):
+            try? await server.sendNotification(.protocolCancelRequest(.init(id: id)))
+        case let .stringId(id):
+            try? await server.sendNotification(.protocolCancelRequest(.init(id: id)))
+        }
+    }
+
     public func sendRequest<Response: Codable>(
         _ request: ClientRequest,
         completionHandler: @escaping (ServerResult<Response>) -> Void

Tool/Sources/GitHubCopilotService/Services/GitHubCopilotChatService.swift

@@ -50,7 +50,10 @@ public final class GitHubCopilotChatService: BuiltinExtensionChatServiceType {
             let startTimestamp = Date()
 
             continuation.onTermination = { _ in
-                Task { service.unregisterNotificationHandler(id: id) }
+                Task {
+                    service.unregisterNotificationHandler(id: id)
+                    await service.cancelOngoingTask(workDoneToken: workDoneToken)
+                }
             }
 
             service.registerNotificationHandler(id: id) { notification, data in
@@ -71,12 +74,20 @@ public final class GitHubCopilotChatService: BuiltinExtensionChatServiceType {
                         if let reply = progress.value.reply, progress.value.kind == "report" {
                             continuation.yield(reply)
                         } else if progress.value.kind == "end" {
-                            if let error = progress.value.error,
+                            if let error = progress.value.error?.message,
                                progress.value.cancellationReason == nil
                             {
-                                continuation.finish(
-                                    throwing: GitHubCopilotError.chatEndsWithError(error)
-                                )
+                                if error.contains("400") {
+                                    continuation.finish(
+                                        throwing: GitHubCopilotError.chatEndsWithError(
+                                            "\(error). Please try enabling pretend IDE as VSCode."
+                                        )
+                                    )
+                                } else {
+                                    continuation.finish(
+                                        throwing: GitHubCopilotError.chatEndsWithError(error)
+                                    )
+                                }
                             } else {
                                 continuation.finish()
                             }
@@ -198,6 +209,11 @@ extension GitHubCopilotChatService {
                 var message: String
             }
 
+            struct Error: Decodable {
+                var responseIsIncomplete: Bool?
+                var message: String?
+            }
+
             var kind: String
             var title: String?
             var conversationId: String
@@ -209,7 +225,7 @@ extension GitHubCopilotChatService {
             var annotations: [String]?
             var hideText: Bool?
             var cancellationReason: String?
-            var error: String?
+            var error: Error?
         }
 
         var token: String