文件系统
LangChain 提供了与本地文件系统交互的工具。这本笔记本将介绍其中的一些。
注意： 这些工具不建议在沙盒环境之外使用！
%pip install -qU langchain-community
首先，我们将导入这些工具。
from tempfile import TemporaryDirectory
from langchain_community.agent_toolkits import FileManagementToolkit
# 我们将创建一个临时目录以避免杂乱
working_directory = TemporaryDirectory()
文件管理工具包
如果您想为您的代理提供所有文件工具，使用该工具包非常简单。我们将临时目录作为根目录传入，作为 LLM 的工作空间。
建议始终传入根目录，因为如果没有根目录，LLM 很容易污染工作目录，并且没有根目录就无法进行简单的提示注入验证。
toolkit = FileManagementToolkit(
root_dir=str(working_directory.name)
) # 如果您不提供 root_dir，操作将默认为当前工作目录
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_tool.invoke({})
'example.txt'