SupplyAgent/agents/generate_demand_agent/tools/__init__.py

"""
generate_demand_agent 工具包
"""
from __future__ import annotations

from collections.abc import Callable
from typing import Any

from agents.generate_demand_agent.tools.query_category_leaves_by_dim import (
    query_category_leaves_by_dim,
)
from agents.generate_demand_agent.tools.query_category_tree_by_dim import (
    query_category_tree_by_dim,
)
from supply_agent.tools.registry import ToolRegistry

ALL_TOOLS: list[Callable[..., Any]] = [
    query_category_tree_by_dim,
    query_category_leaves_by_dim,
]

__all__ = [
    "ALL_TOOLS",
    "query_category_leaves_by_dim",
    "query_category_tree_by_dim",
    "register_all_tools",
]


def register_all_tools(registry: ToolRegistry) -> ToolRegistry:
    """将 generate_demand_agent 包内的所有工具注册到 ToolRegistry。"""
    return registry.from_decorated(*ALL_TOOLS)