tools/__init__.py

"""
Centralized tools module for GAIA benchmark agents.
Import tools from their respective modules.
"""

from .file_tools import read_excel_file, read_csv_file, calculate_column_sum
from .research_tools import web_search
from .math_tools import calculate_expression, percentage_calculation, currency_format, statistical_summary
from .database_tools import search_similar_gaia_questions, get_exact_answer_if_highly_similar

# File processing tools
FILE_TOOLS = [read_excel_file, read_csv_file, calculate_column_sum]

# Research tools  
RESEARCH_TOOLS = [web_search]

# Mathematical tools
MATH_TOOLS = [calculate_expression, percentage_calculation, currency_format, statistical_summary]

# Database retrieval tools
DATABASE_TOOLS = [search_similar_gaia_questions, get_exact_answer_if_highly_similar]

# All tools combined
ALL_TOOLS = FILE_TOOLS + RESEARCH_TOOLS + MATH_TOOLS + DATABASE_TOOLS

__all__ = [
    "FILE_TOOLS",
    "RESEARCH_TOOLS", 
    "MATH_TOOLS",
    "DATABASE_TOOLS",
    "ALL_TOOLS"
]