mirror of
https://github.com/nomic-ai/gpt4all
synced 2024-11-02 09:40:42 +00:00
54 lines
1.7 KiB
C++
54 lines
1.7 KiB
C++
#ifndef LOCALDOCS_H
|
|
#define LOCALDOCS_H
|
|
|
|
#include "localdocsmodel.h"
|
|
#include "database.h"
|
|
|
|
#include <QObject>
|
|
|
|
class LocalDocs : public QObject
|
|
{
|
|
Q_OBJECT
|
|
Q_PROPERTY(LocalDocsModel *localDocsModel READ localDocsModel NOTIFY localDocsModelChanged)
|
|
Q_PROPERTY(int chunkSize READ chunkSize WRITE setChunkSize NOTIFY chunkSizeChanged)
|
|
Q_PROPERTY(int retrievalSize READ retrievalSize WRITE setRetrievalSize NOTIFY retrievalSizeChanged)
|
|
|
|
public:
|
|
static LocalDocs *globalInstance();
|
|
|
|
LocalDocsModel *localDocsModel() const { return m_localDocsModel; }
|
|
|
|
Q_INVOKABLE void addFolder(const QString &collection, const QString &path);
|
|
Q_INVOKABLE void removeFolder(const QString &collection, const QString &path);
|
|
void requestRetrieve(const QString &uid, const QList<QString> &collections, const QString &text);
|
|
|
|
int chunkSize() const;
|
|
void setChunkSize(int chunkSize);
|
|
|
|
int retrievalSize() const;
|
|
void setRetrievalSize(int retrievalSize);
|
|
|
|
Q_SIGNALS:
|
|
void requestAddFolder(const QString &collection, const QString &path);
|
|
void requestRemoveFolder(const QString &collection, const QString &path);
|
|
void requestRetrieveFromDB(const QString &uid, const QList<QString> &collections, const QString &text, int retrievalSize);
|
|
void requestChunkSizeChange(int chunkSize);
|
|
void receivedResult(const QString &uid, const QList<ResultInfo> &result);
|
|
void localDocsModelChanged();
|
|
void chunkSizeChanged();
|
|
void retrievalSizeChanged();
|
|
|
|
private:
|
|
int m_chunkSize;
|
|
int m_retrievalSize;
|
|
LocalDocsModel *m_localDocsModel;
|
|
Database *m_database;
|
|
|
|
private:
|
|
explicit LocalDocs();
|
|
~LocalDocs() {}
|
|
friend class MyLocalDocs;
|
|
};
|
|
|
|
#endif // LOCALDOCS_H
|