From 965af4a945c713c508cb4d38905aeaaca9bb083c Mon Sep 17 00:00:00 2001
From: Petr Mironychev <9195189+Palm1r@users.noreply.github.com>
Date: Mon, 14 Oct 2024 21:35:17 +0200
Subject: [PATCH] Add Llama chat support

---
 qodeassist.cpp              | 1 +
 templates/CodeLlamaChat.hpp | 6 ++++++
 2 files changed, 7 insertions(+)

diff --git a/qodeassist.cpp b/qodeassist.cpp
index 77e37f6..2ab2f4b 100644
--- a/qodeassist.cpp
+++ b/qodeassist.cpp
@@ -94,6 +94,7 @@ public:
         templateManager.registerTemplate<Templates::DeepSeekCoderChat>();
         templateManager.registerTemplate<Templates::CodeLlamaChat>();
         templateManager.registerTemplate<Templates::QwenChat>();
+        templateManager.registerTemplate<Templates::LlamaChat>();
 
         Utils::Icon QCODEASSIST_ICON(
             {{":/resources/images/qoderassist-icon.png", Utils::Theme::IconsBaseColor}});
diff --git a/templates/CodeLlamaChat.hpp b/templates/CodeLlamaChat.hpp
index 2342e3c..c7c9dac 100644
--- a/templates/CodeLlamaChat.hpp
+++ b/templates/CodeLlamaChat.hpp
@@ -46,4 +46,10 @@ public:
     }
 };
 
+class LlamaChat : public CodeLlamaChat
+{
+public:
+    QString name() const override { return "Llama Chat"; }
+};
+
 } // namespace QodeAssist::Templates