You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
18 lines
618 B
18 lines
618 B
from langchain.agents.agent_toolkits import FileManagementToolkit
|
|
from tempfile import TemporaryDirectory
|
|
|
|
# We'll make a temporary directory to avoid clutter
|
|
working_directory = TemporaryDirectory()
|
|
|
|
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()
|
|
|
|
file_management_tools = FileManagementToolkit(
|
|
root_dir=str(working_directory.name),
|
|
selected_tools=["read_file", "write_file", "list_directory"],
|
|
).get_tools()
|
|
|
|
read_tool, write_tool, list_tool = file_management_tools
|