mirror of
https://github.com/nomic-ai/gpt4all
synced 2024-11-04 12:00:10 +00:00
0ea31487e3
for applicable models in the directory of the executable given a pattern match and then loads the first one it finds. Also, add a busy indicator for model loading.
97 lines
2.3 KiB
C++
97 lines
2.3 KiB
C++
#ifndef LLM_H
|
|
#define LLM_H
|
|
|
|
#include <QObject>
|
|
#include <QThread>
|
|
#include "gptj.h"
|
|
|
|
class GPTJObject : public QObject
|
|
{
|
|
Q_OBJECT
|
|
Q_PROPERTY(bool isModelLoaded READ isModelLoaded NOTIFY isModelLoadedChanged)
|
|
Q_PROPERTY(QString response READ response NOTIFY responseChanged)
|
|
Q_PROPERTY(QString modelName READ modelName NOTIFY modelNameChanged)
|
|
|
|
public:
|
|
|
|
GPTJObject();
|
|
|
|
bool loadModel();
|
|
bool isModelLoaded() const;
|
|
void resetResponse();
|
|
void resetContext();
|
|
void stopGenerating() { m_stopGenerating = true; }
|
|
|
|
QString response() const;
|
|
QString modelName() const;
|
|
|
|
public Q_SLOTS:
|
|
bool prompt(const QString &prompt);
|
|
|
|
Q_SIGNALS:
|
|
void isModelLoadedChanged();
|
|
void responseChanged();
|
|
void responseStarted();
|
|
void responseStopped();
|
|
void modelNameChanged();
|
|
|
|
private:
|
|
bool handleResponse(const std::string &response);
|
|
|
|
private:
|
|
GPTJ *m_gptj;
|
|
std::string m_response;
|
|
QString m_modelName;
|
|
QThread m_llmThread;
|
|
std::atomic<bool> m_stopGenerating;
|
|
};
|
|
|
|
class LLM : public QObject
|
|
{
|
|
Q_OBJECT
|
|
Q_PROPERTY(bool isModelLoaded READ isModelLoaded NOTIFY isModelLoadedChanged)
|
|
Q_PROPERTY(QString response READ response NOTIFY responseChanged)
|
|
Q_PROPERTY(QString modelName READ modelName NOTIFY modelNameChanged)
|
|
Q_PROPERTY(bool responseInProgress READ responseInProgress NOTIFY responseInProgressChanged)
|
|
public:
|
|
|
|
static LLM *globalInstance();
|
|
|
|
Q_INVOKABLE bool isModelLoaded() const;
|
|
Q_INVOKABLE void prompt(const QString &prompt);
|
|
Q_INVOKABLE void resetContext();
|
|
Q_INVOKABLE void resetResponse();
|
|
Q_INVOKABLE void stopGenerating();
|
|
|
|
QString response() const;
|
|
bool responseInProgress() const { return m_responseInProgress; }
|
|
|
|
QString modelName() const;
|
|
|
|
Q_INVOKABLE bool checkForUpdates() const;
|
|
|
|
Q_SIGNALS:
|
|
void isModelLoadedChanged();
|
|
void responseChanged();
|
|
void responseInProgressChanged();
|
|
void promptRequested(const QString &prompt);
|
|
void resetResponseRequested();
|
|
void resetContextRequested();
|
|
void modelNameChanged();
|
|
|
|
private Q_SLOTS:
|
|
void responseStarted();
|
|
void responseStopped();
|
|
|
|
private:
|
|
GPTJObject *m_gptj;
|
|
bool m_responseInProgress;
|
|
|
|
private:
|
|
explicit LLM();
|
|
~LLM() {}
|
|
friend class MyLLM;
|
|
};
|
|
|
|
#endif // LLM_H
|