Skip to content

Agent

Agent

Bases: Structure

Source code in griptape/griptape/structures/agent.py 
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
@define
class Agent(Structure):
    input_template: str = field(default=PromptTask.DEFAULT_INPUT_TEMPLATE)
    tools: list[BaseTool] = field(factory=list, kw_only=True)
    max_meta_memory_entries: Optional[int] = field(default=20, kw_only=True)

    def __attrs_post_init__(self) -> None:
        if len(self.tasks) == 0:
            if self.tools:
                task = ToolkitTask(
                    self.input_template, tools=self.tools, max_meta_memory_entries=self.max_meta_memory_entries
                )
            else:
                task = PromptTask(self.input_template, max_meta_memory_entries=self.max_meta_memory_entries)

            self.add_task(task)

        super().__attrs_post_init__()

    @property
    def task(self) -> BaseTask:
        return self.tasks[0]

    def add_task(self, task: BaseTask) -> BaseTask:
        self.tasks.clear()

        task.preprocess(self)

        self.tasks.append(task)

        return task

    def add_tasks(self, *tasks: BaseTask) -> list[BaseTask]:
        if len(tasks) > 1:
            raise ValueError("Agents can only have one task.")
        return super().add_tasks(*tasks)

    def try_run(self, *args) -> Agent:
        self._execution_args = args

        self.task.reset()

        self.task.execute()

        if self.conversation_memory:
            run = Run(input=self.task.input.to_text(), output=self.task.output.to_text())

            self.conversation_memory.add_run(run)

        self._execution_args = ()

        return self

input_template: str = field(default=PromptTask.DEFAULT_INPUT_TEMPLATE) class-attribute instance-attribute

max_meta_memory_entries: Optional[int] = field(default=20, kw_only=True) class-attribute instance-attribute

task: BaseTask property

tools: list[BaseTool] = field(factory=list, kw_only=True) class-attribute instance-attribute

__attrs_post_init__()

Source code in griptape/griptape/structures/agent.py 
19
20
21
22
23
24
25
26
27
28
29
30
def __attrs_post_init__(self) -> None:
    if len(self.tasks) == 0:
        if self.tools:
            task = ToolkitTask(
                self.input_template, tools=self.tools, max_meta_memory_entries=self.max_meta_memory_entries
            )
        else:
            task = PromptTask(self.input_template, max_meta_memory_entries=self.max_meta_memory_entries)

        self.add_task(task)

    super().__attrs_post_init__()

add_task(task)

Source code in griptape/griptape/structures/agent.py 
36
37
38
39
40
41
42
43
def add_task(self, task: BaseTask) -> BaseTask:
    self.tasks.clear()

    task.preprocess(self)

    self.tasks.append(task)

    return task

add_tasks(*tasks)

Source code in griptape/griptape/structures/agent.py 
45
46
47
48
def add_tasks(self, *tasks: BaseTask) -> list[BaseTask]:
    if len(tasks) > 1:
        raise ValueError("Agents can only have one task.")
    return super().add_tasks(*tasks)

try_run(*args)

Source code in griptape/griptape/structures/agent.py 
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
def try_run(self, *args) -> Agent:
    self._execution_args = args

    self.task.reset()

    self.task.execute()

    if self.conversation_memory:
        run = Run(input=self.task.input.to_text(), output=self.task.output.to_text())

        self.conversation_memory.add_run(run)

    self._execution_args = ()

    return self