griptape

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

4

5
class TestToolkitTask:
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
        import os
13
        from griptape.structures import Agent
14
        from griptape.tools import WebScraper, WebSearch, TaskMemoryClient
15

16
        return StructureTester(
17
            Agent(
18
                tools=[
19
                    WebSearch(
20
                        google_api_key=os.environ["GOOGLE_API_KEY"],
21
                        google_api_search_id=os.environ["GOOGLE_API_SEARCH_ID"],
22
                        off_prompt=False,
23
                    ),
24
                    WebScraper(off_prompt=True),
25
                    TaskMemoryClient(off_prompt=False),
26
                ],
27
                conversation_memory=None,
28
                prompt_driver=request.param,
29
            )
30
        )
31

32
    def test_toolkit_task(self, structure_tester):
33
        structure_tester.run("Give me a summary of the top 2 search results about parrot facts.")
34