SupplyAgent/agents/demand_video_expand_agent/tools/__init__.py

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

from collections.abc import Callable
from typing import Any

from agents.demand_video_expand_agent.tools.batch_save_demand_expansions import (
    batch_save_demand_expansions,
)
from supply_agent.tools.registry import ToolRegistry

ALL_TOOLS: list[Callable[..., Any]] = [
    batch_save_demand_expansions,
]

__all__ = [
    "ALL_TOOLS",
    "batch_save_demand_expansions",
    "register_all_tools",
]


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