mirror/QodeAssist
1
0
Fork 0
mirror of https://github.com/Palm1r/QodeAssist.git synced 2025-07-19 13:34:46 -04:00
Code Issues Packages Projects Releases Wiki Activity

feat: Add similarity search

Browse Source
This commit is contained in:
Petr Mironychev
2025-01-26 20:00:56 +01:00
parent 5a426b4d9f
commit 7b73d7af7b
8 changed files with 259 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
ChatView/ChatRootView.cpp
View File

@ -452,17 +452,36 @@ void ChatRootView::openChatHistoryFolder()
void ChatRootView::testRAG()
{
auto project = ProjectExplorer::ProjectTree::currentProject();
if (project) {
auto files = Context::ContextManager::instance().getProjectSourceFiles(project);
auto future = Context::RAGManager::instance().processFiles(project, files);
connect(
&Context::RAGManager::instance(),
&Context::RAGManager::vectorizationProgress,
this,
[](int processed, int total) {
qDebug() << "Processed" << processed << "of" << total << "files";
});
if (!project) {
qDebug() << "No active project found";
return;
}
const QString TEST_QUERY = "";
qDebug() << "Starting RAG test with query:";
qDebug() << TEST_QUERY;
qDebug() << "\nFirst, processing project files...";
auto files = Context::ContextManager::instance().getProjectSourceFiles(project);
auto future = Context::RAGManager::instance().processFiles(project, files);
connect(
&Context::RAGManager::instance(),
&Context::RAGManager::vectorizationProgress,
this,
[](int processed, int total) {
qDebug() << QString("Vectorization progress: %1 of %2 files").arg(processed).arg(total);
});
connect(
&Context::RAGManager::instance(),
&Context::RAGManager::vectorizationFinished,
this,
[this, project, TEST_QUERY]() {
qDebug() << "\nVectorization completed. Starting similarity search...\n";
Context::RAGManager::instance().searchSimilarDocuments(TEST_QUERY, project, 5);
});
}
void ChatRootView::updateInputTokensCount()