%1%2"; }
+ QStringList stopWords() const override
+ {
+ return QStringList() << "" << "" << "
" << "" << "< EOT >" << "\\end"
+ << "" << "" << "##";
+ }
+ void prepareRequest(QJsonObject &request, const LLMCore::ContextData &context) const override
+ {
+ QString formattedPrompt = promptTemplate().arg(context.suffix, context.prefix);
+ request["prompt"] = formattedPrompt;
+ }
+ QString description() const override
+ {
+ return "The message will contain the following tokens: %1%2";
+ }
+};
+
+} // namespace QodeAssist::Templates
diff --git a/templates/Templates.hpp b/templates/Templates.hpp
index 80f215c..872b58d 100644
--- a/templates/Templates.hpp
+++ b/templates/Templates.hpp
@@ -25,6 +25,7 @@
#include "templates/ChatML.hpp"
#include "templates/Claude.hpp"
#include "templates/CodeLlamaFim.hpp"
+#include "templates/CodeLlamaQMLFim.hpp"
#include "templates/CustomFimTemplate.hpp"
#include "templates/DeepSeekCoderFim.hpp"
#include "templates/Llama2.hpp"
@@ -53,6 +54,7 @@ inline void registerTemplates()
templateManager.registerTemplate();
templateManager.registerTemplate();
templateManager.registerTemplate();
+ templateManager.registerTemplate();
}
} // namespace QodeAssist::Templates