Source code for langchain_community.agent_toolkits.file_management.toolkit

from __future__ import annotations

from typing import Dict, List, Optional, Type

from langchain_core.pydantic_v1 import root_validator
from import BaseToolkit

from import BaseTool
from import CopyFileTool
from import DeleteFileTool
from import FileSearchTool
from import ListDirectoryTool
from import MoveFileTool
from import ReadFileTool
from import WriteFileTool

_FILE_TOOLS: List[Type[BaseTool]] = [
_FILE_TOOLS_MAP: Dict[str, Type[BaseTool]] = {
    tool_cls.__fields__["name"].default: tool_cls for tool_cls in _FILE_TOOLS

[docs]class FileManagementToolkit(BaseToolkit): """Toolkit for interacting with local files. *Security Notice*: This toolkit provides methods to interact with local files. If providing this toolkit to an agent on an LLM, ensure you scope the agent's permissions to only include the necessary permissions to perform the desired operations. By **default** the agent will have access to all files within the root dir and will be able to Copy, Delete, Move, Read, Write and List files in that directory. Consider the following: - Limit access to particular directories using `root_dir`. - Use filesystem permissions to restrict access and permissions to only the files and directories required by the agent. - Limit the tools available to the agent to only the file operations necessary for the agent's intended use. - Sandbox the agent by running it in a container. See for more information. """ root_dir: Optional[str] = None """If specified, all file operations are made relative to root_dir.""" selected_tools: Optional[List[str]] = None """If provided, only provide the selected tools. Defaults to all.""" @root_validator def validate_tools(cls, values: dict) -> dict: selected_tools = values.get("selected_tools") or [] for tool_name in selected_tools: if tool_name not in _FILE_TOOLS_MAP: raise ValueError( f"File Tool of name {tool_name} not supported." f" Permitted tools: {list(_FILE_TOOLS_MAP)}" ) return values
[docs] def get_tools(self) -> List[BaseTool]: """Get the tools in the toolkit.""" allowed_tools = self.selected_tools or _FILE_TOOLS_MAP tools: List[BaseTool] = [] for tool in allowed_tools: tool_cls = _FILE_TOOLS_MAP[tool] tools.append(tool_cls(root_dir=self.root_dir)) # type: ignore[call-arg] return tools
__all__ = ["FileManagementToolkit"]