"""Custom tools package for the LangGraph demo agent."""

from .poem_tool import print_poem
from .ssh_tool import SSHTool
from langchain_community.tools.shell.tool import ShellTool

# Pre-configured SSH tool for your server - only connects when actually used
# TODO: Update these connection details for your actual server
configured_ssh_tool = SSHTool(
    host="your-server.example.com",  # Replace with your server
    username="admin",                # Replace with your username
    key_filename="~/.ssh/id_rsa",    # Replace with your key path
    ask_human_input=True             # Safety confirmation
)

__all__ = ["print_poem", "SSHTool", "ShellTool", "configured_ssh_tool"]