✨ feat: Add OpenRouter provider

CMakeLists.txt

@@ -56,6 +56,7 @@ add_qtc_plugin(QodeAssist
     providers/OllamaProvider.hpp providers/OllamaProvider.cpp
     providers/LMStudioProvider.hpp providers/LMStudioProvider.cpp
     providers/OpenAICompatProvider.hpp providers/OpenAICompatProvider.cpp
+    providers/OpenRouterProvider.hpp providers/OpenRouterProvider.cpp
     QodeAssist.qrc
     LSPCompletion.hpp
     LLMSuggestion.hpp LLMSuggestion.cpp

llmcore/OpenAIMessage.cpp

@@ -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();

llmcore/OpenAIMessage.hpp

@@ -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

providers/LMStudioProvider.cpp

@@ -107,16 +107,37 @@ bool LMStudioProvider::handleResponse(QNetworkReply *reply, QString &accumulated
         return false;
     }
 
-    auto message = LLMCore::OpenAIMessage::fromJson(data);
+    QByteArrayList chunks = data.split('\n');
+    for (const QByteArray &chunk : chunks) {
+        if (chunk.trimmed().isEmpty() || chunk == "data: [DONE]") {
+            continue;
+        }
+
+        QByteArray jsonData = chunk;
+        if (chunk.startsWith("data: ")) {
+            jsonData = chunk.mid(6);
+        }
+
+        QJsonParseError error;
+        QJsonDocument doc = QJsonDocument::fromJson(jsonData, &error);
+
+        if (doc.isNull()) {
+            continue;
+        }
+
+        auto message = LLMCore::OpenAIMessage::fromJson(doc.object());
         if (message.hasError()) {
-        LOG_MESSAGE("Error in OpenAI response: " + message.error);
-        return false;
+            LOG_MESSAGE("Error in LMStudioProvider response: " + message.error);
+            continue;
         }
 
         accumulatedResponse += message.getContent();
         return message.isDone();
     }
 
+    return false;
+}
+
 QList<QString> LMStudioProvider::getInstalledModels(const QString &url)
 {
     QList<QString> models;

providers/OpenAICompatProvider.cpp

@@ -26,6 +26,7 @@
 #include <QJsonObject>
 #include <QNetworkReply>
 
+#include "llmcore/OpenAIMessage.hpp"
 #include "logger/Logger.hpp"
 
 namespace QodeAssist::Providers {
@@ -100,74 +101,40 @@ void OpenAICompatProvider::prepareRequest(QJsonObject &request, LLMCore::Request
 
 bool OpenAICompatProvider::handleResponse(QNetworkReply *reply, QString &accumulatedResponse)
 {
-    bool isComplete = false;
-    QString tempResponse = accumulatedResponse;
-
-    while (reply->canReadLine()) {
-        QByteArray line = reply->readLine().trimmed();
-        if (line.isEmpty()) {
-            continue;
-        }
-
-        if (!line.startsWith("data:")) {
-            continue;
-        }
-
-        line = line.mid(6);
-
-        if (line == "[DONE]") {
-            isComplete = true;
-            break;
-        }
-
-        QJsonDocument jsonResponse = QJsonDocument::fromJson(line);
-        if (jsonResponse.isNull()) {
-            LOG_MESSAGE(
-                "Invalid JSON response from OpenAI compatible provider: " + QString::fromUtf8(line));
-            continue;
-        }
-
-        QJsonObject responseObj = jsonResponse.object();
-
-        if (responseObj.contains("error")) {
-            LOG_MESSAGE(
-                "OpenAI compatible provider error: "
-                + QString::fromUtf8(QJsonDocument(responseObj).toJson(QJsonDocument::Indented)));
+    QByteArray data = reply->readAll();
+    if (data.isEmpty()) {
         return false;
     }
 
-        if (responseObj.contains("choices")) {
-            QJsonArray choices = responseObj["choices"].toArray();
-            if (!choices.isEmpty()) {
-                QJsonObject choice = choices.first().toObject();
-                QJsonObject delta = choice["delta"].toObject();
-                if (delta.contains("content")) {
-                    QString completion = delta["content"].toString();
-                    if (!completion.isEmpty()) {
-                        tempResponse += completion;
-                    }
-                }
-                QString finishReason = choice["finish_reason"].toString();
-                if (!finishReason.isNull() && finishReason == "stop") {
-                    isComplete = true;
-                }
-            }
+    QByteArrayList chunks = data.split('\n');
+    for (const QByteArray &chunk : chunks) {
+        if (chunk.trimmed().isEmpty() || chunk == "data: [DONE]") {
+            continue;
         }
 
-        if (responseObj.contains("usage")) {
-            QJsonObject usage = responseObj["usage"].toObject();
-            LOG_MESSAGE(QString("Token usage - Prompt: %1, Completion: %2, Total: %3")
-                            .arg(usage["prompt_tokens"].toInt())
-                            .arg(usage["completion_tokens"].toInt())
-                            .arg(usage["total_tokens"].toInt()));
-        }
+        QByteArray jsonData = chunk;
+        if (chunk.startsWith("data: ")) {
+            jsonData = chunk.mid(6);
         }
 
-    if (!tempResponse.isEmpty()) {
-        accumulatedResponse = tempResponse;
+        QJsonParseError error;
+        QJsonDocument doc = QJsonDocument::fromJson(jsonData, &error);
+
+        if (doc.isNull()) {
+            continue;
         }
 
-    return isComplete;
+        auto message = LLMCore::OpenAIMessage::fromJson(doc.object());
+        if (message.hasError()) {
+            LOG_MESSAGE("Error in OpenAI response: " + message.error);
+            continue;
+        }
+
+        accumulatedResponse += message.getContent();
+        return message.isDone();
+    }
+
+    return false;
 }
 
 QList<QString> OpenAICompatProvider::getInstalledModels(const QString &url)

providers/OpenRouterProvider.cpp

@@ -0,0 +1,126 @@
+/* 
+ * Copyright (C) 2024 Petr Mironychev
+ *
+ * This file is part of QodeAssist.
+ *
+ * QodeAssist is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * QodeAssist is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with QodeAssist. If not, see <https://www.gnu.org/licenses/>.
+ */
+
+#include "OpenRouterProvider.hpp"
+#include "settings/ChatAssistantSettings.hpp"
+#include "settings/CodeCompletionSettings.hpp"
+
+#include <QJsonArray>
+#include <QJsonDocument>
+#include <QJsonObject>
+#include <QNetworkReply>
+
+#include "llmcore/OpenAIMessage.hpp"
+#include "logger/Logger.hpp"
+
+namespace QodeAssist::Providers {
+
+OpenRouterProvider::OpenRouterProvider() {}
+
+QString OpenRouterProvider::name() const
+{
+    return "OpenRouter";
+}
+
+QString OpenRouterProvider::url() const
+{
+    return "https://openrouter.ai/api";
+}
+
+void OpenRouterProvider::prepareRequest(QJsonObject &request, LLMCore::RequestType type)
+{
+    auto prepareMessages = [](QJsonObject &req) -> QJsonArray {
+        QJsonArray messages;
+        if (req.contains("system")) {
+            messages.append(
+                QJsonObject{{"role", "system"}, {"content", req.take("system").toString()}});
+        }
+        if (req.contains("prompt")) {
+            messages.append(
+                QJsonObject{{"role", "user"}, {"content", req.take("prompt").toString()}});
+        }
+        return messages;
+    };
+
+    auto applyModelParams = [&request](const auto &settings) {
+        request["max_tokens"] = settings.maxTokens();
+        request["temperature"] = settings.temperature();
+
+        if (settings.useTopP())
+            request["top_p"] = settings.topP();
+        if (settings.useTopK())
+            request["top_k"] = settings.topK();
+        if (settings.useFrequencyPenalty())
+            request["frequency_penalty"] = settings.frequencyPenalty();
+        if (settings.usePresencePenalty())
+            request["presence_penalty"] = settings.presencePenalty();
+    };
+
+    QJsonArray messages = prepareMessages(request);
+    if (!messages.isEmpty()) {
+        request["messages"] = std::move(messages);
+    }
+
+    if (type == LLMCore::RequestType::Fim) {
+        applyModelParams(Settings::codeCompletionSettings());
+    } else {
+        applyModelParams(Settings::chatAssistantSettings());
+    }
+}
+
+bool OpenRouterProvider::handleResponse(QNetworkReply *reply, QString &accumulatedResponse)
+{
+    QByteArray data = reply->readAll();
+    if (data.isEmpty()) {
+        return false;
+    }
+
+    QByteArrayList chunks = data.split('\n');
+    for (const QByteArray &chunk : chunks) {
+        if (chunk.trimmed().isEmpty() || chunk.contains("OPENROUTER PROCESSING")
+            || chunk == "data: [DONE]") {
+            continue;
+        }
+
+        QByteArray jsonData = chunk;
+        if (chunk.startsWith("data: ")) {
+            jsonData = chunk.mid(6);
+        }
+
+        QJsonParseError error;
+        QJsonDocument doc = QJsonDocument::fromJson(jsonData, &error);
+
+        if (doc.isNull()) {
+            continue;
+        }
+
+        auto message = LLMCore::OpenAIMessage::fromJson(doc.object());
+        if (message.hasError()) {
+            LOG_MESSAGE("Error in OpenRouter response: " + message.error);
+            continue;
+        }
+
+        accumulatedResponse += message.getContent();
+        return message.isDone();
+    }
+
+    return false;
+}
+
+} // namespace QodeAssist::Providers

providers/OpenrouterProvider.hpp

@@ -0,0 +1,38 @@
+/* 
+ * Copyright (C) 2024 Petr Mironychev
+ *
+ * This file is part of QodeAssist.
+ *
+ * QodeAssist is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * QodeAssist is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with QodeAssist. If not, see <https://www.gnu.org/licenses/>.
+ */
+
+#pragma once
+
+#include "llmcore/Provider.hpp"
+#include "providers/OpenAICompatProvider.hpp"
+
+namespace QodeAssist::Providers {
+
+class OpenRouterProvider : public OpenAICompatProvider
+{
+public:
+    OpenRouterProvider();
+
+    QString name() const override;
+    QString url() const override;
+    void prepareRequest(QJsonObject &request, LLMCore::RequestType type) override;
+    bool handleResponse(QNetworkReply *reply, QString &accumulatedResponse) override;
+};
+
+} // namespace QodeAssist::Providers

providers/Providers.hpp

@@ -23,6 +23,7 @@
 #include "providers/LMStudioProvider.hpp"
 #include "providers/OllamaProvider.hpp"
 #include "providers/OpenAICompatProvider.hpp"
+#include "providers/OpenRouterProvider.hpp"
 
 namespace QodeAssist::Providers {
 
@@ -32,6 +33,7 @@ inline void registerProviders()
     providerManager.registerProvider<OllamaProvider>();
     providerManager.registerProvider<LMStudioProvider>();
     providerManager.registerProvider<OpenAICompatProvider>();
+    providerManager.registerProvider<OpenRouterProvider>();
 }
 
 } // namespace QodeAssist::Providers