griptape

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

4

5
class TestToolTask:
6
    @pytest.fixture(
7
        autouse=True, params=StructureTester.TOOL_TASK_CAPABLE_PROMPT_DRIVERS, ids=StructureTester.prompt_driver_id_fn
8
    )
9
    def structure_tester(self, request):
10
        from griptape.structures import Agent
11
        from griptape.tasks import ToolTask
12
        from griptape.tools import Calculator
13

14
        return StructureTester(
15
            Agent(
16
                tasks=[ToolTask(tool=Calculator(off_prompt=False))],
17
                conversation_memory=None,
18
                prompt_driver=request.param,
19
            )
20
        )
21

22
    def test_tool_task(self, structure_tester):
23
        structure_tester.run("What is 7 times 3 divided by 5 plus 10.")
24