LongArticleTaskServer/app/domains/mcp/_handler_map.py

from typing import Any, Awaitable, Callable, Dict

from app.core.database import DatabaseManager
from app.core.observability import LogService


HandlerType = Callable[
    [DatabaseManager, LogService, Dict[str, Any] | None], Awaitable[Any]
]


async def _get_decode_response_wrapper(
    pool: DatabaseManager,
    log_service: LogService,
    params: Dict[str, Any] | None,
):
    # 这里只是为 handler_map 统一签名做一层薄封装
    return await get_decode_response(pool=pool, log_service=log_service, params=params)


HANDLER_MAP: Dict[str, HandlerType] = {
    "get_decode_response": _get_decode_response_wrapper,
}


__all__ = ["HANDLER_MAP"]