Skip to content

Commit 5e9e9ec

Browse files
authored
fix contextMerging not accounting userInput (#3701)
1 parent 6f3bb46 commit 5e9e9ec

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

jetbrains-core/src/software/aws/toolkits/jetbrains/services/codewhisperer/service/CodeWhispererRecommendationManager.kt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,7 @@ class CodeWhispererRecommendationManager {
119119
): List<DetailContext> {
120120
val seen = mutableSetOf<String>()
121121
return recommendations.map {
122-
val isDiscardedByUserInput = !it.content().startsWith(userInput)
122+
val isDiscardedByUserInput = !it.content().startsWith(userInput) || it.content() == userInput
123123
if (isDiscardedByUserInput) {
124124
return@map DetailContext(requestId, it, it, isDiscarded = true, isTruncatedOnRight = false, rightOverlap = "")
125125
}
@@ -135,13 +135,13 @@ class CodeWhispererRecommendationManager {
135135
val truncated = it.toBuilder()
136136
.content(truncatedContent)
137137
.build()
138-
val isDiscardedByUserInputForTruncated = !truncated.content().startsWith(userInput)
138+
val isDiscardedByUserInputForTruncated = !truncated.content().startsWith(userInput) || truncated.content() == userInput
139139
if (isDiscardedByUserInputForTruncated) {
140140
return@map DetailContext(requestId, it, truncated, isDiscarded = true, isTruncatedOnRight = true, rightOverlap = overlap)
141141
}
142142

143143
val reformatted = reformat(requestContext, truncated)
144-
val isDiscardedByRightContextTruncationDedupe = truncated.content().isEmpty() || !seen.add(reformatted.content())
144+
val isDiscardedByRightContextTruncationDedupe = !seen.add(reformatted.content())
145145
DetailContext(
146146
requestId,
147147
it,

0 commit comments

Comments
 (0)