17 lines
683 B
Python
17 lines
683 B
Python
"""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"]
|