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.
25 lines
840 B
25 lines
840 B
######################## ######################################################## file system
|
|
|
|
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
|
|
write_tool.run({"file_path": "example.txt", "text": "Hello World!"})
|
|
|
|
# List files in the working directory
|
|
list_tool.run({})
|
|
|