code_execution_task

Suggest edits
  • T = TypeVar('T', bound=BaseArtifact) module-attribute

Bases: BaseTask[T]

Source Code in griptape/tasks/code_execution_task.py
@define
class CodeExecutionTask(BaseTask[T]):
    DEFAULT_INPUT_TEMPLATE = "{{ args[0] }}"
    _input: Union[str, TextArtifact, Callable[[BaseTask], TextArtifact]] = field(
        default=DEFAULT_INPUT_TEMPLATE,
        alias="input",
    )
    on_run: Callable[[CodeExecutionTask[T]], T] = field(kw_only=True)

    @property
    def input(self) -> TextArtifact:
        if isinstance(self._input, TextArtifact):
            return self._input
        if callable(self._input):
            return self._input(self)
        return TextArtifact(J2().render_from_string(self._input, **self.full_context))

    def try_run(self) -> T:
        return self.on_run(self)

  • DEFAULT_INPUT_TEMPLATE = '{{ args[0] }}' class-attribute instance-attribute

  • _input = field(default=DEFAULT_INPUT_TEMPLATE, alias='input') class-attribute instance-attribute

  • input property

  • on_run = field(kw_only=True) class-attribute instance-attribute

try_run()

Source Code in griptape/tasks/code_execution_task.py
def try_run(self) -> T:
    return self.on_run(self)
← Prev

branch_task

↑ Up

tasks

Next →

extraction_task

Could this page be better? Report a problem or suggest an addition!