36 lines
880 B
Python
36 lines
880 B
Python
"""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'
|
|
] |