add markdown parsing + latex + tree-sitter highlighting for codeblocks in llm responses
This commit is contained in:
@@ -0,0 +1,70 @@
|
||||
#pragma once
|
||||
|
||||
#include <QJsonArray>
|
||||
#include <QJsonObject>
|
||||
#include <QList>
|
||||
#include <QObject>
|
||||
#include <QString>
|
||||
|
||||
#include <functional>
|
||||
|
||||
namespace ZShell::llm {
|
||||
|
||||
// A capability the model may invoke mid-turn. Tools run asynchronously
|
||||
// and report exactly one result: `{"output": ...}` on success or
|
||||
// `{"error": ...}` on failure.
|
||||
class LlmTool : public QObject {
|
||||
Q_OBJECT
|
||||
|
||||
public:
|
||||
explicit LlmTool(QObject* parent = nullptr);
|
||||
~LlmTool() override;
|
||||
|
||||
[[nodiscard]] virtual QString name() const = 0;
|
||||
[[nodiscard]] virtual QString description() const = 0;
|
||||
// JSON Schema describing the tool's `arguments` object.
|
||||
[[nodiscard]] virtual QJsonObject parameters() const = 0;
|
||||
|
||||
// Runs the tool; `done` is invoked exactly once, with
|
||||
// `{"output": ...}` on success or `{"error": ...}` on failure.
|
||||
// `done` must be invoked asynchronously (on a later event loop
|
||||
// iteration), never synchronously within execute().
|
||||
virtual void execute(
|
||||
const QJsonObject& args,
|
||||
std::function<void(const QJsonObject& result)> done) = 0;
|
||||
// Abandons in-flight work, if any.
|
||||
virtual void cancel();
|
||||
|
||||
// The OpenAI-compatible `tools` entry for this tool.
|
||||
[[nodiscard]] QJsonObject specification() const;
|
||||
};
|
||||
|
||||
// Owns the set of tools available to the model.
|
||||
class ToolRegistry : public QObject {
|
||||
Q_OBJECT
|
||||
|
||||
Q_PROPERTY(bool enabled READ enabled WRITE setEnabled NOTIFY enabledChanged)
|
||||
|
||||
public:
|
||||
explicit ToolRegistry(QObject* parent = nullptr);
|
||||
|
||||
[[nodiscard]] bool enabled() const { return m_enabled; }
|
||||
void setEnabled(bool value);
|
||||
|
||||
// Takes ownership; tools become children of the registry.
|
||||
void registerTool(LlmTool* tool);
|
||||
[[nodiscard]] LlmTool* tool(const QString& name) const;
|
||||
// The request body's `tools` array; empty while disabled.
|
||||
[[nodiscard]] QJsonArray specifications() const;
|
||||
// Abandons in-flight work in every tool.
|
||||
void cancelAll();
|
||||
|
||||
Q_SIGNALS:
|
||||
void enabledChanged();
|
||||
|
||||
private:
|
||||
bool m_enabled = true;
|
||||
QList<LlmTool*> m_tools;
|
||||
};
|
||||
|
||||
} // namespace ZShell::llm
|
||||
Reference in New Issue
Block a user