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

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


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