SupplyAgent/agents/find_agent/tools/__init__.py

"""
find_agent 工具包

在此注册本 Agent 专属的工具。
"""
from __future__ import annotations

from collections.abc import Callable
from typing import Any

from agents.find_agent.tools.douyin_detail import douyin_detail
from agents.find_agent.tools.douyin_search import douyin_search
from agents.find_agent.tools.douyin_search_tikhub import douyin_search_tikhub
from agents.find_agent.tools.douyin_user_videos import douyin_user_videos
from agents.find_agent.tools.decision_support import (
    normalize_age_portraits,
)
from agents.find_agent.tools.hotspot_profile import (
    batch_fetch_portraits,
    get_account_fans_portrait,
    get_content_fans_portrait,
)
from agents.find_agent.tools.video_discovery_store import (
    audit_video_discovery_run,
    batch_save_video_candidate_evaluations,
    create_video_discovery_run,
    query_video_discovery_state,
    record_video_search_page,
)
from supply_agent.tools.registry import ToolRegistry

ALL_TOOLS: list[Callable[..., Any]] = [
    douyin_search,
    douyin_search_tikhub,
    douyin_user_videos,
    douyin_detail,
    get_content_fans_portrait,
    get_account_fans_portrait,
    batch_fetch_portraits,
    normalize_age_portraits,
    create_video_discovery_run,
    record_video_search_page,
    batch_save_video_candidate_evaluations,
    audit_video_discovery_run,
    query_video_discovery_state,
]

__all__ = [
    "ALL_TOOLS",
    "douyin_search",
    "douyin_search_tikhub",
    "douyin_user_videos",
    "douyin_detail",
    "get_content_fans_portrait",
    "get_account_fans_portrait",
    "batch_fetch_portraits",
    "normalize_age_portraits",
    "create_video_discovery_run",
    "record_video_search_page",
    "batch_save_video_candidate_evaluations",
    "audit_video_discovery_run",
    "query_video_discovery_state",
    "register_all_tools",
]


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