mirror/QodeAssist
1
0
Fork 0
mirror of https://github.com/Palm1r/QodeAssist.git synced 2025-12-20 16:22:48 -05: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

19
context/RAGManager.hpp
View File

@ -40,11 +40,28 @@ class RAGManager : public QObject
public:
static RAGManager &instance();
struct SearchResult
{
QString filePath;
float l2Score;
float cosineScore;
bool operator<(const SearchResult &other) const;
};
// Process and vectorize files
QFuture<void> processFiles(ProjectExplorer::Project *project, const QStringList &filePaths);
std::optional<RAGVector> loadVectorFromStorage(
ProjectExplorer::Project *project, const QString &filePath);
QStringList getStoredFiles(ProjectExplorer::Project *project) const;
bool isFileStorageOutdated(ProjectExplorer::Project *project, const QString &filePath) const;
RAGVectorizer *getVectorizer() const { return m_vectorizer.get(); }
// Search functionality
QFuture<QList<SearchResult>> search(
const QString &text, ProjectExplorer::Project *project, int topK = 5);
void searchSimilarDocuments(const QString &text, ProjectExplorer::Project *project, int topK = 5);
void logSearchResults(const QList<SearchResult> &results) const;
signals:
void vectorizationProgress(int processed, int total);
@ -53,6 +70,8 @@ signals:
private:
RAGManager(QObject *parent = nullptr);
~RAGManager();
RAGManager(const RAGManager &) = delete;
RAGManager &operator=(const RAGManager &) = delete;
QFuture<bool> processFile(ProjectExplorer::Project *project, const QString &filePath);
void processNextBatch(