tool_registry.h

#pragma once
 
#include <agents-cpp/tool.h>
#include <agents-cpp/tools/wiki_tool.h>
#include <map>
#include <memory>
#include <vector>
 
namespace agents {
namespace tools {

class ToolRegistry {
public:
    ToolRegistry() = default;
    ~ToolRegistry() = default;

    void registerTool(std::shared_ptr<Tool> tool);

    std::shared_ptr<Tool> getTool(const String& name) const;

    std::vector<std::shared_ptr<Tool>> getAllTools() const;

    bool hasTool(const String& name) const;

    void removeTool(const String& name);

    void clear();

    JsonObject getToolSchemas() const;

    static ToolRegistry& global();
 
private:
    std::map<String, std::shared_ptr<Tool>> tools_;
};

void registerStandardTools(ToolRegistry& registry);

std::shared_ptr<Tool> createShellCommandTool();

std::shared_ptr<Tool> createWebSearchTool();

std::shared_ptr<Tool> createWikipediaTool();

std::shared_ptr<Tool> createPythonTool();

std::shared_ptr<Tool> createFileReadTool();

std::shared_ptr<Tool> createFileWriteTool();
 
} // namespace tools
} // namespace agents