2024-03-27 10:12:28 +08:00

12 lines
311 B
Python

# LangChain 的 Shell 工具
from chatchat.server.pydantic_v1 import BaseModel, Field
from langchain_community.tools import ShellTool
def shell(query: str):
tool = ShellTool()
return tool.run(tool_input=query)
class ShellInput(BaseModel):
query: str = Field(description="The command to execute")