Skip to content

Commit 63e2ef2

Browse files
committed
fix: highlighted text referencing in prompts
1 parent 7a54bf8 commit 63e2ef2

File tree

1 file changed

+13
-2
lines changed

1 file changed

+13
-2
lines changed

src/main/java/ee/carlrobert/codegpt/toolwindow/chat/ChatToolWindowTabPanel.java

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -300,8 +300,19 @@ private Unit handleSubmit(String text, List<? extends AppliedActionInlay> applie
300300
}
301301
promptBuilder.append(remainingText);
302302

303-
message.setUserMessage(promptBuilder.toString());
304-
message.setPrompt(promptBuilder.toString());
303+
String highlightedTextMd = "";
304+
if (editor != null) {
305+
var selectionModel = editor.getSelectionModel();
306+
var selectedText = selectionModel.getSelectedText();
307+
if (selectedText != null && !selectedText.isEmpty()) {
308+
var fileExtension = FileUtil.getFileExtension(editor.getVirtualFile().getName());
309+
highlightedTextMd = format("\n```%s\n%s\n```\n", fileExtension, selectedText);
310+
selectionModel.removeSelection();
311+
}
312+
}
313+
314+
message.setUserMessage(highlightedTextMd + promptBuilder);
315+
message.setPrompt(highlightedTextMd + promptBuilder);
305316

306317
sendMessage(message, ConversationType.DEFAULT, processEditorSelection(editor, message));
307318
return Unit.INSTANCE;

0 commit comments

Comments
 (0)