extraction_task

Contents

extraction_task#

Bases: BaseTextInputTask[ListArtifact “)

Source Code in griptape/tasks/extraction_task.py

@define
class ExtractionTask(BaseTextInputTask[ListArtifact]):
    extraction_engine: BaseExtractionEngine = field(kw_only=True)
    args: dict = field(kw_only=True, factory=dict)

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

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

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

try_run()#

Source Code in griptape/tasks/extraction_task.py

def try_run(self) -> ListArtifact:
    return self.extraction_engine.extract_artifacts(ListArtifact([self.input]), rulesets=self.rulesets, **self.args)
Contents