mirror of
https://github.com/Palm1r/QodeAssist.git
synced 2026-05-30 02:49:12 -04:00
33 lines
1.1 KiB
C++
33 lines
1.1 KiB
C++
// Copyright (C) 2024-2026 Petr Mironychev
|
||
// SPDX-License-Identifier: GPL-3.0-or-later
|
||
|
||
#pragma once
|
||
|
||
#include "llmcore/PromptTemplate.hpp"
|
||
|
||
namespace QodeAssist::Templates {
|
||
|
||
class DeepSeekCoderFim : public LLMQore::PromptTemplate
|
||
{
|
||
public:
|
||
LLMQore::TemplateType type() const override { return LLMQore::TemplateType::Fim; }
|
||
QString name() const override { return "DeepSeekCoder FIM"; }
|
||
QString promptTemplate() const override
|
||
{
|
||
return "<｜fim▁begin｜>%1<｜fim▁hole｜>%2<｜fim▁end｜>";
|
||
}
|
||
QStringList stopWords() const override { return QStringList(); }
|
||
void prepareRequest(QJsonObject &request, const LLMQore::ContextData &context) const override
|
||
{
|
||
QString formattedPrompt = promptTemplate().arg(context.prefix, context.suffix);
|
||
request["prompt"] = formattedPrompt;
|
||
}
|
||
QString description() const override
|
||
{
|
||
return "The message will contain the following tokens: "
|
||
"<｜fim▁begin｜>%1<｜fim▁hole｜>%2<｜fim▁end｜>";
|
||
}
|
||
};
|
||
|
||
} // namespace QodeAssist::Templates
|