gpt4all/gpt4all-chat/localdocs.h

#ifndef LOCALDOCS_H
#define LOCALDOCS_H

#include "database.h"
#include "localdocsmodel.h" // IWYU pragma: keep

#include <QObject>
#include <QString>
#include <QStringList>

class LocalDocs : public QObject
{
    Q_OBJECT
    Q_PROPERTY(bool databaseValid READ databaseValid NOTIFY databaseValidChanged)
    Q_PROPERTY(LocalDocsModel *localDocsModel READ localDocsModel NOTIFY localDocsModelChanged)

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);
    Q_INVOKABLE void forceIndexing(const QString &collection);

    Database *database() const { return m_database; }

    bool databaseValid() const { return m_database->isValid(); }

public Q_SLOTS:
    void handleChunkSizeChanged();
    void handleFileExtensionsChanged();
    void aboutToQuit();

Q_SIGNALS:
    void requestStart();
    void requestForceIndexing(const QString &collection, const QString &embedding_model);
    void forceRebuildFolder(const QString &path);
    void requestAddFolder(const QString &collection, const QString &path, const QString &embedding_model);
    void requestRemoveFolder(const QString &collection, const QString &path);
    void requestChunkSizeChange(int chunkSize);
    void requestFileExtensionsChange(const QStringList &extensions);
    void localDocsModelChanged();
    void databaseValidChanged();

private:
    LocalDocsModel *m_localDocsModel;
    Database *m_database;

private:
    explicit LocalDocs();
    friend class MyLocalDocs;
};

#endif // LOCALDOCS_H