forked from Archives/langchain
334c162f16
Add other File Utilities, include - List Directory - Search for file - Move - Copy - Remove file Bundle as toolkit Add a notebook that connects to the Chat Agent, which somewhat supports multi-arg input tools Update original read/write files to return the original dir paths and better handle unsupported file paths. Add unit tests
39 lines
1.4 KiB
Python
39 lines
1.4 KiB
Python
"""Test the WriteFile tool."""
|
|
|
|
from pathlib import Path
|
|
from tempfile import TemporaryDirectory
|
|
|
|
from langchain.tools.file_management.utils import (
|
|
INVALID_PATH_TEMPLATE,
|
|
)
|
|
from langchain.tools.file_management.write import WriteFileTool
|
|
|
|
|
|
def test_write_file_with_root_dir() -> None:
|
|
"""Test the WriteFile tool when a root dir is specified."""
|
|
with TemporaryDirectory() as temp_dir:
|
|
tool = WriteFileTool(root_dir=temp_dir)
|
|
tool.run({"file_path": "file.txt", "text": "Hello, world!"})
|
|
assert (Path(temp_dir) / "file.txt").exists()
|
|
assert (Path(temp_dir) / "file.txt").read_text() == "Hello, world!"
|
|
|
|
|
|
def test_write_file_errs_outside_root_dir() -> None:
|
|
"""Test the WriteFile tool when a root dir is specified."""
|
|
with TemporaryDirectory() as temp_dir:
|
|
tool = WriteFileTool(root_dir=temp_dir)
|
|
result = tool.run({"file_path": "../file.txt", "text": "Hello, world!"})
|
|
assert result == INVALID_PATH_TEMPLATE.format(
|
|
arg_name="file_path", value="../file.txt"
|
|
)
|
|
|
|
|
|
def test_write_file() -> None:
|
|
"""Test the WriteFile tool."""
|
|
with TemporaryDirectory() as temp_dir:
|
|
file_path = str(Path(temp_dir) / "file.txt")
|
|
tool = WriteFileTool()
|
|
tool.run({"file_path": file_path, "text": "Hello, world!"})
|
|
assert (Path(temp_dir) / "file.txt").exists()
|
|
assert (Path(temp_dir) / "file.txt").read_text() == "Hello, world!"
|