hxf/backend/th_agenter/services/tools/__init__.py

"""Agent tools package."""

from .weather import WeatherQueryTool
from .search import TavilySearchTool
from .datetime_tool import DateTimeTool

# Try to import PostgreSQL MCP tool
try:
    from th_agenter.services.mcp.postgresql_mcp import PostgreSQLMCPTool
except ImportError:
    PostgreSQLMCPTool = None

# Try to import MySQL MCP tool
try:
    from th_agenter.services.mcp.mysql_mcp import MySQLMCPTool
except ImportError:
    MySQLMCPTool = None


# Try to import LangChain native tools if available
try:
    from .langchain_native_tools import LANGCHAIN_NATIVE_TOOLS
except ImportError:
    LANGCHAIN_NATIVE_TOOLS = []

__all__ = [
    'WeatherQueryTool',
    'TavilySearchTool',
    'DateTimeTool',
] + ([
    'PostgreSQLMCPTool',
] if PostgreSQLMCPTool is not None else []) + ([
    'MySQLMCPTool',
] if MySQLMCPTool is not None else []) + [
    'LANGCHAIN_NATIVE_TOOLS'
]