Spaces:
Sleeping
Sleeping
File size: 1,502 Bytes
2bf863e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 | """MCP tool registry for Polyglot-Optima.
Exposes 9 tools per plan §9. The TOOL_REGISTRY dict is loaded by the environment
at startup and dispatched from PolyglotOptimaEnvironment._dispatch_tool.
Each tool is a plain Python callable (tool_args: dict, state: OptimizationState) -> dict.
The @tool decorator (Hour 22 deployment-time wrapper) adds Pydantic schema
validation, mode tagging, and async dispatch — for now, plain functions.
"""
from __future__ import annotations
from .hardware_profiler import get_hardware_profile_tool
from .python_analyzer import (
profile_python_hotspots_tool,
analyze_complexity_tool,
check_memory_access_tool,
)
from .cpp_compiler import compile_and_benchmark_tool
from .verifier import verify_equivalence_tool
from .portability_checker import check_portability_tool
from .bottleneck_reporter import get_bottleneck_report_tool
from .submit import submit_optimization_tool
TOOL_REGISTRY = {
"get_hardware_profile": get_hardware_profile_tool,
"profile_python_hotspots": profile_python_hotspots_tool,
"analyze_complexity": analyze_complexity_tool,
"check_memory_access": check_memory_access_tool,
"compile_and_benchmark": compile_and_benchmark_tool,
"verify_equivalence": verify_equivalence_tool,
"check_portability": check_portability_tool,
"get_bottleneck_report": get_bottleneck_report_tool,
"submit_optimization": submit_optimization_tool,
}
__all__ = ["TOOL_REGISTRY"]
|