AutoMetabuilder/backend/autometabuilder/utils/tool_map_builder.py

"""Build tool map from registry entries."""
from ..loaders.callable_loader import load_callable


def build_tool_map(gh, registry_entries: list) -> dict:
    """Build tool name to callable map."""
    tool_map = {}
    for entry in registry_entries:
        name = entry.get("name")
        provider = entry.get("provider")
        if not name:
            continue
        if provider == "github":
            method = entry.get("method")
            tool_map[name] = getattr(gh, method) if gh and method else None
            continue
        if provider == "module":
            path = entry.get("callable")
            tool_map[name] = load_callable(path) if path else None
            continue
        tool_map[name] = None
    return tool_map