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

fix: highlighted text referencing in prompts

Browse source
This commit is contained in:
Carl-Robert Linnupuu 2024-09-12 02:16:12 +03:00
parent 7a54bf8788
commit 63e2ef2a2d
1 changed files with 13 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/main/java/ee/carlrobert/codegpt/toolwindow/chat/ChatToolWindowTabPanel.java
View file

@ -300,8 +300,19 @@ public class ChatToolWindowTabPanel implements Disposable {
}
promptBuilder.append(remainingText);
message.setUserMessage(promptBuilder.toString());
message.setPrompt(promptBuilder.toString());
String highlightedTextMd = "";
if (editor != null) {
var selectionModel = editor.getSelectionModel();
var selectedText = selectionModel.getSelectedText();
if (selectedText != null && !selectedText.isEmpty()) {
var fileExtension = FileUtil.getFileExtension(editor.getVirtualFile().getName());
highlightedTextMd = format("\n```%s\n%s\n```\n", fileExtension, selectedText);
selectionModel.removeSelection();
}
}
message.setUserMessage(highlightedTextMd + promptBuilder);
message.setPrompt(highlightedTextMd + promptBuilder);
sendMessage(message, ConversationType.DEFAULT, processEditorSelection(editor, message));
return Unit.INSTANCE;