mirror of
https://github.com/Palm1r/QodeAssist.git
synced 2025-05-28 03:10:28 -04:00
refactor: Reuse extractFilePathFromRequest() more (#117)
This commit is contained in:
parent
7c483f89cd
commit
c9a3cdaf25
@ -29,6 +29,7 @@
|
|||||||
#include "CodeHandler.hpp"
|
#include "CodeHandler.hpp"
|
||||||
#include "context/ContextManager.hpp"
|
#include "context/ContextManager.hpp"
|
||||||
#include "context/DocumentContextReader.hpp"
|
#include "context/DocumentContextReader.hpp"
|
||||||
|
#include "context/Utils.hpp"
|
||||||
#include "llmcore/PromptTemplateManager.hpp"
|
#include "llmcore/PromptTemplateManager.hpp"
|
||||||
#include "llmcore/ProvidersManager.hpp"
|
#include "llmcore/ProvidersManager.hpp"
|
||||||
#include "logger/Logger.hpp"
|
#include "logger/Logger.hpp"
|
||||||
@ -37,14 +38,6 @@
|
|||||||
|
|
||||||
namespace QodeAssist {
|
namespace QodeAssist {
|
||||||
|
|
||||||
QString extractFilePathFromRequest(const QJsonObject &request)
|
|
||||||
{
|
|
||||||
QJsonObject params = request["params"].toObject();
|
|
||||||
QJsonObject doc = params["doc"].toObject();
|
|
||||||
QString uri = doc["uri"].toString();
|
|
||||||
return QUrl(uri).toLocalFile();
|
|
||||||
}
|
|
||||||
|
|
||||||
LLMClientInterface::LLMClientInterface(
|
LLMClientInterface::LLMClientInterface(
|
||||||
const Settings::GeneralSettings &generalSettings,
|
const Settings::GeneralSettings &generalSettings,
|
||||||
const Settings::CodeCompletionSettings &completeSettings)
|
const Settings::CodeCompletionSettings &completeSettings)
|
||||||
@ -263,20 +256,20 @@ LLMCore::ContextData LLMClientInterface::prepareContext(
|
|||||||
QJsonObject doc = params["doc"].toObject();
|
QJsonObject doc = params["doc"].toObject();
|
||||||
QJsonObject position = doc["position"].toObject();
|
QJsonObject position = doc["position"].toObject();
|
||||||
|
|
||||||
Utils::FilePath filePath = Utils::FilePath::fromString(extractFilePathFromRequest(request));
|
auto filePath = Context::extractFilePathFromRequest(request);
|
||||||
TextEditor::TextDocument *textDocument = TextEditor::TextDocument::textDocumentForFilePath(
|
TextEditor::TextDocument *textDocument = TextEditor::TextDocument::textDocumentForFilePath(
|
||||||
filePath);
|
Utils::FilePath::fromString(filePath));
|
||||||
|
|
||||||
if (!textDocument) {
|
if (!textDocument) {
|
||||||
LOG_MESSAGE("Error: Document is not available for" + filePath.toString());
|
LOG_MESSAGE("Error: Document is not available for" + filePath);
|
||||||
return LLMCore::ContextData{};
|
return LLMCore::ContextData{};
|
||||||
}
|
}
|
||||||
|
|
||||||
int cursorPosition = position["character"].toInt();
|
int cursorPosition = position["character"].toInt();
|
||||||
int lineNumber = position["line"].toInt();
|
int lineNumber = position["line"].toInt();
|
||||||
|
|
||||||
Context::DocumentContextReader reader(
|
Context::DocumentContextReader
|
||||||
textDocument->document(), textDocument->mimeType(), textDocument->filePath().toString());
|
reader(textDocument->document(), textDocument->mimeType(), filePath);
|
||||||
return reader.prepareContext(lineNumber, cursorPosition, m_completeSettings);
|
return reader.prepareContext(lineNumber, cursorPosition, m_completeSettings);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -306,7 +299,7 @@ void LLMClientInterface::sendCompletionToClient(
|
|||||||
QString processedCompletion
|
QString processedCompletion
|
||||||
= promptTemplate->type() == LLMCore::TemplateType::Chat
|
= promptTemplate->type() == LLMCore::TemplateType::Chat
|
||||||
&& m_completeSettings.smartProcessInstuctText()
|
&& m_completeSettings.smartProcessInstuctText()
|
||||||
? CodeHandler::processText(completion, extractFilePathFromRequest(request))
|
? CodeHandler::processText(completion, Context::extractFilePathFromRequest(request))
|
||||||
: completion;
|
: completion;
|
||||||
|
|
||||||
completionItem[LanguageServerProtocol::textKey] = processedCompletion;
|
completionItem[LanguageServerProtocol::textKey] = processedCompletion;
|
||||||
|
@ -26,6 +26,7 @@
|
|||||||
|
|
||||||
#include "GeneralSettings.hpp"
|
#include "GeneralSettings.hpp"
|
||||||
#include "Logger.hpp"
|
#include "Logger.hpp"
|
||||||
|
#include "Utils.hpp"
|
||||||
#include <texteditor/textdocument.h>
|
#include <texteditor/textdocument.h>
|
||||||
#include <utils/filepath.h>
|
#include <utils/filepath.h>
|
||||||
|
|
||||||
@ -72,16 +73,12 @@ ContentFile ContextManager::createContentFile(const QString &filePath) const
|
|||||||
|
|
||||||
ProgrammingLanguage ContextManager::getDocumentLanguage(const QJsonObject &request)
|
ProgrammingLanguage ContextManager::getDocumentLanguage(const QJsonObject &request)
|
||||||
{
|
{
|
||||||
QJsonObject params = request["params"].toObject();
|
auto filePath = extractFilePathFromRequest(request);
|
||||||
QJsonObject doc = params["doc"].toObject();
|
|
||||||
QString uri = doc["uri"].toString();
|
|
||||||
|
|
||||||
Utils::FilePath filePath = Utils::FilePath::fromString(QUrl(uri).toLocalFile());
|
|
||||||
TextEditor::TextDocument *textDocument = TextEditor::TextDocument::textDocumentForFilePath(
|
TextEditor::TextDocument *textDocument = TextEditor::TextDocument::textDocumentForFilePath(
|
||||||
filePath);
|
Utils::FilePath::fromString(filePath));
|
||||||
|
|
||||||
if (!textDocument) {
|
if (!textDocument) {
|
||||||
LOG_MESSAGE("Error: Document is not available for" + filePath.toFSPathString());
|
LOG_MESSAGE("Error: Document is not available for" + filePath);
|
||||||
return Context::ProgrammingLanguage::Unknown;
|
return Context::ProgrammingLanguage::Unknown;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
34
context/Utils.hpp
Normal file
34
context/Utils.hpp
Normal file
@ -0,0 +1,34 @@
|
|||||||
|
/*
|
||||||
|
* Copyright (C) 2025 Povilas Kanapickas <povilas@radix.lt>
|
||||||
|
*
|
||||||
|
* This file is part of QodeAssist.
|
||||||
|
*
|
||||||
|
* QodeAssist is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by
|
||||||
|
* the Free Software Foundation, either version 3 of the License, or
|
||||||
|
* (at your option) any later version.
|
||||||
|
*
|
||||||
|
* QodeAssist is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
* GNU General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License
|
||||||
|
* along with QodeAssist. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <QJsonObject>
|
||||||
|
|
||||||
|
namespace QodeAssist::Context {
|
||||||
|
|
||||||
|
inline QString extractFilePathFromRequest(const QJsonObject &request)
|
||||||
|
{
|
||||||
|
QJsonObject params = request["params"].toObject();
|
||||||
|
QJsonObject doc = params["doc"].toObject();
|
||||||
|
QString uri = doc["uri"].toString();
|
||||||
|
return QUrl(uri).toLocalFile();
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace QodeAssist::Context
|
Loading…
Reference in New Issue
Block a user