vrr/ProxyAI
2
0
Fork 0
mirror of https://github.com/carlrobertoh/ProxyAI.git synced 2026-05-20 01:02:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: next edit calls upon empty completion

Browse source
This commit is contained in:
Carl-Robert Linnupuu 2025-05-13 11:03:39 +01:00
parent b7f7f22fc3
commit f9f1282ab4
1 changed files with 10 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

21
src/main/kotlin/ee/carlrobert/codegpt/codecompletions/CodeCompletionEventListener.kt
View file

@ -122,25 +122,17 @@ class CodeCompletionEventListener(
val parsedContent = parseOutput(firstLine + remainingContent)
if (parsedContent.isNotEmpty()) {
cache?.setCache(prefix, suffix, firstLine + parsedContent)
cache?.setCache(prefix, suffix, parsedContent)
CodeGPTKeys.REMAINING_CODE_COMPLETION.set(
editor,
PartialCodeCompletionResponse.newBuilder()
.setPartialCompletion(remainingContent)
.setPartialCompletion(parsedContent.removePrefix(firstLine ?: ""))
.build()
)
}
} else {
val formattedLine = CodeCompletionFormatter(editor).format(finalResult.toString())
if (formattedLine.isEmpty()) {
editor.project?.service<GrpcClientService>()?.getNextEdit(
editor,
prefix + suffix,
runReadAction { editor.caretModel.offset })
return
}
if (isNotAllowed(formattedLine)) {
return
}
@ -181,6 +173,13 @@ class CodeCompletionEventListener(
private fun handleCompleted() {
setLoading(false)
if (messageBuilder.isEmpty()) {
editor.project?.service<GrpcClientService>()?.getNextEdit(
editor,
prefix + suffix,
runReadAction { editor.caretModel.offset })
}
}
private fun setLoading(loading: Boolean) {
@ -196,7 +195,7 @@ class CodeCompletionEventListener(
return CodeCompletionParserFactory
.getParserForFileExtension(editor.virtualFile.extension)
.parse(prefix, suffix, (firstLine ?: "") + input)
.parse(prefix, suffix, input)
.trimEnd()
}
}