"""需求分级统筹规划 Agent 的工具。"""
from __future__ import annotations

from collections.abc import Callable
from typing import Any

from agents.demand_grade_orchestrator_agent.tools.query_global_heat_tree import query_global_heat_tree
from agents.demand_grade_orchestrator_agent.tools.query_heat_node_group import query_heat_node_group
from agents.demand_grade_orchestrator_agent.tools.save_grade_plan import save_grade_plan
from supply_agent.tools.registry import ToolRegistry

ALL_TOOLS: list[Callable[..., Any]] = [query_global_heat_tree, query_heat_node_group, save_grade_plan]


def register_all_tools(registry: ToolRegistry) -> ToolRegistry:
    return registry.from_decorated(*ALL_TOOLS)
