This repository has been archived on 2024-10-24. You can view files and clone it, but cannot push or open issues or pull requests.
NG_2024_PacketAnalyzer/llmmanager.cpp

31 lines
916 B
C++

#include "llmmanager.h"
LlmManager::LlmManager(QObject *parent)
: QObject{parent}
{
connect (m_manager, &QNetworkAccessManager::finished, this, &LlmManager::llmResponse);
}
void LlmManager::llmRequest(QString prompt)
{
QNetworkRequest request(QUrl("http://127.0.0.1:11434/api/generate"));
request.setHeader(QNetworkRequest::ContentTypeHeader, "application/json");
QJsonObject json;
json["model"] = "llama3.1:8b";
json["prompt"] = prompt;
json["stream"] = false;
m_manager->post(request, QJsonDocument(json).toJson());
}
void LlmManager::llmResponse(QNetworkReply *reply)
{
if (reply->error() == QNetworkReply::NoError) {
QByteArray textReply = reply->readAll();
QString responseText = QJsonDocument::fromJson(textReply).object().value("response").toString();
emit responseReceived(responseText);
} else {
emit errorReceived();
}
}