griptape

1
from tests.utils.structure_tester import StructureTester
2
import pytest
3

4

5
class TestFileManager:
6
    @pytest.fixture(
7
        autouse=True,
8
        params=StructureTester.TOOLKIT_TASK_CAPABLE_PROMPT_DRIVERS,
9
        ids=StructureTester.prompt_driver_id_fn,
10
    )
11
    def structure_tester(self, request):
12
        from griptape.structures import Agent
13
        from griptape.tools import FileManager
14

15
        return StructureTester(
16
            Agent(tools=[FileManager(off_prompt=False)], conversation_memory=None, prompt_driver=request.param)
17
        )
18

19
    def test_save_content_to_disk(self, structure_tester):
20
        structure_tester.run('Write the content "Hello World!" to a file called "poem.txt".')
21

22
    def test_load_files_from_disk(self, structure_tester):
23
        structure_tester.run("Read the content of the file called 'poem.txt'.")
24