mirror of
https://github.com/Palm1r/QodeAssist.git
synced 2026-05-30 02:49:12 -04:00
37 lines
1.2 KiB
C++
37 lines
1.2 KiB
C++
// Copyright (C) 2024-2026 Petr Mironychev
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
#pragma once
|
|
|
|
#include <QJsonObject>
|
|
#include <QList>
|
|
#include <QString>
|
|
|
|
#include "ContextData.hpp"
|
|
#include "ProviderID.hpp"
|
|
|
|
namespace QodeAssist::PluginLLMCore {
|
|
|
|
enum class TemplateType { Chat, FIM, FIMOnChat };
|
|
|
|
class PromptTemplate
|
|
{
|
|
public:
|
|
virtual ~PromptTemplate() = default;
|
|
virtual TemplateType type() const = 0;
|
|
virtual QString name() const = 0;
|
|
virtual QStringList stopWords() const = 0;
|
|
virtual void prepareRequest(QJsonObject &request, const ContextData &context) const = 0;
|
|
virtual QString description() const = 0;
|
|
virtual bool isSupportProvider(ProviderID id) const = 0;
|
|
|
|
// Endpoint path this template expects to be sent to. Empty string
|
|
// (default) means "let the provider's client use its standard chat
|
|
// path" (/chat/completions, /api/chat, /v1/messages, ...). Templates
|
|
// producing non-chat payload shapes (e.g. {prompt, suffix} for
|
|
// Mistral FIM, {input_prefix, input_suffix} for llama.cpp infill)
|
|
// must override this to the path their payload is valid for.
|
|
virtual QString endpoint() const { return {}; }
|
|
};
|
|
} // namespace QodeAssist::PluginLLMCore
|