mirror of
https://github.com/Palm1r/QodeAssist.git
synced 2026-05-30 02:49:12 -04:00
feat: Add agents and agents settings
This commit is contained in:
75
sources/templates/JsonPromptTemplate.hpp
Normal file
75
sources/templates/JsonPromptTemplate.hpp
Normal file
@@ -0,0 +1,75 @@
|
||||
// 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
|
||||
Reference in New Issue
Block a user