mirror/QodeAssist
1
0
Fork 0
mirror of https://github.com/Palm1r/QodeAssist.git synced 2025-07-18 21:14:34 -04:00
Code Issues Packages Projects Releases Wiki Activity

✨ feat: Add OpenRouter provider

Browse Source
This commit is contained in:
Petr Mironychev
2024-12-10 21:27:56 +01:00
parent 882047d7b2
commit c070fd5cfd
8 changed files with 233 additions and 111 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
llmcore/OpenAIMessage.cpp
View File

@ -20,46 +20,13 @@
#include "OpenAIMessage.hpp"
#include <QJsonArray>
#include <QJsonDocument>
#include <QJsonObject>
namespace QodeAssist::LLMCore {
OpenAIMessage OpenAIMessage::fromJson(const QByteArray &data)
OpenAIMessage OpenAIMessage::fromJson(const QJsonObject &obj)
{
OpenAIMessage msg;
QByteArrayList lines = data.split('\n');
QByteArray jsonData;
for (const QByteArray &line : lines) {
if (line.trimmed().isEmpty()) {
continue;
}
if (line.trimmed() == "data: [DONE]") {
msg.done = true;
continue;
}
if (line.startsWith("data: ")) {
jsonData = line.mid(6);
break;
}
}
if (jsonData.isEmpty()) {
jsonData = data;
}
QJsonParseError error;
QJsonDocument doc = QJsonDocument::fromJson(jsonData, &error);
if (doc.isNull()) {
msg.error = QString("Invalid JSON response: %1").arg(error.errorString());
return msg;
}
QJsonObject obj = doc.object();
if (obj.contains("error")) {
msg.error = obj["error"].toObject()["message"].toString();
return msg;
@ -70,10 +37,12 @@ OpenAIMessage OpenAIMessage::fromJson(const QByteArray &data)
if (!choices.isEmpty()) {
auto choiceObj = choices[0].toObject();
if (choiceObj.contains("message")) {
msg.choice.content = choiceObj["message"].toObject()["content"].toString();
} else if (choiceObj.contains("delta")) {
msg.choice.content = choiceObj["delta"].toObject()["content"].toString();
if (choiceObj.contains("delta")) {
QJsonObject delta = choiceObj["delta"].toObject();
msg.choice.content = delta["content"].toString();
} else if (choiceObj.contains("message")) {
QJsonObject message = choiceObj["message"].toObject();
msg.choice.content = message["content"].toString();
}
msg.choice.finishReason = choiceObj["finish_reason"].toString();

4
llmcore/OpenAIMessage.hpp
View File

@ -46,13 +46,11 @@ public:
bool done{false};
Usage usage;
static OpenAIMessage fromJson(const QByteArray &data);
QString getContent() const;
bool hasError() const;
bool isDone() const;
private:
static OpenAIMessage fromJsonObject(const QJsonObject &obj);
static OpenAIMessage fromJson(const QJsonObject &obj);
};
} // namespace QodeAssist::LLMCore