fix: Bring back old behavior of readStrings{After,Before}Cursor setting (#97)

This commit is contained in:
Povilas Kanapickas 2025-03-05 21:27:16 +02:00 committed by GitHub
parent 69a8aa80d9
commit 5dceb5cd19
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 8 additions and 6 deletions

View File

@ -253,10 +253,12 @@ LLMCore::ContextData DocumentContextReader::prepareContext(
contextBefore = readWholeFileBefore(lineNumber, cursorPosition);
contextAfter = readWholeFileAfter(lineNumber, cursorPosition);
} else {
// Note that readStrings{After,Before}Cursor include current line, but linesCount argument of
// getContext{After,Before} do not
contextBefore
= getContextBefore(lineNumber, cursorPosition, settings.readStringsBeforeCursor());
= getContextBefore(lineNumber, cursorPosition, settings.readStringsBeforeCursor() + 1);
contextAfter
= getContextAfter(lineNumber, cursorPosition, settings.readStringsAfterCursor());
= getContextAfter(lineNumber, cursorPosition, settings.readStringsAfterCursor() + 1);
}
QString fileContext;

View File

@ -283,15 +283,15 @@ TEST_F(DocumentContextReaderTest, testPrepareContext)
EXPECT_EQ(
reader.prepareContext(2, 3, *createSettingsForLines(1, 1)),
(ContextData{
.prefix = "Lin",
.suffix = "e 3",
.prefix = "Line 2\nLin",
.suffix = "e 3\nLine 4",
.fileContext = "\n Language: (MIME: text/python) filepath: /path/to/file()\n\n\n "}));
EXPECT_EQ(
reader.prepareContext(2, 3, *createSettingsForLines(2, 2)),
(ContextData{
.prefix = "Line 2\nLin",
.suffix = "e 3\nLine 4",
.prefix = "Line 1\nLine 2\nLin",
.suffix = "e 3\nLine 4\nLine 5",
.fileContext = "\n Language: (MIME: text/python) filepath: /path/to/file()\n\n\n "}));
}