Skip to content

Extraction task

ExtractionTask

Bases: BaseTextInputTask

Source code in griptape/tasks/extraction_task.py 
 8
 9
10
11
12
13
14
15
16
17
18
@define
class ExtractionTask(BaseTextInputTask):
    _extraction_engine: BaseExtractionEngine = field(kw_only=True, default=None, alias="extraction_engine")
    args: dict = field(kw_only=True)

    @property
    def extraction_engine(self) -> BaseExtractionEngine:
        return self._extraction_engine

    def run(self) -> ListArtifact | ErrorArtifact:
        return self.extraction_engine.extract(self.input.to_text(), rulesets=self.all_rulesets, **self.args)

args: dict = field(kw_only=True) class-attribute instance-attribute

extraction_engine: BaseExtractionEngine property

run()

Source code in griptape/tasks/extraction_task.py 
17
18
def run(self) -> ListArtifact | ErrorArtifact:
    return self.extraction_engine.extract(self.input.to_text(), rulesets=self.all_rulesets, **self.args)