feat: Add Google AI provider and template

templates/GoogleAI.hpp

@@ -0,0 +1,72 @@
+/* 
+ * 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 <QJsonArray>
+#include <QJsonObject>
+
+#include "llmcore/PromptTemplate.hpp"
+
+namespace QodeAssist::Templates {
+
+class GoogleAI : public LLMCore::PromptTemplate
+{
+public:
+    LLMCore::TemplateType type() const override { return LLMCore::TemplateType::Chat; }
+    QString name() const override { return "Google AI"; }
+    QStringList stopWords() const override { return QStringList(); }
+
+    void prepareRequest(QJsonObject &request, const LLMCore::ContextData &context) const override
+    {
+        QJsonArray contents;
+
+        if (context.systemPrompt && !context.systemPrompt->isEmpty()) {
+            request["system_instruction"] = QJsonObject{
+                {"parts", QJsonObject{{"text", context.systemPrompt.value()}}}};
+        }
+
+        for (const auto &msg : context.history.value()) {
+            QJsonObject content;
+            QJsonArray parts;
+
+            parts.append(QJsonObject{{"text", msg.content}});
+
+            QString role = msg.role;
+            if (role == "assistant") {
+                role = "model";
+            }
+
+            content["role"] = role;
+            content["parts"] = parts;
+            contents.append(content);
+        }
+
+        request["contents"] = contents;
+    }
+
+    QString description() const override { return "Google AI (Gemini)"; }
+
+    bool isSupportProvider(LLMCore::ProviderID id) const override
+    {
+        return id == QodeAssist::LLMCore::ProviderID::GoogleAI;
+    }
+};
+
+} // namespace QodeAssist::Templates

templates/Templates.hpp

@@ -31,6 +31,7 @@
 #include "templates/OpenAICompatible.hpp"
 // #include "templates/CustomFimTemplate.hpp"
 // #include "templates/DeepSeekCoderFim.hpp"
+#include "templates/GoogleAI.hpp"
 #include "templates/Llama2.hpp"
 #include "templates/Llama3.hpp"
 #include "templates/Qwen.hpp"
@@ -58,6 +59,7 @@ inline void registerTemplates()
     templateManager.registerTemplate<QwenFim>();
     templateManager.registerTemplate<OpenAICompatible>();
     templateManager.registerTemplate<Alpaca>();
+    templateManager.registerTemplate<GoogleAI>();
 }
 
 } // namespace QodeAssist::Templates