trends-mcp/trends_mcp/aliases.py

from __future__ import annotations

import json
from pathlib import Path

_ALIASES_PATH = Path(__file__).with_name("entity_map.json")


def load_aliases() -> dict[str, str]:
    return {k.lower(): v for k, v in json.loads(_ALIASES_PATH.read_text()).items()}


def normalize_entity(name: str) -> str:
    aliases = load_aliases()
    key = " ".join(str(name).strip().lower().split())
    return aliases.get(key, str(name).strip())