Skip to content

extraction_task

ExtractionTask

Bases: BaseTextInputTask

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

    def try_run(self) -> ListArtifact | ErrorArtifact:
        return self.extraction_engine.extract_artifacts(ListArtifact([self.input]), rulesets=self.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

try_run()

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