Add extended context

This commit is contained in:
Petr Mironychev
2024-08-28 10:10:31 +02:00
parent 989718017a
commit fc58c38f63
6 changed files with 58 additions and 10 deletions

View File

@ -98,8 +98,8 @@ QString LLMClientInterface::сontextBefore(TextEditor::TextEditorWidget *widget,
if (!widget)
return QString();
QTextDocument *doc = widget->document();
DocumentContextReader reader(doc);
DocumentContextReader reader(widget->textDocument());
QString languageAndFileInfo = reader.getLanguageAndFileInfo();
QString contextBefore;
if (settings().readFullFile()) {
@ -110,7 +110,10 @@ QString LLMClientInterface::сontextBefore(TextEditor::TextEditorWidget *widget,
settings().readStringsBeforeCursor());
}
return contextBefore;
return QString("%1\n%2\n%3")
.arg(reader.getSpecificInstructions())
.arg(reader.getLanguageAndFileInfo())
.arg(contextBefore);
}
QString LLMClientInterface::сontextAfter(TextEditor::TextEditorWidget *widget,
@ -120,8 +123,7 @@ QString LLMClientInterface::сontextAfter(TextEditor::TextEditorWidget *widget,
if (!widget)
return QString();
QTextDocument *doc = widget->document();
DocumentContextReader reader(doc);
DocumentContextReader reader(widget->textDocument());
QString contextAfter;
if (settings().readFullFile()) {