tool_registry.cpp

Go to the documentation of this file.

// SPDX-License-Identifier: Apache-2.0
#include <entropic/mcp/tool_registry.h>
#include <entropic/mcp/server_base.h>
#include <entropic/types/logging.h>
 
#include <nlohmann/json.hpp>
 
static auto logger = entropic::log::get("mcp.tool_registry");
 
namespace entropic {
 
void ToolRegistry::register_tool(ToolBase* tool) {
    if (tool == nullptr) {
        logger->warn("Attempted to register null tool");
        return;
    }
    const auto& name = tool->name();
    if (tools_.count(name) > 0) {
        logger->warn("Tool '{}' already registered — replacing", name);
    }
    tools_[name] = tool;
    logger->info("Registered tool: {}", name);
}
 
bool ToolRegistry::has_tool(const std::string& name) const {
    return tools_.count(name) > 0;
}
 
std::string ToolRegistry::get_tools_json() const {
    auto arr = nlohmann::json::array();
    for (const auto& [name, tool] : tools_) {
        nlohmann::json entry;
        entry["name"] = tool->definition().name;
        entry["description"] = tool->definition().description;
        entry["inputSchema"] = nlohmann::json::parse(
            tool->definition().input_schema);
        arr.push_back(std::move(entry));
    }
    return arr.dump();
}
 
std::vector<const ToolDefinition*> ToolRegistry::get_definitions() const {
    std::vector<const ToolDefinition*> defs;
    defs.reserve(tools_.size());
    for (const auto& [name, tool] : tools_) {
        defs.push_back(&tool->definition());
    }
    return defs;
}
 
ServerResponse ToolRegistry::dispatch(
    const std::string& name,
    const std::string& args_json) {
    auto it = tools_.find(name);
    if (it == tools_.end()) {
        logger->warn("Unknown tool: {}", name);
        ServerResponse resp;
        resp.result = "Error: Unknown tool '" + name + "'";
        return resp;
    }
    logger->info("Dispatch: tool='{}'", name);
    return it->second->execute(args_json);
}
 
ToolBase* ToolRegistry::get_tool(const std::string& name) const {
    auto it = tools_.find(name);
    if (it == tools_.end()) {
        return nullptr;
    }
    return it->second;
}
 
} // namespace entropic