Skip to content

extraction_task

ExtractionTask

Bases: BaseTextInputTask

Source code in griptape/tasks/extraction_task.py 
15
16
17
18
19
20
21
22
23
@define
class ExtractionTask(BaseTextInputTask):
    extraction_engine: BaseExtractionEngine = field(kw_only=True)
    args: dict = field(kw_only=True, factory=dict)

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

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

extraction_engine: BaseExtractionEngine = field(kw_only=True) class-attribute instance-attribute

run()

Source code in griptape/tasks/extraction_task.py 
20
21
22
23
def run(self) -> ListArtifact | ErrorArtifact:
    return self.extraction_engine.extract_artifacts(
        ListArtifact([self.input]), rulesets=self.all_rulesets, **self.args
    )