ide message content

continuedev · Jan 21, 2025 · 794a8f0 · 794a8f0
1 parent cdf7b01
commit 794a8f0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/...ain/kotlin/com/github/continuedev/continueintellijextension/continue/IdeProtocolClient.kt b/...ain/kotlin/com/github/continuedev/continueintellijextension/continue/IdeProtocolClient.kt
@@ -439,8 +439,11 @@ class IdeProtocolClient(
                                     null,
                                     null
                                 ) { response ->
-                                    val result = (response as Map<String, Any>)["result"] as Map<String, Any>
+                                    val responseObject = response as Map<*, *>
+                                    val responseContent = responseObject["content"] as Map<*, *>
+                                    val result = responseContent["result"] as Map<*, *>
                                     val config = result["config"] as Map<String, Any>
+
                                     val applyCodeBlockModel = getModelByRole(config, "applyCodeBlock")
 
                                     if (applyCodeBlockModel != null) {
@@ -451,7 +454,6 @@ class IdeProtocolClient(
                                         config["models"] as List<Map<String, Any>>
                                     val curSelectedModel = models.find { it["title"] == params.curSelectedModelTitle }
 
-//                                    continuation.resume(curSelectedModel)
                                     if (curSelectedModel == null) {
                                         return@request
                                     } else {