Skip to content

Commit da650e8

Browse files
committed
Addressing comments and adding cwsprChatProgrammingLanguage to commons package
1 parent fc6a8a1 commit da650e8

File tree

3 files changed

+7
-10
lines changed

3 files changed

+7
-10
lines changed

gradle/libs.versions.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ mockitoKotlin = "5.4.0"
2525
mockk = "1.13.10"
2626
nimbus-jose-jwt = "9.40"
2727
node-gradle = "7.0.2"
28-
telemetryGenerator = "1.0.262"
28+
telemetryGenerator = "1.0.271"
2929
testLogger = "4.0.0"
3030
testRetry = "1.5.10"
3131
# test-only; platform provides slf4j transitively at runtime

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/cwc/controller/chat/messenger/ChatPromptHandler.kt

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,8 @@ class ChatPromptHandler(private val telemetryHelper: TelemetryHelper) {
3333
private val codeReferences = mutableListOf<CodeReference>()
3434
private var requestId: String = ""
3535
private var statusCode: Int = 0
36-
private var codeBlockLanguage: String = "plaintext"
36+
private val defaultTestGenResponseLanguage: String = "plaintext"
37+
private var codeBlockLanguage: String = defaultTestGenResponseLanguage
3738

3839
companion object {
3940
private val CODE_BLOCK_PATTERN = Regex("<pre>\\s*<code")
@@ -148,18 +149,18 @@ class ChatPromptHandler(private val telemetryHelper: TelemetryHelper) {
148149
// This fulfills both the cases of unit test generation(java, python) and general use case(Non java and Non python) languages.
149150
val codeBlockStart = message.indexOf("```")
150151
if (codeBlockStart == -1) {
151-
return "plaintext"
152+
return defaultTestGenResponseLanguage
152153
}
153154

154155
val languageStart = codeBlockStart + 3
155156
val languageEnd = message.indexOf('\n', languageStart)
156157

157158
if (languageEnd == -1) {
158-
return "plaintext"
159+
return defaultTestGenResponseLanguage
159160
}
160161

161162
val language = message.substring(languageStart, languageEnd).trim()
162-
return if (language.isNotEmpty()) language else "plaintext"
163+
return if (language.isNotEmpty()) language else defaultTestGenResponseLanguage
163164
}
164165

165166
private fun processChatEvent(
@@ -228,7 +229,7 @@ class ChatPromptHandler(private val telemetryHelper: TelemetryHelper) {
228229
} else {
229230
responseText.toString()
230231
}
231-
if (codeBlockLanguage == "plaintext") {
232+
if (codeBlockLanguage == defaultTestGenResponseLanguage) {
232233
// To get the language of generated code in Q chat.
233234
codeBlockLanguage = extractCodeBlockLanguage(message)
234235
}

plugins/core/jetbrains-community/resources/telemetryOverride.json

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -692,10 +692,6 @@
692692
{
693693
"type": "cwsprChatHasProjectContext",
694694
"required": false
695-
},
696-
{
697-
"type": "cwsprChatProgrammingLanguage",
698-
"required": false
699695
}
700696
]
701697
},

0 commit comments

Comments
 (0)