mirror of
https://github.com/Palm1r/QodeAssist.git
synced 2026-05-30 10:59:30 -04:00
76 lines
2.4 KiB
C++
76 lines
2.4 KiB
C++
// Copyright (C) 2024-2026 Petr Mironychev
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
#pragma once
|
|
|
|
#include <memory>
|
|
#include <mutex>
|
|
#include <optional>
|
|
|
|
#include <QJsonObject>
|
|
#include <QString>
|
|
|
|
#include <inja/inja.hpp>
|
|
|
|
#include "PromptTemplate.hpp"
|
|
|
|
namespace QodeAssist {
|
|
struct AgentConfig;
|
|
}
|
|
|
|
namespace QodeAssist::Templates {
|
|
|
|
// Renderer for the request-body jinja template embedded in an
|
|
// AgentConfig. One per Agent — built inline from the config (no shared
|
|
// template registry, no model/provider filtering).
|
|
class JsonPromptTemplate : public PromptTemplate
|
|
{
|
|
public:
|
|
// Build a renderer from an already-parsed agent config. Compiles
|
|
// the jinja source via inja once. On failure returns nullptr and
|
|
// populates `*error` (existing content preserved; pass nullptr to
|
|
// discard).
|
|
static std::unique_ptr<JsonPromptTemplate> fromConfig(
|
|
const AgentConfig &cfg, QString *error = nullptr);
|
|
|
|
QString name() const override { return m_name; }
|
|
QString description() const override { return m_description; }
|
|
|
|
// Standalone-template filters are gone — each template is built for
|
|
// exactly one agent, so it always matches its owner's provider/model.
|
|
bool isSupportProvider(Providers::ProviderID) const override { return true; }
|
|
bool isSupportModel(const QString &) const override { return true; }
|
|
PromptShape shape() const override { return PromptShape::Chat; }
|
|
|
|
void prepareRequest(QJsonObject &request, const ContextData &context) const override;
|
|
|
|
[[nodiscard]] bool buildFullRequest(
|
|
QJsonObject &request,
|
|
const ContextData &context,
|
|
bool thinkingEnabled = false) const override;
|
|
|
|
const QJsonObject &sampling() const { return m_sampling; }
|
|
|
|
private:
|
|
JsonPromptTemplate() = default;
|
|
|
|
std::optional<QJsonObject> renderBody(const ContextData &context) const;
|
|
void applySampling(QJsonObject &request, bool thinkingEnabled) const;
|
|
|
|
QString m_name;
|
|
QString m_description;
|
|
|
|
// m_env is populated once in fromConfig() and never mutated again.
|
|
// It is `mutable` only because inja::Environment::render() is not a
|
|
// const member; m_renderMutex serialises those render() calls since
|
|
// inja's render path is not internally re-entrant on one Environment.
|
|
mutable inja::Environment m_env;
|
|
inja::Template m_template;
|
|
mutable std::mutex m_renderMutex;
|
|
|
|
QJsonObject m_sampling;
|
|
QJsonObject m_thinking;
|
|
};
|
|
|
|
} // namespace QodeAssist::Templates
|