Addressing comments

laileni-aws · Apr 23, 2024 · ac2b63d · ac2b63d
1 parent 7cb364f
commit ac2b63d
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/...ftware/aws/toolkits/jetbrains/services/cwc/controller/chat/messenger/ChatPromptHandler.kt b/...ftware/aws/toolkits/jetbrains/services/cwc/controller/chat/messenger/ChatPromptHandler.kt
@@ -33,13 +33,16 @@ class ChatPromptHandler(private val telemetryHelper: TelemetryHelper) {
     private var requestId: String = ""
     private var statusCode: Int = 0
 
+    companion object {
+        private val CODE_BLOCK_PATTERN = Regex("<pre>\\s*<code")
+    }
+
     private fun countTotalNumberOfCodeBlocks(message: StringBuilder): Int {
         if (message.isEmpty()) {
             return 0
         }
         val htmlInString = convertMarkdownToHTML(message.toString())
-        val patternOfCodeBlock = Regex("<pre>\\s*<code")
-        return patternOfCodeBlock.findAll(htmlInString).count()
+        return CODE_BLOCK_PATTERN.findAll(htmlInString).count()
     }
 
     fun handle(