__init__.py

"""nse-orchestrator plugin for Hermes Agent."""

from __future__ import annotations

import sys
from pathlib import Path

_HERE = Path(__file__).resolve().parent
if str(_HERE) not in sys.path:
    sys.path.insert(0, str(_HERE))

from .tools import (  # noqa: E402
    NSE_CHECK_SCHEMA,
    NSE_HEALTH_SCHEMA,
    NSE_INIT_SCHEMA,
    NSE_STATUS_SCHEMA,
    handle_nse_check,
    handle_nse_health,
    handle_nse_init,
    handle_nse_status,
)

_TOOLS = (
    ("nse_init",   NSE_INIT_SCHEMA,   handle_nse_init,   "🧠"),
    ("nse_check",  NSE_CHECK_SCHEMA,  handle_nse_check,  "🚦"),
    ("nse_health", NSE_HEALTH_SCHEMA, handle_nse_health, "❤️"),
    ("nse_status", NSE_STATUS_SCHEMA, handle_nse_status, "📊"),
)


def register(ctx) -> None:
    for name, schema, handler, emoji in _TOOLS:
        ctx.register_tool(
            name=name,
            toolset="nse",
            schema=schema,
            handler=handler,
            emoji=emoji,
        )