文件系统
LangChain 提供开箱即用的与本地文件系统交互的工具。此笔记本将介绍其中的一些工具。
注意：不建议在沙盒环境之外使用这些工具！
%pip install -qU langchain-community
首先，我们将导入工具。
from tempfile import TemporaryDirectory
from langchain_community.agent_toolkits import FileManagementToolkit
# We'll make a temporary directory to avoid clutter
working_directory = TemporaryDirectory()
API 参考：FileManagementToolkit
FileManagementToolkit
如果您想为您的代理提供所有文件工具，可以使用工具包轻松做到这一点。我们将临时目录作为根目录传递，作为 LLM 的工作区。
建议始终传递根目录，因为如果没有它，LLM 容易污染工作目录，并且如果没有它，就没有针对简单提示注入的任何验证。
toolkit = FileManagementToolkit(
root_dir=str(working_directory.name)
) # If you don't provide a root_dir, operations will default to the current working directory
toolkit.get_tools()
[CopyFileTool(root_dir='/tmp/tmprdvsw3tg'),
DeleteFileTool(root_dir='/tmp/tmprdvsw3tg'),
FileSearchTool(root_dir='/tmp/tmprdvsw3tg'),
MoveFileTool(root_dir='/tmp/tmprdvsw3tg'),
ReadFileTool(root_dir='/tmp/tmprdvsw3tg'),
WriteFileTool(root_dir='/tmp/tmprdvsw3tg'),
ListDirectoryTool(root_dir='/tmp/tmprdvsw3tg')]
选择文件系统工具
如果您只想选择某些工具，您可以在初始化工具包时将它们作为参数传递，或者可以分别初始化所需的工具。
tools = FileManagementToolkit(
root_dir=str(working_directory.name),
selected_tools=["read_file", "write_file", "list_directory"],
).get_tools()
tools
[ReadFileTool(root_dir='/tmp/tmprdvsw3tg'),
WriteFileTool(root_dir='/tmp/tmprdvsw3tg'),
ListDirectoryTool(root_dir='/tmp/tmprdvsw3tg')]
read_tool, write_tool, list_tool = tools
write_tool.invoke({"file_path": "example.txt", "text": "Hello World!"})
'File written successfully to example.txt.'
# List files in the working directory
list_tool.invoke({})
'example.txt'