mirror of
https://github.com/nomic-ai/gpt4all.git
synced 2024-09-11 13:25:35 +03:00
127 lines
3.4 KiB
C++
127 lines
3.4 KiB
C++
|
#include "network.h"
|
||
|
#include "llm.h"
|
||
|
|
||
|
#include <QCoreApplication>
|
||
|
#include <QUuid>
|
||
|
#include <QJsonDocument>
|
||
|
#include <QJsonArray>
|
||
|
#include <QJsonObject>
|
||
|
#include <QSettings>
|
||
|
#include <QNetworkRequest>
|
||
|
|
||
|
//#define DEBUG
|
||
|
|
||
|
class MyNetwork: public Network { };
|
||
|
Q_GLOBAL_STATIC(MyNetwork, networkInstance)
|
||
|
Network *Network::globalInstance()
|
||
|
{
|
||
|
return networkInstance();
|
||
|
}
|
||
|
|
||
|
Network::Network()
|
||
|
: QObject{nullptr}
|
||
|
, m_isActive(false)
|
||
|
{
|
||
|
QSettings settings;
|
||
|
settings.sync();
|
||
|
m_isActive = settings.value("network/isActive", false).toBool();
|
||
|
m_uniqueId = settings.value("uniqueId", generateUniqueId()).toString();
|
||
|
settings.setValue("uniqueId", m_uniqueId);
|
||
|
settings.sync();
|
||
|
emit activeChanged();
|
||
|
}
|
||
|
|
||
|
void Network::setActive(bool b)
|
||
|
{
|
||
|
QSettings settings;
|
||
|
settings.setValue("network/isActive", b);
|
||
|
settings.sync();
|
||
|
m_isActive = b;
|
||
|
emit activeChanged();
|
||
|
}
|
||
|
|
||
|
QString Network::generateUniqueId() const
|
||
|
{
|
||
|
return QUuid::createUuid().toString(QUuid::WithoutBraces);
|
||
|
}
|
||
|
|
||
|
bool Network::packageAndSendJson(const QString &json)
|
||
|
{
|
||
|
if (!m_isActive)
|
||
|
return false;
|
||
|
|
||
|
QJsonParseError err;
|
||
|
QJsonDocument doc = QJsonDocument::fromJson(json.toUtf8(), &err);
|
||
|
if (err.error != QJsonParseError::NoError) {
|
||
|
qDebug() << "Couldn't parse: " << json << err.errorString();
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
Q_ASSERT(doc.isObject());
|
||
|
QJsonObject object = doc.object();
|
||
|
object.insert("source", "gpt4all-chat");
|
||
|
object.insert("agent_id", LLM::globalInstance()->modelName());
|
||
|
object.insert("submitter_id", m_uniqueId);
|
||
|
|
||
|
QSettings settings;
|
||
|
settings.sync();
|
||
|
QString attribution = settings.value("attribution", QString()).toString();
|
||
|
if (!attribution.isEmpty())
|
||
|
object.insert("attribution", attribution);
|
||
|
|
||
|
QJsonDocument newDoc;
|
||
|
newDoc.setObject(object);
|
||
|
|
||
|
#if defined(DEBUG)
|
||
|
printf("%s", qPrintable(newDoc.toJson(QJsonDocument::Indented)));
|
||
|
fflush(stdout);
|
||
|
#endif
|
||
|
|
||
|
QUrl jsonUrl("http://localhost/v1/ingest/chat");
|
||
|
QNetworkRequest request(jsonUrl);
|
||
|
QByteArray body(newDoc.toJson());
|
||
|
request.setHeader(QNetworkRequest::ContentTypeHeader, "application/json");
|
||
|
QNetworkReply *jsonReply = m_networkManager.post(request, body);
|
||
|
connect(jsonReply, &QNetworkReply::finished, this, &Network::handleJsonUploadFinished);
|
||
|
m_activeUploads.append(jsonReply);
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
void Network::handleJsonUploadFinished()
|
||
|
{
|
||
|
QNetworkReply *jsonReply = qobject_cast<QNetworkReply *>(sender());
|
||
|
if (!jsonReply)
|
||
|
return;
|
||
|
|
||
|
m_activeUploads.removeAll(jsonReply);
|
||
|
|
||
|
QVariant response = jsonReply->attribute(QNetworkRequest::HttpStatusCodeAttribute);
|
||
|
Q_ASSERT(response.isValid());
|
||
|
bool ok;
|
||
|
int code = response.toInt(&ok);
|
||
|
if (!ok)
|
||
|
qWarning() << "ERROR: Invalid response.";
|
||
|
if (code != 200)
|
||
|
qWarning() << "ERROR: response != 200 code:" << code;
|
||
|
|
||
|
QByteArray jsonData = jsonReply->readAll();
|
||
|
QJsonParseError err;
|
||
|
QJsonDocument document = QJsonDocument::fromJson(jsonData, &err);
|
||
|
if (err.error != QJsonParseError::NoError) {
|
||
|
qDebug() << "ERROR: Couldn't parse: " << jsonData << err.errorString();
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
#if defined(DEBUG)
|
||
|
printf("%s", qPrintable(document.toJson(QJsonDocument::Indented)));
|
||
|
fflush(stdout);
|
||
|
#endif
|
||
|
|
||
|
jsonReply->deleteLater();
|
||
|
}
|
||
|
|
||
|
bool Network::sendConversation(const QString &conversation)
|
||
|
{
|
||
|
return packageAndSendJson(conversation);
|
||
|
}
|