CodeWizard/groqai.cpp at master · AdamJosephMather/CodeWizard · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
#include "groqai.h"
#include <qboxlayout.h>
#include <qdialog.h>
#include <qlabel.h>
#include "mytextedit.h"
#include <QtNetwork/QNetworkAccessManager>
#include <QtNetwork/QNetworkRequest>
#include <QtNetwork/QNetworkReply>
#include <QtNetwork/QtNetwork>

GroqAI::GroqAI(QObject *parent)
	: QObject(parent),
	  m_networkManager(new QNetworkAccessManager(this))
{
	connect(m_networkManager, &QNetworkAccessManager::finished, this, &GroqAI::onNetworkReply);
}

GroqAI::~GroqAI()
{
}

void GroqAI::setApiKey(const QString &apiKey)
{
	m_apiKey = apiKey;
}

void GroqAI::setModel(const QString &model)
{
	m_model = model;
}

void GroqAI::setTextEdit(MyTextEdit* tE)
{
	textEdit = tE;
}

void GroqAI::openHelpMenu(QString text) {
	qDebug() << "openHelpMenu";

	// Create a simple help dialog
	QDialog *helpDialog = new QDialog();
	helpDialog->setWindowTitle("Help Menu");

	helpDialog->resize(700, 100); // seems to resize to the apropriate size given the text or something - it's the label that causes this I think

	QLabel *helpText = new QLabel(text, helpDialog);
	helpText->setWordWrap(true);
	helpText->setFont(textEdit->font());
	helpText->setTextInteractionFlags(Qt::TextSelectableByMouse);

	QVBoxLayout *layout = new QVBoxLayout(helpDialog);
	layout->addWidget(helpText);

	helpDialog->setLayout(layout);
	helpDialog->exec();
}

void GroqAI::generateResponse(const QList<QPair<QString, QString>> &conversation)
{
	if (m_apiKey.isEmpty() || m_model.isEmpty()) {
		openHelpMenu("API key or model is not set for Groq AI.");
		return;
	}

	QUrl apiUrl("https://api.groq.com/openai/v1/chat/completions");
	QNetworkRequest request(apiUrl);
	request.setHeader(QNetworkRequest::ContentTypeHeader, "application/json");
	request.setRawHeader("Authorization", "Bearer " + m_apiKey.toUtf8());

	QJsonObject payload;
	payload["model"] = m_model;
	payload["messages"] = formatConversation(conversation);

	QJsonDocument doc(payload);
	QByteArray data = doc.toJson();

	m_networkManager->post(request, data);
}

void GroqAI::generateResponse(const QStringList &senders, const QStringList &messages)
{
	if (senders.size() != messages.size()) {
		qDebug() << "Senders != Messages length";
		return;
	}

	QList<QPair<QString, QString>> conversation;
	for (int i = 0; i < senders.size(); ++i) {
		conversation.append(qMakePair(senders[i], messages[i]));
	}

	generateResponse(conversation);
}

QJsonArray GroqAI::formatConversation(const QList<QPair<QString, QString>> &conversation)
{
	QJsonArray messages;
	for (const QPair<QString, QString> &entry : conversation) {
		QJsonObject messageObject;
		messageObject["role"] = entry.first;
		messageObject["content"] = entry.second;
		messages.append(messageObject);
	}
	return messages;
}

QJsonArray GroqAI::formatConversation(const QStringList &senders, const QStringList &messages)
{
	QJsonArray formatted;
	for (int i = 0; i < senders.size(); ++i) {
		QJsonObject messageObject;
		messageObject["sender"] = senders[i];
		messageObject["content"] = messages[i];
		formatted.append(messageObject);
	}
	return formatted;
}

void GroqAI::onNetworkReply(QNetworkReply *reply)
{
	if (reply->error() != QNetworkReply::NoError) {
		openHelpMenu(reply->errorString());
		qDebug() << reply->errorString();
		reply->deleteLater();
		return;
	}

	QByteArray responseData = reply->readAll();
	QJsonDocument jsonResponse = QJsonDocument::fromJson(responseData);
	if (jsonResponse.isObject()) {
		QJsonObject responseObject = jsonResponse.object();
		if (responseObject.contains("choices")) {
			qDebug() << responseObject["choices"];
			qDebug() << responseObject["choices"].toArray()[0];
			qDebug() << responseObject["choices"].toArray()[0].toObject()["message"];
			qDebug() << responseObject["choices"].toArray()[0].toObject()["message"].toObject()["content"];

			emit responseReceived(responseObject["choices"].toArray()[0].toObject()["message"].toObject()["content"].toString());
		} else {
			qDebug() << "Invalid response format." << responseData << "\n\n" << jsonResponse;
		}
	} else {
		qDebug() << "Failed to parse JSON response.\n\n" << responseData;
	}

	reply->deleteLater();
}