tool.h

#pragma once
 
#include <agents-cpp/types.h>
#include <functional>
 
namespace agents {

struct ToolResult {
    bool success;
    String content;
    JsonObject data;
};

using ToolCallback = std::function<ToolResult(const JsonObject&)>;

class Tool {
public:
    Tool(const String& name, const String& description);

    virtual ~Tool() = default;

    const String& getName() const;

    const String& getDescription() const;

    const ParameterMap& getParameters() const;

    const JsonObject& getSchema() const;

    void addParameter(const Parameter& param);

    void setCallback(ToolCallback callback);

    virtual ToolResult execute(const JsonObject& params) const;

    bool validateParameters(const JsonObject& params) const;

protected:
    String name_;
    String description_;
    ParameterMap parameters_;
    ToolCallback callback_;
    JsonObject schema_;

    void updateSchema();
};

std::shared_ptr<Tool> createTool(
    const String& name,
    const String& description,
    const std::vector<Parameter>& parameters,
    ToolCallback callback
);
 
} // namespace agents