zach
cd3da1941b
C++ / fmt (pull_request) Successful in 4s
C++ / build (pull_request) Failing after 2m6s
C++ / clang-tidy (pull_request) Failing after 2m5s
JS/TS / fmt (pull_request) Failing after 3m8s
Python / static (pull_request) Successful in 24s
Python / verify (pull_request) Successful in 1m21s
JS/TS / lint (pull_request) Successful in 5m10s
Rust / fmt (pull_request) Successful in 30s
Rust / build (pull_request) Successful in 56s
Rust / clippy (pull_request) Successful in 46s
62 lines
1.4 KiB
C++
62 lines
1.4 KiB
C++
#pragma once
|
|
|
|
#include "tool.hpp"
|
|
|
|
#include <QByteArray>
|
|
#include <QList>
|
|
#include <QNetworkAccessManager>
|
|
|
|
#include <functional>
|
|
|
|
class QNetworkReply;
|
|
class QTimer;
|
|
|
|
namespace ZShell::llm {
|
|
|
|
class WebFetchTool : public LlmTool {
|
|
Q_OBJECT
|
|
|
|
public:
|
|
static constexpr int MaxResponseBytes = 5 * 1024 * 1024;
|
|
static constexpr int DefaultTimeoutSeconds = 30;
|
|
static constexpr int MaxTimeoutSeconds = 120;
|
|
static const QString StoragePath;
|
|
|
|
explicit WebFetchTool(QObject* parent = nullptr);
|
|
~WebFetchTool() override;
|
|
|
|
QString name() const override;
|
|
QString description() const override;
|
|
QJsonObject parameters() const override;
|
|
void execute(
|
|
const QString& toolCallId,
|
|
const QJsonObject& args,
|
|
int outputBudgetChars,
|
|
std::function<void(const QJsonObject& result)> done) override;
|
|
void cancel() override;
|
|
|
|
static QString extractTextFromHtml(const QString& html);
|
|
static QString decodeEntities(const QString& text);
|
|
|
|
private:
|
|
struct Job {
|
|
QNetworkReply* reply = nullptr;
|
|
QTimer* timer = nullptr;
|
|
QByteArray body;
|
|
bool tooLarge = false;
|
|
QString format;
|
|
QString toolCallId;
|
|
int budgetChars = 0;
|
|
std::function<void(const QJsonObject& result)> done;
|
|
};
|
|
|
|
void completeJob(Job* job, QJsonObject result);
|
|
QString saveToFile(
|
|
const Job& job, const QString& content, const QString& mime) const;
|
|
|
|
QNetworkAccessManager m_manager;
|
|
QList<Job*> m_jobs;
|
|
};
|
|
|
|
} // namespace ZShell::llm
|