agent/tools/__init__.py

"""
Hugging Face tools for the agent
"""

from agent.tools.dataset_tools import (
    HF_INSPECT_DATASET_TOOL_SPEC,
    hf_inspect_dataset_handler,
)
from agent.tools.github_find_examples import (
    GITHUB_FIND_EXAMPLES_TOOL_SPEC,
    github_find_examples_handler,
)
from agent.tools.github_list_repos import (
    GITHUB_LIST_REPOS_TOOL_SPEC,
    github_list_repos_handler,
)
from agent.tools.github_read_file import (
    GITHUB_READ_FILE_TOOL_SPEC,
    github_read_file_handler,
)
from agent.tools.jobs_tool import HF_JOBS_TOOL_SPEC, HfJobsTool, hf_jobs_handler
from agent.tools.types import ToolResult

# New tools for enhanced functionality
from agent.tools.slides_tool import SLIDES_TOOL_SPEC, create_slides_handler
from agent.tools.document_tool import DOCUMENT_TOOL_SPEC, create_document_handler
from agent.tools.web_search_tool import WEB_SEARCH_TOOL_SPEC, web_search_handler
from agent.tools.image_gen_tool import IMAGE_GEN_TOOL_SPEC, generate_image_handler

__all__ = [
    "ToolResult",
    "HF_JOBS_TOOL_SPEC",
    "hf_jobs_handler",
    "HfJobsTool",
    "GITHUB_FIND_EXAMPLES_TOOL_SPEC",
    "github_find_examples_handler",
    "GITHUB_LIST_REPOS_TOOL_SPEC",
    "github_list_repos_handler",
    "GITHUB_READ_FILE_TOOL_SPEC",
    "github_read_file_handler",
    "HF_INSPECT_DATASET_TOOL_SPEC",
    "hf_inspect_dataset_handler",
    # New tools
    "SLIDES_TOOL_SPEC",
    "create_slides_handler",
    "DOCUMENT_TOOL_SPEC",
    "create_document_handler",
    "WEB_SEARCH_TOOL_SPEC",
    "web_search_handler",
    "IMAGE_GEN_TOOL_SPEC",
    "generate_image_handler",
]