文件系统

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'

工具概念指南
工具操作指南

FileManagementToolkit

选择文件系统工具

相关