14 lines
419 B
Python
14 lines
419 B
Python
# LangChain 的 Shell 工具
|
|
from langchain_community.tools import ShellTool
|
|
|
|
from chatchat.server.pydantic_v1 import Field
|
|
|
|
from .tools_registry import BaseToolOutput, regist_tool
|
|
|
|
|
|
# @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))
|