# LangChain 的 Shell 工具
from langchain.tools.shell import ShellTool

from chatchat.server.pydantic_v1 import Field
from .tools_registry import regist_tool


@regist_tool
def shell(query: str = Field(description="The command to execute")):
    '''Use Shell to execute system shell commands'''
    tool = ShellTool()
    return tool.run(tool_input=query)